diff --git a/CosmoTech_Acceleration_Library/__init__.py b/CosmoTech_Acceleration_Library/__init__.py index 4715b5b3..f2400559 100644 --- a/CosmoTech_Acceleration_Library/__init__.py +++ b/CosmoTech_Acceleration_Library/__init__.py @@ -1,4 +1,4 @@ # Copyright (C) - 2023 - 2025 - Cosmo Tech # Licensed under the MIT license. -__version__ = '0.9.3' +__version__ = '0.9.4' diff --git a/cosmotech/coal/utils/postgresql.py b/cosmotech/coal/utils/postgresql.py index 7266a76a..38f6ea6e 100644 --- a/cosmotech/coal/utils/postgresql.py +++ b/cosmotech/coal/utils/postgresql.py @@ -8,6 +8,7 @@ from typing import Optional from urllib.parse import quote +import adbc_driver_manager import pyarrow as pa from adbc_driver_postgresql import dbapi from pyarrow import Table @@ -66,18 +67,11 @@ def get_postgresql_table_schema( with (dbapi.connect(postgresql_full_uri) as conn): try: - catalog = conn.adbc_get_objects(depth="tables", - catalog_filter=postgres_db, - db_schema_filter=postgres_schema, - table_name_filter=target_table_name).read_all().to_pylist()[0] - schema = catalog["catalog_db_schemas"][0] - table = schema["db_schema_tables"][0] - if table["table_name"] == target_table_name: - return conn.adbc_get_table_schema( - target_table_name, - db_schema_filter=postgres_schema, - ) - except IndexError: + return conn.adbc_get_table_schema( + target_table_name, + db_schema_filter=postgres_schema, + ) + except adbc_driver_manager.ProgrammingError: LOGGER.warning(f"Table {postgres_schema}.{target_table_name} not found") return None