40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from rest_framework import serializers
|
|
from system.accounts.models import User
|
|
|
|
class UserSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = User
|
|
fields = ('username', 'email', 'first_name', 'last_name', 'password', 'bio', 'image')
|
|
extra_kwargs = {'password': {'write_only': True}}
|
|
|
|
def create(self, validated_data):
|
|
password = validated_data.pop('password')
|
|
user = User(
|
|
**validated_data
|
|
)
|
|
user.set_password(password)
|
|
user.save()
|
|
return user
|
|
|
|
def update(self, instance, validated_data):
|
|
for key, value in validated_data.items():
|
|
if key == 'password':
|
|
instance.set_password(value)
|
|
else:
|
|
setattr(instance, key, value)
|
|
instance.save()
|
|
return instance
|
|
|
|
|
|
class ProfileSerializer(serializers.ModelSerializer):
|
|
following = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ('username', 'first_name', 'last_name', 'bio', 'image', 'following')
|
|
|
|
def get_following(self, obj):
|
|
user = self.context.get('request').user
|
|
if user.is_authenticated:
|
|
return obj.followers.filter(pk=user.id).exists()
|
|
return False |