mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-04 07:34:44 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from datahub.sdk import DataHubClient
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
# Initialize the client
|
||
|
|
client = DataHubClient(
|
||
|
|
server="https://your-datahub-cloud-instance.com", token="your-token"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Subscribe to all assertion changes for a dataset
|
||
|
|
client.subscriptions.subscribe(
|
||
|
|
urn="urn:li:dataset:(urn:li:dataPlatform:snowflake,purchases,PROD)",
|
||
|
|
subscriber_urn="urn:li:corpuser:john.doe",
|
||
|
|
# entity_change_types defaults to all available change types for datasets
|
||
|
|
)
|
||
|
|
log.info("Successfully subscribed to dataset notifications")
|
||
|
|
|
||
|
|
# Subscribe to specific assertion changes
|
||
|
|
client.subscriptions.subscribe(
|
||
|
|
urn="urn:li:assertion:your-assertion-id",
|
||
|
|
subscriber_urn="urn:li:corpuser:john.doe",
|
||
|
|
entity_change_types=["ASSERTION_PASSED", "ASSERTION_FAILED"],
|
||
|
|
)
|
||
|
|
log.info("Successfully subscribed to specific assertion changes")
|
||
|
|
|
||
|
|
# Subscribe a group to assertion changes
|
||
|
|
client.subscriptions.subscribe(
|
||
|
|
urn="urn:li:assertion:your-assertion-id",
|
||
|
|
subscriber_urn="urn:li:corpGroup:data-team",
|
||
|
|
entity_change_types=["ASSERTION_FAILED", "ASSERTION_ERROR"],
|
||
|
|
)
|
||
|
|
log.info("Successfully subscribed group to assertion failures and errors")
|