mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-03 15:17:58 +00:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from datahub.emitter.mce_builder import make_dataset_urn
|
||
|
|
|
||
|
|
# read-modify-write requires access to the DataHubGraph (RestEmitter is not enough)
|
||
|
|
from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
|
||
|
|
|
||
|
|
# Imports for metadata model classes
|
||
|
|
from datahub.metadata.schema_classes import (
|
||
|
|
DataPlatformInstanceClass,
|
||
|
|
DatasetKeyClass,
|
||
|
|
StatusClass,
|
||
|
|
)
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
|
||
|
|
dataset_urn = make_dataset_urn(platform="hive", name="realestate_db.sales", env="PROD")
|
||
|
|
|
||
|
|
gms_endpoint = "http://localhost:8080"
|
||
|
|
graph = DataHubGraph(DatahubClientConfig(server=gms_endpoint))
|
||
|
|
|
||
|
|
# Query multiple aspects from entity
|
||
|
|
result = graph.get_aspects_for_entity(
|
||
|
|
entity_urn=dataset_urn,
|
||
|
|
aspects=["status", "dataPlatformInstance", "datasetKey"],
|
||
|
|
aspect_types=[StatusClass, DataPlatformInstanceClass, DatasetKeyClass],
|
||
|
|
)
|
||
|
|
|
||
|
|
# result are typed according to their class if exist
|
||
|
|
if result is not None:
|
||
|
|
if result["datasetKey"]:
|
||
|
|
log.info(result["datasetKey"].name)
|