mirror of
https://github.com/datahub-project/datahub.git
synced 2025-06-27 05:03:31 +00:00
37 lines
1.2 KiB
Python
37 lines
1.2 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"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Unsubscribe from all changes for a dataset
|
||
|
|
client.subscriptions.unsubscribe(
|
||
|
|
urn="urn:li:dataset:(urn:li:dataPlatform:snowflake,purchases,PROD)",
|
||
|
|
subscriber_urn="urn:li:corpuser:john.doe",
|
||
|
|
# entity_change_types defaults to all existing change types
|
||
|
|
)
|
||
|
|
log.info("Successfully unsubscribed from all dataset notifications")
|
||
|
|
|
||
|
|
# Unsubscribe from specific assertion change types
|
||
|
|
client.subscriptions.unsubscribe(
|
||
|
|
urn="urn:li:assertion:your-assertion-id",
|
||
|
|
subscriber_urn="urn:li:corpuser:john.doe",
|
||
|
|
entity_change_types=[
|
||
|
|
"ASSERTION_PASSED"
|
||
|
|
], # Keep ASSERTION_FAILED and ASSERTION_ERROR
|
||
|
|
)
|
||
|
|
log.info("Successfully unsubscribed from specific assertion change types")
|
||
|
|
|
||
|
|
# Unsubscribe a group from assertion changes
|
||
|
|
client.subscriptions.unsubscribe(
|
||
|
|
urn="urn:li:assertion:your-assertion-id",
|
||
|
|
subscriber_urn="urn:li:corpGroup:data-team",
|
||
|
|
entity_change_types=["ASSERTION_FAILED", "ASSERTION_ERROR"],
|
||
|
|
)
|
||
|
|
log.info("Successfully unsubscribed group from assertion notifications")
|