..

nfx · nfx · commit dab56cb91edf · 2023-11-07T13:28:52.000+01:00
diff --git a/src/databricks/labs/ucx/cli.py b/src/databricks/labs/ucx/cli.py
@@ -1,18 +1,19 @@
+import json
 import logging
-import sys, json
+import sys
 import webbrowser
 
 from databricks.sdk import WorkspaceClient
 
 from databricks.labs.ucx.install import WorkspaceInstaller
 
-logger = logging.getLogger('databricks.labs.ucx')
+logger = logging.getLogger("databricks.labs.ucx")
 
 
 def workflows():
     ws = WorkspaceClient()
     installer = WorkspaceInstaller(ws)
-    logger.info('Fetching deployed jobs...')
+    logger.info("Fetching deployed jobs...")
     print(json.dumps(installer.latest_job_status()))
 
 
@@ -25,23 +26,24 @@ def open_remote_config():
 
 
 MAPPING = {
-    'open-remote-config': open_remote_config,
-    'workflows': workflows,
+    "open-remote-config": open_remote_config,
+    "workflows": workflows,
 }
 
 
 def main(raw):
     payload = json.loads(raw)
-    command = payload['command']
+    command = payload["command"]
     if command not in MAPPING:
-        raise KeyError(f'cannot find command: {command}')
-    flags = payload['flags']
-    log_level = flags.pop('log_level')
-    if log_level != 'disabled':
+        msg = f"cannot find command: {command}"
+        raise KeyError(msg)
+    flags = payload["flags"]
+    log_level = flags.pop("log_level")
+    if log_level != "disabled":
         databricks_logger = logging.getLogger("databricks")
         databricks_logger.setLevel(log_level.upper())
 
-    kwargs = {k.replace('-', '_'): v for k,v in flags.items()}
+    kwargs = {k.replace("-", "_"): v for k, v in flags.items()}
     MAPPING[command](**kwargs)
 
 
diff --git a/src/databricks/labs/ucx/install.py b/src/databricks/labs/ucx/install.py
@@ -822,11 +822,13 @@ def latest_job_status(self) -> list[dict]:
         latest_status = []
         for step, job_id in self.deployed_steps().items():
             job_runs = list(self._ws.jobs.list_runs(job_id=job_id, limit=1))
-            latest_status.append({
-                'step': step,
-                'state': 'UNKNOWN' if not job_runs else str(job_runs[0].state.result_state),
-                'started': '' if not job_runs else job_runs[0].start_time
-            })
+            latest_status.append(
+                {
+                    "step": step,
+                    "state": "UNKNOWN" if not job_runs else str(job_runs[0].state.result_state),
+                    "started": "" if not job_runs else job_runs[0].start_time,
+                }
+            )
         return latest_status
 
 
