adding fetching to query status (#15064)

apache · Jun 9, 2021 · 2d3f552 · 2d3f552
1 parent 256e145
commit 2d3f552
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/superset-frontend/src/SqlLab/components/QueryTable/index.jsx b/superset-frontend/src/SqlLab/components/QueryTable/index.jsx
@@ -120,6 +120,14 @@ const statusAttributes = {
       status: 'queued',
     },
   },
+  error: {
+    color: ({ theme }) => theme.colors.error.base,
+    config: {
+      name: 'error',
+      label: t('Unknown Status'),
+      status: 'unknown status!',
+    },
+  },
 };
 
 const StatusIcon = styled(Icon, {
@@ -172,6 +180,8 @@ const QueryTable = props => {
     return props.queries
       .map(query => {
         const q = { ...query };
+        const status = statusAttributes[q.state] || statusAttributes.error;
+
         if (q.endDttm) {
           q.duration = fDuration(q.startDttm, q.endDttm);
         }
@@ -268,14 +278,11 @@ const QueryTable = props => {
             />
           );
         q.state = (
-          <Tooltip
-            title={statusAttributes[q.state].config.label}
-            placement="bottom"
-          >
+          <Tooltip title={status.config.label} placement="bottom">
             <span>
               <StatusIcon
-                name={statusAttributes[q.state].config.name}
-                status={statusAttributes[q.state].config.status}
+                name={status.config.name}
+                status={status.config.status}
               />
             </span>
           </Tooltip>

diff --git a/superset-frontend/src/SqlLab/types.ts b/superset-frontend/src/SqlLab/types.ts
@@ -30,6 +30,7 @@ export type QueryState =
   | 'running'
   | 'scheduled'
   | 'success'
+  | 'fetching'
   | 'timed_out';
 
 export type Query = {

diff --git a/superset/utils/core.py b/superset/utils/core.py
@@ -263,6 +263,7 @@ class QueryStatus(str, Enum):  # pylint: disable=too-few-public-methods
     RUNNING: str = "running"
     SCHEDULED: str = "scheduled"
     SUCCESS: str = "success"
+    FETCHING: str = "fetching"
     TIMED_OUT: str = "timed_out"