diff --git a/ingestion/src/metadata/ingestion/source/database/snowflake/query_parser.py b/ingestion/src/metadata/ingestion/source/database/snowflake/query_parser.py index 418f638c480..4b8bfcc7e0d 100644 --- a/ingestion/src/metadata/ingestion/source/database/snowflake/query_parser.py +++ b/ingestion/src/metadata/ingestion/source/database/snowflake/query_parser.py @@ -24,7 +24,6 @@ from metadata.generated.schema.metadataIngestion.workflow import ( from metadata.generated.schema.type.tableQuery import TableQuery from metadata.ingestion.api.steps import InvalidSourceException from metadata.ingestion.ometa.ometa_api import OpenMetadata -from metadata.ingestion.source.connections import get_connection from metadata.ingestion.source.database.query_parser_source import QueryParserSource from metadata.ingestion.source.database.snowflake.queries import ( SNOWFLAKE_SESSION_TAG_QUERY, @@ -89,24 +88,8 @@ class SnowflakeQueryParserSource(QueryParserSource, ABC): ) def get_table_query(self) -> Iterable[TableQuery]: - database = self.config.serviceConnection.__root__.config.database - if database: - use_db_query = f"USE DATABASE {database}" - self.engine.execute(use_db_query) - self.set_session_query_tag() - yield from super().get_table_query() - else: - query = "SHOW DATABASES" - results = self.engine.execute(query) - for res in results: - row = list(res) - use_db_query = f"USE DATABASE {row[1]}" - self.engine.execute(use_db_query) - logger.info(f"Ingesting from database: {row[1]}") - self.config.serviceConnection.__root__.config.database = row[1] - self.engine = get_connection(self.service_connection) - self.set_session_query_tag() - yield from super().get_table_query() + self.set_session_query_tag() + yield from super().get_table_query() def get_database_name(self, data: dict) -> str: # pylint: disable=arguments-differ """