yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-24 17:59:52 +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.location import Location, LocationType
from metadata.generated.schema.entity.data.pipeline import Pipeline, PipelineStatus from metadata.generated.schema.entity.data.pipeline import Pipeline, PipelineStatus
from metadata.generated.schema.entity.data.table import Table 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 ( from metadata.generated.schema.entity.services.connections.database.sampleDataConnection import (
SampleDataConnection, SampleDataConnection,
) )
@ -183,6 +184,7 @@ class SampleDataSource(Source[Entity]):
self.service_connection = config.serviceConnection.__root__.config self.service_connection = config.serviceConnection.__root__.config
self.metadata_config = metadata_config self.metadata_config = metadata_config
self.metadata = OpenMetadata(metadata_config) self.metadata = OpenMetadata(metadata_config)
self.list_policies = []
self.storage_service_json = json.load( self.storage_service_json = json.load(
open( open(
@ -599,11 +601,15 @@ class SampleDataSource(Source[Entity]):
description=f"This is {user['teams']} description.", 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 = ( roles = (
[ [
CreateRoleRequest( CreateRoleRequest(
name=role, name=role,
description=f"This is {role} description.", description=f"This is {role} description.",
policies=[EntityReference(id=role_ref_id, type="policies")],
) )
for role in user["roles"] for role in user["roles"]
] ]