Remove comment. Add party_code range

edivorce/apps/core/models.py

@@ -112,9 +112,6 @@ class UserResponse(models.Model):
 
 
 class Document(models.Model):
-    """
-    This is only a POC model and should not be loaded on a production system.
-    """
     filename = models.CharField(max_length=128, null=True)  # saving the original filename separately
     """ File name and extension """
 

edivorce/apps/core/serializer.py

@@ -25,7 +25,7 @@ class UserResponseSerializer(serializers.ModelSerializer):
 
 class DocumentSerializer(serializers.ModelSerializer):
     doc_type = serializers.CharField()
-    party_code = serializers.IntegerField()
+    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)