Commit | Line | Data |
---|---|---|
22422846 MM |
1 | #!/usr/bin/env python3 |
2 | ||
3 | ||
4 | class DiaspyError(Exception): | |
5 | """Base exception for all errors | |
6 | raised by diaspy. | |
7 | """ | |
8 | pass | |
9 | ||
10 | ||
11 | class LoginError(DiaspyError): | |
12 | """Exception raised when something | |
13 | bad happens while performing actions | |
14 | related to logging in. | |
15 | """ | |
16 | pass | |
17 | ||
18 | ||
65b1f099 MM |
19 | class UserError(DiaspyError): |
20 | """Exception raised when something related to users goes wrong. | |
21 | """ | |
22 | pass | |
23 | ||
24 | ||
7ebc8b0d MM |
25 | class ConversationError(DiaspyError): |
26 | """Exception raised when something related to conversations goes wrong. | |
27 | """ | |
28 | pass | |
29 | ||
30 | ||
22422846 MM |
31 | def react(r, message='', accepted=[200, 201, 202, 203, 204, 205, 206], exception=DiaspyError): |
32 | """This method tries to decides how to react | |
33 | to a response code passed to it. If it's an | |
34 | error code it will raise an exception (it will | |
35 | call `throw()` method. | |
36 | ||
37 | If response code is not accepted AND cannot | |
38 | be matched to any exception, generic exception | |
39 | (DiaspyError) is raised (provided that `exception` | |
40 | param was left untouched). | |
41 | ||
42 | By default `accepted` param contains all HTTP | |
43 | success codes. | |
44 | ||
45 | User can force type of exception to raise by passing | |
46 | `exception` param. | |
47 | ||
48 | :param r: response code | |
49 | :type r: int | |
50 | :param message: message for the exception | |
51 | :type message: str | |
52 | :param accepted: list of accepted error codes | |
53 | :type accepted: list | |
54 | :param exception: preferred exception to raise | |
55 | :type exception: valid exception type (default: DiaspyError) | |
56 | """ | |
57 | if r in accepted: e = None | |
58 | else: e = DiaspyError | |
59 | ||
60 | if e is not None: e = exception | |
61 | throw(e, message=message) | |
62 | ||
63 | ||
64 | def throw(e, message=''): | |
65 | """This function throws an error with given message. | |
66 | If None is passed as `e` throw() will not raise | |
67 | anything. | |
68 | ||
69 | :param e: exception to throw | |
70 | :type e: any valid exception type or None | |
71 | :param message: message for exception | |
72 | :type message: str | |
73 | """ | |
74 | if e is None: pass | |
75 | else: raise e(message) |