OnlineJudge/utils/api/_serializers.py

15 lines
474 B
Python
Raw Normal View History

2016-11-19 04:32:23 +00:00
from rest_framework import serializers
2017-01-26 04:33:00 +00:00
class UsernameSerializer(serializers.Serializer):
id = serializers.IntegerField()
username = serializers.CharField()
real_name = serializers.SerializerMethodField()
def __init__(self, *args, **kwargs):
self.is_admin_role = kwargs.pop("is_admin_role", False)
super().__init__(*args, **kwargs)
def get_real_name(self, obj):
return obj.userprofile.real_name if self.is_admin_role else None