mirror of
https://git.v0id.ovh/n3wt-innov/n3wt-school.git
synced 2026-01-29 07:53:23 +00:00
107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
from rest_framework import serializers
|
|
from Auth.models import Profile, ProfileRole
|
|
from Establishment.models import Establishment
|
|
|
|
class ProfileSerializer(serializers.ModelSerializer):
|
|
id = serializers.IntegerField(required=False)
|
|
password = serializers.CharField(write_only=True)
|
|
|
|
class Meta:
|
|
model = Profile
|
|
fields = ['id', 'password', 'email', 'code', 'datePeremption', 'username']
|
|
extra_kwargs = {'password': {'write_only': True}}
|
|
|
|
def create(self, validated_data):
|
|
user = Profile(
|
|
username=validated_data['username'],
|
|
email=validated_data['email'],
|
|
code=validated_data.get('code', ''),
|
|
datePeremption=validated_data.get('datePeremption', '')
|
|
)
|
|
user.set_password(validated_data['password'])
|
|
user.full_clean()
|
|
user.save()
|
|
return user
|
|
|
|
def update(self, instance, validated_data):
|
|
password = validated_data.pop('password', None)
|
|
instance = super().update(instance, validated_data)
|
|
|
|
if password:
|
|
instance.set_password(password)
|
|
|
|
instance.full_clean()
|
|
instance.save()
|
|
return instance
|
|
|
|
def to_representation(self, instance):
|
|
ret = super().to_representation(instance)
|
|
ret['password'] = '********'
|
|
return ret
|
|
|
|
class ProfileRoleSerializer(serializers.ModelSerializer):
|
|
profile = ProfileSerializer()
|
|
establishment = serializers.PrimaryKeyRelatedField(queryset=Establishment.objects.all())
|
|
|
|
class Meta:
|
|
model = ProfileRole
|
|
fields = ['role_type', 'establishment', 'is_active', 'profile']
|
|
|
|
def create(self, validated_data):
|
|
profile_data = validated_data.pop('profile')
|
|
profile_serializer = ProfileSerializer(data=profile_data)
|
|
profile_serializer.is_valid(raise_exception=True)
|
|
profile = profile_serializer.save()
|
|
profile_role = ProfileRole.objects.create(profile=profile, **validated_data)
|
|
return profile_role
|
|
|
|
def update(self, instance, validated_data):
|
|
profile_data = validated_data.pop('profile')
|
|
profile_serializer = ProfileSerializer(instance.profile, data=profile_data)
|
|
profile_serializer.is_valid(raise_exception=True)
|
|
profile = profile_serializer.save()
|
|
|
|
instance.role_type = validated_data.get('role_type', instance.role_type)
|
|
instance.establishment_id = validated_data.get('establishment', instance.establishment.id)
|
|
instance.is_active = validated_data.get('is_active', instance.is_active)
|
|
instance.save()
|
|
return instance
|
|
|
|
class ProfilUpdateSerializer(serializers.ModelSerializer):
|
|
roles = ProfileRoleSerializer(many=True, required=False)
|
|
|
|
class Meta:
|
|
model = Profile
|
|
fields = ['id', 'password', 'email', 'code', 'datePeremption', 'username', 'roles']
|
|
extra_kwargs = {
|
|
'password': {'write_only': True, 'required': False}
|
|
}
|
|
|
|
def update(self, instance, validated_data):
|
|
roles_data = validated_data.pop('roles', [])
|
|
password = validated_data.pop('password', None)
|
|
instance = super().update(instance, validated_data)
|
|
|
|
if password:
|
|
instance.set_password(password)
|
|
|
|
instance.full_clean()
|
|
instance.save()
|
|
|
|
for role_data in roles_data:
|
|
ProfileRole.objects.update_or_create(
|
|
profile=instance,
|
|
establishment_id=role_data.get('establishment_id'),
|
|
defaults={
|
|
'role_type': role_data.get('role_type'),
|
|
'is_active': role_data.get('is_active', True)
|
|
}
|
|
)
|
|
|
|
return instance
|
|
|
|
def to_representation(self, instance):
|
|
ret = super().to_representation(instance)
|
|
ret['password'] = '********'
|
|
return ret
|