start to write contract apis
This commit is contained in:
39
core/apps/contracts/serializers/contract.py
Normal file
39
core/apps/contracts/serializers/contract.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from django.db import transaction
|
||||
|
||||
from rest_framework import serializers
|
||||
|
||||
from core.apps.contracts.models.contract import Contract
|
||||
from core.apps.contracts.serializers.contract_side import ContractSideCreateSerializer
|
||||
|
||||
|
||||
class ContractCreateSerializer(serializers.Serializer):
|
||||
file = serializers.FileField(max_length=None, allow_empty_file=False)
|
||||
contract_number = serializers.IntegerField()
|
||||
name = serializers.CharField()
|
||||
sides = serializers.ChoiceField(choices=('two_or_more', 'customer_only', 'only_company'))
|
||||
face_id = serializers.BooleanField()
|
||||
attach_file = serializers.BooleanField()
|
||||
add_folder = serializers.BooleanField()
|
||||
add_notification = serializers.BooleanField()
|
||||
|
||||
def create(self, validated_data):
|
||||
with transaction.atomic():
|
||||
contract = Contract.objects.create(
|
||||
file=validated_data.pop('file'),
|
||||
contract_number=validated_data.pop('contract_number'),
|
||||
name=validated_data.pop('name'),
|
||||
sides=validated_data.pop('sides'),
|
||||
face_id=validated_data.pop('face_id'),
|
||||
attach_file=validated_data.pop('attach_file'),
|
||||
add_folder=validated_data.pop('add_folder'),
|
||||
add_notification=validated_data.pop('add_notification'),
|
||||
)
|
||||
return contract
|
||||
|
||||
|
||||
class ContractListSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Contract
|
||||
fields = [
|
||||
'id', 'name', 'file', 'contract_number', 'sides', 'face_id', 'add_folder', 'attach_file', 'add_notification', 'created_at'
|
||||
]
|
||||
Reference in New Issue
Block a user