Move EditStatus to gn3.db.case_attributes.

Signed-off-by: Munyoki Kilyungi <me@bonfacemunyoki.com>

2 files changed, 14 insertions, 11 deletions

diff --git a/gn3/case_attributes.py b/gn3/case_attributes.py
index 8312f0d..dae6def 100644
--- a/gn3/case_attributes.py
+++ b/gn3/case_attributes.py
@@ -7,7 +7,7 @@ import tempfile
 import lmdb
 import pickle
 from typing import Union
-from enum import Enum, auto
+
 from pathlib import Path
 from functools import reduce
 from datetime import datetime
@@ -15,7 +15,7 @@ from urllib.parse import urljoin
 
 import requests
 from MySQLdb.cursors import DictCursor
-from gn3.db.case_attributes import CaseAttributeEdit
+from gn3.db.case_attributes import CaseAttributeEdit, EditStatus
 from authlib.integrations.flask_oauth2.errors import _HTTPException
 from flask import (
     jsonify,
@@ -44,15 +44,6 @@ class NoDiffError(ValueError):
         super().__init__(
             self, "No difference between existing data and sent data.")
 
-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
 
 class CAJSONEncoder(json.JSONEncoder):
     """Encoder for CaseAttribute-specific data"""
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: