Merge pull request #100 from sangoma/wheel_client_api

Add a wheel client api

pepper/libpepper.py

@@ -4,10 +4,8 @@
 (Specifically the rest_cherrypy netapi module.)
 
 '''
-import functools
 import json
 import logging
-import os
 import ssl
 try:
     ssl._create_default_https_context = ssl._create_stdlib_context
@@ -90,9 +88,8 @@ def req_get(self, path):
         print(api.login('salt','salt','pam'))
         print(api.req_get('/keys'))
         '''
-
         import requests
-        
+
         headers = {
             'Accept': 'application/json',
             'Content-Type': 'application/json',
@@ -110,7 +107,7 @@ def req_get(self, path):
                   'verify': self._ssl_verify == True,
                   }
         try:
-            resp = requests.get(**params)           
+            resp = requests.get(**params)
 
             if resp.status_code == 401:
                 raise PepperException(str(resp.status_code) + ':Authentication denied')
@@ -127,7 +124,7 @@ def req_get(self, path):
             print(e)
             return
         return resp.json()
-        
+
     def req(self, path, data=None):
         '''
         A thin wrapper around urllib2 to send requests and return the response
@@ -361,6 +358,27 @@ def runner(self, fun, **kwargs):
 
         return self.low([low], path='/')
 
+    def wheel(self, fun, arg=None, kwarg=None, **kwargs):
+        '''
+        Run a single command using the ``wheel`` client
+
+        Usage::
+          wheel('key.accept', match='myminion')
+        '''
+        low = {
+            'client': 'wheel',
+            'fun': fun,
+        }
+
+        if arg:
+            low['arg'] = arg
+        if kwarg:
+            low['kwarg'] = kwarg
+
+        low.update(kwargs)
+
+        return self.low([low], path='/')
+
     def login(self, username, password, eauth):
         '''
         Authenticate with salt-api and return the user permissions and