You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.8 KiB
49 lines
1.8 KiB
from django.db import IntegrityError
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from .models import Document, UserResponse
|
|
|
|
|
|
class UserResponseSerializer(serializers.ModelSerializer):
|
|
user = serializers.PrimaryKeyRelatedField(read_only=True)
|
|
question = serializers.PrimaryKeyRelatedField(read_only=True)
|
|
|
|
class Meta:
|
|
model = UserResponse
|
|
fields = ('user', 'question', 'value')
|
|
|
|
def create(self, validated_data):
|
|
response = UserResponse(**validated_data)
|
|
response.save()
|
|
return response
|
|
|
|
def update(self, instance, validated_data):
|
|
instance.value = validated_data['value']
|
|
instance.save()
|
|
|
|
|
|
class DocumentSerializer(serializers.ModelSerializer):
|
|
doc_type = serializers.CharField()
|
|
party_code = serializers.IntegerField(min_value=0, max_value=2)
|
|
file = serializers.FileField()
|
|
filename = serializers.CharField(read_only=True)
|
|
size = serializers.IntegerField(read_only=True)
|
|
rotation = serializers.IntegerField(read_only=True)
|
|
sort_order = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
model = Document
|
|
fields = ('file', 'doc_type', 'party_code', 'filename', 'size', 'rotation', 'sort_order')
|
|
|
|
def create(self, validated_data):
|
|
filename = validated_data['file'].name
|
|
size = validated_data['file'].size
|
|
user = self.context['request'].user
|
|
order = Document.objects.filter(bceid_user=user, doc_type=validated_data['doc_type'], party_code=validated_data['party_code']).count() + 1
|
|
response = Document(bceid_user=user, filename=filename, size=size, sort_order=order, **validated_data)
|
|
try:
|
|
response.save()
|
|
except IntegrityError:
|
|
raise ValidationError("You have already uploaded that file")
|
|
return response
|