datahub_python_sdk_emit_dataset_metadata_rest_api.py

1import datahub.emitter.mce_builder as builder
2from datahub.emitter.rest_emitter import DatahubRestEmitter
3
4# Create an emitter to the GMS REST API.
5# Ensure you have DataHub running locally or update the GMS server URL.
6emitter = DatahubRestEmitter(gms_server="http://localhost:8080")
7
8# Construct a Metadata Change Event (MCE) for a dataset
9metadata_event = builder.make_dataset_description_update_event(
10    dataset_urn=builder.make_dataset_urn(platform="snowflake", name="orders", env="PROD"),
11    description="This table contains customer orders from the snowflake platform."
12)
13
14# Emit the metadata to DataHub
15try:
16    emitter.emit_mce(metadata_event)
17    print("Successfully emitted metadata to DataHub!")
18except Exception as e:
19    print(f"Failed to emit metadata: {e}")