yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-11-07 23:13:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ingest): support multiple entity_registry sections (#7066)

Browse Source
This commit is contained in:
Harshal Sheth 2023-01-18 01:42:08 -08:00 committed by GitHub
parent a3f4c40422
commit f39ed5244c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
metadata-ingestion/scripts/avro_codegen.py
View File

@ -272,7 +272,13 @@ def generate(entity_registry: str, schemas_path: str, outdir: str) -> None:
aspect = aspects[entity.keyAspect]
# This requires that entities cannot share a keyAspect.
assert "keyForEntity" not in aspect["Aspect"]
if (
"keyForEntity" in aspect["Aspect"]
and aspect["Aspect"]["keyForEntity"] != entity.name
):
raise ValueError(
f'Entity key {entity.keyAspect} is used by {aspect["Aspect"]["keyForEntity"]} and {entity.name}'
)
aspect["Aspect"]["keyForEntity"] = entity.name
aspect["Aspect"]["entityCategory"] = entity.category