mirror of
https://github.com/datahub-project/datahub.git
synced 2025-10-31 18:59:23 +00:00
13 lines
443 B
Python
13 lines
443 B
Python
|
from datahub.metadata.urns import DataFlowUrn, DataJobUrn, DatasetUrn
|
||
|
|
from datahub.sdk.main_client import DataHubClient
|
||
|
|
|
||
|
|
client = DataHubClient.from_env()
|
||
|
|
|
||
|
|
dataset_urn = DatasetUrn(platform="snowflake", name="upstream_table")
|
||
|
|
datajob_urn = DataJobUrn(
|
||
|
|
job_id="example_datajob",
|
||
|
|
flow=DataFlowUrn(orchestrator="airflow", flow_id="example_dag", cluster="PROD"),
|
||
|
|
)
|
||
|
|
|
||
|
|
client.lineage.add_lineage(upstream=datajob_urn, downstream=dataset_urn)
|