Load API key from config

StackStorm · May 25, 2016 · 17d69d3 · 17d69d3
1 parent e2ebe44
commit 17d69d3
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 2 deletions.
diff --git a/st2client/st2client/client.py b/st2client/st2client/client.py
@@ -37,7 +37,7 @@
 
 class Client(object):
     def __init__(self, base_url=None, auth_url=None, api_url=None, api_version=None, cacert=None,
-                 debug=False, token=None):
+                 debug=False, token=None, apikey=None):
         # Get CLI options. If not given, then try to get it from the environment.
         self.endpoints = dict()
 
@@ -79,6 +79,11 @@ def __init__(self, base_url=None, auth_url=None, api_url=None, api_version=None,
 
         self.token = token
 
+        if apikey:
+            os.environ['ST2_API_KEY'] = apikey
+
+        self.apikey = apikey
+
         # Instantiate resource managers and assign appropriate API endpoint.
         self.managers = dict()
         self.managers['Token'] = ResourceManager(

diff --git a/st2client/st2client/config_parser.py b/st2client/st2client/config_parser.py
@@ -79,6 +79,10 @@
         'password': {
             'type': 'string',
             'default': None
+        },
+        'apikey': {
+            'type': 'string',
+            'default': None
         }
     },
     'api': {

diff --git a/st2client/st2client/shell.py b/st2client/st2client/shell.py
@@ -79,6 +79,7 @@
     'auth_url': ['auth', 'url'],
     'api_url': ['api', 'url'],
     'api_version': ['general', 'api_version'],
+    'apikey': ['credentials', 'apikey'],
     'cacert': ['general', 'cacert'],
     'debug': ['cli', 'debug']
 }

diff --git a/st2client/tests/unit/test_config_parser.py b/st2client/tests/unit/test_config_parser.py
@@ -55,7 +55,8 @@ def test_parse(self):
             },
             'credentials': {
                 'username': 'test1',
-                'password': 'test1'
+                'password': 'test1',
+                'apikey': None
             },
             'api': {
                 'url': 'http://127.0.0.1:9101/v1'