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.
75 lines
3.4 KiB
75 lines
3.4 KiB
from django.db import IntegrityError
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from .models import Document, UserResponse
|
|
from .validators import file_scan_validation, valid_file_extension, valid_doc_type, valid_rotation
|
|
|
|
|
|
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 CreateDocumentSerializer(serializers.ModelSerializer):
|
|
doc_type = serializers.CharField(required=True, validators=[valid_doc_type])
|
|
party_code = serializers.IntegerField(min_value=0, max_value=2, required=True)
|
|
file = serializers.FileField(required=True, validators=[valid_file_extension,file_scan_validation])
|
|
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)
|
|
file_url = serializers.URLField(source='get_file_url', read_only=True)
|
|
|
|
class Meta:
|
|
model = Document
|
|
fields = ('file', 'doc_type', 'party_code', 'filename', 'size',
|
|
'rotation', 'sort_order', 'file_url', 'filing_type')
|
|
|
|
def create(self, validated_data):
|
|
filename = validated_data['file'].name
|
|
size = validated_data['file'].size
|
|
user = self.context['request'].user
|
|
existing_docs = Document.objects.filter(bceid_user=user, doc_type=validated_data['doc_type'], party_code=validated_data['party_code'])
|
|
for other_doc in existing_docs:
|
|
if other_doc.is_pdf:
|
|
raise ValidationError("PDF documents cannot be combined with images. Only a single PDF or multiple images can be uploaded into one form.")
|
|
|
|
sort_order = existing_docs.count() + 1
|
|
document = Document(bceid_user=user, filename=filename, size=size, sort_order=sort_order, **validated_data)
|
|
if document.is_pdf and existing_docs.count() > 0:
|
|
raise ValidationError("Only one PDF is allowed per form, and PDF documents cannot be combined with images.")
|
|
try:
|
|
document.save()
|
|
except IntegrityError:
|
|
raise ValidationError("This file appears to have already been uploaded for this document. Duplicate filename: " + filename)
|
|
return document
|
|
|
|
|
|
class DocumentMetadataSerializer(serializers.ModelSerializer):
|
|
doc_type = serializers.CharField(read_only=True)
|
|
party_code = serializers.IntegerField(read_only=True)
|
|
filing_type = serializers.CharField(read_only=True)
|
|
filename = serializers.CharField(read_only=True)
|
|
size = serializers.IntegerField(read_only=True)
|
|
rotation = serializers.IntegerField(min_value=0, max_value=270, validators=[valid_rotation])
|
|
sort_order = serializers.IntegerField(read_only=True)
|
|
file_url = serializers.URLField(source='get_file_url', read_only=True)
|
|
|
|
class Meta:
|
|
model = Document
|
|
fields = ('doc_type', 'party_code', 'filename', 'size',
|
|
'rotation', 'sort_order', 'file_url', 'filing_type')
|