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)
|
