diff --git a/api/api/db/tables/account.py b/api/api/db/tables/account.py index ed4fa8f..84c689e 100644 --- a/api/api/db/tables/account.py +++ b/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" diff --git a/api/api/db/tables/events.py b/api/api/db/tables/events.py index c4d8ab2..fba2e58 100644 --- a/api/api/db/tables/events.py +++ b/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( diff --git a/api/api/db/tables/process.py b/api/api/db/tables/process.py index e2f8c68..379663c 100644 --- a/api/api/db/tables/process.py +++ b/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(