yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-02 03:29:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix #2620: pk_columns & unique_columns fixed (#2621)

Browse Source 
Co-authored-by: Mayur SIngal <mayursingal@Mayurs-MacBook-Pro.local>
This commit is contained in:
Mayur Singal 2022-02-05 02:08:12 +05:30 committed by GitHub
parent 5a6e7d743e
commit 6128fa30dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ingestion/src/metadata/ingestion/source/sql_source.py
View File

@ -454,16 +454,15 @@ class SQLSource(Source[OMetaDatabaseAndTable]):
unique_constraints = []
pk_columns = (
pk_constraints["column_constraints"]
if len(pk_constraints) > 0 and "column_constraints" in pk_constraints.keys()
pk_constraints.get("constrained_columns")
if len(pk_constraints) > 0 and pk_constraints.get("constrained_columns")
else {}
)
unique_columns = [
constraint["column_names"]
for constraint in unique_constraints
if "column_names" in constraint.keys()
]
unique_columns = []
for constraint in unique_constraints:
if constraint.get("column_names"):
unique_columns.extend(constraint.get("column_names"))
table_columns = []
columns = inspector.get_columns(table, schema)