mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-26 11:00:06 +00:00
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from datahub.api.entities.corpgroup.corpgroup import (
|
||
|
|
CorpGroup,
|
||
|
|
CorpGroupGenerationConfig,
|
||
|
|
)
|
||
|
|
from datahub.ingestion.graph.client import DataHubGraph, DataHubGraphConfig
|
||
|
|
from datahub.utilities.urns.corpuser_urn import CorpuserUrn
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
|
||
|
|
group_email = "foogroup@acryl.io"
|
||
|
|
group = CorpGroup(
|
||
|
|
id=group_email,
|
||
|
|
admins=[str(CorpuserUrn.create_from_id("datahub"))],
|
||
|
|
members=[
|
||
|
|
str(CorpuserUrn.create_from_id("bar@acryl.io")),
|
||
|
|
str(CorpuserUrn.create_from_id("joe@acryl.io")),
|
||
|
|
],
|
||
|
|
groups=[],
|
||
|
|
display_name="Foo Group",
|
||
|
|
email=group_email,
|
||
|
|
description="Software engineering team",
|
||
|
|
slack="@foogroup",
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create graph client
|
||
|
|
datahub_graph = DataHubGraph(DataHubGraphConfig(server="http://localhost:8080"))
|
||
|
|
|
||
|
|
for event in group.generate_mcp(
|
||
|
|
generation_config=CorpGroupGenerationConfig(
|
||
|
|
override_editable=False, datahub_graph=datahub_graph
|
||
|
|
)
|
||
|
|
):
|
||
|
|
datahub_graph.emit(event)
|
||
|
|
log.info(f"Upserted group {group.urn}")
|