23 lines
813 B
Python
23 lines
813 B
Python
from rest_framework.serializers import ModelSerializer, SerializerMethodField
|
|
|
|
from room.serializers import RoomUserSerializer
|
|
from .models import CustomUser
|
|
|
|
|
|
class UserSerializer(ModelSerializer):
|
|
room = SerializerMethodField()
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = ('email', 'last_login', 'date_joined', 'is_staff', 'username', "id_code", 'room', 'first_name', 'last_name')
|
|
|
|
def get_room(self, obj):
|
|
try:
|
|
return RoomUserSerializer(obj.room_set.all(), many=True, context={'userId': obj.id_code}).data
|
|
except:
|
|
return RoomUserSerializer(obj.room_set.all(), many=True).data
|
|
|
|
|
|
class UserEditSerializer(ModelSerializer):
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = ('email', 'username', 'last_name', 'first_name') |