openmetadata-ingestion 1.2.5.2__py3-none-any.whl → 1.3.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of openmetadata-ingestion might be problematic. Click here for more details.
- airflow_provider_openmetadata/lineage/runner.py +44 -5
- airflow_provider_openmetadata/lineage/status.py +1 -1
- metadata/antlr/split_listener.py +1 -4
- metadata/applications/auto_tagger.py +212 -0
- metadata/cli/app.py +47 -0
- metadata/cli/db_dump.py +6 -3
- metadata/clients/aws_client.py +4 -0
- metadata/clients/domo_client.py +0 -2
- metadata/cmd.py +24 -13
- metadata/data_insight/processor/reports/entity_report_data_processor.py +8 -2
- metadata/data_insight/processor/reports/web_analytic_report_data_processor.py +4 -0
- metadata/data_insight/source/metadata.py +11 -4
- metadata/data_quality/interface/sqlalchemy/databricks/test_suite_interface.py +29 -0
- metadata/data_quality/interface/sqlalchemy/snowflake/test_suite_interface.py +29 -0
- metadata/data_quality/interface/sqlalchemy/sqa_test_suite_interface.py +6 -6
- metadata/data_quality/interface/sqlalchemy/unity_catalog/test_suite_interface.py +35 -0
- metadata/data_quality/interface/test_suite_interface_factory.py +40 -0
- metadata/data_quality/processor/test_case_runner.py +17 -8
- metadata/data_quality/source/test_suite.py +8 -2
- metadata/data_quality/validations/base_test_handler.py +21 -17
- metadata/data_quality/validations/column/base/columnValueLengthsToBeBetween.py +38 -1
- metadata/data_quality/validations/column/base/columnValuesToBeBetween.py +36 -1
- metadata/data_quality/validations/column/base/columnValuesToBeInSet.py +27 -0
- metadata/data_quality/validations/column/base/columnValuesToBeNotInSet.py +27 -0
- metadata/data_quality/validations/column/base/columnValuesToBeNotNull.py +27 -0
- metadata/data_quality/validations/column/base/columnValuesToBeUnique.py +15 -0
- metadata/data_quality/validations/column/base/columnValuesToMatchRegex.py +27 -0
- metadata/data_quality/validations/column/base/columnValuesToNotMatchRegex.py +27 -0
- metadata/data_quality/validations/column/pandas/columnValueLengthsToBeBetween.py +23 -0
- metadata/data_quality/validations/column/pandas/columnValuesToBeBetween.py +23 -0
- metadata/data_quality/validations/column/pandas/columnValuesToBeInSet.py +11 -0
- metadata/data_quality/validations/column/pandas/columnValuesToBeNotInSet.py +11 -0
- metadata/data_quality/validations/column/pandas/columnValuesToBeNotNull.py +11 -0
- metadata/data_quality/validations/column/pandas/columnValuesToMatchRegex.py +11 -0
- metadata/data_quality/validations/column/pandas/columnValuesToNotMatchRegex.py +11 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValueLengthsToBeBetween.py +27 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToBeBetween.py +23 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToBeInSet.py +11 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToBeNotInSet.py +11 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToBeNotNull.py +11 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToMatchRegex.py +11 -0
- metadata/data_quality/validations/column/sqlalchemy/columnValuesToNotMatchRegex.py +11 -0
- metadata/data_quality/validations/mixins/pandas_validator_mixin.py +13 -2
- metadata/data_quality/validations/mixins/sqa_validator_mixin.py +40 -0
- metadata/examples/workflows/databricks.yaml +0 -1
- metadata/examples/workflows/mongodb.yaml +4 -2
- metadata/examples/workflows/mstr.yaml +24 -0
- metadata/examples/workflows/sas.yaml +28 -0
- metadata/examples/workflows/unity_catalog.yaml +27 -0
- metadata/examples/workflows/unity_catalog_lineage.yaml +18 -0
- metadata/examples/workflows/unity_catalog_usage.yaml +35 -0
- metadata/generated/antlr/EntityLinkLexer.py +391 -292
- metadata/generated/antlr/EntityLinkListener.py +12 -12
- metadata/generated/antlr/EntityLinkParser.py +228 -103
- metadata/generated/schema/analytics/__init__.py +1 -1
- metadata/generated/schema/analytics/basic.py +1 -1
- metadata/generated/schema/analytics/reportData.py +1 -1
- metadata/generated/schema/analytics/reportDataType/__init__.py +1 -1
- metadata/generated/schema/analytics/reportDataType/aggregatedCostAnalysisReportData.py +1 -1
- metadata/generated/schema/analytics/reportDataType/entityReportData.py +1 -1
- metadata/generated/schema/analytics/reportDataType/rawCostAnalysisReportData.py +1 -1
- metadata/generated/schema/analytics/reportDataType/webAnalyticEntityViewReportData.py +1 -1
- metadata/generated/schema/analytics/reportDataType/webAnalyticUserActivityReportData.py +1 -1
- metadata/generated/schema/analytics/webAnalyticEvent.py +1 -1
- metadata/generated/schema/analytics/webAnalyticEventData.py +1 -1
- metadata/generated/schema/analytics/webAnalyticEventType/__init__.py +1 -1
- metadata/generated/schema/analytics/webAnalyticEventType/customEvent.py +1 -1
- metadata/generated/schema/analytics/webAnalyticEventType/pageViewEvent.py +1 -1
- metadata/generated/schema/api/__init__.py +1 -1
- metadata/generated/schema/api/addGlossaryToAssetsRequest.py +1 -1
- metadata/generated/schema/api/analytics/__init__.py +1 -1
- metadata/generated/schema/api/analytics/createWebAnalyticEvent.py +1 -1
- metadata/generated/schema/api/automations/__init__.py +1 -1
- metadata/generated/schema/api/automations/createWorkflow.py +1 -1
- metadata/generated/schema/api/bulkAssets.py +1 -1
- metadata/generated/schema/api/classification/__init__.py +1 -1
- metadata/generated/schema/api/classification/createClassification.py +1 -1
- metadata/generated/schema/api/classification/createTag.py +1 -1
- metadata/generated/schema/api/classification/loadTags.py +1 -1
- metadata/generated/schema/api/createBot.py +2 -3
- metadata/generated/schema/api/createEventPublisherJob.py +1 -1
- metadata/generated/schema/api/createType.py +1 -1
- metadata/generated/schema/api/data/__init__.py +1 -1
- metadata/generated/schema/api/data/createChart.py +1 -1
- metadata/generated/schema/api/data/createContainer.py +2 -2
- metadata/generated/schema/api/data/createCustomProperty.py +2 -3
- metadata/generated/schema/api/data/createDashboard.py +1 -1
- metadata/generated/schema/api/data/createDashboardDataModel.py +1 -1
- metadata/generated/schema/api/data/createDatabase.py +2 -3
- metadata/generated/schema/api/data/createDatabaseSchema.py +2 -3
- metadata/generated/schema/api/data/createGlossary.py +2 -3
- metadata/generated/schema/api/data/createGlossaryTerm.py +2 -3
- metadata/generated/schema/api/data/createMlModel.py +1 -1
- metadata/generated/schema/api/data/createPipeline.py +2 -2
- metadata/generated/schema/api/data/createQuery.py +2 -2
- metadata/generated/schema/api/data/createSearchIndex.py +1 -1
- metadata/generated/schema/api/data/createStoredProcedure.py +2 -4
- metadata/generated/schema/api/data/createTable.py +2 -2
- metadata/generated/schema/api/data/createTableProfile.py +1 -1
- metadata/generated/schema/api/data/createTopic.py +1 -1
- metadata/generated/schema/api/data/loadGlossary.py +1 -1
- metadata/generated/schema/api/data/restoreEntity.py +1 -1
- metadata/generated/schema/api/dataInsight/__init__.py +1 -1
- metadata/generated/schema/api/dataInsight/createDataInsightChart.py +1 -1
- metadata/generated/schema/api/dataInsight/kpi/__init__.py +1 -1
- metadata/generated/schema/api/dataInsight/kpi/createKpiRequest.py +1 -1
- metadata/generated/schema/api/docStore/__init__.py +1 -1
- metadata/generated/schema/api/docStore/createDocument.py +1 -1
- metadata/generated/schema/api/domains/__init__.py +1 -1
- metadata/generated/schema/api/domains/createDataProduct.py +1 -1
- metadata/generated/schema/api/domains/createDomain.py +1 -1
- metadata/generated/schema/api/feed/__init__.py +1 -1
- metadata/generated/schema/api/feed/closeTask.py +8 -1
- metadata/generated/schema/api/feed/createPost.py +1 -1
- metadata/generated/schema/api/feed/createSuggestion.py +30 -0
- metadata/generated/schema/api/feed/createThread.py +1 -1
- metadata/generated/schema/api/feed/resolveTask.py +12 -1
- metadata/generated/schema/api/feed/threadCount.py +18 -14
- metadata/generated/schema/api/lineage/__init__.py +1 -1
- metadata/generated/schema/api/lineage/addLineage.py +1 -1
- metadata/generated/schema/api/openMetadataServerVersion.py +1 -1
- metadata/generated/schema/api/policies/__init__.py +1 -1
- metadata/generated/schema/api/policies/createPolicy.py +1 -1
- metadata/generated/schema/api/services/__init__.py +1 -1
- metadata/generated/schema/api/services/createDashboardService.py +1 -1
- metadata/generated/schema/api/services/createDatabaseService.py +1 -1
- metadata/generated/schema/api/services/createMessagingService.py +1 -1
- metadata/generated/schema/api/services/createMetadataService.py +1 -1
- metadata/generated/schema/api/services/createMlModelService.py +1 -1
- metadata/generated/schema/api/services/createPipelineService.py +1 -1
- metadata/generated/schema/api/services/createSearchService.py +1 -1
- metadata/generated/schema/api/services/createStorageService.py +1 -1
- metadata/generated/schema/api/services/ingestionPipelines/__init__.py +1 -1
- metadata/generated/schema/api/services/ingestionPipelines/createIngestionPipeline.py +1 -1
- metadata/generated/schema/api/setOwner.py +1 -1
- metadata/generated/schema/api/teams/__init__.py +1 -1
- metadata/generated/schema/api/teams/createPersona.py +1 -1
- metadata/generated/schema/api/teams/createRole.py +2 -3
- metadata/generated/schema/api/teams/createTeam.py +1 -1
- metadata/generated/schema/api/teams/createUser.py +2 -2
- metadata/generated/schema/api/tests/__init__.py +1 -1
- metadata/generated/schema/api/tests/createCustomMetric.py +4 -2
- metadata/generated/schema/api/tests/createLogicalTestCases.py +1 -1
- metadata/generated/schema/api/tests/createTestCase.py +4 -1
- metadata/generated/schema/api/tests/createTestCaseResolutionStatus.py +31 -0
- metadata/generated/schema/api/tests/createTestDefinition.py +1 -1
- metadata/generated/schema/api/tests/createTestSuite.py +1 -1
- metadata/generated/schema/api/voteRequest.py +1 -1
- metadata/generated/schema/auth/__init__.py +1 -1
- metadata/generated/schema/auth/basicAuth.py +1 -1
- metadata/generated/schema/auth/basicLoginRequest.py +1 -1
- metadata/generated/schema/auth/changePasswordRequest.py +1 -1
- metadata/generated/schema/auth/createPersonalToken.py +1 -1
- metadata/generated/schema/auth/emailRequest.py +1 -1
- metadata/generated/schema/auth/emailVerificationToken.py +1 -1
- metadata/generated/schema/auth/generateToken.py +1 -1
- metadata/generated/schema/auth/jwtAuth.py +1 -1
- metadata/generated/schema/auth/loginRequest.py +1 -1
- metadata/generated/schema/auth/logoutRequest.py +1 -1
- metadata/generated/schema/auth/passwordResetRequest.py +3 -3
- metadata/generated/schema/auth/passwordResetToken.py +1 -1
- metadata/generated/schema/auth/personalAccessToken.py +1 -1
- metadata/generated/schema/auth/refreshToken.py +1 -1
- metadata/generated/schema/auth/registrationRequest.py +2 -2
- metadata/generated/schema/auth/revokePersonalToken.py +1 -1
- metadata/generated/schema/auth/revokeToken.py +1 -1
- metadata/generated/schema/auth/serviceTokenEnum.py +1 -1
- metadata/generated/schema/auth/ssoAuth.py +1 -1
- metadata/generated/schema/auth/tokenRefreshRequest.py +1 -1
- metadata/generated/schema/configuration/__init__.py +1 -1
- metadata/generated/schema/configuration/appsPrivateConfiguration.py +28 -0
- metadata/generated/schema/configuration/authConfig.py +1 -1
- metadata/generated/schema/configuration/authenticationConfiguration.py +1 -1
- metadata/generated/schema/configuration/authorizerConfiguration.py +1 -1
- metadata/generated/schema/configuration/changeEventConfiguration.py +1 -1
- metadata/generated/schema/configuration/elasticSearchConfiguration.py +7 -1
- metadata/generated/schema/configuration/eventHandlerConfiguration.py +1 -1
- metadata/generated/schema/configuration/fernetConfiguration.py +1 -1
- metadata/generated/schema/configuration/jwtTokenConfiguration.py +1 -1
- metadata/generated/schema/configuration/kafkaEventConfiguration.py +1 -1
- metadata/generated/schema/configuration/ldapConfiguration.py +28 -2
- metadata/generated/schema/configuration/ldapTrustStoreConfig/__init__.py +1 -1
- metadata/generated/schema/configuration/ldapTrustStoreConfig/customTrustManagerConfig.py +1 -1
- metadata/generated/schema/configuration/ldapTrustStoreConfig/hostNameConfig.py +1 -1
- metadata/generated/schema/configuration/ldapTrustStoreConfig/jvmDefaultConfig.py +1 -1
- metadata/generated/schema/configuration/ldapTrustStoreConfig/trustAllConfig.py +1 -1
- metadata/generated/schema/configuration/ldapTrustStoreConfig/truststoreConfig.py +1 -1
- metadata/generated/schema/configuration/loginConfiguration.py +1 -1
- metadata/generated/schema/configuration/logoConfiguration.py +1 -1
- metadata/generated/schema/configuration/pipelineServiceClientConfiguration.py +1 -1
- metadata/generated/schema/configuration/slackAppConfiguration.py +1 -1
- metadata/generated/schema/configuration/taskNotificationConfiguration.py +1 -1
- metadata/generated/schema/configuration/testResultNotificationConfiguration.py +1 -1
- metadata/generated/schema/dataInsight/__init__.py +1 -1
- metadata/generated/schema/dataInsight/dataInsightChart.py +1 -1
- metadata/generated/schema/dataInsight/dataInsightChartResult.py +1 -1
- metadata/generated/schema/dataInsight/kpi/__init__.py +1 -1
- metadata/generated/schema/dataInsight/kpi/basic.py +1 -1
- metadata/generated/schema/dataInsight/kpi/kpi.py +1 -1
- metadata/generated/schema/dataInsight/type/__init__.py +1 -1
- metadata/generated/schema/dataInsight/type/aggregatedUnusedAssetsCount.py +1 -1
- metadata/generated/schema/dataInsight/type/aggregatedUnusedAssetsSize.py +1 -1
- metadata/generated/schema/dataInsight/type/aggregatedUsedVsUnusedAssetsCount.py +1 -1
- metadata/generated/schema/dataInsight/type/aggregatedUsedVsUnusedAssetsSize.py +1 -1
- metadata/generated/schema/dataInsight/type/dailyActiveUsers.py +1 -1
- metadata/generated/schema/dataInsight/type/mostActiveUsers.py +1 -1
- metadata/generated/schema/dataInsight/type/mostViewedEntities.py +1 -1
- metadata/generated/schema/dataInsight/type/pageViewsByEntities.py +1 -1
- metadata/generated/schema/dataInsight/type/percentageOfEntitiesWithDescriptionByType.py +1 -1
- metadata/generated/schema/dataInsight/type/percentageOfEntitiesWithOwnerByType.py +1 -1
- metadata/generated/schema/dataInsight/type/percentageOfServicesWithDescription.py +1 -1
- metadata/generated/schema/dataInsight/type/percentageOfServicesWithOwner.py +1 -1
- metadata/generated/schema/dataInsight/type/totalEntitiesByTier.py +1 -1
- metadata/generated/schema/dataInsight/type/totalEntitiesByType.py +1 -1
- metadata/generated/schema/dataInsight/type/unusedAssets.py +1 -1
- metadata/generated/schema/email/__init__.py +1 -1
- metadata/generated/schema/email/emailRequest.py +1 -1
- metadata/generated/schema/email/smtpSettings.py +1 -1
- metadata/generated/schema/entity/__init__.py +1 -1
- metadata/generated/schema/entity/applications/__init__.py +1 -1
- metadata/generated/schema/entity/applications/app.py +18 -4
- metadata/generated/schema/entity/applications/appRunRecord.py +1 -1
- metadata/generated/schema/entity/applications/configuration/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/applicationConfig.py +30 -0
- metadata/generated/schema/entity/{services/connections/database/mongoDB → applications/configuration/external}/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/autoTaggerAppConfig.py +28 -0
- metadata/generated/schema/entity/applications/configuration/external/metaPilotAppConfig.py +46 -0
- metadata/generated/schema/entity/applications/configuration/internal/__init__.py +3 -0
- metadata/generated/schema/entity/applications/configuration/internal/dataInsightsAppConfig.py +14 -0
- metadata/generated/schema/entity/applications/configuration/internal/dataInsightsReportAppConfig.py +27 -0
- metadata/generated/schema/entity/applications/configuration/{searchIndexingApp.py → internal/searchIndexingAppConfig.py} +13 -3
- metadata/generated/schema/entity/applications/configuration/private/external/__init__.py +3 -0
- metadata/generated/schema/entity/applications/configuration/private/external/metaPilotAppPrivateConfig.py +26 -0
- metadata/generated/schema/entity/applications/createAppRequest.py +1 -1
- metadata/generated/schema/entity/applications/jobStatus.py +3 -3
- metadata/generated/schema/entity/applications/liveExecutionContext.py +1 -1
- metadata/generated/schema/entity/applications/marketplace/__init__.py +1 -1
- metadata/generated/schema/entity/applications/marketplace/appMarketPlaceDefinition.py +12 -3
- metadata/generated/schema/entity/applications/marketplace/createAppMarketPlaceDefinitionReq.py +12 -3
- metadata/generated/schema/entity/applications/scheduledExecutionContext.py +1 -1
- metadata/generated/schema/entity/automations/__init__.py +1 -1
- metadata/generated/schema/entity/automations/testServiceConnection.py +2 -2
- metadata/generated/schema/entity/automations/workflow.py +1 -1
- metadata/generated/schema/entity/bot.py +2 -3
- metadata/generated/schema/entity/classification/__init__.py +1 -1
- metadata/generated/schema/entity/classification/classification.py +1 -1
- metadata/generated/schema/entity/classification/tag.py +2 -2
- metadata/generated/schema/entity/data/__init__.py +1 -1
- metadata/generated/schema/entity/data/chart.py +1 -1
- metadata/generated/schema/entity/data/container.py +4 -9
- metadata/generated/schema/entity/data/dashboard.py +1 -1
- metadata/generated/schema/entity/data/dashboardDataModel.py +1 -1
- metadata/generated/schema/entity/data/database.py +4 -8
- metadata/generated/schema/entity/data/databaseSchema.py +2 -8
- metadata/generated/schema/entity/data/glossary.py +1 -1
- metadata/generated/schema/entity/data/glossaryTerm.py +1 -1
- metadata/generated/schema/entity/data/metrics.py +1 -1
- metadata/generated/schema/entity/data/mlmodel.py +1 -1
- metadata/generated/schema/entity/data/pipeline.py +6 -9
- metadata/generated/schema/entity/data/query.py +2 -2
- metadata/generated/schema/entity/data/report.py +1 -1
- metadata/generated/schema/entity/data/searchIndex.py +1 -1
- metadata/generated/schema/entity/data/storedProcedure.py +3 -9
- metadata/generated/schema/entity/data/table.py +18 -12
- metadata/generated/schema/entity/data/topic.py +1 -1
- metadata/generated/schema/entity/docStore/__init__.py +1 -1
- metadata/generated/schema/entity/docStore/document.py +1 -1
- metadata/generated/schema/entity/domains/__init__.py +1 -1
- metadata/generated/schema/entity/domains/dataProduct.py +1 -1
- metadata/generated/schema/entity/domains/domain.py +1 -1
- metadata/generated/schema/entity/events/__init__.py +1 -1
- metadata/generated/schema/entity/events/webhook.py +6 -3
- metadata/generated/schema/entity/feed/__init__.py +1 -1
- metadata/generated/schema/entity/feed/suggestion.py +61 -0
- metadata/generated/schema/entity/feed/thread.py +6 -1
- metadata/generated/schema/entity/policies/__init__.py +1 -1
- metadata/generated/schema/entity/policies/accessControl/__init__.py +1 -1
- metadata/generated/schema/entity/policies/accessControl/resourceDescriptor.py +1 -1
- metadata/generated/schema/entity/policies/accessControl/resourcePermission.py +1 -1
- metadata/generated/schema/entity/policies/accessControl/rule.py +1 -1
- metadata/generated/schema/entity/policies/filters.py +2 -5
- metadata/generated/schema/entity/policies/policy.py +1 -1
- metadata/generated/schema/entity/services/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/common/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/common/sslCertPaths.py +1 -1
- metadata/generated/schema/entity/services/connections/common/sslCertValues.py +1 -1
- metadata/generated/schema/entity/services/connections/common/sslConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/connectionBasicType.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/customDashboardConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/domoDashboardConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/lightdashConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/lookerConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/metabaseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/modeConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/mstrConnection.py +46 -0
- metadata/generated/schema/entity/services/connections/dashboard/powerBIConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/qlikSenseConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/dashboard/quickSightConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/redashConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/supersetConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/tableauConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/athenaConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/azureSQLConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/bigQueryConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/clickhouseConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/common/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/common/basicAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/database/common/iamAuthConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/common/jwtAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/database/couchbaseConnection.py +6 -1
- metadata/generated/schema/entity/services/connections/database/customDatabaseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/databricksConnection.py +2 -7
- metadata/generated/schema/entity/services/connections/database/datalake/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/datalake/azureConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/datalake/gcsConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/datalake/s3Config.py +1 -1
- metadata/generated/schema/entity/services/connections/database/datalakeConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/db2Connection.py +3 -2
- metadata/generated/schema/entity/services/connections/database/deltaLakeConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/domoDatabaseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/dorisConnection.py +86 -0
- metadata/generated/schema/entity/services/connections/database/druidConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/dynamoDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/glueConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/greenplumConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/hiveConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/iceberg/__init__.py +3 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/dynamoDbCatalogConnection.py +17 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/glueCatalogConnection.py +16 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/hiveCatalogConnection.py +23 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/icebergCatalog.py +43 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/icebergFileSystem.py +20 -0
- metadata/generated/schema/entity/services/connections/database/iceberg/restCatalogConnection.py +69 -0
- metadata/generated/schema/entity/services/connections/database/icebergConnection.py +35 -0
- metadata/generated/schema/entity/services/connections/database/impalaConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/mariaDBConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/mongoDBConnection.py +24 -14
- metadata/generated/schema/entity/services/connections/database/mssqlConnection.py +5 -2
- metadata/generated/schema/entity/services/connections/database/mysqlConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/oracleConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/pinotDBConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/postgresConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/prestoConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/redshiftConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/salesforceConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sapHanaConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/sasConnection.py +48 -0
- metadata/generated/schema/entity/services/connections/database/singleStoreConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/snowflakeConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/sqliteConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/trinoConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/unityCatalogConnection.py +90 -0
- metadata/generated/schema/entity/services/connections/database/verticaConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/messaging/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/customMessagingConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/kafkaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/kinesisConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/pulsarConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/redpandaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/saslMechanismType.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/alationConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/amundsenConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/atlasConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/metadataESConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/openMetadataConnection.py +5 -20
- metadata/generated/schema/entity/services/connections/mlmodel/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/customMlModelConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/mlflowConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/sageMakerConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/sklearnConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/airbyteConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/airflowConnection.py +2 -8
- metadata/generated/schema/entity/services/connections/pipeline/backendConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/customPipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/dagsterConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/databricksPipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/domoPipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/fivetranConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/gluePipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/nifiConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/sparkConnection.py +23 -0
- metadata/generated/schema/entity/services/connections/pipeline/splineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/search/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/search/customSearchConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/search/elasticSearch/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/search/elasticSearch/apiAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/search/elasticSearch/basicAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/search/elasticSearchConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/search/openSearchConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/serviceConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/{adlsConection.py → adlsConnection.py} +2 -2
- metadata/generated/schema/entity/services/connections/storage/customStorageConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/gcsConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/s3Connection.py +1 -1
- metadata/generated/schema/entity/services/connections/testConnectionDefinition.py +1 -1
- metadata/generated/schema/entity/services/connections/testConnectionResult.py +1 -1
- metadata/generated/schema/entity/services/dashboardService.py +4 -1
- metadata/generated/schema/entity/services/databaseService.py +13 -1
- metadata/generated/schema/entity/services/ingestionPipelines/__init__.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/ingestionPipeline.py +27 -22
- metadata/generated/schema/entity/services/ingestionPipelines/pipelineServiceClientResponse.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/status.py +56 -0
- metadata/generated/schema/entity/services/messagingService.py +1 -1
- metadata/generated/schema/entity/services/metadataService.py +1 -1
- metadata/generated/schema/entity/services/mlmodelService.py +1 -1
- metadata/generated/schema/entity/services/pipelineService.py +4 -1
- metadata/generated/schema/entity/services/searchService.py +1 -1
- metadata/generated/schema/entity/services/serviceType.py +1 -1
- metadata/generated/schema/entity/services/storageService.py +10 -3
- metadata/generated/schema/entity/teams/__init__.py +1 -1
- metadata/generated/schema/entity/teams/persona.py +1 -1
- metadata/generated/schema/entity/teams/role.py +2 -6
- metadata/generated/schema/entity/teams/team.py +1 -1
- metadata/generated/schema/entity/teams/teamHierarchy.py +1 -1
- metadata/generated/schema/entity/teams/user.py +3 -10
- metadata/generated/schema/entity/type.py +2 -2
- metadata/generated/schema/entity/utils/__init__.py +1 -1
- metadata/generated/schema/entity/utils/entitiesCount.py +1 -1
- metadata/generated/schema/entity/utils/servicesCount.py +1 -1
- metadata/generated/schema/entity/utils/supersetApiConnection.py +1 -1
- metadata/generated/schema/events/__init__.py +1 -1
- metadata/generated/schema/events/alertMetrics.py +27 -0
- metadata/generated/schema/events/api/__init__.py +1 -1
- metadata/generated/schema/events/api/createEventSubscription.py +14 -12
- metadata/generated/schema/events/emailAlertConfig.py +1 -1
- metadata/generated/schema/events/eventFilterRule.py +13 -2
- metadata/generated/schema/events/eventSubscription.py +89 -49
- metadata/generated/schema/events/eventSubscriptionOffset.py +21 -0
- metadata/generated/schema/events/failedEvent.py +26 -0
- metadata/generated/schema/events/filterResourceDescriptor.py +27 -0
- metadata/generated/schema/events/subscriptionResourceDescriptor.py +11 -10
- metadata/generated/schema/metadataIngestion/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/application.py +40 -0
- metadata/generated/schema/metadataIngestion/applicationPipeline.py +15 -6
- metadata/generated/schema/metadataIngestion/dashboardServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/dataInsightPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/databaseServiceMetadataPipeline.py +6 -1
- metadata/generated/schema/metadataIngestion/databaseServiceProfilerPipeline.py +6 -1
- metadata/generated/schema/metadataIngestion/databaseServiceQueryLineagePipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/databaseServiceQueryUsagePipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtAzureConfig.py +6 -3
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtBucketDetails.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtCloudConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtGCSConfig.py +6 -3
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtHttpConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtLocalConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/dbtconfig/dbtS3Config.py +6 -3
- metadata/generated/schema/metadataIngestion/messagingServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/metadataToElasticSearchPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/mlmodelServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/pipelineServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/searchServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/containerMetadataConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/manifestMetadataConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageBucketDetails.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageMetadataADLSConfig.py +2 -2
- metadata/generated/schema/metadataIngestion/storage/storageMetadataGCSConfig.py +24 -0
- metadata/generated/schema/metadataIngestion/storage/storageMetadataHttpConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageMetadataLocalConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageMetadataS3Config.py +1 -1
- metadata/generated/schema/metadataIngestion/storageServiceMetadataPipeline.py +8 -1
- metadata/generated/schema/metadataIngestion/testSuitePipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/workflow.py +1 -1
- metadata/generated/schema/monitoring/__init__.py +1 -1
- metadata/generated/schema/monitoring/eventMonitorProvider.py +1 -1
- metadata/generated/schema/security/__init__.py +1 -1
- metadata/generated/schema/security/client/__init__.py +1 -1
- metadata/generated/schema/security/client/auth0SSOClientConfig.py +1 -1
- metadata/generated/schema/security/client/azureSSOClientConfig.py +1 -1
- metadata/generated/schema/security/client/customOidcSSOClientConfig.py +1 -1
- metadata/generated/schema/security/client/googleSSOClientConfig.py +1 -1
- metadata/generated/schema/security/client/oktaSSOClientConfig.py +1 -1
- metadata/generated/schema/security/client/openMetadataJWTClientConfig.py +1 -1
- metadata/generated/schema/security/client/samlSSOClientConfig.py +1 -1
- metadata/generated/schema/security/credentials/__init__.py +1 -1
- metadata/generated/schema/security/credentials/accessTokenAuth.py +1 -1
- metadata/generated/schema/security/credentials/apiAccessTokenAuth.py +1 -1
- metadata/generated/schema/security/credentials/awsCredentials.py +1 -1
- metadata/generated/schema/security/credentials/azureCredentials.py +1 -1
- metadata/generated/schema/security/credentials/basicAuth.py +1 -1
- metadata/generated/schema/security/credentials/bitbucketCredentials.py +1 -1
- metadata/generated/schema/security/credentials/gcpCredentials.py +1 -1
- metadata/generated/schema/security/credentials/gcpValues.py +1 -1
- metadata/generated/schema/security/credentials/gitCredentials.py +1 -1
- metadata/generated/schema/security/credentials/githubCredentials.py +1 -1
- metadata/generated/schema/security/secrets/__init__.py +1 -1
- metadata/generated/schema/security/secrets/secretsManagerClientLoader.py +1 -1
- metadata/generated/schema/security/secrets/secretsManagerConfiguration.py +14 -4
- metadata/generated/schema/security/secrets/secretsManagerProvider.py +2 -2
- metadata/generated/schema/security/securityConfiguration.py +1 -1
- metadata/generated/schema/security/ssl/__init__.py +1 -1
- metadata/generated/schema/security/ssl/validateSSLClientConfig.py +1 -1
- metadata/generated/schema/security/ssl/verifySSLConfig.py +4 -1
- metadata/generated/schema/settings/__init__.py +1 -1
- metadata/generated/schema/settings/settings.py +1 -1
- metadata/generated/schema/system/__init__.py +1 -1
- metadata/generated/schema/system/entityError.py +17 -0
- metadata/generated/schema/system/eventPublisherJob.py +5 -19
- metadata/generated/schema/system/indexingError.py +34 -0
- metadata/generated/schema/system/ui/__init__.py +1 -1
- metadata/generated/schema/system/ui/knowledgePanel.py +1 -1
- metadata/generated/schema/system/ui/page.py +1 -1
- metadata/generated/schema/tests/__init__.py +1 -1
- metadata/generated/schema/tests/assigned.py +18 -0
- metadata/generated/schema/tests/basic.py +17 -39
- metadata/generated/schema/tests/customMetric.py +4 -2
- metadata/generated/schema/tests/resolved.py +35 -0
- metadata/generated/schema/tests/testCase.py +11 -2
- metadata/generated/schema/tests/testCaseResolutionStatus.py +63 -0
- metadata/generated/schema/tests/testDefinition.py +5 -1
- metadata/generated/schema/tests/testSuite.py +1 -1
- metadata/generated/schema/type/__init__.py +1 -1
- metadata/generated/schema/type/auditLog.py +1 -1
- metadata/generated/schema/type/basic.py +5 -5
- metadata/generated/schema/type/bulkOperationResult.py +1 -1
- metadata/generated/schema/type/changeEvent.py +5 -75
- metadata/generated/schema/type/changeEventType.py +29 -0
- metadata/generated/schema/type/collectionDescriptor.py +1 -1
- metadata/generated/schema/type/csvDocumentation.py +1 -1
- metadata/generated/schema/type/csvErrorType.py +1 -1
- metadata/generated/schema/type/csvFile.py +1 -1
- metadata/generated/schema/type/csvImportResult.py +1 -1
- metadata/generated/schema/type/dailyCount.py +1 -1
- metadata/generated/schema/type/databaseConnectionConfig.py +1 -1
- metadata/generated/schema/type/entityHistory.py +1 -1
- metadata/generated/schema/type/entityLineage.py +2 -1
- metadata/generated/schema/type/entityReference.py +5 -1
- metadata/generated/schema/type/entityReferenceList.py +1 -1
- metadata/generated/schema/type/entityRelationship.py +1 -1
- metadata/generated/schema/type/entityUsage.py +1 -1
- metadata/generated/schema/type/filterPattern.py +1 -1
- metadata/generated/schema/type/function.py +7 -6
- metadata/generated/schema/type/include.py +1 -1
- metadata/generated/schema/type/jdbcConnection.py +1 -1
- metadata/generated/schema/type/lifeCycle.py +1 -1
- metadata/generated/schema/type/paging.py +1 -1
- metadata/generated/schema/type/profile.py +1 -1
- metadata/generated/schema/type/queryParserData.py +2 -2
- metadata/generated/schema/type/reaction.py +1 -1
- metadata/generated/schema/type/schedule.py +1 -1
- metadata/generated/schema/type/schema.py +1 -1
- metadata/generated/schema/type/tableQuery.py +2 -2
- metadata/generated/schema/type/tableUsageCount.py +1 -1
- metadata/generated/schema/type/tagLabel.py +1 -1
- metadata/generated/schema/type/usageDetails.py +1 -1
- metadata/generated/schema/type/usageRequest.py +1 -1
- metadata/generated/schema/type/votes.py +1 -1
- metadata/great_expectations/action.py +3 -2
- metadata/ingestion/api/delete.py +5 -2
- metadata/ingestion/api/models.py +4 -10
- metadata/ingestion/api/parser.py +3 -37
- metadata/ingestion/api/status.py +15 -4
- metadata/ingestion/api/step.py +39 -6
- metadata/ingestion/api/steps.py +20 -0
- metadata/ingestion/api/topology_runner.py +114 -168
- metadata/ingestion/bulksink/metadata_usage.py +11 -5
- metadata/ingestion/lineage/models.py +4 -0
- metadata/ingestion/lineage/parser.py +4 -5
- metadata/ingestion/lineage/sql_lineage.py +9 -17
- metadata/ingestion/models/custom_pydantic.py +4 -2
- metadata/ingestion/models/lf_tags_model.py +33 -0
- metadata/ingestion/models/patch_request.py +108 -2
- metadata/ingestion/models/tests_data.py +9 -0
- metadata/ingestion/models/topology.py +177 -60
- metadata/ingestion/ometa/auth_provider.py +0 -349
- metadata/ingestion/ometa/mixins/es_mixin.py +17 -0
- metadata/ingestion/ometa/mixins/patch_mixin.py +17 -68
- metadata/ingestion/ometa/mixins/suggestions_mixin.py +41 -0
- metadata/ingestion/ometa/mixins/table_mixin.py +18 -0
- metadata/ingestion/ometa/mixins/tests_mixin.py +24 -3
- metadata/ingestion/ometa/mixins/user_mixin.py +117 -22
- metadata/ingestion/ometa/ometa_api.py +18 -25
- metadata/ingestion/ometa/routes.py +7 -0
- metadata/ingestion/processor/query_parser.py +9 -2
- metadata/ingestion/sink/metadata_rest.py +25 -6
- metadata/ingestion/source/dashboard/dashboard_service.py +38 -61
- metadata/ingestion/source/dashboard/domodashboard/metadata.py +13 -14
- metadata/ingestion/source/dashboard/lightdash/metadata.py +2 -1
- metadata/ingestion/source/dashboard/looker/metadata.py +38 -25
- metadata/ingestion/source/dashboard/metabase/metadata.py +23 -6
- metadata/ingestion/source/dashboard/mode/client.py +10 -23
- metadata/ingestion/source/dashboard/mode/connection.py +6 -1
- metadata/ingestion/source/dashboard/mode/metadata.py +8 -4
- metadata/ingestion/source/dashboard/mstr/client.py +208 -0
- metadata/ingestion/source/dashboard/mstr/connection.py +53 -0
- metadata/ingestion/source/dashboard/mstr/metadata.py +182 -0
- metadata/ingestion/source/dashboard/mstr/models.py +144 -0
- metadata/ingestion/source/dashboard/powerbi/metadata.py +15 -10
- metadata/ingestion/source/dashboard/qliksense/metadata.py +11 -7
- metadata/ingestion/source/dashboard/quicksight/metadata.py +9 -5
- metadata/ingestion/source/dashboard/redash/metadata.py +23 -14
- metadata/ingestion/source/dashboard/superset/api_source.py +11 -8
- metadata/ingestion/source/dashboard/superset/client.py +16 -9
- metadata/ingestion/source/dashboard/superset/connection.py +3 -3
- metadata/ingestion/source/dashboard/superset/db_source.py +14 -10
- metadata/ingestion/source/dashboard/superset/mixin.py +22 -18
- metadata/ingestion/source/dashboard/superset/queries.py +1 -1
- metadata/ingestion/source/dashboard/tableau/client.py +91 -11
- metadata/ingestion/source/dashboard/tableau/connection.py +10 -1
- metadata/ingestion/source/dashboard/tableau/metadata.py +58 -70
- metadata/ingestion/source/dashboard/tableau/models.py +0 -8
- metadata/ingestion/source/dashboard/tableau/queries.py +5 -5
- metadata/ingestion/source/database/athena/client.py +80 -0
- metadata/ingestion/source/database/athena/connection.py +7 -0
- metadata/ingestion/source/database/athena/metadata.py +161 -19
- metadata/ingestion/source/database/azuresql/metadata.py +0 -1
- metadata/ingestion/source/database/bigquery/connection.py +24 -3
- metadata/ingestion/source/database/bigquery/metadata.py +39 -27
- metadata/ingestion/source/database/bigquery/queries.py +11 -4
- metadata/ingestion/source/database/bigquery/query_parser.py +13 -0
- metadata/ingestion/source/database/bigquery/usage.py +1 -3
- metadata/ingestion/source/database/column_helpers.py +0 -10
- metadata/ingestion/source/database/column_type_parser.py +11 -5
- metadata/ingestion/source/database/common_db_source.py +8 -2
- metadata/ingestion/source/database/common_nosql_source.py +8 -4
- metadata/ingestion/source/database/database_service.py +89 -7
- metadata/ingestion/source/database/databricks/client.py +5 -10
- metadata/ingestion/source/database/databricks/connection.py +16 -55
- metadata/ingestion/source/database/databricks/lineage.py +29 -26
- metadata/ingestion/source/database/databricks/metadata.py +448 -11
- metadata/ingestion/source/database/databricks/queries.py +28 -0
- metadata/ingestion/source/database/databricks/query_parser.py +5 -1
- metadata/ingestion/source/database/databricks/usage.py +2 -2
- metadata/ingestion/source/database/datalake/connection.py +31 -4
- metadata/ingestion/source/database/datalake/metadata.py +113 -27
- metadata/ingestion/source/database/dbt/dbt_config.py +5 -0
- metadata/ingestion/source/database/dbt/dbt_service.py +10 -14
- metadata/ingestion/source/database/dbt/dbt_utils.py +3 -1
- metadata/ingestion/source/database/dbt/metadata.py +11 -25
- metadata/ingestion/source/database/deltalake/metadata.py +6 -3
- metadata/ingestion/source/database/domodatabase/metadata.py +7 -6
- metadata/ingestion/source/database/doris/connection.py +72 -0
- metadata/ingestion/source/database/doris/metadata.py +315 -0
- metadata/ingestion/source/database/doris/queries.py +54 -0
- metadata/ingestion/source/database/doris/utils.py +64 -0
- metadata/ingestion/source/database/extended_sample_data.py +532 -0
- metadata/ingestion/source/database/glue/metadata.py +8 -5
- metadata/ingestion/source/database/hive/connection.py +0 -2
- metadata/ingestion/source/database/hive/utils.py +3 -0
- metadata/ingestion/source/database/iceberg/catalog/__init__.py +65 -0
- metadata/ingestion/source/database/iceberg/catalog/base.py +40 -0
- metadata/ingestion/source/database/iceberg/catalog/dynamodb.py +102 -0
- metadata/ingestion/source/database/iceberg/catalog/glue.py +88 -0
- metadata/ingestion/source/database/iceberg/catalog/hive.py +51 -0
- metadata/ingestion/source/database/iceberg/catalog/rest.py +84 -0
- metadata/ingestion/source/database/iceberg/connection.py +68 -0
- metadata/ingestion/source/database/iceberg/fs/__init__.py +52 -0
- metadata/ingestion/source/database/iceberg/fs/azure.py +44 -0
- metadata/ingestion/source/database/iceberg/fs/base.py +30 -0
- metadata/ingestion/source/database/iceberg/fs/s3.py +77 -0
- metadata/ingestion/source/database/iceberg/helper.py +124 -0
- metadata/ingestion/source/database/iceberg/metadata.py +311 -0
- metadata/ingestion/source/database/iceberg/models.py +66 -0
- metadata/ingestion/source/database/life_cycle_query_mixin.py +72 -3
- metadata/ingestion/source/database/mongodb/connection.py +1 -5
- metadata/ingestion/source/database/mssql/lineage.py +2 -0
- metadata/ingestion/source/database/mssql/metadata.py +108 -4
- metadata/ingestion/source/database/mssql/models.py +30 -0
- metadata/ingestion/source/database/mssql/queries.py +179 -1
- metadata/ingestion/source/database/mssql/utils.py +207 -4
- metadata/ingestion/source/database/mysql/metadata.py +0 -2
- metadata/ingestion/source/database/oracle/metadata.py +108 -2
- metadata/ingestion/source/database/oracle/models.py +30 -0
- metadata/ingestion/source/database/oracle/queries.py +98 -17
- metadata/ingestion/source/database/oracle/utils.py +0 -1
- metadata/ingestion/source/database/postgres/lineage.py +32 -14
- metadata/ingestion/source/database/postgres/metadata.py +15 -7
- metadata/ingestion/source/database/postgres/pgspider/lineage.py +0 -1
- metadata/ingestion/source/database/postgres/queries.py +4 -2
- metadata/ingestion/source/database/postgres/query_parser.py +4 -72
- metadata/ingestion/source/database/postgres/usage.py +41 -0
- metadata/ingestion/source/database/postgres/utils.py +34 -0
- metadata/ingestion/source/database/query_parser_source.py +8 -2
- metadata/ingestion/source/database/redshift/metadata.py +14 -4
- metadata/ingestion/source/database/redshift/queries.py +10 -4
- metadata/ingestion/source/database/redshift/query_parser.py +16 -0
- metadata/ingestion/source/database/redshift/usage.py +0 -2
- metadata/ingestion/source/database/salesforce/metadata.py +32 -3
- metadata/ingestion/source/database/sample_data.py +120 -6
- metadata/ingestion/source/database/sas/client.py +184 -0
- metadata/ingestion/source/database/sas/connection.py +47 -0
- metadata/ingestion/source/database/sas/extension_attr.py +103 -0
- metadata/ingestion/source/database/sas/metadata.py +914 -0
- metadata/ingestion/source/database/snowflake/metadata.py +8 -51
- metadata/ingestion/source/database/snowflake/models.py +6 -1
- metadata/ingestion/source/database/snowflake/queries.py +0 -4
- metadata/ingestion/source/database/snowflake/query_parser.py +5 -20
- metadata/ingestion/source/database/snowflake/utils.py +2 -6
- metadata/ingestion/source/database/stored_procedures_mixin.py +12 -8
- metadata/ingestion/source/database/unitycatalog/__init__.py +0 -0
- metadata/ingestion/source/database/unitycatalog/client.py +87 -0
- metadata/ingestion/source/database/unitycatalog/connection.py +97 -0
- metadata/ingestion/source/database/{databricks/unity_catalog → unitycatalog}/lineage.py +11 -11
- metadata/ingestion/source/database/{databricks/unity_catalog → unitycatalog}/metadata.py +42 -49
- metadata/ingestion/source/database/unitycatalog/query_parser.py +60 -0
- metadata/ingestion/source/database/unitycatalog/usage.py +31 -0
- metadata/ingestion/source/database/usage_source.py +3 -2
- metadata/ingestion/source/messaging/common_broker_source.py +6 -4
- metadata/ingestion/source/messaging/kinesis/metadata.py +6 -3
- metadata/ingestion/source/messaging/messaging_service.py +6 -2
- metadata/ingestion/source/metadata/amundsen/metadata.py +10 -7
- metadata/ingestion/source/metadata/atlas/metadata.py +5 -5
- metadata/ingestion/source/mlmodel/mlflow/metadata.py +5 -2
- metadata/ingestion/source/mlmodel/mlmodel_service.py +6 -2
- metadata/ingestion/source/pipeline/airflow/connection.py +0 -12
- metadata/ingestion/source/pipeline/airflow/lineage_parser.py +12 -6
- metadata/ingestion/source/pipeline/airflow/metadata.py +63 -34
- metadata/ingestion/source/pipeline/airflow/models.py +5 -4
- metadata/ingestion/source/pipeline/dagster/metadata.py +7 -4
- metadata/ingestion/source/pipeline/databrickspipeline/metadata.py +12 -9
- metadata/ingestion/source/pipeline/domopipeline/metadata.py +7 -4
- metadata/ingestion/source/pipeline/gluepipeline/metadata.py +5 -2
- metadata/ingestion/source/pipeline/pipeline_service.py +6 -2
- metadata/ingestion/source/pipeline/spline/metadata.py +0 -1
- metadata/ingestion/source/search/elasticsearch/connection.py +4 -1
- metadata/ingestion/source/search/elasticsearch/metadata.py +1 -2
- metadata/ingestion/source/search/search_service.py +6 -2
- metadata/ingestion/source/storage/s3/metadata.py +22 -17
- metadata/ingestion/source/storage/storage_service.py +53 -11
- metadata/ingestion/stage/table_usage.py +9 -2
- metadata/mixins/sqalchemy/sqa_mixin.py +14 -7
- metadata/parsers/protobuf_parser.py +29 -11
- metadata/pii/processor.py +9 -2
- metadata/profiler/api/models.py +19 -1
- metadata/profiler/interface/pandas/profiler_interface.py +59 -18
- metadata/profiler/interface/profiler_interface.py +13 -2
- metadata/profiler/interface/profiler_interface_factory.py +49 -14
- metadata/profiler/interface/sqlalchemy/bigquery/profiler_interface.py +3 -0
- metadata/profiler/interface/sqlalchemy/databricks/profiler_interface.py +26 -0
- metadata/profiler/interface/sqlalchemy/db2/__init__.py +0 -0
- metadata/profiler/interface/sqlalchemy/db2/profiler_interface.py +38 -0
- metadata/profiler/interface/sqlalchemy/mariadb/profiler_interface.py +85 -0
- metadata/profiler/interface/sqlalchemy/profiler_interface.py +77 -34
- metadata/profiler/interface/sqlalchemy/single_store/profiler_interface.py +2 -2
- metadata/profiler/interface/sqlalchemy/snowflake/profiler_interface.py +7 -0
- metadata/profiler/interface/sqlalchemy/trino/profiler_interface.py +2 -2
- metadata/profiler/interface/sqlalchemy/unity_catalog/profiler_interface.py +33 -0
- metadata/profiler/metrics/composed/null_ratio.py +1 -1
- metadata/profiler/metrics/hybrid/histogram.py +1 -0
- metadata/profiler/metrics/static/max.py +4 -1
- metadata/profiler/metrics/static/min.py +4 -1
- metadata/profiler/metrics/system/queries/snowflake.py +89 -17
- metadata/profiler/metrics/system/system.py +62 -20
- metadata/profiler/orm/functions/length.py +1 -0
- metadata/profiler/orm/functions/median.py +7 -0
- metadata/profiler/orm/functions/table_metric_computer.py +462 -0
- metadata/profiler/orm/registry.py +1 -0
- metadata/profiler/processor/core.py +116 -52
- metadata/profiler/processor/default.py +14 -3
- metadata/profiler/processor/handle_partition.py +2 -2
- metadata/profiler/processor/processor.py +9 -4
- metadata/profiler/processor/sampler/sqlalchemy/bigquery/sampler.py +31 -3
- metadata/profiler/processor/sampler/sqlalchemy/sampler.py +29 -6
- metadata/profiler/processor/sampler/sqlalchemy/trino/sampler.py +10 -4
- metadata/profiler/source/base/profiler_source.py +5 -2
- metadata/profiler/source/bigquery/type_mapper.py +0 -1
- metadata/profiler/source/mariadb/functions/median.py +20 -0
- metadata/profiler/source/mariadb/metrics/window/first_quartile.py +10 -0
- metadata/profiler/source/mariadb/metrics/window/median.py +10 -0
- metadata/profiler/source/mariadb/metrics/window/third_quartile.py +10 -0
- metadata/profiler/source/metadata.py +43 -10
- metadata/profiler/source/metadata_ext.py +16 -50
- metadata/py.typed +0 -0
- metadata/readers/dataframe/json.py +5 -4
- metadata/readers/file/api_reader.py +0 -1
- metadata/utils/constants.py +5 -0
- metadata/utils/datalake/datalake_utils.py +363 -129
- metadata/utils/entity_link.py +26 -6
- metadata/utils/fqn.py +20 -0
- metadata/utils/helpers.py +55 -9
- metadata/utils/importer.py +2 -3
- metadata/utils/life_cycle_utils.py +4 -4
- metadata/utils/logger.py +13 -2
- metadata/utils/partition.py +10 -5
- metadata/utils/secrets/client/loader.py +0 -1
- metadata/utils/secrets/noop_secrets_manager.py +4 -3
- metadata/utils/secrets/secrets_manager_factory.py +3 -4
- metadata/utils/{source_hash_utils.py → source_hash.py} +10 -1
- metadata/utils/sqlalchemy_utils.py +21 -0
- metadata/utils/storage_metadata_config.py +42 -1
- metadata/utils/tag_utils.py +5 -2
- metadata/workflow/application.py +154 -0
- metadata/workflow/application_output_handler.py +34 -0
- metadata/workflow/base.py +84 -153
- metadata/workflow/data_insight.py +8 -7
- metadata/workflow/data_quality.py +3 -2
- metadata/workflow/ingestion.py +203 -0
- metadata/workflow/metadata.py +2 -3
- metadata/workflow/output_handler.py +204 -0
- metadata/workflow/profiler.py +2 -2
- metadata/workflow/usage.py +3 -4
- metadata/workflow/workflow_output_handler.py +15 -255
- metadata/workflow/workflow_status_mixin.py +44 -52
- openmetadata_ingestion-1.3.0.0.dist-info/METADATA +749 -0
- {openmetadata_ingestion-1.2.5.2.dist-info → openmetadata_ingestion-1.3.0.0.dist-info}/RECORD +812 -725
- {openmetadata_ingestion-1.2.5.2.dist-info → openmetadata_ingestion-1.3.0.0.dist-info}/WHEEL +1 -1
- metadata/generated/schema/entity/applications/appConfig.py +0 -21
- metadata/generated/schema/entity/applications/configuration/dataInsightsApp.py +0 -17
- metadata/generated/schema/entity/applications/configuration/externalAppIngestionConfig.py +0 -38
- metadata/generated/schema/entity/services/connections/database/mongoDB/mongoDBValues.py +0 -44
- metadata/generated/schema/events/dataInsightAlertConfig.py +0 -17
- metadata/generated/schema/events/entitySpelFilters.py +0 -19
- metadata/ingestion/models/es_documents.py +0 -339
- metadata/ingestion/ometa/mixins/glossary_mixin.py +0 -501
- metadata/ingestion/ometa/provider_registry.py +0 -144
- metadata/ingestion/source/database/databricks/legacy/lineage.py +0 -51
- metadata/ingestion/source/database/databricks/legacy/metadata.py +0 -339
- metadata/ingestion/source/metadata/metadata_elasticsearch/metadata.py +0 -144
- metadata/profiler/orm/functions/table_metric_construct.py +0 -365
- openmetadata_ingestion-1.2.5.2.dist-info/METADATA +0 -426
- /metadata/ingestion/source/{database/databricks/legacy → dashboard/mstr}/__init__.py +0 -0
- /metadata/ingestion/source/database/{databricks/unity_catalog → doris}/__init__.py +0 -0
- /metadata/ingestion/source/{metadata/metadata_elasticsearch → database/sas}/__init__.py +0 -0
- /metadata/ingestion/source/database/{databricks → unitycatalog}/models.py +0 -0
- {openmetadata_ingestion-1.2.5.2.dist-info → openmetadata_ingestion-1.3.0.0.dist-info}/LICENSE +0 -0
- {openmetadata_ingestion-1.2.5.2.dist-info → openmetadata_ingestion-1.3.0.0.dist-info}/entry_points.txt +0 -0
- {openmetadata_ingestion-1.2.5.2.dist-info → openmetadata_ingestion-1.3.0.0.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,532 @@
|
|
|
1
|
+
# Copyright 2021 Collate
|
|
2
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
3
|
+
# you may not use this file except in compliance with the License.
|
|
4
|
+
# You may obtain a copy of the License at
|
|
5
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
6
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
7
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
8
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
9
|
+
# See the License for the specific language governing permissions and
|
|
10
|
+
# limitations under the License.
|
|
11
|
+
"""
|
|
12
|
+
Sample Data source ingestion
|
|
13
|
+
"""
|
|
14
|
+
import json
|
|
15
|
+
from collections import namedtuple
|
|
16
|
+
from typing import Iterable
|
|
17
|
+
|
|
18
|
+
from metadata.generated.schema.api.data.createChart import CreateChartRequest
|
|
19
|
+
from metadata.generated.schema.api.data.createDashboard import CreateDashboardRequest
|
|
20
|
+
from metadata.generated.schema.api.data.createDashboardDataModel import (
|
|
21
|
+
CreateDashboardDataModelRequest,
|
|
22
|
+
)
|
|
23
|
+
from metadata.generated.schema.api.data.createDatabase import CreateDatabaseRequest
|
|
24
|
+
from metadata.generated.schema.api.data.createDatabaseSchema import (
|
|
25
|
+
CreateDatabaseSchemaRequest,
|
|
26
|
+
)
|
|
27
|
+
from metadata.generated.schema.api.data.createGlossary import CreateGlossaryRequest
|
|
28
|
+
from metadata.generated.schema.api.data.createGlossaryTerm import (
|
|
29
|
+
CreateGlossaryTermRequest,
|
|
30
|
+
)
|
|
31
|
+
from metadata.generated.schema.api.data.createTable import CreateTableRequest
|
|
32
|
+
from metadata.generated.schema.api.lineage.addLineage import AddLineageRequest
|
|
33
|
+
from metadata.generated.schema.entity.data.dashboard import Dashboard
|
|
34
|
+
from metadata.generated.schema.entity.data.dashboardDataModel import DashboardDataModel
|
|
35
|
+
from metadata.generated.schema.entity.data.database import Database
|
|
36
|
+
from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
|
|
37
|
+
from metadata.generated.schema.entity.data.table import Table
|
|
38
|
+
from metadata.generated.schema.entity.services.connections.database.customDatabaseConnection import (
|
|
39
|
+
CustomDatabaseConnection,
|
|
40
|
+
)
|
|
41
|
+
from metadata.generated.schema.entity.services.dashboardService import DashboardService
|
|
42
|
+
from metadata.generated.schema.entity.services.databaseService import DatabaseService
|
|
43
|
+
from metadata.generated.schema.metadataIngestion.workflow import (
|
|
44
|
+
Source as WorkflowSource,
|
|
45
|
+
)
|
|
46
|
+
from metadata.generated.schema.type.entityLineage import (
|
|
47
|
+
ColumnLineage,
|
|
48
|
+
EntitiesEdge,
|
|
49
|
+
LineageDetails,
|
|
50
|
+
)
|
|
51
|
+
from metadata.generated.schema.type.entityLineage import Source as LineageSource
|
|
52
|
+
from metadata.generated.schema.type.entityReference import EntityReference
|
|
53
|
+
from metadata.ingestion.api.common import Entity
|
|
54
|
+
from metadata.ingestion.api.models import Either
|
|
55
|
+
from metadata.ingestion.api.steps import InvalidSourceException, Source
|
|
56
|
+
from metadata.ingestion.ometa.ometa_api import OpenMetadata
|
|
57
|
+
from metadata.utils import fqn
|
|
58
|
+
from metadata.utils.constants import UTF_8
|
|
59
|
+
from metadata.utils.helpers import get_standard_chart_type
|
|
60
|
+
from metadata.utils.logger import ingestion_logger
|
|
61
|
+
|
|
62
|
+
logger = ingestion_logger()
|
|
63
|
+
|
|
64
|
+
COLUMN_NAME = "Column"
|
|
65
|
+
KEY_TYPE = "Key type"
|
|
66
|
+
DATA_TYPE = "Data type"
|
|
67
|
+
COL_DESCRIPTION = "Description"
|
|
68
|
+
TableKey = namedtuple("TableKey", ["schema", "table_name"])
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
class InvalidSampleDataException(Exception):
|
|
72
|
+
"""
|
|
73
|
+
Sample data is not valid to be ingested
|
|
74
|
+
"""
|
|
75
|
+
|
|
76
|
+
|
|
77
|
+
class ExtendedSampleDataSource(Source): # pylint: disable=too-many-instance-attributes
|
|
78
|
+
"""
|
|
79
|
+
Loads JSON data and prepares the required
|
|
80
|
+
python objects to be sent to the Sink.
|
|
81
|
+
"""
|
|
82
|
+
|
|
83
|
+
def __init__(self, config: WorkflowSource, metadata: OpenMetadata):
|
|
84
|
+
import faker # pylint: disable=import-outside-toplevel
|
|
85
|
+
|
|
86
|
+
super().__init__()
|
|
87
|
+
|
|
88
|
+
self.fake = faker.Faker(["en-US", "zh_CN", "ja_JP", "th_TH"])
|
|
89
|
+
self.database_service_json = {}
|
|
90
|
+
self.dashboard_service_json = {}
|
|
91
|
+
self.config = config
|
|
92
|
+
self.service_connection = config.serviceConnection.__root__.config
|
|
93
|
+
self.metadata = metadata
|
|
94
|
+
self.list_policies = []
|
|
95
|
+
self.store_table_fqn = set()
|
|
96
|
+
self.store_data_model_fqn = []
|
|
97
|
+
self.store_dashboard_fqn = []
|
|
98
|
+
self.main_glossary = None
|
|
99
|
+
self.glossary_term_list = []
|
|
100
|
+
|
|
101
|
+
sample_data_folder = self.service_connection.connectionOptions.__root__.get(
|
|
102
|
+
"sampleDataFolder"
|
|
103
|
+
)
|
|
104
|
+
self.include_glossary = self.service_connection.connectionOptions.__root__.get(
|
|
105
|
+
"includeGlossary"
|
|
106
|
+
)
|
|
107
|
+
self.include_lineage_stress_testing = (
|
|
108
|
+
self.service_connection.connectionOptions.__root__.get(
|
|
109
|
+
"includeLineageStressTesting"
|
|
110
|
+
)
|
|
111
|
+
)
|
|
112
|
+
extneded_sample_data_folder = (
|
|
113
|
+
self.service_connection.connectionOptions.__root__.get(
|
|
114
|
+
"extendedSampleDataFolder"
|
|
115
|
+
)
|
|
116
|
+
)
|
|
117
|
+
if not sample_data_folder:
|
|
118
|
+
raise InvalidSampleDataException(
|
|
119
|
+
"Cannot get sampleDataFolder from connection options"
|
|
120
|
+
)
|
|
121
|
+
|
|
122
|
+
if not extneded_sample_data_folder:
|
|
123
|
+
raise InvalidSampleDataException(
|
|
124
|
+
"Cannot get ExtendedSampleDataFolder from connection options"
|
|
125
|
+
)
|
|
126
|
+
|
|
127
|
+
self.database_service_json = json.load(
|
|
128
|
+
open( # pylint: disable=consider-using-with
|
|
129
|
+
sample_data_folder + "/datasets/service.json",
|
|
130
|
+
"r",
|
|
131
|
+
encoding=UTF_8,
|
|
132
|
+
)
|
|
133
|
+
)
|
|
134
|
+
|
|
135
|
+
self.tables = json.load(
|
|
136
|
+
open( # pylint: disable=consider-using-with
|
|
137
|
+
sample_data_folder + "/datasets/tables.json",
|
|
138
|
+
"r",
|
|
139
|
+
encoding=UTF_8,
|
|
140
|
+
)
|
|
141
|
+
)
|
|
142
|
+
self.extended_tables = json.load(
|
|
143
|
+
open( # pylint: disable=consider-using-with
|
|
144
|
+
extneded_sample_data_folder + "/datasets/upstream_edge_tables.json",
|
|
145
|
+
"r",
|
|
146
|
+
encoding=UTF_8,
|
|
147
|
+
)
|
|
148
|
+
)
|
|
149
|
+
|
|
150
|
+
self.database_service_json = json.load(
|
|
151
|
+
open( # pylint: disable=consider-using-with
|
|
152
|
+
sample_data_folder + "/datasets/service.json",
|
|
153
|
+
"r",
|
|
154
|
+
encoding=UTF_8,
|
|
155
|
+
)
|
|
156
|
+
)
|
|
157
|
+
self.charts = json.load(
|
|
158
|
+
open( # pylint: disable=consider-using-with
|
|
159
|
+
sample_data_folder + "/dashboards/charts.json",
|
|
160
|
+
"r",
|
|
161
|
+
encoding=UTF_8,
|
|
162
|
+
)
|
|
163
|
+
)
|
|
164
|
+
self.database_service = self.metadata.get_service_or_create(
|
|
165
|
+
entity=DatabaseService, config=WorkflowSource(**self.database_service_json)
|
|
166
|
+
)
|
|
167
|
+
|
|
168
|
+
self.dashboard_service_json = json.load(
|
|
169
|
+
open( # pylint: disable=consider-using-with
|
|
170
|
+
sample_data_folder + "/dashboards/service.json",
|
|
171
|
+
"r",
|
|
172
|
+
encoding=UTF_8,
|
|
173
|
+
)
|
|
174
|
+
)
|
|
175
|
+
|
|
176
|
+
self.data_models = json.load(
|
|
177
|
+
open( # pylint: disable=consider-using-with
|
|
178
|
+
sample_data_folder + "/dashboards/dashboardDataModels.json",
|
|
179
|
+
"r",
|
|
180
|
+
encoding=UTF_8,
|
|
181
|
+
)
|
|
182
|
+
)
|
|
183
|
+
self.dashboards = json.load(
|
|
184
|
+
open( # pylint: disable=consider-using-with
|
|
185
|
+
sample_data_folder + "/dashboards/dashboards.json",
|
|
186
|
+
"r",
|
|
187
|
+
encoding=UTF_8,
|
|
188
|
+
)
|
|
189
|
+
)
|
|
190
|
+
self.dashboard_service = self.metadata.get_service_or_create(
|
|
191
|
+
entity=DashboardService,
|
|
192
|
+
config=WorkflowSource(**self.dashboard_service_json),
|
|
193
|
+
)
|
|
194
|
+
self.db_name = None
|
|
195
|
+
|
|
196
|
+
@classmethod
|
|
197
|
+
def create(cls, config_dict, metadata: OpenMetadata):
|
|
198
|
+
"""Create class instance"""
|
|
199
|
+
config: WorkflowSource = WorkflowSource.parse_obj(config_dict)
|
|
200
|
+
connection: CustomDatabaseConnection = config.serviceConnection.__root__.config
|
|
201
|
+
if not isinstance(connection, CustomDatabaseConnection):
|
|
202
|
+
raise InvalidSourceException(
|
|
203
|
+
f"Expected CustomDatabaseConnection, but got {connection}"
|
|
204
|
+
)
|
|
205
|
+
return cls(config, metadata)
|
|
206
|
+
|
|
207
|
+
def prepare(self):
|
|
208
|
+
"""Nothing to prepare"""
|
|
209
|
+
|
|
210
|
+
def _iter(self, *_, **__) -> Iterable[Entity]:
|
|
211
|
+
yield from self.generate_sample_data()
|
|
212
|
+
|
|
213
|
+
def close(self):
|
|
214
|
+
"""Nothing to close"""
|
|
215
|
+
|
|
216
|
+
def test_connection(self) -> None:
|
|
217
|
+
"""Custom sources don't support testing connections"""
|
|
218
|
+
|
|
219
|
+
def create_depth_nodes(self, from_table, to_table):
|
|
220
|
+
"""Create Depth Nodes"""
|
|
221
|
+
from_col_list = []
|
|
222
|
+
for col in from_table.columns:
|
|
223
|
+
from_col_list.append(col.fullyQualifiedName.__root__)
|
|
224
|
+
to_col = to_table.columns[0].fullyQualifiedName.__root__
|
|
225
|
+
yield Either(
|
|
226
|
+
right=AddLineageRequest(
|
|
227
|
+
edge=EntitiesEdge(
|
|
228
|
+
fromEntity=EntityReference(id=from_table.id.__root__, type="table"),
|
|
229
|
+
toEntity=EntityReference(
|
|
230
|
+
id=to_table.id.__root__,
|
|
231
|
+
type="table",
|
|
232
|
+
),
|
|
233
|
+
lineageDetails=LineageDetails(
|
|
234
|
+
columnsLineage=[
|
|
235
|
+
ColumnLineage(fromColumns=from_col_list, toColumn=to_col)
|
|
236
|
+
]
|
|
237
|
+
),
|
|
238
|
+
)
|
|
239
|
+
)
|
|
240
|
+
)
|
|
241
|
+
|
|
242
|
+
def generate_sample_data(
|
|
243
|
+
self,
|
|
244
|
+
): # pylint: disable=too-many-locals,too-many-statements
|
|
245
|
+
"""
|
|
246
|
+
Generate sample data for dashboard and database service,
|
|
247
|
+
with lineage between them, having long names, special characters and description
|
|
248
|
+
"""
|
|
249
|
+
if self.include_glossary:
|
|
250
|
+
yield from self.create_glossary()
|
|
251
|
+
yield from self.create_glossary_term()
|
|
252
|
+
if self.include_lineage_stress_testing:
|
|
253
|
+
db = self.create_database_request(
|
|
254
|
+
"extended_sample_data", self.generate_text()
|
|
255
|
+
)
|
|
256
|
+
yield Either(right=db)
|
|
257
|
+
|
|
258
|
+
schema = self.create_database_schema_request(
|
|
259
|
+
"extended_sample_database_schema", self.generate_text(), db
|
|
260
|
+
)
|
|
261
|
+
yield Either(right=schema)
|
|
262
|
+
for table in self.extended_tables["tables"]:
|
|
263
|
+
text = self.generate_text()
|
|
264
|
+
table_request = self.create_table_request(
|
|
265
|
+
table["name"], text, schema, table
|
|
266
|
+
)
|
|
267
|
+
yield Either(right=table_request)
|
|
268
|
+
downstream_node_fqn_table = fqn.build(
|
|
269
|
+
self.metadata,
|
|
270
|
+
entity_type=Table,
|
|
271
|
+
service_name=self.database_service.name.__root__,
|
|
272
|
+
database_name=db.name.__root__,
|
|
273
|
+
schema_name=schema.name.__root__,
|
|
274
|
+
table_name=table_request.name.__root__,
|
|
275
|
+
)
|
|
276
|
+
to_table = self.metadata.get_by_name(
|
|
277
|
+
entity=Table, fqn=downstream_node_fqn_table
|
|
278
|
+
)
|
|
279
|
+
main_table = to_table
|
|
280
|
+
self.store_table_fqn.add(downstream_node_fqn_table)
|
|
281
|
+
|
|
282
|
+
for _ in range(40):
|
|
283
|
+
# 40 Dynamic Lineage Depths
|
|
284
|
+
name = self.generate_name()
|
|
285
|
+
table_request = self.create_table_request(name, text, schema, table)
|
|
286
|
+
yield Either(right=table_request)
|
|
287
|
+
upstream_node_fqn_table = fqn.build(
|
|
288
|
+
self.metadata,
|
|
289
|
+
entity_type=Table,
|
|
290
|
+
service_name=self.database_service.name.__root__,
|
|
291
|
+
database_name=db.name.__root__,
|
|
292
|
+
schema_name=schema.name.__root__,
|
|
293
|
+
table_name=table_request.name.__root__,
|
|
294
|
+
)
|
|
295
|
+
from_table = self.metadata.get_by_name(
|
|
296
|
+
entity=Table, fqn=upstream_node_fqn_table
|
|
297
|
+
)
|
|
298
|
+
yield from self.create_depth_nodes(
|
|
299
|
+
from_table=from_table, to_table=to_table
|
|
300
|
+
)
|
|
301
|
+
to_table = from_table
|
|
302
|
+
for _ in range(40):
|
|
303
|
+
name = self.generate_name()
|
|
304
|
+
table_request = self.create_table_request(
|
|
305
|
+
name, text, schema, self.extended_tables["tables"][-1]
|
|
306
|
+
)
|
|
307
|
+
yield Either(right=table_request)
|
|
308
|
+
upstream_node_fqn_table = fqn.build(
|
|
309
|
+
self.metadata,
|
|
310
|
+
entity_type=Table,
|
|
311
|
+
service_name=self.database_service.name.__root__,
|
|
312
|
+
database_name=db.name.__root__,
|
|
313
|
+
schema_name=schema.name.__root__,
|
|
314
|
+
table_name=table_request.name.__root__,
|
|
315
|
+
)
|
|
316
|
+
from_table = self.metadata.get_by_name(
|
|
317
|
+
entity=Table, fqn=upstream_node_fqn_table
|
|
318
|
+
)
|
|
319
|
+
yield from self.create_depth_nodes(
|
|
320
|
+
from_table=from_table, to_table=main_table
|
|
321
|
+
)
|
|
322
|
+
|
|
323
|
+
for table in self.tables["tables"]:
|
|
324
|
+
text = self.generate_text()
|
|
325
|
+
table_request = self.create_table_request(
|
|
326
|
+
table["name"], text, schema, table
|
|
327
|
+
)
|
|
328
|
+
yield Either(right=table_request)
|
|
329
|
+
table_entity_fqn = fqn.build(
|
|
330
|
+
self.metadata,
|
|
331
|
+
entity_type=Table,
|
|
332
|
+
service_name=self.database_service.name.__root__,
|
|
333
|
+
database_name=db.name.__root__,
|
|
334
|
+
schema_name=schema.name.__root__,
|
|
335
|
+
table_name=table_request.name.__root__,
|
|
336
|
+
)
|
|
337
|
+
from_table = self.metadata.get_by_name(
|
|
338
|
+
entity=Table, fqn=table_entity_fqn
|
|
339
|
+
)
|
|
340
|
+
|
|
341
|
+
yield from self.create_depth_nodes(
|
|
342
|
+
from_table=from_table, to_table=to_table
|
|
343
|
+
)
|
|
344
|
+
|
|
345
|
+
self.dashboard_service_json["name"] = name
|
|
346
|
+
self.dashboard_service_json["description"] = text
|
|
347
|
+
for chart in self.charts["charts"]:
|
|
348
|
+
yield Either(
|
|
349
|
+
right=CreateChartRequest(
|
|
350
|
+
name=chart["name"],
|
|
351
|
+
displayName=chart["displayName"],
|
|
352
|
+
description=chart["description"],
|
|
353
|
+
chartType=get_standard_chart_type(chart["chartType"]),
|
|
354
|
+
sourceUrl=chart["sourceUrl"],
|
|
355
|
+
service=self.dashboard_service.fullyQualifiedName,
|
|
356
|
+
)
|
|
357
|
+
)
|
|
358
|
+
for data_model in self.data_models["datamodels"]:
|
|
359
|
+
name = self.generate_name()
|
|
360
|
+
text = self.generate_text()
|
|
361
|
+
data_model_request = self.create_dashboard_data_model_request(
|
|
362
|
+
name, text, data_model
|
|
363
|
+
)
|
|
364
|
+
yield Either(right=data_model_request)
|
|
365
|
+
data_model_entity_fqn = fqn.build(
|
|
366
|
+
self.metadata,
|
|
367
|
+
entity_type=DashboardDataModel,
|
|
368
|
+
service_name=self.dashboard_service.name.__root__,
|
|
369
|
+
data_model_name=data_model_request.name.__root__,
|
|
370
|
+
)
|
|
371
|
+
self.store_data_model_fqn.append(data_model_entity_fqn)
|
|
372
|
+
|
|
373
|
+
for table_fqn in self.store_table_fqn:
|
|
374
|
+
from_table = self.metadata.get_by_name(entity=Table, fqn=table_fqn)
|
|
375
|
+
for dashboard_datamodel_fqn in self.store_data_model_fqn:
|
|
376
|
+
to_datamodel = self.metadata.get_by_name(
|
|
377
|
+
entity=DashboardDataModel, fqn=dashboard_datamodel_fqn
|
|
378
|
+
)
|
|
379
|
+
|
|
380
|
+
yield Either(
|
|
381
|
+
right=AddLineageRequest(
|
|
382
|
+
edge=EntitiesEdge(
|
|
383
|
+
fromEntity=EntityReference(
|
|
384
|
+
id=from_table.id.__root__, type="table"
|
|
385
|
+
),
|
|
386
|
+
toEntity=EntityReference(
|
|
387
|
+
id=to_datamodel.id.__root__,
|
|
388
|
+
type="dashboardDataModel",
|
|
389
|
+
),
|
|
390
|
+
lineageDetails=LineageDetails(
|
|
391
|
+
source=LineageSource.DashboardLineage
|
|
392
|
+
),
|
|
393
|
+
)
|
|
394
|
+
)
|
|
395
|
+
)
|
|
396
|
+
for dashboard in self.dashboards["dashboards"]:
|
|
397
|
+
dashboard_request = CreateDashboardRequest(
|
|
398
|
+
name=dashboard["name"],
|
|
399
|
+
displayName=dashboard["displayName"],
|
|
400
|
+
description=dashboard["description"],
|
|
401
|
+
sourceUrl=dashboard["sourceUrl"],
|
|
402
|
+
charts=dashboard["charts"],
|
|
403
|
+
service=self.dashboard_service.fullyQualifiedName,
|
|
404
|
+
)
|
|
405
|
+
yield Either(right=dashboard_request)
|
|
406
|
+
dashboard_fqn = fqn.build(
|
|
407
|
+
self.metadata,
|
|
408
|
+
entity_type=Dashboard,
|
|
409
|
+
service_name=self.dashboard_service.name.__root__,
|
|
410
|
+
dashboard_name=dashboard_request.name.__root__,
|
|
411
|
+
)
|
|
412
|
+
to_dashboard = self.metadata.get_by_name(
|
|
413
|
+
entity=Dashboard, fqn=dashboard_fqn
|
|
414
|
+
)
|
|
415
|
+
yield Either(
|
|
416
|
+
right=AddLineageRequest(
|
|
417
|
+
edge=EntitiesEdge(
|
|
418
|
+
fromEntity=EntityReference(
|
|
419
|
+
id=to_datamodel.id.__root__,
|
|
420
|
+
type="dashboardDataModel",
|
|
421
|
+
),
|
|
422
|
+
toEntity=EntityReference(
|
|
423
|
+
id=to_dashboard.id.__root__, type="dashboard"
|
|
424
|
+
),
|
|
425
|
+
lineageDetails=LineageDetails(
|
|
426
|
+
source=LineageSource.DashboardLineage
|
|
427
|
+
),
|
|
428
|
+
)
|
|
429
|
+
)
|
|
430
|
+
)
|
|
431
|
+
|
|
432
|
+
def create_glossary(self):
|
|
433
|
+
self.main_glossary = CreateGlossaryRequest(
|
|
434
|
+
name="NestedGlossaryTest",
|
|
435
|
+
displayName="NestedGlossaryTest",
|
|
436
|
+
description="Description of test glossary",
|
|
437
|
+
)
|
|
438
|
+
yield Either(right=self.main_glossary)
|
|
439
|
+
|
|
440
|
+
def create_glossary_term(self):
|
|
441
|
+
"""
|
|
442
|
+
Create Glossary Terms
|
|
443
|
+
"""
|
|
444
|
+
for _ in range(20):
|
|
445
|
+
random_name = self.fake.first_name()
|
|
446
|
+
yield Either(
|
|
447
|
+
right=CreateGlossaryTermRequest(
|
|
448
|
+
glossary="NestedGlossaryTest",
|
|
449
|
+
name=random_name,
|
|
450
|
+
displayName=random_name,
|
|
451
|
+
description="Test glossary term ",
|
|
452
|
+
parent=self.glossary_term_list[-1]
|
|
453
|
+
if len(self.glossary_term_list) > 3
|
|
454
|
+
else None,
|
|
455
|
+
)
|
|
456
|
+
)
|
|
457
|
+
if len(self.glossary_term_list) > 3 and self.glossary_term_list[-1]:
|
|
458
|
+
self.glossary_term_list.append(
|
|
459
|
+
f"{self.glossary_term_list[-1]}.{random_name}"
|
|
460
|
+
)
|
|
461
|
+
else:
|
|
462
|
+
self.glossary_term_list.append(f"NestedGlossaryTest.{random_name}")
|
|
463
|
+
|
|
464
|
+
for _ in range(500):
|
|
465
|
+
random_name = self.fake.first_name()
|
|
466
|
+
yield Either(
|
|
467
|
+
right=CreateGlossaryTermRequest(
|
|
468
|
+
glossary="NestedGlossaryTest",
|
|
469
|
+
name=random_name,
|
|
470
|
+
displayName=random_name,
|
|
471
|
+
description="Test glossary term 1",
|
|
472
|
+
)
|
|
473
|
+
)
|
|
474
|
+
|
|
475
|
+
def generate_name(self):
|
|
476
|
+
return f"Sample-@!3_(%t3st@)%_^{self.fake.name()}"
|
|
477
|
+
|
|
478
|
+
def generate_text(self):
|
|
479
|
+
return f"Sample-@!3_(%m@)%_^{self.fake.text()}"
|
|
480
|
+
|
|
481
|
+
def create_database_request(self, name, text):
|
|
482
|
+
db = CreateDatabaseRequest(
|
|
483
|
+
name=name,
|
|
484
|
+
description=text,
|
|
485
|
+
service=self.database_service.fullyQualifiedName.__root__,
|
|
486
|
+
)
|
|
487
|
+
return db
|
|
488
|
+
|
|
489
|
+
def create_database_schema_request(self, name, text, db):
|
|
490
|
+
self.db_name = db.name.__root__
|
|
491
|
+
db_fqn = fqn.build(
|
|
492
|
+
self.metadata,
|
|
493
|
+
entity_type=Database,
|
|
494
|
+
service_name=self.database_service.name.__root__,
|
|
495
|
+
database_name=db.name.__root__,
|
|
496
|
+
)
|
|
497
|
+
schema = CreateDatabaseSchemaRequest(
|
|
498
|
+
name=name,
|
|
499
|
+
description=text,
|
|
500
|
+
database=db_fqn,
|
|
501
|
+
)
|
|
502
|
+
return schema
|
|
503
|
+
|
|
504
|
+
def create_table_request(self, name, text, schema, table):
|
|
505
|
+
dbschema_fqn = fqn.build(
|
|
506
|
+
self.metadata,
|
|
507
|
+
entity_type=DatabaseSchema,
|
|
508
|
+
service_name=self.database_service.name.__root__,
|
|
509
|
+
database_name=self.db_name,
|
|
510
|
+
schema_name=schema.name.__root__,
|
|
511
|
+
)
|
|
512
|
+
table_request = CreateTableRequest(
|
|
513
|
+
name=name,
|
|
514
|
+
description=text,
|
|
515
|
+
columns=table["columns"],
|
|
516
|
+
databaseSchema=dbschema_fqn,
|
|
517
|
+
tableConstraints=table.get("tableConstraints"),
|
|
518
|
+
tableType=table["tableType"],
|
|
519
|
+
)
|
|
520
|
+
return table_request
|
|
521
|
+
|
|
522
|
+
def create_dashboard_data_model_request(self, name, text, data_model):
|
|
523
|
+
data_model_request = CreateDashboardDataModelRequest(
|
|
524
|
+
name=name,
|
|
525
|
+
description=text,
|
|
526
|
+
columns=data_model["columns"],
|
|
527
|
+
dataModelType=data_model["dataModelType"],
|
|
528
|
+
sql=data_model["sql"],
|
|
529
|
+
serviceType=data_model["serviceType"],
|
|
530
|
+
service=self.dashboard_service.fullyQualifiedName,
|
|
531
|
+
)
|
|
532
|
+
return data_model_request
|
|
@@ -30,13 +30,16 @@ from metadata.generated.schema.entity.data.table import Column, Table, TableType
|
|
|
30
30
|
from metadata.generated.schema.entity.services.connections.database.glueConnection import (
|
|
31
31
|
GlueConnection,
|
|
32
32
|
)
|
|
33
|
+
from metadata.generated.schema.entity.services.ingestionPipelines.status import (
|
|
34
|
+
StackTraceError,
|
|
35
|
+
)
|
|
33
36
|
from metadata.generated.schema.metadataIngestion.databaseServiceMetadataPipeline import (
|
|
34
37
|
DatabaseServiceMetadataPipeline,
|
|
35
38
|
)
|
|
36
39
|
from metadata.generated.schema.metadataIngestion.workflow import (
|
|
37
40
|
Source as WorkflowSource,
|
|
38
41
|
)
|
|
39
|
-
from metadata.ingestion.api.models import Either
|
|
42
|
+
from metadata.ingestion.api.models import Either
|
|
40
43
|
from metadata.ingestion.api.steps import InvalidSourceException
|
|
41
44
|
from metadata.ingestion.models.ometa_classification import OMetaTagAndClassification
|
|
42
45
|
from metadata.ingestion.ometa.ometa_api import OpenMetadata
|
|
@@ -143,7 +146,7 @@ class GlueSource(DatabaseServiceSource):
|
|
|
143
146
|
StackTraceError(
|
|
144
147
|
name=schema.CatalogId,
|
|
145
148
|
error=f"Unexpected exception to get database name [{schema.CatalogId}]: {exc}",
|
|
146
|
-
|
|
149
|
+
stackTrace=traceback.format_exc(),
|
|
147
150
|
)
|
|
148
151
|
)
|
|
149
152
|
|
|
@@ -191,7 +194,7 @@ class GlueSource(DatabaseServiceSource):
|
|
|
191
194
|
StackTraceError(
|
|
192
195
|
name=schema.Name,
|
|
193
196
|
error=f"Unexpected exception to get database schema [{schema.Name}]: {exc}",
|
|
194
|
-
|
|
197
|
+
stackTrace=traceback.format_exc(),
|
|
195
198
|
)
|
|
196
199
|
)
|
|
197
200
|
|
|
@@ -273,7 +276,7 @@ class GlueSource(DatabaseServiceSource):
|
|
|
273
276
|
StackTraceError(
|
|
274
277
|
name=table.Name,
|
|
275
278
|
error=f"Unexpected exception to get table [{table.Name}]: {exc}",
|
|
276
|
-
|
|
279
|
+
stackTrace=traceback.format_exc(),
|
|
277
280
|
)
|
|
278
281
|
)
|
|
279
282
|
|
|
@@ -316,7 +319,7 @@ class GlueSource(DatabaseServiceSource):
|
|
|
316
319
|
left=StackTraceError(
|
|
317
320
|
name=table_name,
|
|
318
321
|
error=f"Unexpected exception to yield table [{table_name}]: {exc}",
|
|
319
|
-
|
|
322
|
+
stackTrace=traceback.format_exc(),
|
|
320
323
|
)
|
|
321
324
|
)
|
|
322
325
|
|
|
@@ -126,7 +126,6 @@ def get_metastore_connection(connection: Any) -> Engine:
|
|
|
126
126
|
|
|
127
127
|
@get_metastore_connection.register
|
|
128
128
|
def _(connection: PostgresConnection):
|
|
129
|
-
|
|
130
129
|
# import required to load sqlalchemy plugin
|
|
131
130
|
# pylint: disable=import-outside-toplevel,unused-import
|
|
132
131
|
from metadata.ingestion.source.database.hive.metastore_dialects.postgres import ( # nopycln: import
|
|
@@ -153,7 +152,6 @@ def _(connection: PostgresConnection):
|
|
|
153
152
|
|
|
154
153
|
@get_metastore_connection.register
|
|
155
154
|
def _(connection: MysqlConnection):
|
|
156
|
-
|
|
157
155
|
# import required to load sqlalchemy plugin
|
|
158
156
|
# pylint: disable=import-outside-toplevel,unused-import
|
|
159
157
|
from metadata.ingestion.source.database.hive.metastore_dialects.mysql import ( # nopycln: import
|
|
@@ -26,6 +26,7 @@ _type_map.update(
|
|
|
26
26
|
"binary": types.BINARY,
|
|
27
27
|
"char": types.CHAR,
|
|
28
28
|
"varchar": types.VARCHAR,
|
|
29
|
+
"decimal": types.DECIMAL,
|
|
29
30
|
}
|
|
30
31
|
)
|
|
31
32
|
|
|
@@ -61,6 +62,8 @@ def get_columns(
|
|
|
61
62
|
if attype == "decimal":
|
|
62
63
|
prec, scale = charlen.split(",")
|
|
63
64
|
args = (int(prec), int(scale))
|
|
65
|
+
elif attype.startswith("struct"):
|
|
66
|
+
args = []
|
|
64
67
|
else:
|
|
65
68
|
args = (int(charlen),)
|
|
66
69
|
coltype = coltype(*args)
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
# Copyright 2021 Collate
|
|
2
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
3
|
+
# you may not use this file except in compliance with the License.
|
|
4
|
+
# You may obtain a copy of the License at
|
|
5
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
6
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
7
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
8
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
9
|
+
# See the License for the specific language governing permissions and
|
|
10
|
+
# limitations under the License.
|
|
11
|
+
|
|
12
|
+
"""
|
|
13
|
+
Iceberg Catalog Factory.
|
|
14
|
+
"""
|
|
15
|
+
from typing import Dict, Type
|
|
16
|
+
|
|
17
|
+
from pyiceberg.catalog import Catalog
|
|
18
|
+
|
|
19
|
+
from metadata.generated.schema.entity.services.connections.database.iceberg.dynamoDbCatalogConnection import (
|
|
20
|
+
DynamoDbCatalogConnection,
|
|
21
|
+
)
|
|
22
|
+
from metadata.generated.schema.entity.services.connections.database.iceberg.glueCatalogConnection import (
|
|
23
|
+
GlueCatalogConnection,
|
|
24
|
+
)
|
|
25
|
+
from metadata.generated.schema.entity.services.connections.database.iceberg.hiveCatalogConnection import (
|
|
26
|
+
HiveCatalogConnection,
|
|
27
|
+
)
|
|
28
|
+
from metadata.generated.schema.entity.services.connections.database.iceberg.icebergCatalog import (
|
|
29
|
+
IcebergCatalog,
|
|
30
|
+
)
|
|
31
|
+
from metadata.generated.schema.entity.services.connections.database.iceberg.restCatalogConnection import (
|
|
32
|
+
RestCatalogConnection,
|
|
33
|
+
)
|
|
34
|
+
from metadata.ingestion.source.database.iceberg.catalog.base import IcebergCatalogBase
|
|
35
|
+
from metadata.ingestion.source.database.iceberg.catalog.dynamodb import (
|
|
36
|
+
IcebergDynamoDbCatalog,
|
|
37
|
+
)
|
|
38
|
+
from metadata.ingestion.source.database.iceberg.catalog.glue import IcebergGlueCatalog
|
|
39
|
+
from metadata.ingestion.source.database.iceberg.catalog.hive import IcebergHiveCatalog
|
|
40
|
+
from metadata.ingestion.source.database.iceberg.catalog.rest import IcebergRestCatalog
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
class IcebergCatalogFactory:
|
|
44
|
+
"""Factory Class to get any PyIceberg implemented Catalog."""
|
|
45
|
+
|
|
46
|
+
catalog_type_map: Dict[str, Type[IcebergCatalogBase]] = {
|
|
47
|
+
RestCatalogConnection.__name__: IcebergRestCatalog,
|
|
48
|
+
HiveCatalogConnection.__name__: IcebergHiveCatalog,
|
|
49
|
+
GlueCatalogConnection.__name__: IcebergGlueCatalog,
|
|
50
|
+
DynamoDbCatalogConnection.__name__: IcebergDynamoDbCatalog,
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
@classmethod
|
|
54
|
+
def from_connection(cls, catalog: IcebergCatalog) -> Catalog:
|
|
55
|
+
"""Returns a PyIceberg Catalog from the given catalog configuration."""
|
|
56
|
+
catalog_type = cls.catalog_type_map.get(
|
|
57
|
+
catalog.connection.__class__.__name__, None
|
|
58
|
+
)
|
|
59
|
+
|
|
60
|
+
if not catalog_type:
|
|
61
|
+
raise NotImplementedError(
|
|
62
|
+
f"Iceberg Catalog of type ['{catalog.connection.__class__.__name__}'] Not Implemmented."
|
|
63
|
+
)
|
|
64
|
+
|
|
65
|
+
return catalog_type.get_catalog(catalog)
|