yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-22 17:01:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

MINOR: Fix unity catalog test connection (#15712)

Browse Source
This commit is contained in:
Mayur Singal 2024-03-27 15:14:26 +05:30 committed by GitHub
parent ec4673587d
commit 6acb47a87a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ingestion/src/metadata/ingestion/source/database/unitycatalog/connection.py
View File

@ -68,16 +68,20 @@ def test_connection(
break
def get_schemas(connection: WorkspaceClient, table_obj: DatabricksTable):
for schema in connection.schemas.list(catalog_name=table_obj.catalog_name):
table_obj.schema_name = schema.name
break
for catalog in connection.catalogs.list():
for schema in connection.schemas.list(catalog_name=catalog.name):
if schema.name:
table_obj.schema_name = schema.name
table_obj.catalog_name = catalog.name
return
def get_tables(connection: WorkspaceClient, table_obj: DatabricksTable):
for table in connection.tables.list(
catalog_name=table_obj.catalog_name, schema_name=table_obj.schema_name
):
table_obj.name = table.name
break
if table_obj.catalog_name and table_obj.schema_name:
for table in connection.tables.list(
catalog_name=table_obj.catalog_name, schema_name=table_obj.schema_name
):
table_obj.name = table.name
break
test_fn = {
"CheckAccess": connection.catalogs.list,