from django.contrib.auth.models import User
from django.contrib.staticfiles.templatetags.staticfiles import static
from rest_framework import serializers
from models import *
from friendship.models import Friend
from friendship.models import Follow
[docs]class UserProfilePictureSerializer(serializers.ModelSerializer):
[docs]class UserSerializer(serializers.ModelSerializer):
profile = UserProfilePictureSerializer(read_only=True)
[docs]class PictureField(serializers.Field):
[docs] def to_representation(self, obj):
if obj:
return obj.url
else:
return static('assets/img/default.jpg')
[docs]class UserProfileSerializer(serializers.ModelSerializer):
user = UserSerializer(read_only=True)
picture = PictureField(read_only=True)
[docs]class UserCoverSerializer(serializers.ModelSerializer):
[docs]class FirstUserProfileSerializer(serializers.ModelSerializer):
user = UserSerializer(read_only=True)
[docs]class FriendShipSerializer(serializers.ModelSerializer):
user = UserSerializer(source='from_user')