Skip to content
Snippets Groups Projects
Commit 2c5c1cd0 authored by Erwan Rouchet's avatar Erwan Rouchet Committed by Bastien Abadie
Browse files

Helper method to get options from environment variables

parent 86304d31
No related branches found
No related tags found
No related merge requests found
Pipeline #28145 passed
from arkindex.client import ArkindexClient # noqa: F401 from arkindex.client import ArkindexClient, options_from_env # noqa: F401
""" """
Arkindex API Client Arkindex API Client
""" """
import os.path import os
import apistar import apistar
import logging import logging
import yaml import yaml
...@@ -16,6 +16,22 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__)) ...@@ -16,6 +16,22 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def options_from_env():
"""
Get API client keyword arguments from environment variables.
"""
options = {}
if 'ARKINDEX_API_TOKEN' in os.environ:
options['token'] = os.environ.get('ARKINDEX_API_TOKEN')
if 'ARKINDEX_API_INTERNAL_URL' in os.environ:
options['base_url'] = os.environ.get('ARKINDEX_API_INTERNAL_URL')
elif 'ARKINDEX_API_URL' in os.environ:
options['base_url'] = os.environ.get('ARKINDEX_API_URL')
return options
class ArkindexClient(apistar.Client): class ArkindexClient(apistar.Client):
""" """
An Arkindex API client. An Arkindex API client.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment