yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-07 16:11:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #15355: fix KeyError issue if not present (#15387)

Browse Source
This commit is contained in:
Ayush Shah 2024-02-28 19:55:03 +05:30 committed by GitHub
parent 3e83bdac3d
commit c31bb98e64
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ingestion/src/metadata/ingestion/source/database/bigquery/metadata.py
View File

@ -252,7 +252,9 @@ class BigquerySource(
test_connection_fn(
self.metadata, inspector_details.engine, self.service_connection
)
if os.environ[GOOGLE_CREDENTIALS]:
# GOOGLE_CREDENTIALS may not have been set,
# to avoid key error, we use `get` for dict
if os.environ.get(GOOGLE_CREDENTIALS):
self.temp_credentials_file_path.append(os.environ[GOOGLE_CREDENTIALS])
def query_table_names_and_types(
@ -443,7 +445,8 @@ class BigquerySource(
inspector_details = get_inspector_details(
database_name=database_name, service_connection=self.service_connection
)
self.temp_credentials_file_path.append(os.environ[GOOGLE_CREDENTIALS])
if os.environ.get(GOOGLE_CREDENTIALS):
self.temp_credentials_file_path.append(os.environ[GOOGLE_CREDENTIALS])
self.client = inspector_details.client
self.engine = inspector_details.engine
self.inspector = inspector_details.inspector