yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-09-15 04:01:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tableau connector when it cannot connect to URI (#4451)

Browse Source
This commit is contained in:
Gabe Lyons 2022-03-18 15:46:46 -07:00 committed by GitHub
parent 89a8fa05eb
commit bf35d4c83e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
metadata-ingestion/src/datahub/ingestion/source/tableau.py
View File

@ -120,6 +120,7 @@ class TableauSource(Source):
self.config = config
self.report = SourceReport()
self.server = None
# This list keeps track of datasource being actively used by workbooks so that we only retrieve those
# when emitting published data sources.
self.datasource_ids_being_used: List[str] = []
@ -130,7 +131,8 @@ class TableauSource(Source):
self._authenticate()
def close(self) -> None:
self.server.auth.sign_out()
if self.server is not None:
self.server.auth.sign_out()
def _authenticate(self):
# https://tableau.github.io/server-client-python/docs/api-ref#authentication
@ -154,11 +156,13 @@ class TableauSource(Source):
self.server = Server(self.config.connect_uri, use_server_version=True)
self.server.auth.sign_in(authentication)
except ServerResponseError as e:
logger.error(e)
self.report.report_failure(
key="tableau-login",
reason=f"Unable to Login with credentials provided" f"Reason: {str(e)}",
)
except Exception as e:
logger.error(e)
self.report.report_failure(
key="tableau-login", reason=f"Unable to Login" f"Reason: {str(e)}"
)
@ -925,6 +929,8 @@ class TableauSource(Source):
return cls(ctx, config)
def get_workunits(self) -> Iterable[MetadataWorkUnit]:
if self.server is None:
return
try:
yield from self.emit_workbooks(self.config.workbooks_page_size)
if self.datasource_ids_being_used: