19 lines
644 B
Python
19 lines
644 B
Python
|
from rest_framework import serializers
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
class UserSerializer(serializers.ModelSerializer):
|
||
|
|
name = serializers.SerializerMethodField()
|
||
|
|
role = serializers.SerializerMethodField()
|
||
|
|
class Meta:
|
||
|
|
model = User
|
||
|
|
fields = ['id', 'email', 'username', 'name', 'role']
|
||
|
|
def get_name(self, obj):
|
||
|
|
return f"{obj.first_name} {obj.last_name}".strip() or obj.username
|
||
|
|
def get_role(self, obj):
|
||
|
|
if obj.is_superuser:
|
||
|
|
return 'superadmin'
|
||
|
|
if obj.is_staff:
|
||
|
|
return 'admin'
|
||
|
|
return getattr(obj, 'role', 'user')
|