from rest_framework import serializers

from .models import User

class UserSerializer(serializers.ModelSerializer):
    user_type = serializers.SerializerMethodField()

    class Meta:
        model = User
        fields = ('id', 'user_type', 'phone', 'date_joined', 'last_login', 'is_active')

    def get_user_type(self, obj):
        if not obj.role==None:
            return obj.role.name
        else:
            return
