openmetadata-ingestion 1.9.0.0__py3-none-any.whl → 1.10.14.4__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.
- airflow_provider_openmetadata/lineage/backend.py +4 -2
- airflow_provider_openmetadata/lineage/config/loader.py +8 -1
- metadata/cli/common.py +0 -1
- metadata/clients/aws_client.py +4 -0
- metadata/data_quality/interface/test_suite_interface.py +9 -3
- metadata/data_quality/validations/models.py +11 -1
- metadata/data_quality/validations/runtime_param_setter/base_diff_params_setter.py +39 -13
- metadata/data_quality/validations/runtime_param_setter/param_setter_factory.py +9 -0
- metadata/data_quality/validations/runtime_param_setter/table_custom_sql_query_params_setter.py +31 -0
- metadata/data_quality/validations/runtime_param_setter/table_diff_params_setter.py +20 -13
- metadata/data_quality/validations/table/base/tableCustomSQLQuery.py +132 -3
- metadata/data_quality/validations/table/pandas/tableCustomSQLQuery.py +38 -0
- metadata/data_quality/validations/table/sqlalchemy/tableCustomSQLQuery.py +320 -0
- metadata/data_quality/validations/table/sqlalchemy/tableDiff.py +4 -0
- metadata/examples/workflows/airbyte.yaml +8 -2
- metadata/examples/workflows/bigquery.yaml +1 -0
- metadata/examples/workflows/confluent_cdc.yaml +70 -0
- metadata/examples/workflows/confluent_cdc_cloud.yaml +86 -0
- metadata/examples/workflows/confluent_cdc_local.yaml +120 -0
- metadata/examples/workflows/databricks.yaml +9 -1
- metadata/examples/workflows/databricks_usage.yaml +7 -1
- metadata/examples/workflows/exasol.yaml +25 -0
- metadata/examples/workflows/grafana.yaml +115 -0
- metadata/examples/workflows/kinesisfirehose.yaml +99 -0
- metadata/examples/workflows/metabase.yaml +1 -0
- metadata/examples/workflows/my_confluent_cloud.yaml +49 -0
- metadata/examples/workflows/unity_catalog.yaml +7 -1
- metadata/examples/workflows/unity_catalog_usage.yaml +7 -1
- metadata/generated/antlr/EntityLinkLexer.py +419 -353
- 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/addTagToAssetsRequest.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 +52 -1
- metadata/generated/schema/api/classification/createTag.py +29 -2
- metadata/generated/schema/api/classification/createTagWithRecognizers.py +67 -0
- metadata/generated/schema/api/classification/loadTags.py +1 -1
- metadata/generated/schema/api/configuration/__init__.py +3 -0
- metadata/generated/schema/api/configuration/rdfConfiguration.py +47 -0
- metadata/generated/schema/api/createBot.py +1 -1
- 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/createAPICollection.py +1 -1
- metadata/generated/schema/api/data/createAPIEndpoint.py +1 -1
- metadata/generated/schema/api/data/createChart.py +1 -1
- metadata/generated/schema/api/data/createContainer.py +1 -1
- metadata/generated/schema/api/data/createCustomProperty.py +1 -1
- metadata/generated/schema/api/data/createDashboard.py +1 -1
- metadata/generated/schema/api/data/createDashboardDataModel.py +5 -1
- metadata/generated/schema/api/data/createDataContract.py +33 -3
- metadata/generated/schema/api/data/createDatabase.py +1 -1
- metadata/generated/schema/api/data/createDatabaseSchema.py +1 -1
- metadata/generated/schema/api/data/createDirectory.py +1 -1
- metadata/generated/schema/api/data/createEntityProfile.py +36 -0
- metadata/generated/schema/api/data/createFile.py +1 -1
- metadata/generated/schema/api/data/createGlossary.py +1 -1
- metadata/generated/schema/api/data/createGlossaryTerm.py +1 -1
- metadata/generated/schema/api/data/createMetric.py +12 -1
- metadata/generated/schema/api/data/createMlModel.py +1 -1
- metadata/generated/schema/api/data/createPipeline.py +1 -1
- metadata/generated/schema/api/data/createQuery.py +1 -1
- metadata/generated/schema/api/data/createQueryCostRecord.py +1 -1
- metadata/generated/schema/api/data/createSearchIndex.py +1 -1
- metadata/generated/schema/api/data/createSpreadsheet.py +9 -1
- metadata/generated/schema/api/data/createStoredProcedure.py +1 -1
- metadata/generated/schema/api/data/createTable.py +1 -1
- metadata/generated/schema/api/data/createTableProfile.py +1 -1
- metadata/generated/schema/api/data/createTopic.py +1 -1
- metadata/generated/schema/api/data/createWorksheet.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/data/updateColumn.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/custom/__init__.py +1 -1
- metadata/generated/schema/api/dataInsight/custom/createDataInsightCustomChart.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 +7 -1
- metadata/generated/schema/api/domains/createDomain.py +1 -1
- metadata/generated/schema/api/entityRelationship/__init__.py +1 -1
- metadata/generated/schema/api/entityRelationship/entityRelationshipDirection.py +1 -1
- metadata/generated/schema/api/entityRelationship/esEntityRelationshipData.py +1 -1
- metadata/generated/schema/api/entityRelationship/relationshipRef.py +1 -1
- metadata/generated/schema/api/entityRelationship/searchEntityRelationshipRequest.py +1 -1
- metadata/generated/schema/api/entityRelationship/searchEntityRelationshipResult.py +1 -1
- metadata/generated/schema/api/entityRelationship/searchSchemaEntityRelationshipResult.py +1 -1
- metadata/generated/schema/api/events/__init__.py +3 -0
- metadata/generated/schema/api/events/createNotificationTemplate.py +61 -0
- metadata/generated/schema/api/events/notificationTemplateValidationRequest.py +30 -0
- metadata/generated/schema/api/events/notificationTemplateValidationResponse.py +36 -0
- metadata/generated/schema/api/feed/__init__.py +1 -1
- metadata/generated/schema/api/feed/closeTask.py +1 -1
- metadata/generated/schema/api/feed/createPost.py +1 -1
- metadata/generated/schema/api/feed/createSuggestion.py +1 -1
- metadata/generated/schema/api/feed/createThread.py +1 -1
- metadata/generated/schema/api/feed/resolveTask.py +1 -1
- metadata/generated/schema/api/feed/threadCount.py +1 -1
- metadata/generated/schema/api/governance/__init__.py +1 -1
- metadata/generated/schema/api/governance/createWorkflowDefinition.py +20 -5
- metadata/generated/schema/api/governance/createWorkflowInstanceState.py +1 -1
- metadata/generated/schema/api/lineage/__init__.py +1 -1
- metadata/generated/schema/api/lineage/addLineage.py +1 -1
- metadata/generated/schema/api/lineage/entityCountLineageRequest.py +66 -0
- metadata/generated/schema/api/lineage/esLineageData.py +1 -1
- metadata/generated/schema/api/lineage/lineageDirection.py +1 -1
- metadata/generated/schema/api/lineage/lineagePaginationInfo.py +57 -0
- metadata/generated/schema/api/lineage/nodeInformation.py +10 -2
- metadata/generated/schema/api/lineage/searchLineageRequest.py +1 -1
- metadata/generated/schema/api/lineage/searchLineageResult.py +1 -1
- metadata/generated/schema/api/mcp/__init__.py +1 -1
- metadata/generated/schema/api/mcp/mcpSearchResponse.py +118 -0
- metadata/generated/schema/api/mcp/mcpToolDefinition.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/rdf/__init__.py +3 -0
- metadata/generated/schema/api/rdf/sparqlQuery.py +57 -0
- metadata/generated/schema/api/rdf/sparqlResponse.py +56 -0
- metadata/generated/schema/api/scim/__init__.py +1 -1
- metadata/generated/schema/api/scim/scimGroup.py +1 -1
- metadata/generated/schema/api/scim/scimPatchOp.py +1 -1
- metadata/generated/schema/api/scim/scimUser.py +1 -1
- metadata/generated/schema/api/search/__init__.py +1 -1
- metadata/generated/schema/api/search/previewSearchRequest.py +1 -1
- metadata/generated/schema/api/services/__init__.py +1 -1
- metadata/generated/schema/api/services/createApiService.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/createDriveService.py +9 -2
- 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/createSecurityService.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 +8 -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 +1 -1
- metadata/generated/schema/api/teams/createTeam.py +1 -1
- metadata/generated/schema/api/teams/createUser.py +1 -1
- metadata/generated/schema/api/tests/__init__.py +1 -1
- metadata/generated/schema/api/tests/createCustomMetric.py +1 -1
- metadata/generated/schema/api/tests/createLogicalTestCases.py +1 -1
- metadata/generated/schema/api/tests/createTestCase.py +5 -1
- metadata/generated/schema/api/tests/createTestCaseResolutionStatus.py +1 -1
- metadata/generated/schema/api/tests/createTestCaseResult.py +1 -1
- metadata/generated/schema/api/tests/createTestDefinition.py +1 -1
- metadata/generated/schema/api/tests/createTestSuite.py +5 -1
- metadata/generated/schema/api/tests/moveGlossaryTermRequest.py +1 -1
- metadata/generated/schema/api/validateGlossaryTagsRequest.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 +1 -1
- 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 +1 -1
- 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/supportToken.py +1 -1
- metadata/generated/schema/auth/tokenRefreshRequest.py +1 -1
- metadata/generated/schema/configuration/__init__.py +1 -1
- metadata/generated/schema/configuration/aiPlatformConfiguration.py +1 -1
- metadata/generated/schema/configuration/appsPrivateConfiguration.py +1 -1
- metadata/generated/schema/configuration/assetCertificationSettings.py +1 -1
- metadata/generated/schema/configuration/authConfig.py +1 -1
- metadata/generated/schema/configuration/authenticationConfiguration.py +49 -13
- metadata/generated/schema/configuration/authorizerConfiguration.py +39 -10
- metadata/generated/schema/configuration/cacheConfiguration.py +1 -1
- metadata/generated/schema/configuration/changeEventConfiguration.py +1 -1
- metadata/generated/schema/configuration/dataQualityConfiguration.py +1 -1
- metadata/generated/schema/configuration/elasticSearchConfiguration.py +19 -1
- metadata/generated/schema/configuration/entityRulesSettings.py +1 -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 +1 -1
- 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/limitsConfiguration.py +1 -1
- metadata/generated/schema/configuration/lineageSettings.py +9 -1
- metadata/generated/schema/configuration/logStorageConfiguration.py +105 -0
- metadata/generated/schema/configuration/loginConfiguration.py +1 -1
- metadata/generated/schema/configuration/logoConfiguration.py +1 -1
- metadata/generated/schema/configuration/openMetadataBaseUrlConfiguration.py +1 -1
- metadata/generated/schema/configuration/opertionalConfiguration.py +1 -1
- metadata/generated/schema/configuration/opsConfig.py +1 -1
- metadata/generated/schema/configuration/pipelineServiceClientConfiguration.py +9 -2
- metadata/generated/schema/configuration/profilerConfiguration.py +2 -1
- metadata/generated/schema/configuration/searchSettings.py +7 -2
- metadata/generated/schema/configuration/securityConfiguration.py +26 -0
- 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/configuration/themeConfiguration.py +1 -1
- metadata/generated/schema/configuration/uiThemePreference.py +1 -1
- metadata/generated/schema/configuration/workflowSettings.py +15 -1
- metadata/generated/schema/dataInsight/__init__.py +1 -1
- metadata/generated/schema/dataInsight/custom/__init__.py +1 -1
- metadata/generated/schema/dataInsight/custom/dataInsightCustomChart.py +1 -1
- metadata/generated/schema/dataInsight/custom/dataInsightCustomChartResult.py +1 -1
- metadata/generated/schema/dataInsight/custom/dataInsightCustomChartResultList.py +1 -1
- metadata/generated/schema/dataInsight/custom/formulaHolder.py +1 -1
- metadata/generated/schema/dataInsight/custom/lineChart.py +1 -1
- metadata/generated/schema/dataInsight/custom/summaryCard.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/unusedAssets.py +1 -1
- metadata/generated/schema/email/__init__.py +1 -1
- metadata/generated/schema/email/emailRequest.py +1 -1
- metadata/generated/schema/email/emailTemplate.py +1 -1
- metadata/generated/schema/email/emailTemplatePlaceholder.py +1 -1
- metadata/generated/schema/email/smtpSettings.py +1 -1
- metadata/generated/schema/email/templateValidationReponse.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 +15 -1
- metadata/generated/schema/entity/applications/appExtension.py +1 -1
- 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 +1 -1
- metadata/generated/schema/entity/applications/configuration/external/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addCustomProperties.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addDataProductAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addDescriptionAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addDomainAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addOwnerAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addTagsAction.py +7 -2
- metadata/generated/schema/entity/applications/configuration/external/automator/addTermsAction.py +48 -0
- metadata/generated/schema/entity/applications/configuration/external/automator/addTestCaseAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/addTierAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/lineagePropagationAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/mlTaggingAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/propagationStopConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeCustomPropertiesAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeDataProductAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeDescriptionAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeDomainAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeOwnerAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeTagsAction.py +6 -2
- metadata/generated/schema/entity/applications/configuration/external/automator/removeTermsAction.py +59 -0
- metadata/generated/schema/entity/applications/configuration/external/automator/removeTestCaseAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automator/removeTierAction.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/automatorAppConfig.py +7 -1
- metadata/generated/schema/entity/applications/configuration/external/collateAIAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterAppConfig.py +103 -0
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterConnectors/__init__.py +3 -0
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterConnectors/bigQueryConnection.py +90 -0
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterConnectors/databricksConnection.py +101 -0
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterConnectors/redshiftConnection.py +76 -0
- metadata/generated/schema/entity/applications/configuration/external/metadataExporterConnectors/snowflakeConnection.py +115 -0
- metadata/generated/schema/entity/applications/configuration/external/slackAppTokenConfiguration.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/autoPilotAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/collateAIQualityAgentAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/collateAITierAgentAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/dataInsightsAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/dataInsightsReportAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/dataRetentionConfiguration.py +15 -1
- metadata/generated/schema/entity/applications/configuration/internal/helloPipelinesConfiguration.py +1 -1
- metadata/generated/schema/entity/applications/configuration/internal/searchIndexingAppConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/external/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/external/collateAIAppPrivateConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/internal/__init__.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/internal/collateAITierAgentAppPrivateConfig.py +1 -1
- metadata/generated/schema/entity/applications/configuration/private/limits.py +1 -1
- metadata/generated/schema/entity/applications/createAppRequest.py +9 -2
- metadata/generated/schema/entity/applications/jobStatus.py +1 -1
- 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 +8 -1
- metadata/generated/schema/entity/applications/marketplace/createAppMarketPlaceDefinitionReq.py +8 -1
- metadata/generated/schema/entity/applications/scheduledExecutionContext.py +1 -1
- metadata/generated/schema/entity/automations/__init__.py +1 -1
- metadata/generated/schema/entity/automations/queryRunnerRequest.py +46 -0
- metadata/generated/schema/entity/automations/response/__init__.py +3 -0
- metadata/generated/schema/entity/automations/response/queryRunnerResponse.py +51 -0
- metadata/generated/schema/entity/automations/testServiceConnection.py +3 -1
- metadata/generated/schema/entity/automations/testSparkEngineConnection.py +31 -0
- metadata/generated/schema/entity/automations/workflow.py +8 -2
- metadata/generated/schema/entity/bot.py +1 -1
- metadata/generated/schema/entity/classification/__init__.py +1 -1
- metadata/generated/schema/entity/classification/classification.py +57 -2
- metadata/generated/schema/entity/classification/tag.py +41 -3
- metadata/generated/schema/entity/data/__init__.py +1 -1
- metadata/generated/schema/entity/data/apiCollection.py +8 -1
- metadata/generated/schema/entity/data/apiEndpoint.py +8 -1
- metadata/generated/schema/entity/data/chart.py +12 -1
- metadata/generated/schema/entity/data/container.py +7 -1
- metadata/generated/schema/entity/data/dashboard.py +6 -1
- metadata/generated/schema/entity/data/dashboardDataModel.py +13 -1
- metadata/generated/schema/entity/data/dataContract.py +238 -2
- metadata/generated/schema/entity/data/database.py +6 -1
- metadata/generated/schema/entity/data/databaseSchema.py +8 -1
- metadata/generated/schema/entity/data/directory.py +6 -1
- metadata/generated/schema/entity/data/file.py +6 -1
- metadata/generated/schema/entity/data/glossary.py +6 -1
- metadata/generated/schema/entity/data/glossaryTerm.py +8 -12
- metadata/generated/schema/entity/data/metric.py +18 -1
- metadata/generated/schema/entity/data/mlmodel.py +6 -1
- metadata/generated/schema/entity/data/pipeline.py +136 -2
- metadata/generated/schema/entity/data/query.py +6 -1
- metadata/generated/schema/entity/data/queryCostRecord.py +1 -1
- metadata/generated/schema/entity/data/queryCostSearchResult.py +1 -1
- metadata/generated/schema/entity/data/report.py +6 -1
- metadata/generated/schema/entity/data/searchIndex.py +8 -1
- metadata/generated/schema/entity/data/spreadsheet.py +10 -4
- metadata/generated/schema/entity/data/storedProcedure.py +9 -1
- metadata/generated/schema/entity/data/table.py +102 -1
- metadata/generated/schema/entity/data/topic.py +6 -1
- metadata/generated/schema/entity/data/worksheet.py +6 -1
- metadata/generated/schema/entity/datacontract/__init__.py +1 -1
- metadata/generated/schema/entity/datacontract/dataContractResult.py +1 -1
- metadata/generated/schema/entity/datacontract/qualityValidation.py +1 -1
- metadata/generated/schema/entity/datacontract/schemaValidation.py +1 -1
- metadata/generated/schema/entity/datacontract/semanticsValidation.py +1 -1
- metadata/generated/schema/entity/datacontract/slaValidation.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 +155 -2
- metadata/generated/schema/entity/domains/domain.py +1 -1
- metadata/generated/schema/entity/events/__init__.py +1 -1
- metadata/generated/schema/entity/events/notificationTemplate.py +111 -0
- metadata/generated/schema/entity/events/webhook.py +1 -1
- metadata/generated/schema/entity/feed/__init__.py +1 -1
- metadata/generated/schema/entity/feed/assets.py +1 -1
- metadata/generated/schema/entity/feed/customProperty.py +1 -1
- metadata/generated/schema/entity/feed/description.py +1 -1
- metadata/generated/schema/entity/feed/domain.py +1 -1
- metadata/generated/schema/entity/feed/entityInfo.py +1 -1
- metadata/generated/schema/entity/feed/owner.py +1 -1
- metadata/generated/schema/entity/feed/suggestion.py +1 -1
- metadata/generated/schema/entity/feed/tag.py +1 -1
- metadata/generated/schema/entity/feed/testCaseResult.py +1 -1
- metadata/generated/schema/entity/feed/thread.py +1 -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 +1 -1
- metadata/generated/schema/entity/policies/policy.py +1 -1
- metadata/generated/schema/entity/services/__init__.py +1 -1
- metadata/generated/schema/entity/services/apiService.py +1 -1
- metadata/generated/schema/entity/services/connections/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/api/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/api/restConnection.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 +2 -2
- metadata/generated/schema/entity/services/connections/dashboard/domoDashboardConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/grafanaConnection.py +77 -0
- metadata/generated/schema/entity/services/connections/dashboard/hexConnection.py +76 -0
- metadata/generated/schema/entity/services/connections/dashboard/lightdashConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/lookerConnection.py +13 -2
- metadata/generated/schema/entity/services/connections/dashboard/metabaseConnection.py +17 -4
- metadata/generated/schema/entity/services/connections/dashboard/microStrategyConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/modeConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerBIConnection.py +9 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerBIReportServerConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerbi/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerbi/azureConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerbi/bucketDetails.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerbi/gcsConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/powerbi/s3Config.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/qlikCloudConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/dashboard/qlikSenseConnection.py +1 -1
- 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/sigmaConnection.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/dashboard/thoughtSpotConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/athenaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/azureSQLConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/bigQueryConnection.py +9 -1
- metadata/generated/schema/entity/services/connections/database/bigTableConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/cassandra/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/cassandra/cloudConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/cassandraConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/clickhouseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/cockroachConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/common/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/common/azureConfig.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/common/noConfigAuthenticationTypes.py +1 -1
- metadata/generated/schema/entity/services/connections/database/couchbaseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/customDatabaseConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/database/databricks/__init__.py +3 -0
- metadata/generated/schema/entity/services/connections/database/databricks/azureAdSetup.py +37 -0
- metadata/generated/schema/entity/services/connections/database/databricks/databricksOAuth.py +30 -0
- metadata/generated/schema/entity/services/connections/database/databricks/personalAccessToken.py +23 -0
- metadata/generated/schema/entity/services/connections/database/databricksConnection.py +14 -6
- 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 +1 -1
- metadata/generated/schema/entity/services/connections/database/db2Connection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/deltaLakeConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/deltalake/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/deltalake/metastoreConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/deltalake/storageConfig.py +1 -1
- metadata/generated/schema/entity/services/connections/database/domoDatabaseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/dorisConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/druidConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/dynamoDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/epicConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/exasolConnection.py +4 -1
- metadata/generated/schema/entity/services/connections/database/glueConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/greenplumConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/hiveConnection.py +16 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/dynamoDbCatalogConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/glueCatalogConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/hiveCatalogConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/icebergCatalog.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/icebergFileSystem.py +1 -1
- metadata/generated/schema/entity/services/connections/database/iceberg/restCatalogConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/icebergConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/impalaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/mariaDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/mongoDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/mssqlConnection.py +26 -1
- metadata/generated/schema/entity/services/connections/database/mysqlConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/oracleConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/pinotDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/postgresConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/prestoConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/redshiftConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/salesforceConnection.py +25 -6
- metadata/generated/schema/entity/services/connections/database/sapErpConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sapHana/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sapHana/sapHanaHDBConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sapHana/sapHanaSQLConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sapHanaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/sasConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/serviceNowConnection.py +90 -0
- metadata/generated/schema/entity/services/connections/database/singleStoreConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/snowflakeConnection.py +10 -2
- metadata/generated/schema/entity/services/connections/database/sqliteConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/ssasConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/synapseConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/teradataConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/timescaleConnection.py +162 -0
- metadata/generated/schema/entity/services/connections/database/trinoConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/database/unityCatalogConnection.py +14 -6
- metadata/generated/schema/entity/services/connections/database/verticaConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/drive/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/drive/customDriveConnection.py +35 -2
- metadata/generated/schema/entity/services/connections/drive/googleDriveConnection.py +34 -1
- metadata/generated/schema/entity/services/connections/drive/sharePointConnection.py +34 -1
- metadata/generated/schema/entity/services/connections/messaging/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/messaging/customMessagingConnection.py +2 -2
- 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/alationSinkConnection.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/collibraConnection.py +73 -0
- metadata/generated/schema/entity/services/connections/metadata/metadataESConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/metadata/openMetadataConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/mlmodel/customMlModelConnection.py +2 -2
- 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/mlmodel/vertexaiConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/airbyte/__init__.py +3 -0
- metadata/generated/schema/entity/services/connections/pipeline/airbyte/basicAuth.py +26 -0
- metadata/generated/schema/entity/services/connections/pipeline/airbyte/oauthClientAuth.py +30 -0
- metadata/generated/schema/entity/services/connections/pipeline/airbyteConnection.py +16 -10
- metadata/generated/schema/entity/services/connections/pipeline/airflowConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/backendConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/customPipelineConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/pipeline/dagsterConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/databricksPipelineConnection.py +11 -1
- metadata/generated/schema/entity/services/connections/pipeline/datafactoryConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/dbtCloudConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/domoPipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/fivetranConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/pipeline/flinkConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/gluePipelineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/kafkaConnectConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/kinesisFirehoseConnection.py +58 -0
- metadata/generated/schema/entity/services/connections/pipeline/matillion/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/matillion/matillionETL.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/matillionConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/nifi/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/nifi/basicAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/nifi/clientCertificateAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/nifiConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/openLineageConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/snowplowConnection.py +93 -0
- metadata/generated/schema/entity/services/connections/pipeline/sparkConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/splineConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/ssisConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/stitchConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/pipeline/wherescapeConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/search/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/search/customSearchConnection.py +2 -2
- 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/security/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/security/ranger/__init__.py +1 -1
- metadata/generated/schema/entity/services/connections/security/ranger/basicAuth.py +1 -1
- metadata/generated/schema/entity/services/connections/security/rangerConnection.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/adlsConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/customStorageConnection.py +2 -2
- metadata/generated/schema/entity/services/connections/storage/gcsConnection.py +1 -1
- metadata/generated/schema/entity/services/connections/storage/s3Connection.py +10 -2
- 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 +7 -1
- metadata/generated/schema/entity/services/databaseService.py +7 -1
- metadata/generated/schema/entity/services/driveService.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/__init__.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/ingestionPipeline.py +8 -1
- metadata/generated/schema/entity/services/ingestionPipelines/pipelineServiceClientResponse.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/reverseIngestionResponse.py +1 -1
- metadata/generated/schema/entity/services/ingestionPipelines/status.py +1 -1
- metadata/generated/schema/entity/services/messagingService.py +1 -1
- metadata/generated/schema/entity/services/metadataService.py +4 -1
- metadata/generated/schema/entity/services/mlmodelService.py +1 -1
- metadata/generated/schema/entity/services/pipelineService.py +7 -1
- metadata/generated/schema/entity/services/searchService.py +1 -1
- metadata/generated/schema/entity/services/securityService.py +1 -1
- metadata/generated/schema/entity/services/serviceType.py +1 -1
- metadata/generated/schema/entity/services/storageService.py +1 -1
- 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 +1 -1
- 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 +1 -1
- metadata/generated/schema/entity/type.py +1 -1
- 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 +1 -1
- metadata/generated/schema/events/api/__init__.py +1 -1
- metadata/generated/schema/events/api/createEventSubscription.py +1 -1
- metadata/generated/schema/events/api/eventSubscriptionDiagnosticInfo.py +1 -1
- metadata/generated/schema/events/api/eventsRecord.py +1 -1
- metadata/generated/schema/events/api/testEventSubscriptionDestination.py +1 -1
- metadata/generated/schema/events/api/typedEvent.py +1 -1
- metadata/generated/schema/events/emailAlertConfig.py +1 -1
- metadata/generated/schema/events/eventFilterRule.py +1 -1
- metadata/generated/schema/events/eventSubscription.py +16 -1
- metadata/generated/schema/events/eventSubscriptionOffset.py +1 -1
- metadata/generated/schema/events/failedEvent.py +1 -1
- metadata/generated/schema/events/failedEventResponse.py +1 -1
- metadata/generated/schema/events/filterResourceDescriptor.py +1 -1
- metadata/generated/schema/events/statusContext.py +1 -1
- metadata/generated/schema/events/subscriptionResourceDescriptor.py +1 -1
- metadata/generated/schema/events/subscriptionStatus.py +1 -1
- metadata/generated/schema/events/testDestinationStatus.py +1 -1
- metadata/generated/schema/governance/workflows/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/edge.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodeSubType.py +4 -1
- metadata/generated/schema/governance/workflows/elements/nodeType.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/checkEntityAttributesTask.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/createAndRunIngestionPipelineTask.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/dataCompletenessTask.py +124 -0
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/rollbackEntityTask.py +54 -0
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/runAppTask.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/setEntityAttributeTask.py +66 -0
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/setEntityCertificationTask.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/automatedTask/setGlossaryTermStatusTask.py +11 -12
- metadata/generated/schema/governance/workflows/elements/nodes/endEvent/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/endEvent/endEvent.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/gateway/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/gateway/parallelGateway.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/startEvent/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/startEvent/startEvent.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/userTask/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/nodes/userTask/userApprovalTask.py +22 -1
- metadata/generated/schema/governance/workflows/elements/triggers/__init__.py +1 -1
- metadata/generated/schema/governance/workflows/elements/triggers/eventBasedEntityTrigger.py +90 -7
- metadata/generated/schema/governance/workflows/elements/triggers/noOpTrigger.py +1 -1
- metadata/generated/schema/governance/workflows/elements/triggers/periodicBatchEntityTrigger.py +80 -7
- metadata/generated/schema/governance/workflows/workflowDefinition.py +1 -1
- metadata/generated/schema/governance/workflows/workflowInstance.py +1 -1
- metadata/generated/schema/governance/workflows/workflowInstanceState.py +1 -1
- metadata/generated/schema/jobs/__init__.py +1 -1
- metadata/generated/schema/jobs/backgroundJob.py +1 -1
- metadata/generated/schema/jobs/enumCleanupArgs.py +1 -1
- metadata/generated/schema/metadataIngestion/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/apiServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/application.py +8 -1
- metadata/generated/schema/metadataIngestion/applicationPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/dashboardServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/dataInsightPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/databaseServiceAutoClassificationPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/databaseServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/databaseServiceProfilerPipeline.py +1 -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 +1 -1
- 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 +1 -1
- 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 +1 -1
- metadata/generated/schema/metadataIngestion/driveServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/engine/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/engine/nativeEngineConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/engine/sparkEngineConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/messagingServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/metadataToElasticSearchPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/mlmodelServiceMetadataPipeline.py +29 -2
- metadata/generated/schema/metadataIngestion/pipelineServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/reverseIngestionPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/reverseingestionconfig/__init__.py +1 -1
- metadata/generated/schema/metadataIngestion/reverseingestionconfig/descriptionConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/reverseingestionconfig/ownerConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/reverseingestionconfig/tagsConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/searchServiceMetadataPipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/securityServiceMetadataPipeline.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 +17 -1
- metadata/generated/schema/metadataIngestion/storage/storageBucketDetails.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageMetadataADLSConfig.py +1 -1
- metadata/generated/schema/metadataIngestion/storage/storageMetadataGCSConfig.py +1 -1
- 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 +1 -1
- metadata/generated/schema/metadataIngestion/testSuitePipeline.py +1 -1
- metadata/generated/schema/metadataIngestion/workflow.py +8 -1
- metadata/generated/schema/monitoring/__init__.py +1 -1
- metadata/generated/schema/monitoring/eventMonitorProvider.py +1 -1
- metadata/generated/schema/scim/__init__.py +1 -1
- metadata/generated/schema/scim/scimConfiguration.py +1 -1
- metadata/generated/schema/search/__init__.py +1 -1
- metadata/generated/schema/search/aggregationRequest.py +1 -1
- metadata/generated/schema/search/searchRequest.py +8 -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/oidcClientConfig.py +6 -10
- 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 +6 -6
- 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 +10 -2
- metadata/generated/schema/security/credentials/gcpCredentials.py +1 -1
- metadata/generated/schema/security/credentials/gcpExternalAccount.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 +10 -2
- metadata/generated/schema/security/credentials/gitlabCredentials.py +10 -2
- metadata/generated/schema/security/credentials/kubernetesCredentials.py +1 -1
- metadata/generated/schema/security/sasl/__init__.py +1 -1
- metadata/generated/schema/security/sasl/saslClientConfig.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 +1 -1
- metadata/generated/schema/security/secrets/secretsManagerProvider.py +1 -1
- 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 +1 -1
- metadata/generated/schema/settings/__init__.py +1 -1
- metadata/generated/schema/settings/settings.py +5 -2
- metadata/generated/schema/system/__init__.py +1 -1
- metadata/generated/schema/system/entityError.py +1 -1
- metadata/generated/schema/system/eventPublisherJob.py +1 -1
- metadata/generated/schema/system/indexingError.py +1 -1
- metadata/generated/schema/system/limitsResponse.py +1 -1
- metadata/generated/schema/system/securityValidationResponse.py +62 -0
- metadata/generated/schema/system/ui/__init__.py +1 -1
- metadata/generated/schema/system/ui/knowledgePanel.py +1 -1
- metadata/generated/schema/system/ui/navigationItem.py +1 -1
- metadata/generated/schema/system/ui/page.py +8 -1
- metadata/generated/schema/system/ui/tab.py +1 -1
- metadata/generated/schema/system/ui/uiCustomization.py +1 -1
- metadata/generated/schema/system/validationResponse.py +8 -1
- metadata/generated/schema/tests/__init__.py +1 -1
- metadata/generated/schema/tests/assigned.py +1 -1
- metadata/generated/schema/tests/basic.py +20 -20
- metadata/generated/schema/tests/customMetric.py +1 -1
- metadata/generated/schema/tests/dataQualityReport.py +1 -1
- metadata/generated/schema/tests/resolved.py +1 -1
- metadata/generated/schema/tests/testCase.py +26 -2
- metadata/generated/schema/tests/testCaseResolutionStatus.py +1 -1
- metadata/generated/schema/tests/testDefinition.py +1 -1
- metadata/generated/schema/tests/testSuite.py +12 -1
- metadata/generated/schema/type/__init__.py +1 -1
- metadata/generated/schema/type/apiSchema.py +1 -1
- metadata/generated/schema/type/assetCertification.py +1 -1
- metadata/generated/schema/type/auditLog.py +1 -1
- metadata/generated/schema/type/basic.py +9 -1
- metadata/generated/schema/type/bulkOperationResult.py +1 -1
- metadata/generated/schema/type/changeEvent.py +1 -1
- metadata/generated/schema/type/changeEventType.py +1 -1
- metadata/generated/schema/type/changeSummaryMap.py +1 -1
- metadata/generated/schema/type/collectionDescriptor.py +1 -1
- metadata/generated/schema/type/contextRecognizer.py +50 -0
- metadata/generated/schema/type/contractExecutionStatus.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/customProperties/__init__.py +1 -1
- metadata/generated/schema/type/customProperties/complexTypes.py +1 -1
- metadata/generated/schema/type/customProperties/enumConfig.py +1 -1
- metadata/generated/schema/type/customProperties/tableConfig.py +1 -1
- metadata/generated/schema/type/customProperty.py +1 -1
- metadata/generated/schema/type/customRecognizer.py +39 -0
- metadata/generated/schema/type/dailyCount.py +1 -1
- metadata/generated/schema/type/databaseConnectionConfig.py +1 -1
- metadata/generated/schema/type/denyListRecognizer.py +37 -0
- metadata/generated/schema/type/entityHierarchy.py +1 -1
- metadata/generated/schema/type/entityHistory.py +1 -1
- metadata/generated/schema/type/entityLineage.py +1 -1
- metadata/generated/schema/type/entityProfile.py +45 -0
- metadata/generated/schema/type/entityReference.py +1 -1
- metadata/generated/schema/type/entityReferenceList.py +1 -1
- metadata/generated/schema/type/entityRelationship/__init__.py +1 -1
- metadata/generated/schema/type/entityRelationship/nodeInformation.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 +1 -1
- metadata/generated/schema/type/include.py +1 -1
- metadata/generated/schema/type/jdbcConnection.py +1 -1
- metadata/generated/schema/type/layerPaging.py +1 -1
- metadata/generated/schema/type/lifeCycle.py +1 -1
- metadata/generated/schema/type/paging.py +1 -1
- metadata/generated/schema/type/patternRecognizer.py +39 -0
- metadata/generated/schema/type/personaPreferences.py +1 -1
- metadata/generated/schema/type/piiEntity.py +48 -0
- metadata/generated/schema/type/predefinedRecognizer.py +85 -0
- metadata/generated/schema/type/profile.py +1 -1
- metadata/generated/schema/type/queryParserData.py +1 -1
- metadata/generated/schema/type/reaction.py +1 -1
- metadata/generated/schema/type/recognizer.py +159 -0
- metadata/generated/schema/type/recognizerFeedback.py +120 -0
- metadata/generated/schema/type/recognizers/__init__.py +3 -0
- metadata/generated/schema/type/recognizers/patterns.py +29 -0
- metadata/generated/schema/type/recognizers/regexFlags.py +29 -0
- metadata/generated/schema/type/schedule.py +1 -1
- metadata/generated/schema/type/schema.py +1 -1
- metadata/generated/schema/type/status.py +16 -0
- metadata/generated/schema/type/tableQuery.py +1 -1
- metadata/generated/schema/type/tableUsageCount.py +1 -1
- metadata/generated/schema/type/tagLabel.py +8 -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/action1xx.py +2 -1
- metadata/ingestion/api/status.py +6 -1
- metadata/ingestion/api/step.py +3 -1
- metadata/ingestion/bulksink/metadata_usage.py +19 -2
- metadata/ingestion/connections/builders.py +5 -1
- metadata/ingestion/lineage/masker.py +5 -0
- metadata/ingestion/lineage/models.py +4 -0
- metadata/ingestion/lineage/parser.py +56 -4
- metadata/ingestion/lineage/sql_lineage.py +243 -103
- metadata/ingestion/models/custom_basemodel_validation.py +169 -43
- metadata/ingestion/models/custom_pydantic.py +6 -19
- metadata/ingestion/models/patch_request.py +2 -0
- metadata/ingestion/ometa/client.py +2 -2
- metadata/ingestion/ometa/mixins/csv_mixin.py +180 -0
- metadata/ingestion/ometa/mixins/domain_mixin.py +48 -0
- metadata/ingestion/ometa/mixins/es_mixin.py +7 -1
- metadata/ingestion/ometa/mixins/ingestion_pipeline_mixin.py +26 -0
- metadata/ingestion/ometa/mixins/lineage_mixin.py +1 -1
- metadata/ingestion/ometa/mixins/logs_mixin.py +316 -0
- metadata/ingestion/ometa/mixins/patch_mixin.py +23 -9
- metadata/ingestion/ometa/mixins/profile_mixin.py +77 -0
- metadata/ingestion/ometa/mixins/tag_glossary_mixin.py +68 -0
- metadata/ingestion/ometa/mixins/tests_mixin.py +1 -1
- metadata/ingestion/ometa/mixins/user_mixin.py +57 -16
- metadata/ingestion/ometa/ometa_api.py +24 -1
- metadata/ingestion/ometa/routes.py +5 -0
- metadata/ingestion/source/api/rest/connection.py +12 -6
- metadata/ingestion/source/api/rest/metadata.py +227 -37
- metadata/ingestion/source/api/rest/parser.py +103 -0
- metadata/ingestion/source/dashboard/domodashboard/metadata.py +19 -9
- metadata/ingestion/source/dashboard/grafana/__init__.py +10 -0
- metadata/ingestion/source/dashboard/grafana/client.py +223 -0
- metadata/ingestion/source/dashboard/grafana/connection.py +66 -0
- metadata/ingestion/source/dashboard/grafana/metadata.py +476 -0
- metadata/ingestion/source/dashboard/grafana/models.py +159 -0
- metadata/ingestion/source/dashboard/grafana/service_spec.py +17 -0
- metadata/ingestion/source/dashboard/hex/client.py +104 -0
- metadata/ingestion/source/dashboard/hex/connection.py +60 -0
- metadata/ingestion/source/dashboard/hex/metadata.py +363 -0
- metadata/ingestion/source/dashboard/hex/models.py +143 -0
- metadata/ingestion/source/dashboard/hex/query_fetcher.py +503 -0
- metadata/ingestion/source/dashboard/hex/service_spec.py +18 -0
- metadata/ingestion/source/dashboard/hex/warehouse_queries.py +225 -0
- metadata/ingestion/source/dashboard/looker/bulk_parser.py +31 -14
- metadata/ingestion/source/dashboard/looker/metadata.py +131 -32
- metadata/ingestion/source/dashboard/looker/models.py +3 -0
- metadata/ingestion/source/dashboard/looker/utils.py +18 -3
- metadata/ingestion/source/dashboard/metabase/client.py +33 -11
- metadata/ingestion/source/dashboard/metabase/metadata.py +4 -0
- metadata/ingestion/source/dashboard/metabase/models.py +49 -1
- metadata/ingestion/source/dashboard/microstrategy/metadata.py +1 -1
- metadata/ingestion/source/dashboard/powerbi/client.py +18 -1
- metadata/ingestion/source/dashboard/powerbi/databricks_parser.py +108 -0
- metadata/ingestion/source/dashboard/powerbi/metadata.py +412 -100
- metadata/ingestion/source/dashboard/powerbi/models.py +20 -1
- metadata/ingestion/source/dashboard/redash/metadata.py +4 -0
- metadata/ingestion/source/dashboard/sigma/metadata.py +4 -0
- metadata/ingestion/source/dashboard/superset/api_source.py +2 -0
- metadata/ingestion/source/dashboard/superset/db_source.py +2 -0
- metadata/ingestion/source/dashboard/superset/mixin.py +2 -0
- metadata/ingestion/source/dashboard/tableau/client.py +29 -14
- metadata/ingestion/source/dashboard/tableau/metadata.py +7 -1
- metadata/ingestion/source/database/athena/utils.py +1 -1
- metadata/ingestion/source/database/bigquery/connection.py +33 -13
- metadata/ingestion/source/database/bigquery/lineage.py +5 -10
- metadata/ingestion/source/database/bigquery/metadata.py +39 -34
- metadata/ingestion/source/database/bigquery/queries.py +6 -1
- metadata/ingestion/source/database/column_type_parser.py +1 -1
- metadata/ingestion/source/database/common/data_diff/databricks_base.py +42 -0
- metadata/ingestion/source/database/common_pg_mappings.py +7 -0
- metadata/ingestion/source/database/databricks/auth.py +99 -0
- metadata/ingestion/source/database/databricks/client.py +201 -51
- metadata/ingestion/source/database/databricks/connection.py +99 -13
- metadata/ingestion/source/database/databricks/data_diff/__init__.py +1 -0
- metadata/ingestion/source/database/databricks/data_diff/data_diff.py +12 -0
- metadata/ingestion/source/database/databricks/queries.py +86 -16
- metadata/ingestion/source/database/databricks/service_spec.py +4 -0
- metadata/ingestion/source/database/dbt/constants.py +0 -1
- metadata/ingestion/source/database/dbt/metadata.py +47 -21
- metadata/ingestion/source/database/dbt/models.py +1 -0
- metadata/ingestion/source/database/exasol/connection.py +6 -0
- metadata/ingestion/source/database/exasol/lineage.py +24 -0
- metadata/ingestion/source/database/exasol/metadata.py +6 -0
- metadata/ingestion/source/database/exasol/queries.py +22 -0
- metadata/ingestion/source/database/exasol/query_parser.py +33 -0
- metadata/ingestion/source/database/exasol/service_spec.py +5 -1
- metadata/ingestion/source/database/glue/metadata.py +2 -2
- metadata/ingestion/source/database/hive/connection.py +41 -2
- metadata/ingestion/source/database/hive/custom_hive_connection.py +195 -0
- metadata/ingestion/source/database/iceberg/metadata.py +44 -1
- metadata/ingestion/source/database/iceberg/models.py +22 -13
- metadata/ingestion/source/database/lineage_processors.py +386 -0
- metadata/ingestion/source/database/lineage_source.py +314 -248
- metadata/ingestion/source/database/mariadb/metadata.py +86 -1
- metadata/ingestion/source/database/mariadb/models.py +36 -0
- metadata/ingestion/source/database/mariadb/queries.py +43 -0
- metadata/ingestion/source/database/mssql/lineage.py +3 -27
- metadata/ingestion/source/database/oracle/connection.py +158 -114
- metadata/ingestion/source/database/oracle/lineage.py +3 -10
- metadata/ingestion/source/database/oracle/queries.py +43 -6
- metadata/ingestion/source/database/oracle/service_spec.py +2 -0
- metadata/ingestion/source/database/postgres/connection.py +6 -0
- metadata/ingestion/source/database/postgres/queries.py +22 -0
- metadata/ingestion/source/database/redshift/lineage.py +4 -10
- metadata/ingestion/source/database/redshift/utils.py +2 -1
- metadata/ingestion/source/database/salesforce/connection.py +14 -6
- metadata/ingestion/source/database/salesforce/metadata.py +14 -16
- metadata/ingestion/source/database/sample_data.py +24 -8
- metadata/ingestion/source/database/saphana/cdata_parser.py +327 -117
- metadata/ingestion/source/database/saphana/lineage.py +15 -2
- metadata/ingestion/source/database/saphana/queries.py +7 -0
- metadata/ingestion/source/database/snowflake/connection.py +7 -0
- metadata/ingestion/source/database/snowflake/constants.py +21 -0
- metadata/ingestion/source/database/snowflake/lineage.py +48 -30
- metadata/ingestion/source/database/snowflake/metadata.py +176 -22
- metadata/ingestion/source/database/snowflake/queries.py +15 -1
- metadata/ingestion/source/database/snowflake/query_parser.py +12 -2
- metadata/ingestion/source/database/snowflake/usage.py +97 -0
- metadata/ingestion/source/database/sql_column_handler.py +11 -3
- metadata/ingestion/source/database/stored_procedures_mixin.py +66 -221
- metadata/ingestion/source/database/timescale/__init__.py +3 -0
- metadata/ingestion/source/database/timescale/connection.py +103 -0
- metadata/ingestion/source/database/timescale/lineage.py +124 -0
- metadata/ingestion/source/database/timescale/metadata.py +228 -0
- metadata/ingestion/source/database/timescale/models.py +60 -0
- metadata/ingestion/source/database/timescale/queries.py +96 -0
- metadata/ingestion/source/database/timescale/query_parser.py +106 -0
- metadata/ingestion/source/database/timescale/service_spec.py +12 -0
- metadata/ingestion/source/database/timescale/usage.py +43 -0
- metadata/ingestion/source/database/timescale/utils.py +545 -0
- metadata/ingestion/source/database/trino/metadata.py +43 -4
- metadata/ingestion/source/database/trino/queries.py +4 -0
- metadata/ingestion/source/database/unitycatalog/client.py +35 -1
- metadata/ingestion/source/database/unitycatalog/connection.py +43 -12
- metadata/ingestion/source/database/unitycatalog/data_diff/__init__.py +1 -0
- metadata/ingestion/source/database/unitycatalog/data_diff/data_diff.py +12 -0
- metadata/ingestion/source/database/unitycatalog/lineage.py +213 -24
- metadata/ingestion/source/database/unitycatalog/metadata.py +5 -4
- metadata/ingestion/source/database/unitycatalog/models.py +18 -2
- metadata/ingestion/source/database/unitycatalog/service_spec.py +4 -0
- metadata/ingestion/source/messaging/kafka/connection.py +13 -4
- metadata/ingestion/source/metadata/amundsen/client.py +6 -6
- metadata/ingestion/source/mlmodel/mlmodel_service.py +75 -1
- metadata/ingestion/source/pipeline/airbyte/client.py +142 -4
- metadata/ingestion/source/pipeline/airbyte/connection.py +18 -5
- metadata/ingestion/source/pipeline/airbyte/metadata.py +102 -2
- metadata/ingestion/source/pipeline/airflow/connection.py +8 -6
- metadata/ingestion/source/pipeline/airflow/metadata.py +5 -1
- metadata/ingestion/source/pipeline/databrickspipeline/connection.py +1 -0
- metadata/ingestion/source/pipeline/databrickspipeline/kafka_parser.py +603 -0
- metadata/ingestion/source/pipeline/databrickspipeline/metadata.py +1019 -90
- metadata/ingestion/source/pipeline/databrickspipeline/models.py +19 -3
- metadata/ingestion/source/pipeline/dbtcloud/client.py +85 -76
- metadata/ingestion/source/pipeline/dbtcloud/metadata.py +135 -84
- metadata/ingestion/source/pipeline/dbtcloud/models.py +4 -0
- metadata/ingestion/source/pipeline/dbtcloud/queries.py +7 -11
- metadata/ingestion/source/pipeline/fivetran/client.py +7 -3
- metadata/ingestion/source/pipeline/fivetran/metadata.py +122 -90
- metadata/ingestion/source/pipeline/kafkaconnect/client.py +388 -107
- metadata/ingestion/source/pipeline/kafkaconnect/metadata.py +1678 -72
- metadata/ingestion/source/pipeline/kafkaconnect/models.py +51 -14
- metadata/ingestion/source/pipeline/nifi/metadata.py +62 -5
- metadata/ingestion/source/pipeline/openlineage/metadata.py +6 -2
- metadata/ingestion/source/search/elasticsearch/connection.py +1 -1
- metadata/ingestion/source/storage/gcs/connection.py +15 -5
- metadata/ingestion/source/storage/gcs/metadata.py +285 -26
- metadata/ingestion/source/storage/gcs/models.py +12 -6
- metadata/ingestion/source/storage/s3/metadata.py +46 -11
- metadata/ingestion/source/storage/storage_service.py +2 -0
- metadata/parsers/json_schema_parser.py +41 -1
- metadata/pii/algorithms/classifiers.py +63 -1
- metadata/pii/algorithms/feature_extraction.py +13 -2
- metadata/pii/algorithms/patterns.py +171 -0
- metadata/pii/algorithms/preprocessing.py +19 -13
- metadata/pii/algorithms/presidio_recognizer_factory.py +302 -0
- metadata/pii/algorithms/presidio_utils.py +205 -24
- metadata/pii/algorithms/utils.py +4 -0
- metadata/pii/base_processor.py +2 -1
- metadata/pii/processor.py +8 -5
- metadata/pii/processor_factory.py +28 -0
- metadata/pii/scanners/custom_ner_scanner.py +290 -0
- metadata/pii/tag_analyzer.py +124 -0
- metadata/pii/tag_processor.py +100 -0
- metadata/profiler/interface/sqlalchemy/bigquery/profiler_interface.py +17 -0
- metadata/profiler/interface/sqlalchemy/databricks/profiler_interface.py +24 -0
- metadata/profiler/interface/sqlalchemy/unity_catalog/profiler_interface.py +16 -2
- metadata/profiler/metrics/hybrid/cardinality_distribution.py +232 -0
- metadata/profiler/metrics/hybrid/histogram.py +1 -1
- metadata/profiler/metrics/registry.py +4 -0
- metadata/profiler/metrics/system/bigquery/system.py +4 -1
- metadata/profiler/metrics/system/databricks/system.py +1 -1
- metadata/profiler/orm/converter/converter_registry.py +2 -0
- metadata/profiler/orm/converter/trino/__init__.py +14 -0
- metadata/profiler/orm/converter/trino/converter.py +39 -0
- metadata/profiler/orm/functions/conn_test.py +5 -0
- metadata/profiler/orm/functions/table_metric_computer.py +5 -18
- metadata/profiler/orm/registry.py +7 -0
- metadata/profiler/orm/types/__init__.py +10 -0
- metadata/profiler/orm/types/trino.py +70 -0
- metadata/profiler/processor/core.py +1 -1
- metadata/profiler/processor/default.py +1 -0
- metadata/readers/dataframe/mf4.py +75 -0
- metadata/readers/dataframe/reader_factory.py +3 -0
- metadata/sampler/sqlalchemy/bigquery/sampler.py +17 -0
- metadata/sampler/sqlalchemy/databricks/sampler.py +13 -1
- metadata/sampler/sqlalchemy/unitycatalog/sampler.py +6 -39
- metadata/sdk/__init__.py +188 -0
- metadata/sdk/api/__init__.py +7 -0
- metadata/sdk/api/lineage.py +421 -0
- metadata/sdk/api/search.py +382 -0
- metadata/sdk/client.py +91 -0
- metadata/sdk/config.py +121 -0
- metadata/sdk/entities/__init__.py +64 -0
- metadata/sdk/entities/apicollections.py +19 -0
- metadata/sdk/entities/apiendpoints.py +19 -0
- metadata/sdk/entities/base.py +487 -0
- metadata/sdk/entities/charts.py +17 -0
- metadata/sdk/entities/classifications.py +21 -0
- metadata/sdk/entities/containers.py +17 -0
- metadata/sdk/entities/custom_properties.py +144 -0
- metadata/sdk/entities/dashboard_services.py +18 -0
- metadata/sdk/entities/dashboarddatamodels.py +21 -0
- metadata/sdk/entities/dashboards.py +17 -0
- metadata/sdk/entities/database_services.py +18 -0
- metadata/sdk/entities/databases.py +17 -0
- metadata/sdk/entities/databaseschemas.py +19 -0
- metadata/sdk/entities/datacontracts.py +19 -0
- metadata/sdk/entities/dataproducts.py +19 -0
- metadata/sdk/entities/domains.py +17 -0
- metadata/sdk/entities/glossaries.py +17 -0
- metadata/sdk/entities/glossary_terms.py +18 -0
- metadata/sdk/entities/glossaryterms.py +19 -0
- metadata/sdk/entities/metrics.py +49 -0
- metadata/sdk/entities/mlmodels.py +17 -0
- metadata/sdk/entities/pipelines.py +17 -0
- metadata/sdk/entities/queries.py +17 -0
- metadata/sdk/entities/searchindexes.py +19 -0
- metadata/sdk/entities/storedprocedures.py +19 -0
- metadata/sdk/entities/tables.py +68 -0
- metadata/sdk/entities/tags.py +17 -0
- metadata/sdk/entities/teams.py +17 -0
- metadata/sdk/entities/testcases.py +17 -0
- metadata/sdk/entities/testdefinitions.py +19 -0
- metadata/sdk/entities/testsuites.py +17 -0
- metadata/sdk/entities/users.py +17 -0
- metadata/sdk/examples/builder_end_to_end.py +387 -0
- metadata/sdk/types.py +15 -0
- metadata/utils/bigquery_utils.py +7 -4
- metadata/utils/datalake/datalake_utils.py +121 -0
- metadata/utils/db_utils.py +20 -6
- metadata/utils/deprecation.py +9 -6
- metadata/utils/dispatch.py +6 -3
- metadata/utils/elasticsearch.py +3 -0
- metadata/utils/fqn.py +104 -7
- metadata/utils/helpers.py +18 -1
- metadata/utils/importer.py +3 -2
- metadata/utils/logger.py +12 -0
- metadata/utils/sqa_utils.py +3 -0
- metadata/utils/ssl_manager.py +93 -5
- metadata/utils/streamable_logger.py +569 -0
- metadata/workflow/base.py +49 -2
- metadata/workflow/classification.py +2 -2
- metadata/workflow/workflow_resource_metrics.py +111 -0
- metadata/workflow/workflow_status_mixin.py +1 -1
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/METADATA +559 -540
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/RECORD +1115 -950
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/LICENSE +0 -0
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/WHEEL +0 -0
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/entry_points.txt +0 -0
- {openmetadata_ingestion-1.9.0.0.dist-info → openmetadata_ingestion-1.10.14.4.dist-info}/top_level.txt +0 -0
|
@@ -19,6 +19,7 @@ from typing import (
|
|
|
19
19
|
Dict,
|
|
20
20
|
Generic,
|
|
21
21
|
Hashable,
|
|
22
|
+
Iterable,
|
|
22
23
|
Mapping,
|
|
23
24
|
Optional,
|
|
24
25
|
Sequence,
|
|
@@ -29,7 +30,7 @@ from typing import (
|
|
|
29
30
|
|
|
30
31
|
from presidio_analyzer import AnalyzerEngine
|
|
31
32
|
|
|
32
|
-
from metadata.generated.schema.entity.data.table import DataType
|
|
33
|
+
from metadata.generated.schema.entity.data.table import Column, DataType
|
|
33
34
|
from metadata.pii.algorithms.column_patterns import get_pii_column_name_patterns
|
|
34
35
|
from metadata.pii.algorithms.feature_extraction import (
|
|
35
36
|
extract_pii_from_column_names,
|
|
@@ -48,6 +49,7 @@ from metadata.pii.algorithms.presidio_utils import (
|
|
|
48
49
|
set_presidio_logger_level,
|
|
49
50
|
)
|
|
50
51
|
from metadata.pii.algorithms.tags import PIISensitivityTag, PIITag
|
|
52
|
+
from metadata.pii.tag_analyzer import TagAnalyzer
|
|
51
53
|
|
|
52
54
|
T = TypeVar("T", bound=Hashable)
|
|
53
55
|
|
|
@@ -72,6 +74,11 @@ class ColumnClassifier(ABC, Generic[T]):
|
|
|
72
74
|
higher scores indicate a higher likelihood of the class for the given inputs.
|
|
73
75
|
"""
|
|
74
76
|
|
|
77
|
+
def classify(self, column: Column, sample_data: Sequence[Any]) -> Mapping[T, float]:
|
|
78
|
+
return self.predict_scores(
|
|
79
|
+
sample_data, column_name=column.name.root, column_data_type=column.dataType
|
|
80
|
+
)
|
|
81
|
+
|
|
75
82
|
|
|
76
83
|
@final
|
|
77
84
|
class HeuristicPIIClassifier(ColumnClassifier[PIITag]):
|
|
@@ -178,3 +185,58 @@ class PIISensitiveClassifier(ColumnClassifier[PIISensitivityTag]):
|
|
|
178
185
|
results[tag] /= counts[tag]
|
|
179
186
|
|
|
180
187
|
return results
|
|
188
|
+
|
|
189
|
+
|
|
190
|
+
@final
|
|
191
|
+
class TagClassifier(ColumnClassifier[str]):
|
|
192
|
+
"""
|
|
193
|
+
Heuristic PII Column Classifier
|
|
194
|
+
"""
|
|
195
|
+
|
|
196
|
+
def __init__(
|
|
197
|
+
self,
|
|
198
|
+
*,
|
|
199
|
+
tag_analyzers: Iterable[TagAnalyzer],
|
|
200
|
+
column_name_contribution: float = 0.5,
|
|
201
|
+
score_cutoff: float = 0.1,
|
|
202
|
+
relative_cardinality_cutoff: float = 0.01,
|
|
203
|
+
):
|
|
204
|
+
set_presidio_logger_level()
|
|
205
|
+
|
|
206
|
+
self._analyzers = tag_analyzers
|
|
207
|
+
|
|
208
|
+
self._column_name_contribution = column_name_contribution
|
|
209
|
+
self._score_cutoff = score_cutoff
|
|
210
|
+
self._relative_cardinality_cutoff = relative_cardinality_cutoff
|
|
211
|
+
|
|
212
|
+
def predict_scores(
|
|
213
|
+
self,
|
|
214
|
+
sample_data: Sequence[Any],
|
|
215
|
+
column_name: Optional[str] = None,
|
|
216
|
+
column_data_type: Optional[DataType] = None,
|
|
217
|
+
) -> Mapping[str, float]:
|
|
218
|
+
str_values = preprocess_values(sample_data)
|
|
219
|
+
|
|
220
|
+
if not str_values:
|
|
221
|
+
return {}
|
|
222
|
+
|
|
223
|
+
# Relative cardinality test
|
|
224
|
+
unique_values = set(str_values)
|
|
225
|
+
if len(unique_values) / len(str_values) < self._relative_cardinality_cutoff:
|
|
226
|
+
return {}
|
|
227
|
+
|
|
228
|
+
results: dict[str, float] = defaultdict(float)
|
|
229
|
+
for analyzer in self._analyzers:
|
|
230
|
+
content_score = analyzer.analyze_content(values=str_values)
|
|
231
|
+
column_score = 0.0
|
|
232
|
+
if column_name is not None:
|
|
233
|
+
column_score = analyzer.analyze_column()
|
|
234
|
+
column_score *= max(column_score, self._column_name_contribution)
|
|
235
|
+
|
|
236
|
+
total_score = content_score + column_score
|
|
237
|
+
if total_score > self._score_cutoff:
|
|
238
|
+
results[str(analyzer.tag.fullyQualifiedName)] = (
|
|
239
|
+
content_score + column_score
|
|
240
|
+
)
|
|
241
|
+
|
|
242
|
+
return results
|
|
@@ -86,11 +86,22 @@ def split_column_name(column_name: str) -> List[str]:
|
|
|
86
86
|
"""
|
|
87
87
|
Split a column name into its components.
|
|
88
88
|
This is used for passing column names to the analyzer as context.
|
|
89
|
+
Handles both delimiters (_-. /) and camelCase.
|
|
89
90
|
"""
|
|
90
|
-
#
|
|
91
|
+
# First split by common delimiters
|
|
91
92
|
delimiters = ["_", "-", " ", ".", "/"]
|
|
92
93
|
regex_pattern = "|".join(map(re.escape, delimiters))
|
|
93
|
-
|
|
94
|
+
parts = re.split(regex_pattern, column_name)
|
|
95
|
+
|
|
96
|
+
# Then split each part by camelCase
|
|
97
|
+
result: List[str] = []
|
|
98
|
+
for part in parts:
|
|
99
|
+
if not part:
|
|
100
|
+
continue
|
|
101
|
+
camel_parts = re.sub("([a-z])([A-Z])", r"\1 \2", part).split()
|
|
102
|
+
result.extend([p.lower() for p in camel_parts])
|
|
103
|
+
|
|
104
|
+
return result
|
|
94
105
|
|
|
95
106
|
|
|
96
107
|
def extract_pii_from_column_names(
|
|
@@ -0,0 +1,171 @@
|
|
|
1
|
+
from presidio_analyzer import Pattern
|
|
2
|
+
|
|
3
|
+
credit_cards = [
|
|
4
|
+
Pattern("Credit Card Number", pattern, 0.7)
|
|
5
|
+
for pattern in (
|
|
6
|
+
# Visa: 13 or 16 digits, starts with 4
|
|
7
|
+
r"^4\d{12}(?:\d{3})?$",
|
|
8
|
+
# Mastercard: 16 digits
|
|
9
|
+
# Includes both old range (51-55) and new range (2221-2720)
|
|
10
|
+
r"^(?:5[1-5]\d{2}|2(?:22[1-9]|2[3-9]\d|[3-6]\d{2}|7[01]\d|720))\d{12}$",
|
|
11
|
+
# American Express: 15 digits, starts with 34 or 37
|
|
12
|
+
r"^3[47]\d{13}$",
|
|
13
|
+
# Diners Club: 14 digits, starts with 300-305, 36, or 38
|
|
14
|
+
r"^3(?:0[0-5]|[68]\d)\d{11}$",
|
|
15
|
+
# Discover: 16 digits
|
|
16
|
+
# Starts with 6011, 622126-622925, 644-649, or 65
|
|
17
|
+
r"^(?:6011|65\d{2}|64[4-9]\d|622(?:1(?:2[6-9]|[3-9]\d)|[2-8]\d{2}|9(?:[01]\d|2[0-5])))\d{12}$",
|
|
18
|
+
# JCB: 16 digits, starts with 3528-3589
|
|
19
|
+
r"^35(?:2[89]|[3-8]\d)\d{12}$",
|
|
20
|
+
# Maestro: 12-19 digits
|
|
21
|
+
# Starts with 5018, 5020, 5038, 5893, 6304, 6759, 6761-6763
|
|
22
|
+
r"^(?:50(?:18|20|38)|5893|6(?:304|759|76[1-3]))\d{8,15}$",
|
|
23
|
+
# UnionPay: 16-19 digits, starts with 62
|
|
24
|
+
r"^62\d{14,17}$",
|
|
25
|
+
)
|
|
26
|
+
]
|
|
27
|
+
|
|
28
|
+
us_driving_license = [
|
|
29
|
+
Pattern("US Driving License", pattern, 0.3)
|
|
30
|
+
for pattern in (
|
|
31
|
+
r"^\d{7}$",
|
|
32
|
+
r"^[a-zA-Z]\d{8}$",
|
|
33
|
+
r"^\d{9}$",
|
|
34
|
+
r"^9\d{8}$",
|
|
35
|
+
r"^[a-zA-Z]\d{7}$",
|
|
36
|
+
r"^\d{2}-\d{3}-\d{4}$",
|
|
37
|
+
r"^[a-zA-Z] \d{3} \d{3} \d{3} \d{3}$",
|
|
38
|
+
r"^[a-zA-Z]\d{12}$",
|
|
39
|
+
r"^[a-zA-Z]\d{3}-\d{3}-\d{2}-\d{3}-\d$",
|
|
40
|
+
r"^[a-zA-Z]-\d{3}-\d{3}-\d{3}-\d{3}$",
|
|
41
|
+
r"^[a-zA-Z]\s\d{3}\s\d{3}\s\d{3}\s\d{3}$",
|
|
42
|
+
r"^[a-zA-Z]\d{12}$",
|
|
43
|
+
r"^[a-zA-Z]{2}\d{6}[a-zA-Z]$",
|
|
44
|
+
r"^[a-zA-Z]\d{3}-\d{4}-\d{4}$",
|
|
45
|
+
r"^[a-zA-Z]\d{11}$",
|
|
46
|
+
r"^\d{4}-\d{2}-\d{4}$",
|
|
47
|
+
r"^\d{3}[a-zA-Z]{2}\d{4}$",
|
|
48
|
+
r"^[a-zA-Z]\d{2}-\d{2}-\d{4}$",
|
|
49
|
+
r"^[a-zA-Z]\d{2}-\d{3}-\d{3}$",
|
|
50
|
+
r"^[a-zA-Z]\d{9}$",
|
|
51
|
+
r"^\d{3}-\d{2}-\d{4}$",
|
|
52
|
+
r"^[a-zA-Z]\d{9}$",
|
|
53
|
+
r"^(([0][1-9]|[1][0-2])\d{3}([1-9][0-9]{3})41([0][1-9]|[1][0-9]|[3][0-1]))\d{10}$",
|
|
54
|
+
r"^([0][1-9]|[1][0-2])[a-zA-Z]{3}\d{2}(0[1-9]|[1-2][0-9]|3[0-1])\d$",
|
|
55
|
+
r"^[a-zA-Z]\d{4}-\d{5}-\d{5}$",
|
|
56
|
+
r"^[a-zA-Z]\d{14}$",
|
|
57
|
+
r"^\d{3} \d{3} \d{3}$",
|
|
58
|
+
r"^\d{12}$",
|
|
59
|
+
r"^[a-zA-Z]{3}-\d{2}-\d{4}$",
|
|
60
|
+
r"^[a-zA-Z]{1}[0-9]{4,8}$",
|
|
61
|
+
r"^[a-zA-Z]{2}[0-9]{3,7}$",
|
|
62
|
+
r"^\d{2}\s\d{3}\s\d{3}$",
|
|
63
|
+
r"^[1-9]{2}\d{5}$",
|
|
64
|
+
r"^\d{8}$",
|
|
65
|
+
r"^\d{7,9}$",
|
|
66
|
+
r"^\d{7}[a-zA-Z]$",
|
|
67
|
+
r"^[a-zA-Z]\d{8}$",
|
|
68
|
+
r"^[a-zA-Z*]{1,6}[a-zA-Z]{2}\d[a-zA-Z0-9]{4,6}$",
|
|
69
|
+
r"^[a-zA-Z]\d{6}$",
|
|
70
|
+
r"^[a-zA-Z]\d{3}-\d{4}-\d{4}-\d{2}$",
|
|
71
|
+
r"^\d{6}-\d{3}$",
|
|
72
|
+
r"^\d{2}[a-zA-Z]{3}\d{5}$",
|
|
73
|
+
r"^\d{13}$",
|
|
74
|
+
r"^\d[A-Z]{2}\d{4}$", # #??####
|
|
75
|
+
r"^\d{2}[A-Z]{2}\d{3}$", # ##??###
|
|
76
|
+
r"^\d{3}\s[A-Z]{3}$", # ### ???
|
|
77
|
+
r"^\d{4}$", # ####
|
|
78
|
+
r"^[A-Z]{3}\d{4}$", # ???####
|
|
79
|
+
r"^\d{3}\s?[A-Z]{3}$", # ### ??? or ###???
|
|
80
|
+
r"^\d[A-Z]{3}\d{3}$", # #???###
|
|
81
|
+
r"^\d{3}-[A-Z]{3}$", # ###-???
|
|
82
|
+
r"^[A-Z]{3}-\d{3}$", # ???-###
|
|
83
|
+
r"^\d{6}$", # ######
|
|
84
|
+
r"^[A-Z]{2}-\d{4}$", # ??-####
|
|
85
|
+
r"^[A-Z]{3}\s[A-Z]\d{2}$", # ??? ?##
|
|
86
|
+
r"^[A-Z]\d{2}\s\d[A-Z]{2}$", # ?## #??
|
|
87
|
+
r"^\d{3}\s\d[A-Z]{2}$", # ### #??
|
|
88
|
+
r"^[A-Z]{2}\s\d{4}$", # ?? ####
|
|
89
|
+
r"^[A-Z]{3}\s\d{3}$", # ??? ###
|
|
90
|
+
r"^[HZKLMhzklm][A-Z]{2}\s\d{3}$", # H?? ### (etc)
|
|
91
|
+
r"^[A-Z]\s\d{6}$", # ? ######
|
|
92
|
+
r"^\d[A-Z]\s\d{5}$", # #? #####
|
|
93
|
+
r"^\d[A-Z]\s[A-Z]\d{4}$", # #? ?####
|
|
94
|
+
r"^\d[A-Z]\s[A-Z]{2}\d{3}$", # #? ??###
|
|
95
|
+
r"^\d[A-Z]\s\d[A-Z]\d[A-Z]{3}$", # #? #?#???
|
|
96
|
+
r"^\d[A-Z]\s\d{4}[A-Z]$", # #? ####?
|
|
97
|
+
r"^\d{2}[A-Z]\s\d{4}$", # ##? ####
|
|
98
|
+
r"^[A-Z]{2}\s\d{5}$", # ?? #####
|
|
99
|
+
r"^[A-Z]{2}\d\s\d{4}$", # ??# ####
|
|
100
|
+
r"^\d{3}[A-Z]{1,3}$", # ###?, ###??, ###???
|
|
101
|
+
r"^\d{4}\s[A-Z]{2}$", # #### ??
|
|
102
|
+
r"^\d[A-Z]{3}\s\d{2}$", # #??? ##
|
|
103
|
+
r"^\d[A-Z]{2}\s\d{3}$", # #?? ###
|
|
104
|
+
r"^\d{3}\s[A-Z]{2}\d$", # ### ??#
|
|
105
|
+
r"^\d{2}[A-Z]\s[A-Z]\d{2}$", # ##? ?##
|
|
106
|
+
r"^\d[A-Z]{2}\s[A-Z]\d{2}$", # #?? ?##
|
|
107
|
+
r"^\d{3}-[A-Z]{3}$", # ###-???
|
|
108
|
+
r"^[A-Z]{2}\d\s[A-Z]\d[A-Z]$", # ??# ?#?
|
|
109
|
+
r"^\d-\d{5}[A-Z]$", # #-#####?
|
|
110
|
+
r"^\d{2}-\d{4}[A-Z]$", # ##-####?
|
|
111
|
+
r"^\d-[A-Z]\d{4}$", # #-?####
|
|
112
|
+
r"^\d{2}-[A-Z]\d{3}$", # ##-?###
|
|
113
|
+
r"^\d{2}-[A-Z]{2}\d{2}$", # ##-??##
|
|
114
|
+
r"^\d{2}[A-Z]•\d{3}$", # ##?•###
|
|
115
|
+
r"^\d{3}\s\d{4}$", # ### ####
|
|
116
|
+
r"^[A-Z]\d{2}-[A-Z]{3}$", # ?##-???
|
|
117
|
+
r"^[A-Z]{3}-\d{4}$", # ???-####
|
|
118
|
+
r"^\d{3}-[A-Z]{4}$", # ###-????
|
|
119
|
+
r"^[A-Z]{3}\s\d{4}$", # ??? ####
|
|
120
|
+
r"^\d{3}\s\d[A-Z]{2}$", # ### #??
|
|
121
|
+
r"^\d[A-Z]{2}\s\d{3}$", # #?? ###
|
|
122
|
+
r"^\d[A-Z]{2}\s[A-Z]\d{2}$", # #?? ?##
|
|
123
|
+
r"^\d{2}[A-Z]\s\d{3}$", # ##? ###
|
|
124
|
+
r"^\d{2}[A-Z]\s[A-Z]\d{2}$", # ##? ?##
|
|
125
|
+
r"^\d{2}[A-Z]\s[A-Z]{2}\d$", # ##? ??#
|
|
126
|
+
r"^[A-Z]\d{2}-\d{2}[A-Z]$", # ?##-##?
|
|
127
|
+
r"^[A-Z]\d{2}\s\d[A-Z]{2}$", # ?## #??
|
|
128
|
+
r"^\d{2}[A-Z]{2}\d$", # ##??#
|
|
129
|
+
r"^\d[A-Z]{2}\d{2}$", # #??##
|
|
130
|
+
r"^\d{3}[A-Z]\d$", # ###?#
|
|
131
|
+
r"^\d[A-Z]\d{3}$", # #?###
|
|
132
|
+
r"^[A-Z]{3}\d{4}$", # ???####
|
|
133
|
+
r"^\d[A-Z]{2}\s\d{3}$", # #?? ###
|
|
134
|
+
r"^\d-\d{5}$", # #-#####
|
|
135
|
+
r"^\d-\d{4}[A-Z]$", # #-####?
|
|
136
|
+
r"^\d{2}-\d{5}$", # ##-#####
|
|
137
|
+
)
|
|
138
|
+
]
|
|
139
|
+
|
|
140
|
+
au_tfn_number = [
|
|
141
|
+
Pattern(
|
|
142
|
+
"TFN (High)",
|
|
143
|
+
r"\bTFN:\s?\d{3}\s\d{3}\s\d{3}\b",
|
|
144
|
+
0.7,
|
|
145
|
+
),
|
|
146
|
+
Pattern(
|
|
147
|
+
"TFN (High)",
|
|
148
|
+
r"\bTFN:\s?\d{3}-\d{3}-\d{3}\b",
|
|
149
|
+
0.7,
|
|
150
|
+
),
|
|
151
|
+
Pattern(
|
|
152
|
+
"TFN (High)",
|
|
153
|
+
r"\bTFN:\s?\d{9}\b",
|
|
154
|
+
0.7,
|
|
155
|
+
),
|
|
156
|
+
Pattern(
|
|
157
|
+
"TFN (Medium)",
|
|
158
|
+
r"\b\d{3}\s\d{3}\s\d{3}\b",
|
|
159
|
+
0.1,
|
|
160
|
+
),
|
|
161
|
+
Pattern(
|
|
162
|
+
"TFN (Medium)",
|
|
163
|
+
r"\b\d{3}-\d{3}-\d{3}\b",
|
|
164
|
+
0.1,
|
|
165
|
+
),
|
|
166
|
+
Pattern(
|
|
167
|
+
"TFN (Low)",
|
|
168
|
+
r"\b\d{9}\b",
|
|
169
|
+
0.01,
|
|
170
|
+
),
|
|
171
|
+
]
|
|
@@ -12,8 +12,7 @@
|
|
|
12
12
|
Preprocessing functions for the classification tasks.
|
|
13
13
|
"""
|
|
14
14
|
import datetime
|
|
15
|
-
import
|
|
16
|
-
from typing import Any, List, Mapping, Optional, Sequence
|
|
15
|
+
from typing import Any, List, Mapping, Optional, Sequence, Union, cast
|
|
17
16
|
|
|
18
17
|
from metadata.utils.logger import pii_logger
|
|
19
18
|
|
|
@@ -21,7 +20,7 @@ logger = pii_logger()
|
|
|
21
20
|
|
|
22
21
|
|
|
23
22
|
# pylint: disable=too-many-return-statements
|
|
24
|
-
def convert_to_str(value: Any) -> Optional[str]:
|
|
23
|
+
def convert_to_str(value: Any) -> Optional[Union[List[str], str]]:
|
|
25
24
|
"""
|
|
26
25
|
Convert the given value to a string. This is a conversion
|
|
27
26
|
tailored to our use case, not a generic one.
|
|
@@ -32,14 +31,18 @@ def convert_to_str(value: Any) -> Optional[str]:
|
|
|
32
31
|
# Values we want to convert to string out of the box
|
|
33
32
|
return str(value)
|
|
34
33
|
if isinstance(value, bytes):
|
|
35
|
-
|
|
34
|
+
# Don't classify binary columns, which might contain misleading or outright invalid strings
|
|
35
|
+
return None
|
|
36
36
|
if isinstance(value, (Sequence, Mapping)):
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
37
|
+
if isinstance(value, Mapping):
|
|
38
|
+
value = list(value.values())
|
|
39
|
+
converted = [convert_to_str(el) for el in cast(List[Any], value)]
|
|
40
|
+
return [
|
|
41
|
+
item
|
|
42
|
+
for sublist in converted
|
|
43
|
+
for item in (sublist if isinstance(sublist, list) else [sublist])
|
|
44
|
+
if item is not None
|
|
45
|
+
]
|
|
43
46
|
if value is None:
|
|
44
47
|
# We want to skip None values, not convert them to "None"
|
|
45
48
|
return None
|
|
@@ -53,10 +56,13 @@ def preprocess_values(values: Sequence[Any]) -> List[str]:
|
|
|
53
56
|
if converted_value is None:
|
|
54
57
|
# Skip None values
|
|
55
58
|
continue
|
|
59
|
+
|
|
60
|
+
if not isinstance(converted_value, list):
|
|
61
|
+
converted_value = [converted_value]
|
|
62
|
+
|
|
56
63
|
# skip empty strings
|
|
57
|
-
|
|
58
|
-
continue
|
|
64
|
+
converted_value = [el.strip() for el in converted_value if el.strip()]
|
|
59
65
|
# Add the converted value as is, without any further processing
|
|
60
|
-
result.
|
|
66
|
+
result.extend(converted_value)
|
|
61
67
|
|
|
62
68
|
return result
|
|
@@ -0,0 +1,302 @@
|
|
|
1
|
+
# Copyright 2025 Collate
|
|
2
|
+
# Licensed under the Collate Community License, Version 1.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
|
+
# https://github.com/open-metadata/OpenMetadata/blob/main/ingestion/LICENSE
|
|
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
|
+
Factory for creating Presidio recognizers from OpenMetadata recognizer configurations.
|
|
13
|
+
"""
|
|
14
|
+
import re
|
|
15
|
+
from typing import Dict, List, Optional, cast
|
|
16
|
+
|
|
17
|
+
from presidio_analyzer import EntityRecognizer
|
|
18
|
+
from presidio_analyzer import Pattern as PresidioPattern
|
|
19
|
+
from presidio_analyzer import PatternRecognizer as PresidioPatternRecognizer
|
|
20
|
+
from presidio_analyzer import predefined_recognizers
|
|
21
|
+
|
|
22
|
+
from metadata.generated.schema.entity.classification.tag import Tag
|
|
23
|
+
from metadata.generated.schema.type.contextRecognizer import ContextRecognizer
|
|
24
|
+
from metadata.generated.schema.type.customRecognizer import CustomRecognizer
|
|
25
|
+
from metadata.generated.schema.type.denyListRecognizer import DenyListRecognizer
|
|
26
|
+
from metadata.generated.schema.type.patternRecognizer import PatternRecognizer
|
|
27
|
+
from metadata.generated.schema.type.predefinedRecognizer import PredefinedRecognizer
|
|
28
|
+
from metadata.generated.schema.type.recognizer import Recognizer
|
|
29
|
+
from metadata.generated.schema.type.recognizers.regexFlags import RegexFlags
|
|
30
|
+
from metadata.pii.algorithms.presidio_utils import apply_confidence_threshold
|
|
31
|
+
from metadata.utils.logger import pii_logger
|
|
32
|
+
|
|
33
|
+
logger = pii_logger()
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
class PresidioRecognizerFactory:
|
|
37
|
+
"""Factory for creating Presidio recognizers from OpenMetadata configurations."""
|
|
38
|
+
|
|
39
|
+
@staticmethod
|
|
40
|
+
def create_recognizer(recognizer_config: Recognizer) -> Optional[EntityRecognizer]:
|
|
41
|
+
"""
|
|
42
|
+
Create a Presidio recognizer from an OpenMetadata recognizer configuration.
|
|
43
|
+
|
|
44
|
+
Args:
|
|
45
|
+
recognizer_config: The recognizer configuration from OpenMetadata
|
|
46
|
+
tag_name: The name of the tag this recognizer belongs to
|
|
47
|
+
|
|
48
|
+
Returns:
|
|
49
|
+
A Presidio EntityRecognizer or None if creation fails
|
|
50
|
+
"""
|
|
51
|
+
if not recognizer_config.enabled:
|
|
52
|
+
return None
|
|
53
|
+
|
|
54
|
+
config = recognizer_config.recognizerConfig.root
|
|
55
|
+
|
|
56
|
+
if isinstance(config, PatternRecognizer):
|
|
57
|
+
recognizer = PresidioRecognizerFactory._create_pattern_recognizer(
|
|
58
|
+
config, recognizer_config
|
|
59
|
+
)
|
|
60
|
+
elif isinstance(config, DenyListRecognizer):
|
|
61
|
+
recognizer = PresidioRecognizerFactory._create_deny_list_recognizer(
|
|
62
|
+
config, recognizer_config
|
|
63
|
+
)
|
|
64
|
+
elif isinstance(config, ContextRecognizer):
|
|
65
|
+
recognizer = PresidioRecognizerFactory._create_context_recognizer(
|
|
66
|
+
config, recognizer_config
|
|
67
|
+
)
|
|
68
|
+
elif isinstance(config, CustomRecognizer):
|
|
69
|
+
recognizer = PresidioRecognizerFactory._create_custom_recognizer(
|
|
70
|
+
config, recognizer_config
|
|
71
|
+
)
|
|
72
|
+
elif isinstance(
|
|
73
|
+
config, PredefinedRecognizer
|
|
74
|
+
): # pyright: ignore[reportUnnecessaryIsInstance]
|
|
75
|
+
recognizer = PresidioRecognizerFactory._create_predefined_recognizer(
|
|
76
|
+
config, recognizer_config
|
|
77
|
+
)
|
|
78
|
+
else:
|
|
79
|
+
logger.warning(f"Unknown recognizer type for {recognizer_config.name}")
|
|
80
|
+
return None
|
|
81
|
+
|
|
82
|
+
if recognizer and (threshold := recognizer_config.confidenceThreshold):
|
|
83
|
+
patch_analyze = apply_confidence_threshold(threshold)
|
|
84
|
+
recognizer = patch_analyze(recognizer)
|
|
85
|
+
|
|
86
|
+
return recognizer
|
|
87
|
+
|
|
88
|
+
@staticmethod
|
|
89
|
+
def _get_regex_flags(flags: Optional[RegexFlags]) -> Optional[int]:
|
|
90
|
+
if flags is None:
|
|
91
|
+
return re.IGNORECASE | re.DOTALL | re.MULTILINE
|
|
92
|
+
|
|
93
|
+
re_flags = 0
|
|
94
|
+
if flags.ignoreCase:
|
|
95
|
+
re_flags |= re.IGNORECASE
|
|
96
|
+
if flags.dotAll:
|
|
97
|
+
re_flags |= re.DOTALL
|
|
98
|
+
if flags.multiline:
|
|
99
|
+
re_flags |= re.MULTILINE
|
|
100
|
+
|
|
101
|
+
return re_flags
|
|
102
|
+
|
|
103
|
+
@staticmethod
|
|
104
|
+
def _create_pattern_recognizer(
|
|
105
|
+
config: PatternRecognizer,
|
|
106
|
+
recognizer_config: Recognizer,
|
|
107
|
+
) -> PresidioPatternRecognizer:
|
|
108
|
+
"""Create a pattern-based recognizer."""
|
|
109
|
+
patterns: List[PresidioPattern] = []
|
|
110
|
+
for pattern_config in config.patterns:
|
|
111
|
+
patterns.append(
|
|
112
|
+
PresidioPattern(
|
|
113
|
+
name=pattern_config.name,
|
|
114
|
+
regex=pattern_config.regex,
|
|
115
|
+
score=pattern_config.score or 0.8,
|
|
116
|
+
)
|
|
117
|
+
)
|
|
118
|
+
|
|
119
|
+
return PresidioPatternRecognizer(
|
|
120
|
+
supported_entity=config.supportedEntity.value,
|
|
121
|
+
patterns=patterns,
|
|
122
|
+
name=recognizer_config.name.root,
|
|
123
|
+
supported_language=config.supportedLanguage,
|
|
124
|
+
global_regex_flags=PresidioRecognizerFactory._get_regex_flags(
|
|
125
|
+
config.regexFlags
|
|
126
|
+
),
|
|
127
|
+
)
|
|
128
|
+
|
|
129
|
+
@staticmethod
|
|
130
|
+
def _create_deny_list_recognizer(
|
|
131
|
+
config: DenyListRecognizer, recognizer_config: Recognizer
|
|
132
|
+
) -> PresidioPatternRecognizer:
|
|
133
|
+
"""Create a deny list recognizer using patterns."""
|
|
134
|
+
patterns: List[PresidioPattern] = []
|
|
135
|
+
for value in config.denyList:
|
|
136
|
+
# Escape special regex characters in the value
|
|
137
|
+
escaped_value = re.escape(value)
|
|
138
|
+
|
|
139
|
+
patterns.append(
|
|
140
|
+
PresidioPattern(
|
|
141
|
+
name=f"deny_{value}",
|
|
142
|
+
regex=escaped_value,
|
|
143
|
+
score=0.9, # High confidence for exact matches
|
|
144
|
+
)
|
|
145
|
+
)
|
|
146
|
+
|
|
147
|
+
return PresidioPatternRecognizer(
|
|
148
|
+
supported_entity=config.supportedEntity.value,
|
|
149
|
+
patterns=patterns,
|
|
150
|
+
name=recognizer_config.name.root,
|
|
151
|
+
supported_language=config.supportedLanguage,
|
|
152
|
+
global_regex_flags=PresidioRecognizerFactory._get_regex_flags(
|
|
153
|
+
config.regexFlags
|
|
154
|
+
),
|
|
155
|
+
)
|
|
156
|
+
|
|
157
|
+
@staticmethod
|
|
158
|
+
def _create_context_recognizer(
|
|
159
|
+
config: ContextRecognizer, recognizer_config: Recognizer
|
|
160
|
+
) -> PresidioPatternRecognizer:
|
|
161
|
+
"""Create a context-aware recognizer."""
|
|
162
|
+
# For context recognizers, we can use a pattern recognizer with context words
|
|
163
|
+
# or implement a custom recognizer that uses NLP
|
|
164
|
+
context_patterns: List[PresidioPattern] = []
|
|
165
|
+
|
|
166
|
+
# Create patterns that look for context words near potential entities
|
|
167
|
+
for context_word in config.contextWords:
|
|
168
|
+
# Pattern to match words near context words
|
|
169
|
+
pattern = f"(?i)(?:{context_word})\\s+\\w+|\\w+\\s+(?:{context_word})"
|
|
170
|
+
context_patterns.append(
|
|
171
|
+
PresidioPattern(
|
|
172
|
+
name=f"context_{context_word}",
|
|
173
|
+
regex=pattern,
|
|
174
|
+
score=(config.minScore + config.maxScore) / 2
|
|
175
|
+
if config.minScore and config.maxScore
|
|
176
|
+
else 0.6,
|
|
177
|
+
)
|
|
178
|
+
)
|
|
179
|
+
|
|
180
|
+
return PresidioPatternRecognizer(
|
|
181
|
+
supported_entity=config.supportedEntity.value,
|
|
182
|
+
patterns=context_patterns,
|
|
183
|
+
name=recognizer_config.name.root,
|
|
184
|
+
supported_language=config.supportedLanguage,
|
|
185
|
+
)
|
|
186
|
+
|
|
187
|
+
@staticmethod
|
|
188
|
+
def _create_custom_recognizer(
|
|
189
|
+
config: CustomRecognizer, # pyright: ignore[reportUnusedParameter]
|
|
190
|
+
recognizer_config: Recognizer,
|
|
191
|
+
) -> Optional[EntityRecognizer]:
|
|
192
|
+
"""
|
|
193
|
+
Create a custom recognizer with user-defined logic.
|
|
194
|
+
|
|
195
|
+
For security reasons, we don't execute arbitrary Python code.
|
|
196
|
+
Instead, we could support a limited set of custom behaviors
|
|
197
|
+
or require custom recognizers to be implemented as plugins.
|
|
198
|
+
"""
|
|
199
|
+
logger.warning(
|
|
200
|
+
f"Custom recognizer {recognizer_config.name} requires implementation. "
|
|
201
|
+
+ "Consider using pattern, deny_list, or context recognizers instead."
|
|
202
|
+
)
|
|
203
|
+
return None
|
|
204
|
+
|
|
205
|
+
@staticmethod
|
|
206
|
+
def _create_predefined_recognizer(
|
|
207
|
+
config: PredefinedRecognizer,
|
|
208
|
+
recognizer: Recognizer, # pyright: ignore[reportUnusedParameter]
|
|
209
|
+
) -> Optional[EntityRecognizer]:
|
|
210
|
+
"""Create a custom recognizer with user-defined logic."""
|
|
211
|
+
try:
|
|
212
|
+
predefined_class = getattr(predefined_recognizers, config.name.value)
|
|
213
|
+
except AttributeError:
|
|
214
|
+
logger.error(f"Recognizer {config.name} not found")
|
|
215
|
+
return None
|
|
216
|
+
|
|
217
|
+
args = {}
|
|
218
|
+
if supported_language := config.supportedLanguage:
|
|
219
|
+
args["supported_language"] = supported_language
|
|
220
|
+
if context := config.context:
|
|
221
|
+
args["context"] = context
|
|
222
|
+
|
|
223
|
+
return predefined_class(**args)
|
|
224
|
+
|
|
225
|
+
@staticmethod
|
|
226
|
+
def create_recognizers_for_tag(tag: Tag) -> List[EntityRecognizer]:
|
|
227
|
+
"""
|
|
228
|
+
Create all enabled recognizers for a given tag.
|
|
229
|
+
|
|
230
|
+
Args:
|
|
231
|
+
tag: The tag containing recognizer configurations
|
|
232
|
+
|
|
233
|
+
Returns:
|
|
234
|
+
List of Presidio EntityRecognizer instances
|
|
235
|
+
"""
|
|
236
|
+
recognizers: List[EntityRecognizer] = []
|
|
237
|
+
|
|
238
|
+
if not tag.autoClassificationEnabled or not tag.recognizers:
|
|
239
|
+
return recognizers
|
|
240
|
+
|
|
241
|
+
for recognizer_config in tag.recognizers:
|
|
242
|
+
recognizer = PresidioRecognizerFactory.create_recognizer(recognizer_config)
|
|
243
|
+
if recognizer:
|
|
244
|
+
recognizers.append(recognizer)
|
|
245
|
+
logger.info(
|
|
246
|
+
f"Created recognizer {recognizer_config.name} for tag {tag.name}"
|
|
247
|
+
)
|
|
248
|
+
|
|
249
|
+
return recognizers
|
|
250
|
+
|
|
251
|
+
|
|
252
|
+
class RecognizerRegistry:
|
|
253
|
+
"""Registry for managing custom recognizers from OpenMetadata."""
|
|
254
|
+
|
|
255
|
+
def __init__(self):
|
|
256
|
+
self.recognizers: Dict[str, List[EntityRecognizer]] = {}
|
|
257
|
+
self.tag_priority: Dict[str, int] = {}
|
|
258
|
+
self.tag_confidence_threshold: Dict[str, float] = {}
|
|
259
|
+
|
|
260
|
+
def register_tag_recognizers(self, tag: Tag) -> None:
|
|
261
|
+
"""
|
|
262
|
+
Register all recognizers for a tag.
|
|
263
|
+
|
|
264
|
+
Args:
|
|
265
|
+
tag: The tag with recognizer configurations
|
|
266
|
+
"""
|
|
267
|
+
if not tag.autoClassificationEnabled:
|
|
268
|
+
return
|
|
269
|
+
|
|
270
|
+
tag_fqn = cast(str, tag.fullyQualifiedName)
|
|
271
|
+
self.recognizers[
|
|
272
|
+
tag_fqn
|
|
273
|
+
] = PresidioRecognizerFactory.create_recognizers_for_tag(tag)
|
|
274
|
+
self.tag_priority[tag_fqn] = tag.autoClassificationPriority or 50
|
|
275
|
+
|
|
276
|
+
# Calculate minimum confidence from all recognizers
|
|
277
|
+
min_confidence = 1.0
|
|
278
|
+
for recognizer_config in tag.recognizers or []:
|
|
279
|
+
if recognizer_config.confidenceThreshold:
|
|
280
|
+
min_confidence = min(
|
|
281
|
+
min_confidence, recognizer_config.confidenceThreshold
|
|
282
|
+
)
|
|
283
|
+
self.tag_confidence_threshold[tag_fqn] = min_confidence
|
|
284
|
+
|
|
285
|
+
def get_recognizers_for_tag(self, tag_fqn: str) -> List[EntityRecognizer]:
|
|
286
|
+
"""Get all recognizers registered for a tag."""
|
|
287
|
+
return self.recognizers.get(tag_fqn, [])
|
|
288
|
+
|
|
289
|
+
def get_all_recognizers(self) -> List[EntityRecognizer]:
|
|
290
|
+
"""Get all registered recognizers across all tags."""
|
|
291
|
+
all_recognizers: list[EntityRecognizer] = []
|
|
292
|
+
for recognizers in self.recognizers.values():
|
|
293
|
+
all_recognizers.extend(recognizers)
|
|
294
|
+
return all_recognizers
|
|
295
|
+
|
|
296
|
+
def get_tag_priority(self, tag_fqn: str) -> int:
|
|
297
|
+
"""Get the priority for a tag (used in conflict resolution)."""
|
|
298
|
+
return self.tag_priority.get(tag_fqn, 50)
|
|
299
|
+
|
|
300
|
+
def get_tag_confidence_threshold(self, tag_fqn: str) -> float:
|
|
301
|
+
"""Get the minimum confidence threshold for a tag."""
|
|
302
|
+
return self.tag_confidence_threshold.get(tag_fqn, 0.6)
|