From 861a5ecf59372ae43bca444d8cc96392b29a8510 Mon Sep 17 00:00:00 2001 From: Teddy Date: Thu, 30 Jun 2022 14:58:30 +0200 Subject: [PATCH] Fixes #5800 profiler workflow for engine with catalog + modulo for Presto (#5801) --- ingestion/src/metadata/orm_profiler/api/workflow.py | 9 ++++++++- .../src/metadata/orm_profiler/orm/functions/modulo.py | 1 + 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/orm_profiler/api/workflow.py b/ingestion/src/metadata/orm_profiler/api/workflow.py index dcffc1ba894..56c697c5aa8 100644 --- a/ingestion/src/metadata/orm_profiler/api/workflow.py +++ b/ingestion/src/metadata/orm_profiler/api/workflow.py @@ -198,7 +198,14 @@ class ProfilerWorkflow: if hasattr( self.config.source.serviceConnection.__root__.config, "supportsDatabase" ): - copy_service_connection_config.database = database.name.__root__ + if hasattr( + self.config.source.serviceConnection.__root__.config, "database" + ): + copy_service_connection_config.database = database.name.__root__ + if hasattr( + self.config.source.serviceConnection.__root__.config, "catalog" + ): + copy_service_connection_config.catalog = database.name.__root__ self.create_processor(copy_service_connection_config) diff --git a/ingestion/src/metadata/orm_profiler/orm/functions/modulo.py b/ingestion/src/metadata/orm_profiler/orm/functions/modulo.py index 6f300f217ec..b6f64dbaebc 100644 --- a/ingestion/src/metadata/orm_profiler/orm/functions/modulo.py +++ b/ingestion/src/metadata/orm_profiler/orm/functions/modulo.py @@ -60,6 +60,7 @@ def _(element, compiler, **kw): @compiles(ModuloFn, Dialects.Athena) @compiles(ModuloFn, Dialects.MySQL) @compiles(ModuloFn, Dialects.Oracle) +@compiles(ModuloFn, Dialects.Presto) def _(element, compiler, **kw): value, base = validate_and_compile(element, compiler, **kw)