Enable Primary Key Constraint for DB2 (#16755)

This commit is contained in:
IceS2 2024-06-21 09:45:43 +02:00 committed by GitHub
parent 9f6e3181c0
commit 7411f9e0e1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -12,8 +12,7 @@
import traceback
from typing import Optional
from ibm_db_sa.base import DB2Dialect, ischema_names
from sqlalchemy.engine import reflection
from ibm_db_sa.base import ischema_names
from sqlalchemy.engine.reflection import Inspector
from sqlalchemy.engine.row import LegacyRow
from sqlalchemy.sql.sqltypes import BOOLEAN
@ -32,15 +31,6 @@ from metadata.utils.logger import ingestion_logger
logger = ingestion_logger()
@reflection.cache
def get_pk_constraint(
self, bind, table_name, schema=None, **kw
): # pylint: disable=unused-argument
return {"constrained_columns": [], "name": "undefined"}
DB2Dialect.get_pk_constraint = get_pk_constraint
ischema_names.update({"BOOLEAN": BOOLEAN})