From 6b7e9238f789b9abef1e8eb2fd917d94e08f1f22 Mon Sep 17 00:00:00 2001 From: IceS2 Date: Wed, 19 Jun 2024 14:03:02 +0200 Subject: [PATCH] Update ischema_names for db2 to take BOOLEAN into account (#16718) --- .../src/metadata/ingestion/source/database/db2/metadata.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/ingestion/source/database/db2/metadata.py b/ingestion/src/metadata/ingestion/source/database/db2/metadata.py index 3a41d7f06fa..cc1a562c170 100644 --- a/ingestion/src/metadata/ingestion/source/database/db2/metadata.py +++ b/ingestion/src/metadata/ingestion/source/database/db2/metadata.py @@ -12,10 +12,11 @@ import traceback from typing import Optional -from ibm_db_sa.base import DB2Dialect +from ibm_db_sa.base import DB2Dialect, ischema_names from sqlalchemy.engine import reflection from sqlalchemy.engine.reflection import Inspector from sqlalchemy.engine.row import LegacyRow +from sqlalchemy.sql.sqltypes import BOOLEAN from metadata.generated.schema.entity.services.connections.database.db2Connection import ( Db2Connection, @@ -40,6 +41,8 @@ def get_pk_constraint( DB2Dialect.get_pk_constraint = get_pk_constraint +ischema_names.update({"BOOLEAN": BOOLEAN}) + class Db2Source(CommonDbSourceService): """