late-notebook-97260
05/09/2023, 4:58 AMfrom datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
from datahub.metadata.schema_classes import SchemaMetadataClass
from datahub.metadata.schema_classes import UpstreamLineageClass
table_urn = "urn:li:dataset:(urn:li:dataPlatform:looker,Revenue_SOX.explore.payments_promos_SOX,PROD)"
graph = DataHubGraph(DatahubClientConfig(server=gms_server , token= datahub_gms_api_token))
current_schema_metadata = graph.get_aspect(
entity_urn=table_urn,
aspect_type=UpstreamLineageClass)
lively-cat-88289
05/09/2023, 4:58 AMmammoth-bear-12532
from datahub.ingestion.graph.client import get_default_graph, DataHubGraph
import sys
if __name__ == "__main__":
dataset_urn = sys.argv[1]
graph: DataHubGraph
with get_default_graph() as graph:
print("Downstream entities")
for entity in graph.get_related_entities(dataset_urn, relationship_types=["DownstreamOf"], direction=DataHubGraph.RelationshipDirection.INCOMING):
print(entity)
this should worklate-notebook-97260
05/10/2023, 5:54 AM