cleanup

databrickslabs · Sep 21, 2023 · 993cbac · 993cbac
1 parent b397806
commit 993cbac
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 5 deletions.
diff --git a/src/databricks/labs/ucx/hive_metastore/__init__.py b/src/databricks/labs/ucx/hive_metastore/__init__.py
@@ -1,5 +1,5 @@
 from databricks.labs.ucx.hive_metastore.grants import GrantsCrawler
-from databricks.labs.ucx.hive_metastore.list_mounts import Mounts
+from databricks.labs.ucx.hive_metastore.mounts import Mounts
 from databricks.labs.ucx.hive_metastore.tables import TablesCrawler
 
 __all__ = ["TablesCrawler", "GrantsCrawler", "Mounts"]
diff --git a/src/databricks/labs/ucx/hive_metastore/data_objects.py b/src/databricks/labs/ucx/hive_metastore/data_objects.py
@@ -5,7 +5,7 @@
 from databricks.sdk import WorkspaceClient
 
 from databricks.labs.ucx.framework.crawlers import CrawlerBase, SqlBackend
-from databricks.labs.ucx.hive_metastore.list_mounts import Mounts
+from databricks.labs.ucx.hive_metastore.mounts import Mounts
 
 
 @dataclass

diff --git a/...ks/labs/ucx/hive_metastore/list_mounts.py → ...abricks/labs/ucx/hive_metastore/mounts.py b/...ks/labs/ucx/hive_metastore/list_mounts.py → ...abricks/labs/ucx/hive_metastore/mounts.py
diff --git a/src/databricks/labs/ucx/runtime.py b/src/databricks/labs/ucx/runtime.py
@@ -10,7 +10,7 @@
 from databricks.labs.ucx.framework.tasks import task, trigger
 from databricks.labs.ucx.hive_metastore import GrantsCrawler, TablesCrawler
 from databricks.labs.ucx.hive_metastore.data_objects import ExternalLocationCrawler
-from databricks.labs.ucx.hive_metastore.list_mounts import Mounts
+from databricks.labs.ucx.hive_metastore.mounts import Mounts
 from databricks.labs.ucx.workspace_access import GroupMigrationToolkit
 
 logger = logging.getLogger(__name__)
@@ -197,6 +197,12 @@ def delete_backup_groups(cfg: MigrationConfig):
     toolkit.delete_backup_groups()
 
 
+@task("destroy-schema")
+def destroy_schema(cfg: MigrationConfig):
+    """Removes the `$inventory` database"""
+    RuntimeBackend().execute(f"DROP DATABASE {cfg.inventory_database} CASCADE")
+
+
 def main():
     trigger(*sys.argv)
 

diff --git a/tests/unit/assessment/test_assessment.py b/tests/unit/assessment/test_assessment.py
@@ -5,7 +5,7 @@
 
 from databricks.labs.ucx.assessment.crawlers import ClustersCrawler, JobsCrawler
 from databricks.labs.ucx.hive_metastore.data_objects import ExternalLocationCrawler
-from databricks.labs.ucx.hive_metastore.list_mounts import Mount
+from databricks.labs.ucx.hive_metastore.mounts import Mount
 from databricks.labs.ucx.mixins.sql import Row
 from tests.unit.framework.mocks import MockBackend
 

diff --git a/tests/unit/hive_metastore/test_mounts.py b/tests/unit/hive_metastore/test_mounts.py
@@ -2,7 +2,7 @@
 
 from databricks.sdk.dbutils import MountInfo
 
-from databricks.labs.ucx.hive_metastore.list_mounts import Mount, Mounts
+from databricks.labs.ucx.hive_metastore.mounts import Mount, Mounts
 from tests.unit.framework.mocks import MockBackend