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
"""
import os.path
import os
import apistar
import logging
import yaml
......@@ -16,6 +16,22 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
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):
"""
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