add new enums for datasource types (apache#20015)

ValtechMobility · Jun 9, 2022 · c2469e7 · c2469e7
1 parent 539bfd8
commit c2469e7
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/superset/utils/core.py b/superset/utils/core.py
@@ -174,8 +174,16 @@ class GenericDataType(IntEnum):
     # ROW = 7
 
 
+class DatasourceType(Enum):
+    SQLATABLE = "sqlatable"
+    TABLE = "table"
+    DATASET = "dataset"
+    QUERY = "query"
+    SAVEDQUERY = "saved_query"
+
+
 class DatasourceDict(TypedDict):
-    type: str
+    type: str  # todo(hugh): update this to be DatasourceType
     id: int