yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-03 03:59:12 +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 = [] unique_constraints = []
pk_columns = ( pk_columns = (
pk_constraints["column_constraints"] pk_constraints.get("constrained_columns")
if len(pk_constraints) > 0 and "column_constraints" in pk_constraints.keys() if len(pk_constraints) > 0 and pk_constraints.get("constrained_columns")
else {} else {}
) )
unique_columns = [ unique_columns = []
constraint["column_names"] for constraint in unique_constraints:
for constraint in unique_constraints if constraint.get("column_names"):
if "column_names" in constraint.keys() unique_columns.extend(constraint.get("column_names"))
]
table_columns = [] table_columns = []
columns = inspector.get_columns(table, schema) columns = inspector.get_columns(table, schema)