diff --git a/config.py b/config.py index d33ddd3..f2de118 100644 --- a/config.py +++ b/config.py @@ -14,6 +14,8 @@ sqlite_db_path = os.getenv('SQLITE_DB_PATH', 'squads_stat.sqlite3') log_level = os.getenv('LOG_LEVEL', 'DEBUG').upper() +capi_endpoint = os.getenv("DEMB_CAPI_ENDPOINT", "https://capi.demb.uk/random_token") + time_between_requests = os.getenv("TIME_BETWEEN_REQUESTS") sqlite2postgres_sqlite_location = os.getenv('SQLITE2POSTGRES_SQLITE_LOCATION') diff --git a/utils.py b/utils.py index eff4b56..3df1b4d 100644 --- a/utils.py +++ b/utils.py @@ -125,14 +125,14 @@ def _get_bearer() -> str: :return: bearer token as str """ bearer_request: requests.Response = requests.get( - url='https://capi.demb.uk/random_token', headers={'auth': os.environ['DEMB_CAPI_AUTH']}, timeout=30 + url=config.capi_endpoint, headers={'auth': os.environ['DEMB_CAPI_AUTH']}, timeout=30 ) try: bearer: str = bearer_request.json()['access_token'] except Exception as e: - logger.exception(f'Unable to parse capi.demb.uk answer\nrequested: {bearer_request.url!r}\n' + logger.exception(f'Unable to parse {config.capi_endpoint} answer\nrequested: {bearer_request.url!r}\n' f'code: {bearer_request.status_code!r}\nresponse: {bearer_request.content!r}', exc_info=e) raise e