about summary refs log tree commit diff
path: root/gn3/db
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/db')
-rw-r--r--gn3/db/case_attributes.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/gn3/db/case_attributes.py b/gn3/db/case_attributes.py
index 7b56bc4..a5fdc53 100644
--- a/gn3/db/case_attributes.py
+++ b/gn3/db/case_attributes.py
@@ -1,6 +1,7 @@
 """Module that contains functions for editing case-attribute data"""
 from typing import Any, Optional, Tuple
 from dataclasses import dataclass
+from enum import Enum, auto
 
 import json
 import MySQLdb
@@ -23,6 +24,17 @@ class CaseAttributeEdit:
     changes: dict
 
 
+class EditStatus(Enum):
+    """Enumeration for the status of the edits."""
+    review = auto()   # pylint: disable=[invalid-name]
+    approved = auto()  # pylint: disable=[invalid-name]
+    rejected = auto()  # pylint: disable=[invalid-name]
+
+    def __str__(self):
+        """Print out human-readable form."""
+        return self.name
+
+
 def get_case_attributes(conn) -> Optional[Tuple]:
     """Get all the case attributes from the database."""
     with conn.cursor() as cursor: