Generating filetype and count to be EnumFields

Author: mfixstsci

jwql/website/apps/jwql/monitor_models/common.py

@@ -177,8 +177,12 @@ class Meta:
 # Feel free to rename the models, but don't rename db_table values or field names.
 from django.db import models
 from django.contrib.postgres.fields import ArrayField
+from django_enum import EnumField
+from enum import Enum
 
-from jwql.utils.constants import DEFAULT_MODEL_CHARFIELD, MAX_LEN_FILTER
+from jwql.utils.constants import DEFAULT_MODEL_CHARFIELD, MAX_LEN_FILTER, FILE_SUFFIX_TYPES
+
+FILE_SUFFIX_CLASS = Enum('FILE_SUFFIX_CLASS', FILE_SUFFIX_TYPES)
 
 
 class Monitor(models.Model):
@@ -244,9 +248,8 @@ class Meta:
 
 class FilesystemInstrument(models.Model):
     date = models.DateTimeField()
-    instrument = models.TextField()  # This field type is a guess.
-    filetype = models.TextField()  # This field type is a guess.
-    count = models.IntegerField()
+    filetype = EnumField(FILE_SUFFIX_CLASS)  # This field type is a guess.
+    count = EnumField(FILE_SUFFIX_CLASS)
     size = models.FloatField()
 
     class Meta:

pyproject.toml

@@ -27,6 +27,7 @@ dependencies = [
     "celery>=5.3.6,<6",
     "crds>=11.17.19,<12",
     "django>=5.0.3,<6",
+    "django-enum>=2.0.2,<3",
     "gunicorn>=22.0.0,<23.0.0",
     "inflection>=0.5.1,<0.6",
     "jsonschema>=4.21.1,<5",