Skip to content

Commit 6ab4ae2

Browse files
authored
Refactor installer to separate workflows methods from the installer class (#1055)
1 parent 3cdf2ce commit 6ab4ae2

File tree

6 files changed

+918
-846
lines changed

6 files changed

+918
-846
lines changed

src/databricks/labs/ucx/cli.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
from databricks.labs.ucx.hive_metastore.table_migrate import TablesMigrate
2727
from databricks.labs.ucx.hive_metastore.table_move import TableMove
2828
from databricks.labs.ucx.install import WorkspaceInstallation
29+
from databricks.labs.ucx.installer.workflows import WorkflowsInstallation
2930
from databricks.labs.ucx.workspace_access.clusters import ClusterAccess
3031
from databricks.labs.ucx.workspace_access.groups import GroupManager
3132

@@ -41,7 +42,7 @@
4142
@ucx.command
4243
def workflows(w: WorkspaceClient):
4344
"""Show deployed workflows and their state"""
44-
installation = WorkspaceInstallation.current(w)
45+
installation = WorkflowsInstallation.current(w)
4546
logger.info("Fetching deployed jobs...")
4647
print(json.dumps(installation.latest_job_status()))
4748

@@ -162,7 +163,7 @@ def repair_run(w: WorkspaceClient, step):
162163
"""Repair Run the Failed Job"""
163164
if not step:
164165
raise KeyError("You did not specify --step")
165-
installation = WorkspaceInstallation.current(w)
166+
installation = WorkflowsInstallation.current(w)
166167
logger.info(f"Repair Running {step} Job")
167168
installation.repair_run(step)
168169

0 commit comments

Comments
 (0)