36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from rest_framework.serializers import ModelSerializer, CharField, SerializerMethodField, ListField, IntegerField
|
|
from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
|
|
from account.models import MyUser
|
|
|
|
|
|
class MyUserSerializer(ModelSerializer):
|
|
|
|
def create(self, validated_data):
|
|
user = MyUser.objects.create_user(**validated_data)
|
|
return user
|
|
|
|
class Meta:
|
|
model = MyUser
|
|
fields = ('id', 'email', 'password',
|
|
'first_name', 'last_name',
|
|
'phone', 'date_joined',
|
|
'last_login', 'INN_us',
|
|
'urid_adress', 'organization'
|
|
)
|
|
extra_kwargs = {
|
|
'password': {'write_only': True},
|
|
}
|
|
|
|
|
|
|
|
class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
|
|
|
|
@classmethod
|
|
def get_token(cls, user):
|
|
token = super().get_token(user)
|
|
token['email'] = user.email
|
|
token['first_name'] = user.first_name
|
|
token['last_name'] = user.last_name
|
|
|
|
return token
|