add support for TRAINS_NO_DEFAULT_SERVER issue #209

This commit is contained in:
Shaked Klein Orbach 2020-10-01 15:44:35 +03:00 committed by Allegro AI
parent 8adfa3ebd2
commit e1c5aeed06
3 changed files with 6 additions and 2 deletions

View File

@ -10,3 +10,4 @@ ENV_SECRET_KEY = EnvEntry("TRAINS_API_SECRET_KEY", "ALG_API_SECRET_KEY")
ENV_VERBOSE = EnvEntry("TRAINS_API_VERBOSE", "ALG_API_VERBOSE", type=bool, default=False)
ENV_HOST_VERIFY_CERT = EnvEntry("TRAINS_API_HOST_VERIFY_CERT", "ALG_API_HOST_VERIFY_CERT", type=bool, default=True)
ENV_OFFLINE_MODE = EnvEntry("TRAINS_OFFLINE_MODE", "ALG_OFFLINE_MODE", type=bool, converter=safe_text_to_bool)
ENV_TRAINS_NO_DEFAULT_SERVER = EnvEntry("TRAINS_NO_DEFAULT_SERVER", "ALG_NO_DEFAULT_SERVER", type=bool, default=False)

View File

@ -12,7 +12,7 @@ from six.moves.urllib.parse import urlparse, urlunparse
from .callresult import CallResult
from .defs import ENV_VERBOSE, ENV_HOST, ENV_ACCESS_KEY, ENV_SECRET_KEY, ENV_WEB_HOST, \
ENV_FILES_HOST, ENV_OFFLINE_MODE
ENV_FILES_HOST, ENV_OFFLINE_MODE, ENV_TRAINS_NO_DEFAULT_SERVER
from .request import Request, BatchRequest # noqa: F401
from .token_manager import TokenManager
from ..config import load
@ -140,6 +140,9 @@ class Session(TokenManager):
if not host:
raise ValueError("host is required in init or config")
if ENV_TRAINS_NO_DEFAULT_SERVER.get() and host == self.default_host:
raise ValueError("Configuration file or environment could not be located and default demo server is disabled")
self._ssl_error_count_verbosity = self.config.get(
"api.ssl_error_count_verbosity", self._ssl_error_count_verbosity)

View File

@ -1 +1 @@
__version__ = '0.16.2rc3'
__version__ = '0.16.2rc4'