Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
handle view column issues
  • Loading branch information
hsheth2 committed Jul 1, 2021
commit aaa5fefd29c8cbd5a8f3e3c2a6cc7a75b7ba528b
18 changes: 13 additions & 5 deletions metadata-ingestion/src/datahub/ingestion/source/sql_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,7 +336,16 @@ def loop_views(
self.report.report_dropped(dataset_name)
continue

columns = inspector.get_columns(view, schema)
try:
columns = inspector.get_columns(view, schema)
except KeyError:
# For certain types of views, we are unable to fetch the list of columns.
schema_metadata = None
else:
schema_metadata = get_schema_metadata(
self.report, dataset_name, self.platform, columns
)

try:
# SQLALchemy stubs are incomplete and missing this method.
# PR: https://github.com/dropbox/sqlalchemy-stubs/pull/223.
Expand Down Expand Up @@ -374,10 +383,9 @@ def loop_views(
# uri=dataset_name,
)
dataset_snapshot.aspects.append(dataset_properties)
schema_metadata = get_schema_metadata(
self.report, dataset_name, self.platform, columns
)
dataset_snapshot.aspects.append(schema_metadata)

if schema_metadata:
dataset_snapshot.aspects.append(schema_metadata)

mce = MetadataChangeEvent(proposedSnapshot=dataset_snapshot)
wu = SqlWorkUnit(id=dataset_name, mce=mce)
Expand Down