53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
|
|
from rest_framework import serializers
|
|
from .models import CustomUser
|
|
from django.contrib.auth.models import Group
|
|
from api.customs.models import Importador
|
|
|
|
class CustomUserSerializer(serializers.ModelSerializer):
|
|
"""
|
|
Serializer for the CustomUser model.
|
|
"""
|
|
|
|
password = serializers.CharField(write_only=True)
|
|
groups = serializers.PrimaryKeyRelatedField(queryset=Group.objects.all(), many=True, required=False)
|
|
rfc = serializers.PrimaryKeyRelatedField(
|
|
queryset=Importador.objects.all(),
|
|
many=True,
|
|
required=False,
|
|
pk_field=serializers.CharField(),
|
|
)
|
|
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = ['id', 'username', 'email', 'first_name', 'last_name', 'password', 'profile_picture', 'organizacion', 'is_importador', 'rfc', 'is_active', 'is_superuser', 'groups']
|
|
read_only_fields = ['id', 'organizacion', 'is_superuser']
|
|
|
|
def create(self, validated_data):
|
|
groups = validated_data.pop('groups', [])
|
|
rfcs = validated_data.pop('rfc', [])
|
|
password = validated_data.pop('password')
|
|
user = CustomUser(**validated_data)
|
|
user.set_password(password)
|
|
user.save()
|
|
if groups:
|
|
user.groups.set(groups)
|
|
if rfcs:
|
|
user.rfc.set(rfcs)
|
|
return user
|
|
|
|
def update(self, instance, validated_data):
|
|
groups = validated_data.pop('groups', None)
|
|
rfcs = validated_data.pop('rfc', None)
|
|
password = validated_data.pop('password', None)
|
|
for attr, value in validated_data.items():
|
|
setattr(instance, attr, value)
|
|
if password:
|
|
instance.set_password(password)
|
|
instance.save()
|
|
if groups is not None:
|
|
instance.groups.set(groups)
|
|
if rfcs is not None:
|
|
instance.rfc.set(rfcs)
|
|
return instance
|