refactor(ingest): Add support for group-owners in dataflow entities (#8154)

Co-authored-by: Dominik Schüssele <dominik.schuessele@inovex.de>
Co-authored-by: Andrew Sikowitz <andrew.sikowitz@acryl.io>
This commit is contained in:
dominik s 2023-09-06 21:12:14 +02:00 committed by GitHub
parent 94fa514700
commit 25148f4a65
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -65,6 +65,7 @@ class DataFlow:
url: Optional[str] = None
tags: Set[str] = field(default_factory=set)
owners: Set[str] = field(default_factory=set)
group_owners: Set[str] = field(default_factory=set)
platform_instance: Optional[str] = None
env: Optional[str] = None
@ -92,17 +93,20 @@ class DataFlow:
)
def generate_ownership_aspect(self):
owners = set([builder.make_user_urn(owner) for owner in self.owners]) | set(
[builder.make_group_urn(owner) for owner in self.group_owners]
)
ownership = OwnershipClass(
owners=[
OwnerClass(
owner=builder.make_user_urn(owner),
owner=urn,
type=OwnershipTypeClass.DEVELOPER,
source=OwnershipSourceClass(
type=OwnershipSourceTypeClass.SERVICE,
# url=dag.filepath,
),
)
for owner in (self.owners or [])
for urn in (owners or [])
],
lastModified=AuditStampClass(
time=0, actor=builder.make_user_urn(self.orchestrator)