2026-03-24 14:46:03 +00:00
|
|
|
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()
|
2026-04-22 10:30:58 +05:30
|
|
|
partner = serializers.PrimaryKeyRelatedField(read_only=True)
|
2026-03-24 14:46:03 +00:00
|
|
|
class Meta:
|
|
|
|
|
model = User
|
2026-04-22 10:30:58 +05:30
|
|
|
fields = ['id', 'email', 'username', 'name', 'role', 'partner']
|
2026-03-24 14:46:03 +00:00
|
|
|
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')
|