34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
# django
|
|
from django.db import transaction
|
|
|
|
# rest framework
|
|
from rest_framework import serializers
|
|
|
|
# shared
|
|
from core.apps.shared.models import Support, District
|
|
|
|
|
|
class SupportCreateSerializer(serializers.Serializer):
|
|
district_id = serializers.IntegerField(required=False)
|
|
problem = serializers.CharField()
|
|
date = serializers.DateField()
|
|
type = serializers.ChoiceField(choices=Support.TYPE)
|
|
|
|
def validate(self, data):
|
|
if data.get('district_id'):
|
|
district = District.objects.filter(id=data['district_id']).first()
|
|
if not district:
|
|
raise serializers.ValidationError({'district': "district not found"})
|
|
data['district'] = district
|
|
return data
|
|
|
|
def create(self, validated_data):
|
|
with transaction.atomic():
|
|
return Support.objects.create(
|
|
user=self.context.get('user'),
|
|
district=validated_data.get('district'),
|
|
date=validated_data.get('date'),
|
|
type=validated_data.get('type'),
|
|
problem=validated_data.get('problem' )
|
|
)
|