refactor(tables): change enum, str to StrEnum

api/api/db/tables/account.py

@@ -1,3 +1,5 @@
+import enum
+
 from sqlalchemy import Table, Column, String, Enum as SQLAEnum, JSON, ForeignKey, DateTime, Index
 from sqlalchemy.sql import func
 
@@ -7,14 +9,14 @@ from api.db.sql_types import UnsignedInt
 from api.db import metadata
 
 
-class AccountRole(str, Enum):
+class AccountRole(enum.StrEnum):
     OWNER = "OWNER"
     ADMIN = "ADMIN"
     EDITOR = "EDITOR"
     VIEWER = "VIEWER"
 
 
-class AccountStatus(str, Enum):
+class AccountStatus(enum.StrEnum):
     ACTIVE = "ACTIVE"
     DISABLED = "DISABLED"
     BLOCKED = "BLOCKED"
@@ -39,14 +41,14 @@ account_table = Table(
 )
 
 
-class KeyType(str, Enum):
+class KeyType(enum.StrEnum):
     PASSWORD = "PASSWORD"
     ACCESS_TOKEN = "ACCESS_TOKEN"
     REFRESH_TOKEN = "REFRESH_TOKEN"
     API_KEY = "API_KEY"
 
 
-class KeyStatus(str, Enum):
+class KeyStatus(enum.StrEnum):
     ACTIVE = "ACTIVE"
     EXPIRED = "EXPIRED"
     DELETED = "DELETED"

api/api/db/tables/events.py

@@ -1,3 +1,5 @@
+import enum
+
 from sqlalchemy import Table, Column, Integer, String, Enum as SQLAEnum, JSON, ForeignKey, DateTime, Index
 from sqlalchemy.sql import func
 from enum import Enum, auto
@@ -7,15 +9,15 @@ from api.db.sql_types import UnsignedInt
 from api.db import metadata
 
 
-class EventState(str, Enum):
-    AUTO = auto()
-    DESCRIPTED = auto()
+class EventState(enum.StrEnum):
+    AUTO = "AUTO"
+    DESCRIPTED = "DESCRIPTED"
 
 
-class EventStatus(str, Enum):
-    ACTIVE = auto()
-    DISABLED = auto()
-    DELETED = auto()
+class EventStatus(enum.StrEnum):
+    ACTIVE = "ACTIVE"
+    DISABLED = "DISABLED"
+    DELETED = "DELETED"
 
 
 list_events_table = Table(

api/api/db/tables/process.py

@@ -1,3 +1,5 @@
+import enum
+
 from sqlalchemy import (
     Table,
     Column,
@@ -19,11 +21,11 @@ from api.db.sql_types import UnsignedInt
 from api.db import metadata
 
 
-class ProcessStatus(str, Enum):
-    ACTIVE = auto()
-    STOPPING = auto()
-    STOPPED = auto()
-    DELETED = auto()
+class ProcessStatus(enum.StrEnum):
+    ACTIVE = "ACTIVE"
+    STOPPING = "STOPPING"
+    STOPPED = "STOPPED"
+    DELETED = "DELETED"
 
 
 process_schema_table = Table(
@@ -57,10 +59,10 @@ process_version_archive_table = Table(
 )
 
 
-class NodeStatus(str, Enum):
-    ACTIVE = auto()
-    DISABLED = auto()
-    DELETED = auto()
+class NodeStatus(enum.StrEnum):
+    ACTIVE = "ACTIVE"
+    DISABLED = "DISABLED"
+    DELETED = "DELETED"
 
 
 class NodeType(Enum):
@@ -83,11 +85,11 @@ ps_node_table = Table(
 )
 
 
-class NodeLinkStatus(str, Enum):
-    ACTIVE = auto()
-    STOPPING = auto()
-    STOPPED = auto()
-    DELETED = auto()
+class NodeLinkStatus(enum.StrEnum):
+    ACTIVE = "ACTIVE"
+    STOPPING = "STOPPING"
+    STOPPED = "STOPPED"
+    DELETED = "DELETED"
 
 
 node_link_table = Table(