mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-06 16:49:03 +00:00
23 lines
532 B
Python
23 lines
532 B
Python
from datahub.metadata.urns import TagUrn
|
|
from datahub.sdk import DataFlow, DataHubClient, DataJob
|
|
|
|
client = DataHubClient.from_env()
|
|
|
|
# datajob will inherit the platform and platform instance from the flow
|
|
|
|
dataflow = DataFlow(
|
|
platform="airflow",
|
|
name="example_dag",
|
|
platform_instance="PROD",
|
|
description="example dataflow",
|
|
tags=[TagUrn(name="tag1"), TagUrn(name="tag2")],
|
|
)
|
|
|
|
datajob = DataJob(
|
|
name="example_datajob",
|
|
flow=dataflow,
|
|
)
|
|
|
|
client.entities.upsert(dataflow)
|
|
client.entities.upsert(datajob)
|