aboutsummaryrefslogtreecommitdiffstats
path: root/moodle.py
blob: 7f3ccf51fb2daef4c3cd28da5a99b47af9388e33 (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
60
61
62
63
64
65
66
67
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)