MCP Builder

These classes and methods make it easier to construct MetadataChangeProposals.

MetadataChangeProposalWrapper

class datahub.emitter.mcp.MetadataChangeProposalWrapper(entityType = 'ENTITY_TYPE_UNSET', changeType = 'UPSERT', entityUrn = None, entityKeyAspect = None, auditHeader = None, aspectName = None, aspect = None, systemMetadata = None, headers = None)

Bases: object

• Parameters:
  • entityType (str)
  • changeType (Union[str, ChangeTypeClass]) –
  • entityUrn (Optional[str])
  • entityKeyAspect (Optional[_Aspect])
  • auditHeader (Optional[KafkaAuditHeaderClass]) –
  • aspectName (Optional[str])
  • aspect (Optional[_Aspect])
  • systemMetadata (Optional[SystemMetadataClass]) –
  • headers (Optional[Dict[str, str]])

as_workunit(*, treat_errors_as_warnings=False, is_primary_source=True)

• Parameters:
  • treat_errors_as_warnings (bool)
  • is_primary_source (bool)
• Return type:MetadataWorkUnit

aspect : Optional[_Aspect] = None

aspectName : Optional[str] = None

auditHeader : Optional[KafkaAuditHeaderClass] = None

changeType : Union[str, ChangeTypeClass] = 'UPSERT'

classmethod construct_many(entityUrn, aspects)

• Parameters:
  • entityUrn (str)
  • aspects (Sequence[Optional[_Aspect]])
• Return type:List[MetadataChangeProposalWrapper]

entityKeyAspect : Optional[_Aspect] = None

entityType : str = 'ENTITY_TYPE_UNSET'

entityUrn : Optional[str] = None

classmethod from_obj(obj, tuples=False)

Attempt to deserialize into an MCPW, but fall back to a standard MCP if we’re missing codegen’d classes for the entity key or aspect.

• Parameters:
  • obj (dict)
  • tuples (bool)
• Return type:Union[MetadataChangeProposalWrapper, MetadataChangeProposalClass]

classmethod from_obj_require_wrapper(obj, tuples=False)

• Parameters:
  • obj (dict)
  • tuples (bool)
• Return type:MetadataChangeProposalWrapper

headers : Optional[Dict[str, str]] = None

make_mcp()

• Return type:MetadataChangeProposalClass

systemMetadata : Optional[SystemMetadataClass] = None

to_obj(tuples=False, simplified_structure=False)

• Parameters:
  • tuples (bool)
  • simplified_structure (bool)
• Return type:dict

classmethod try_from_mcl(mcl)

• Parameters:mcl (MetadataChangeLogClass) –
• Return type:Union[MetadataChangeProposalWrapper, MetadataChangeProposalClass]

classmethod try_from_mcpc(mcpc)

Attempts to create a MetadataChangeProposalWrapper from a MetadataChangeProposalClass. Neatly handles unsupported, expected cases, such as unknown aspect types or non-json content type.

• Raises:Exception if the generic aspect is invalid**,** e.g. contains invalid json. –
• Parameters:mcpc (MetadataChangeProposalClass) –
• Return type:Optional[MetadataChangeProposalWrapper]

validate()

• Return type:bool

BigQueryDatasetKey

class datahub.emitter.mcp_builder.BigQueryDatasetKey(**data)

Bases: ProjectIdKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • project_id (str)
  • dataset_id (str)

dataset_id : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

BucketKey

class datahub.emitter.mcp_builder.BucketKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • bucket_name (str)

bucket_name : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

CatalogKey

class datahub.emitter.mcp_builder.CatalogKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • catalog (str)

catalog : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

CatalogKeyWithMetastore

class datahub.emitter.mcp_builder.CatalogKeyWithMetastore(**data)

Bases: MetastoreKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • metastore (str)
  • catalog (str)

catalog : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

ContainerKey

class datahub.emitter.mcp_builder.ContainerKey(**data)

Bases: DatahubKey

Base class for container guid keys. Most users should use one of the subclasses instead.

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)

as_urn()

• Return type:str

as_urn_typed()

• Return type:ContainerUrn

backcompat_env_as_instance : bool

env : Optional[str]

guid_dict()

• Return type:Dict[str, str]

instance : Optional[str]

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

parent_key()

• Return type:Optional[ContainerKey]

platform : str

property_dict()

• Return type:Dict[str, str]

DatabaseKey

class datahub.emitter.mcp_builder.DatabaseKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • database (str)

database : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

DatahubKey

class datahub.emitter.mcp_builder.DatahubKey(**data)

Bases: BaseModel

• Parameters:data (Any)

guid()

• Return type:str

guid_dict()

• Return type:Dict[str, str]

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

ExperimentKey

class datahub.emitter.mcp_builder.ExperimentKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • id (str)

id : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

FolderKey

class datahub.emitter.mcp_builder.FolderKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • folder_abs_path (str)

folder_abs_path : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

MetastoreKey

class datahub.emitter.mcp_builder.MetastoreKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • metastore (str)

metastore : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

NamespaceKey

class datahub.emitter.mcp_builder.NamespaceKey(**data)

Bases: ContainerKey

For Iceberg namespaces (databases/schemas)

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • namespace (str)

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

namespace : str

NotebookKey

class datahub.emitter.mcp_builder.NotebookKey(**data)

Bases: DatahubKey

• Parameters:
  • data (Any)
  • notebook_id (int)
  • platform (str)
  • instance (str | None)

as_urn()

• Return type:str

instance : Optional[str]

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

notebook_id : int

platform : str

PlatformKey

datahub.emitter.mcp_builder.PlatformKey()

alias of ContainerKey

ProjectIdKey

class datahub.emitter.mcp_builder.ProjectIdKey(**data)

Bases: ContainerKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • project_id (str)

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

project_id : str

SchemaKey

class datahub.emitter.mcp_builder.SchemaKey(**data)

Bases: DatabaseKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • database (str)
  • schema (str)

db_schema : str

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

UnitySchemaKey

class datahub.emitter.mcp_builder.UnitySchemaKey(**data)

Bases: CatalogKey

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • catalog (str)
  • unity_schema (str)

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

unity_schema : str

UnitySchemaKeyWithMetastore

class datahub.emitter.mcp_builder.UnitySchemaKeyWithMetastore(**data)

Bases: CatalogKeyWithMetastore

• Parameters:
  • data (Any)
  • platform (str)
  • instance (str | None)
  • env (str | None)
  • backcompat_env_as_instance (bool)
  • metastore (str)
  • catalog (str)
  • unity_schema (str)

model_config : ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

unity_schema : str

add_dataset_to_container

datahub.emitter.mcp_builder.add_dataset_to_container(container_key, dataset_urn)

• Parameters:
  • container_key (TypeVar(KeyType, bound= ContainerKey)) –
  • dataset_urn (str)
• Return type:Iterable[MetadataWorkUnit]

add_domain_to_entity_wu

datahub.emitter.mcp_builder.add_domain_to_entity_wu(entity_urn, domain_urn)

• Parameters:
  • entity_urn (str)
  • domain_urn (str)
• Return type:Iterable[MetadataWorkUnit]

add_entity_to_container

datahub.emitter.mcp_builder.add_entity_to_container(container_key, entity_type, entity_urn)

• Parameters:
  • container_key (TypeVar(KeyType, bound= ContainerKey)) –
  • entity_type (str)
  • entity_urn (str)
• Return type:Iterable[MetadataWorkUnit]

add_owner_to_entity_wu

datahub.emitter.mcp_builder.add_owner_to_entity_wu(entity_type, entity_urn, owner_urn)

• Parameters:
  • entity_type (str)
  • entity_urn (str)
  • owner_urn (str)
• Return type:Iterable[MetadataWorkUnit]

add_structured_properties_to_entity_wu

datahub.emitter.mcp_builder.add_structured_properties_to_entity_wu(entity_urn, structured_properties)

• Parameters:
  • entity_urn (str)
  • structured_properties (Dict[StructuredPropertyUrn, str]) –
• Return type:Iterable[MetadataWorkUnit]

add_tags_to_entity_wu

datahub.emitter.mcp_builder.add_tags_to_entity_wu(entity_type, entity_urn, tags)

• Parameters:
  • entity_type (str)
  • entity_urn (str)
  • tags (List[str])
• Return type:Iterable[MetadataWorkUnit]

create_embed_mcp

datahub.emitter.mcp_builder.create_embed_mcp(urn, embed_url)

• Parameters:
  • urn (str)
  • embed_url (str)
• Return type:MetadataChangeProposalWrapper

entity_supports_aspect

datahub.emitter.mcp_builder.entity_supports_aspect(entity_type, aspect_type)

• Parameters:
  • entity_type (str)
  • aspect_type (Type[TypeVar(Aspect, bound= _Aspect)]) –
• Return type:bool

gen_containers

datahub.emitter.mcp_builder.gen_containers(container_key, name, sub_types, parent_container_key = None, extra_properties = None, structured_properties = None, domain_urn = None, description = None, owner_urn = None, external_url = None, tags = None, qualified_name = None, created = None, last_modified = None)

• Parameters:
  • container_key (TypeVar(KeyType, bound= ContainerKey)) –
  • name (str)
  • sub_types (List[str])
  • parent_container_key (Optional[ContainerKey]) –
  • extra_properties (Optional[Dict[str, str]])
  • structured_properties (Optional[Dict[StructuredPropertyUrn, str]]) –
  • domain_urn (Optional[str])
  • description (Optional[str])
  • owner_urn (Optional[str])
  • external_url (Optional[str])
  • tags (Optional[List[str]])
  • qualified_name (Optional[str])
  • created (Optional[int])
  • last_modified (Optional[int])
• Return type:Iterable[MetadataWorkUnit]

mcps_from_mce

datahub.emitter.mcp_builder.mcps_from_mce(mce)

• Parameters:mce (MetadataChangeEventClass) –
• Return type:Iterable[MetadataChangeProposalWrapper]