-
Notifications
You must be signed in to change notification settings - Fork 1
/
rest.py
128 lines (104 loc) · 3.79 KB
/
rest.py
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#! /usr/bin/env python
# -*- coding: utf8 -*-
from settings import NODE, SUPERVISOR
from six.moves.http_client import FORBIDDEN, NOT_FOUND, OK
from sys import exit
from requests import ConnectionError
import logging
import os
import requests
import json
class NotFoundException(Exception):
"""Raised when there is no print request."""
pass
class UnauthorizedException(Exception):
"""Raised when the Node is unauthorised."""
pass
class UnknownErrorException(Exception):
"""Raised when an unknown error occured."""
pass
class RESTConnection(object):
"""Class gathering all the REST queries and requests."""
@classmethod
def __get(cls, url):
"""Perform a GET request on the given URL."""
try:
return requests.get(
SUPERVISOR['HOST'] + url + '?format=json',
verify=True,
headers={'X-token': NODE['TOKEN']}
)
except ConnectionError as e:
logging.critical("Connection failed: {}".format(e))
exit(1)
@classmethod
def __put(cls, url, *args, **kwargs):
"""Perform a POST request on the given URL."""
headers = {'X-token': NODE['TOKEN']}
headers.update(kwargs.get('headers', {}))
del kwargs['headers']
try:
return requests.put(
SUPERVISOR['HOST'] + url + '?format=json',
verify=True,
headers=headers,
*args, **kwargs
)
except ConnectionError as e:
logging.critical("Connection failed: {}".format(e))
exit(1)
@classmethod
def get_print_request(cls):
"""Get a new print request for printing."""
# Prepare the URL
url = 'printrequest/for_printing/'
response = cls.__get(url)
if response.status_code == OK:
# If everything is okay then parse the data and return it
try:
data = json.loads(response.text)
except Exception as e:
logging.error(
"Error while parsing response: {}\n{}.".format(data, e)
)
raise
logging.info(
"Recived print request id {}.".format(data['id'])
)
return data
elif response.status_code == NOT_FOUND:
logging.info("No print request recived.")
raise NotFoundException()
elif response.status_code == FORBIDDEN:
logging.warning("Node unauthorized.")
raise UnauthorizedException()
else:
logging.error("Unknown error status code: {}".format(
response.status_code))
raise UnknownErrorException()
@classmethod
def post_print_request(cls, printRequestId, printRequest):
"""Post a printed print request to the Supervisor."""
url = 'printrequest/{}/'.format(printRequestId)
# Prepare the print request
try:
data = json.dumps(printRequest)
except:
logging.critical("Data malformed {}.".format(printRequest))
raise
response = cls.__put(
url,
data=data,
headers={'Content-Type': 'application/json'}
)
if response.status_code == OK and not printRequest['error']:
logging.info("The results have been sent.")
elif response.status_code == OK and printRequest['error']:
logging.info("Supervisor has been notified about the failure.")
elif response.status_code == FORBIDDEN:
logging.warning("Node unauthorized.")
raise UnauthorizedException()
else:
logging.error("Unknown error status code: {}".format(
response.status_code))
raise UnknownErrorException()