28 lines
836 B
Python
28 lines
836 B
Python
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from .models import Account
|
||
|
|
|
||
|
|
|
||
|
|
class AccountSerializer(serializers.HyperlinkedModelSerializer):
|
||
|
|
class Meta:
|
||
|
|
model = Account
|
||
|
|
fields = (
|
||
|
|
'url', 'id', 'username', 'type', 'language', 'short_name', 'full_name', 'date_joined', 'last_login',
|
||
|
|
'is_active'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class AccountSerializerMixin(object):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super(AccountSerializerMixin, self).__init__(*args, **kwargs)
|
||
|
|
self.account = self.get_account()
|
||
|
|
|
||
|
|
def get_account(self):
|
||
|
|
request = self.context.get('request')
|
||
|
|
if request:
|
||
|
|
return request.user
|
||
|
|
|
||
|
|
def create(self, validated_data):
|
||
|
|
validated_data['account'] = self.get_account()
|
||
|
|
return super(AccountSerializerMixin, self).create(validated_data)
|