aboutsummaryrefslogtreecommitdiffstats
path: root/moodle.py
blob: 5d0f9bd85a87b624679d192080fa0ab280197ef2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/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)