2021-11-18 23:30:25 -08:00
|
|
|
from typing import List
|
|
|
|
|
|
|
|
|
|
import datahub.emitter.mce_builder as builder
|
|
|
|
|
from datahub.emitter.mcp import MetadataChangeProposalWrapper
|
|
|
|
|
from datahub.emitter.rest_emitter import DatahubRestEmitter
|
|
|
|
|
from datahub.metadata.com.linkedin.pegasus2avro.dashboard import DashboardInfoClass
|
2022-12-30 01:26:47 -05:00
|
|
|
from datahub.metadata.schema_classes import ChangeAuditStampsClass
|
2021-11-18 23:30:25 -08:00
|
|
|
|
|
|
|
|
# Construct the DashboardInfo aspect with the charts -> dashboard lineage.
|
|
|
|
|
charts_in_dashboard: List[str] = [
|
|
|
|
|
builder.make_chart_urn(platform="looker", name="chart_1"),
|
|
|
|
|
builder.make_chart_urn(platform="looker", name="chart_2"),
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
last_modified = ChangeAuditStampsClass()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
dashboard_info = DashboardInfoClass(
|
|
|
|
|
title="My Dashboard 1",
|
|
|
|
|
description="Sample dashboard",
|
|
|
|
|
lastModified=last_modified,
|
|
|
|
|
charts=charts_in_dashboard,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Construct a MetadataChangeProposalWrapper object with the DashboardInfo aspect.
|
|
|
|
|
# NOTE: This will overwrite all of the existing dashboard aspect information associated with this dashboard.
|
|
|
|
|
chart_info_mcp = MetadataChangeProposalWrapper(
|
|
|
|
|
entityUrn=builder.make_dashboard_urn(platform="looker", name="my_dashboard_1"),
|
|
|
|
|
aspect=dashboard_info,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Create an emitter to the GMS REST API.
|
|
|
|
|
emitter = DatahubRestEmitter("http://localhost:8080")
|
|
|
|
|
|
|
|
|
|
# Emit metadata!
|
|
|
|
|
emitter.emit_mcp(chart_info_mcp)
|
