yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-31 10:39:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed Sample Data (#4162)

Browse Source
This commit is contained in:
Ayush Shah 2022-04-15 07:47:11 -07:00 committed by GitHub
parent 3ceea5e425
commit 076921cce2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ingestion/src/metadata/ingestion/source/sample_data.py
View File

@ -35,6 +35,7 @@ from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
from metadata.generated.schema.entity.data.location import Location, LocationType
from metadata.generated.schema.entity.data.pipeline import Pipeline, PipelineStatus
from metadata.generated.schema.entity.data.table import Table
from metadata.generated.schema.entity.policies.policy import Policy
from metadata.generated.schema.entity.services.connections.database.sampleDataConnection import (
SampleDataConnection,
)
@ -183,6 +184,7 @@ class SampleDataSource(Source[Entity]):
self.service_connection = config.serviceConnection.__root__.config
self.metadata_config = metadata_config
self.metadata = OpenMetadata(metadata_config)
self.list_policies = []
self.storage_service_json = json.load(
open(
@ -599,11 +601,15 @@ class SampleDataSource(Source[Entity]):
description=f"This is {user['teams']} description.",
)
]
if not self.list_policies:
self.list_policies = self.metadata.list_entities(entity=Policy)
role_ref_id = self.list_policies.entities[0].id.__root__
roles = (
[
CreateRoleRequest(
name=role,
description=f"This is {role} description.",
policies=[EntityReference(id=role_ref_id, type="policies")],
)
for role in user["roles"]
]