ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-02-07 13:26:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add error message when Trains API Server is not accessible on startup

Browse Source
This commit is contained in:
allegroai 2020-07-11 01:44:45 +03:00
parent 17d2bf2a3e
commit b13cc1e8e7
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
trains_agent/backend_api/session/session.py
View File

@ -134,7 +134,6 @@ class Session(TokenManager):
"api.http.retries", ConfigTree() "api.http.retries", ConfigTree()
).as_plain_ordered_dict() ).as_plain_ordered_dict()
http_retries_config["status_forcelist"] = self._retry_codes http_retries_config["status_forcelist"] = self._retry_codes
self.__http_session = get_http_session_with_retry(**http_retries_config)
self.__worker = worker or gethostname() self.__worker = worker or gethostname()
@ -144,7 +143,14 @@ class Session(TokenManager):
self.client = client or "api-{}".format(__version__) self.client = client or "api-{}".format(__version__)
# limit the reconnect retries, so we get an error if we are starting the session
http_no_retries_config = dict(**http_retries_config)
http_no_retries_config['connect'] = 3
self.__http_session = get_http_session_with_retry(**http_no_retries_config)
# try to connect with the server
self.refresh_token() self.refresh_token()
# create the default session with many retries
self.__http_session = get_http_session_with_retry(**http_retries_config)
# update api version from server response # update api version from server response
try: try:
@ -546,6 +552,9 @@ class Session(TokenManager):
else: else:
raise LoginError("Response data mismatch: No 'token' in 'data' value from res, receive : {}, " raise LoginError("Response data mismatch: No 'token' in 'data' value from res, receive : {}, "
"exception: {}".format(res, ex)) "exception: {}".format(res, ex))
except requests.ConnectionError as ex:
raise ValueError('Connection Error: it seems *api_server* is misconfigured. '
'Is this the TRAINS API server {} ?'.format('/'.join(ex.request.url.split('/')[:3])))
except Exception as ex: except Exception as ex:
raise LoginError('Unrecognized Authentication Error: {} {}'.format(type(ex), ex)) raise LoginError('Unrecognized Authentication Error: {} {}'.format(type(ex), ex))