41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# django
|
|
from django.db import transaction
|
|
|
|
# rest framework
|
|
from rest_framework import serializers
|
|
|
|
# accounts
|
|
from core.apps.accounts.models import User
|
|
# shared
|
|
from core.apps.shared.models import Region
|
|
|
|
|
|
class UserCreateSerializer(serializers.Serializer):
|
|
first_name = serializers.CharField()
|
|
last_name = serializers.CharField()
|
|
telegram_id = serializers.CharField()
|
|
region = serializers.IntegerField()
|
|
|
|
def validate(self, data):
|
|
if User.objects.filter(username=data['telegram_id']).exists():
|
|
raise serializers.ValidationError("User mavjud")
|
|
region = Region.objects.filter(id=data['region']).first()
|
|
if not region:
|
|
raise serializers.ValidationError({"region": "Region topilmadi"})
|
|
data['region_obj'] = region
|
|
return data
|
|
|
|
def create(self, validated_data):
|
|
with transaction.atomic():
|
|
user = User.objects.create(
|
|
first_name=validated_data.get('first_name'),
|
|
last_name=validated_data.get('last_name'),
|
|
telegram_id=validated_data.get('telegram_id'),
|
|
region=validated_data.get('region_obj'),
|
|
is_active=False,
|
|
username=validated_data.get('telegram_id'),
|
|
)
|
|
user.region.users_count += 1
|
|
user.region.save()
|
|
return user
|