Idoit-Tests 01

idoit_scaleup/base.py

@@ -0,0 +1,54 @@
+import json
+import requests
+from pprint import pprint
+from .api_log import IDoitApiLog
+from json.decoder import JSONDecodeError
+
+
+class IDoitApiBase:
+    def __init__(self, cfg):
+        self.cfg = cfg
+        self.debug = False
+
+    def set_debug_mode(self, debug=True):
+        self.debug = debug
+
+    def xml_rpc_call(self, method, params, debug=False):
+        headers = {'content-type': 'application/json'}
+        payload = {
+            "method": method,
+            "params": params,
+            "jsonrpc": "2.0",
+            "id": 1,
+        }
+        basic_auth = (self.cfg['user'], self.cfg['password'])
+        params['apikey'] = self.cfg['api_key']
+        try:
+            response = requests.post(
+                self.cfg['jrpc_url'],
+                data=json.dumps(payload),
+                auth=basic_auth,
+                headers=headers
+            ).json()
+        except JSONDecodeError:
+            print('JSON DECODE ERROR')
+            print('Url')
+            print(self.cfg['jrpc_url'])
+            print('Payload')
+            payload['params']['apikey'] = 'xxx'
+            pprint(payload)
+            response = {}
+
+        IDoitApiLog.instance().append_api_log(
+            self.cfg['jrpc_url'], payload, response)
+
+        if self.debug or debug or 'error' in response.keys():
+            print('Url')
+            print(self.cfg['jrpc_url'])
+            print('Payload')
+            payload['params']['apikey'] = 'xxx'
+            pprint(payload)
+            print('Resonse')
+            pprint(response)
+
+        return response