Add status message

Author: kesmit13

singlestoredb/fusion/handlers/export.py

@@ -217,10 +217,19 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         wsg = get_workspace_group({})
         out = ExportStatus(params['export_id'], wsg)
 
+        status = out._info()
+
         res = FusionSQLResult()
         res.add_field('ExportID', result.STRING)
         res.add_field('Status', result.STRING)
-        res.set_rows([(params['export_id'], out.status)])
+        res.add_field('Message', result.STRING)
+        res.set_rows([
+            (
+                params['export_id'],
+                status.get('status', 'Unknown'),
+                status.get('statusMsg', ''),
+            ),
+        ])
 
         return res
 

singlestoredb/management/export.py

@@ -279,8 +279,7 @@ def __init__(self, export_id: str, workspace_group: WorkspaceGroup):
         self.workspace_group = workspace_group
         self._manager: Optional[WorkspaceManager] = workspace_group._manager
 
-    @property
-    def status(self) -> str:
+    def _info(self) -> Dict[str, Any]:
         """Return export status."""
         if self._manager is None:
             raise ManagementError(
@@ -292,9 +291,17 @@ def status(self) -> str:
             json=dict(egressID=self.export_id),
         )
 
-        print(out.json())
+        return out.json()
+
+    @property
+    def status(self) -> str:
+        """Return export status."""
+        return self._info().get('status', 'Unknown')
 
-        return list(out.json().values())[0]
+    @property
+    def message(self) -> str:
+        """Return export status message."""
+        return self._info().get('statusMsg', '')
 
     def __str__(self) -> str:
         return self.status