Update ischema_names for db2 to take BOOLEAN into account (#16718)

This commit is contained in:
IceS2 2024-06-19 14:03:02 +02:00 committed by GitHub
parent c79c5da00a
commit 6b7e9238f7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -12,10 +12,11 @@
import traceback import traceback
from typing import Optional 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 import reflection
from sqlalchemy.engine.reflection import Inspector from sqlalchemy.engine.reflection import Inspector
from sqlalchemy.engine.row import LegacyRow from sqlalchemy.engine.row import LegacyRow
from sqlalchemy.sql.sqltypes import BOOLEAN
from metadata.generated.schema.entity.services.connections.database.db2Connection import ( from metadata.generated.schema.entity.services.connections.database.db2Connection import (
Db2Connection, Db2Connection,
@ -40,6 +41,8 @@ def get_pk_constraint(
DB2Dialect.get_pk_constraint = get_pk_constraint DB2Dialect.get_pk_constraint = get_pk_constraint
ischema_names.update({"BOOLEAN": BOOLEAN})
class Db2Source(CommonDbSourceService): class Db2Source(CommonDbSourceService):
""" """