add new enums for datasource types (#20015)

apache · May 10, 2022 · 1f0daa9 · 1f0daa9
1 parent bba40d7
commit 1f0daa9
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