mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-03 23:28:11 +00:00
24 lines
754 B
Python
24 lines
754 B
Python
|
import os
|
||
|
|
from datahub.api.entities.dataproduct.dataproduct import DataProduct
|
||
|
|
from datahub.ingestion.source.metadata.business_glossary import BusinessGlossaryConfig
|
||
|
|
import click
|
||
|
|
|
||
|
|
@click.command()
|
||
|
|
@click.option("--out-dir", type=str, required=True)
|
||
|
|
def generate_specs(
|
||
|
|
out_dir: str
|
||
|
|
) -> None:
|
||
|
|
print(out_dir)
|
||
|
|
schemas_dir = f"{out_dir}/schemas"
|
||
|
|
os.makedirs(schemas_dir, exist_ok=True)
|
||
|
|
concept_class_map = {
|
||
|
|
"dataproduct": DataProduct,
|
||
|
|
"businessglossary": BusinessGlossaryConfig
|
||
|
|
}
|
||
|
|
for concept, concept_class in concept_class_map.items():
|
||
|
|
with open(f"{schemas_dir}/{concept}_schema.json", "w") as f:
|
||
|
|
f.write(concept_class.schema_json(indent=2))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
generate_specs()
|