diff --git a/edivorce/apps/core/serializer.py b/edivorce/apps/core/serializer.py index 644ab410..019e3542 100644 --- a/edivorce/apps/core/serializer.py +++ b/edivorce/apps/core/serializer.py @@ -3,7 +3,7 @@ from rest_framework import serializers from rest_framework.exceptions import ValidationError from .models import Document, UserResponse -from .validators import * +from .validators import file_scan_validation, valid_file_extension, valid_doc_type, valid_rotation class UserResponseSerializer(serializers.ModelSerializer): diff --git a/edivorce/apps/core/tests/test_upload.py b/edivorce/apps/core/tests/test_upload.py index 1ce14564..2c7863f9 100644 --- a/edivorce/apps/core/tests/test_upload.py +++ b/edivorce/apps/core/tests/test_upload.py @@ -5,7 +5,6 @@ from unittest import mock from django.test import TestCase from django.core.files.uploadedfile import SimpleUploadedFile from rest_framework import serializers -from rest_framework.test import APIRequestFactory from ..validators import file_scan_validation from ..models import Document diff --git a/edivorce/apps/core/validators.py b/edivorce/apps/core/validators.py index ded51f8b..c0309fb2 100644 --- a/edivorce/apps/core/validators.py +++ b/edivorce/apps/core/validators.py @@ -2,7 +2,6 @@ import logging import clamd import sys -from rest_framework import serializers from rest_framework.exceptions import ValidationError from django.conf import settings @@ -59,4 +58,4 @@ def valid_doc_type(value): def valid_rotation(value): if value % 90 != 0: - raise serializers.ValidationError('Rotation must be 0, 90, 180, or 270') + raise ValidationError('Rotation must be 0, 90, 180, or 270')