1 files changed, 68 insertions, 0 deletions
diff --git a/muddle/moodle.py b/muddle/moodle.py
new file mode 100644
index 0000000..7f3ccf5
--- /dev/null
+++ b/muddle/moodle.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+import requests
+import logging
+
+log = logging.getLogger("muddle.moodle")
+
+#
+# magic moodle api wrapper
+#
+
+
+def request_token(url, user, password):
+    token_url = f"{url}/login/token.php"
+    data = {
+        "username": user,
+        "password": password,
+        "service": "moodle_mobile_app"
+    }
+    log.debug(f"requesting token with POST to {api_url} with DATA {data}")
+    return requests.post(token_url, data=data)
+
+
+def api_call(url, token, function, **kwargs):
+    api_url = f"{url}/webservice/rest/server.php?moodlewsrestformat=json"
+    data = {"wstoken": token, "wsfunction": function}
+    for k, v in kwargs.items():
+        data[str(k)] = v
+
+    log.debug(f"calling api with POST to {api_url} with DATA {data}")
+    try:
+        req = requests.post(api_url, data=data)
+        req.raise_for_status()
+        return req
+    except requests.HTTPError:
+        log.warn(f"Error code returned by HTTP(s) request")
+        return req
+    except (requests.ConnectionError, requests.Timeout, requests.ReadTimeout) as e:
+        log.error(f"Failed to connect for POST request:\n{str(e)}")
+        return None
+
+
+class RestApi:
+    def __init__(self, instance_url, token):
+        self._url = instance_url
+        self._token = token
+
+    def __getattr__(self, key):
+        return lambda **kwargs: api_call(self._url, self._token, str(key), **kwargs)
+
+
+class ApiHelper:
+    def __init__(self, api):
+        self.api = api
+
+    def get_userid(self):
+        req = self.api.core_webservice_get_site_info()
+        if req:
+            return req.json()["userid"]
+        else:
+            return None
+
+    def get_file(self, url, local_path):
+        with requests.post(url, data={"token": self.api._token}, stream=True) as r:
+            r.raise_for_status()
+            with open(local_path, "wb") as f:
+                for chunk in r.iter_content(chunk_size=8192):
+                    if chunk:
+                        f.write(chunk)