53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
"""Back account serializers"""
|
|
from rest_framework import serializers
|
|
from account import models
|
|
from account.models import User
|
|
|
|
|
|
class RoleSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.Role
|
|
fields = [
|
|
'role',
|
|
'country'
|
|
]
|
|
|
|
|
|
class BackUserSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = User
|
|
fields = '__all__'
|
|
extra_kwargs = {
|
|
'password': {'write_only': True}
|
|
}
|
|
read_only_fields = ('old_password', 'last_login', 'date_joined')
|
|
|
|
def create(self, validated_data):
|
|
user = super().create(validated_data)
|
|
user.set_password(validated_data['password'])
|
|
user.save()
|
|
return user
|
|
|
|
|
|
class BackDetailUserSerializer(BackUserSerializer):
|
|
class Meta:
|
|
model = User
|
|
exclude = ('password',)
|
|
read_only_fields = ('old_password', 'last_login', 'date_joined')
|
|
|
|
def create(self, validated_data):
|
|
user = super().create(validated_data)
|
|
user.set_password(validated_data['password'])
|
|
user.save()
|
|
return user
|
|
|
|
|
|
class UserRoleSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.UserRole
|
|
fields = [
|
|
'role',
|
|
'user',
|
|
'establishment'
|
|
]
|