|
3 | 3 | class CloudFlareError(Exception): |
4 | 4 | """ errors for Cloudflare API""" |
5 | 5 |
|
6 | | - def __init__(self, code, message): |
| 6 | + class _code_message(object): |
| 7 | + """ a small class to save away an interger and string (the code and the message)""" |
| 8 | + |
| 9 | + def __init__(self, code, message): |
| 10 | + self.code = code |
| 11 | + self.message = message |
| 12 | + def __int__(self): |
| 13 | + return self.code |
| 14 | + def __str__(self): |
| 15 | + return self.message |
| 16 | + |
| 17 | + def __init__(self, code, message, error_chain=None): |
7 | 18 | """ errors for Cloudflare API""" |
8 | 19 |
|
9 | | - self.code = code |
10 | | - self.message = message |
| 20 | + self.e = self._code_message(int(code), str(message)) |
| 21 | + self.error_chain = None |
| 22 | + if error_chain != None: |
| 23 | + self.error_chain = [] |
| 24 | + for e in error_chain: |
| 25 | + self.error_chain.append(self._code_message(int(e['code']), str(e['message']))) |
| 26 | + # self.error_chain.append({'code': self.code, 'message': str(self.message)}) |
11 | 27 |
|
12 | 28 | def __int__(self): |
13 | | - """ errors for Cloudflare API""" |
| 29 | + """ integer value for Cloudflare API errors""" |
14 | 30 |
|
15 | | - return self.code |
| 31 | + return int(self.e) |
16 | 32 |
|
17 | 33 | def __str__(self): |
18 | | - """ errors for Cloudflare API""" |
| 34 | + """ string value for Cloudflare API errors""" |
| 35 | + |
| 36 | + return str(self.e) |
| 37 | + |
| 38 | + def __len__(self): |
| 39 | + """ Cloudflare API errors can contain a chain of errors""" |
| 40 | + |
| 41 | + if self.error_chain == None: |
| 42 | + return 0 |
| 43 | + else: |
| 44 | + return len(self.error_chain) |
| 45 | + |
| 46 | + def __getitem__(self, ii): |
| 47 | + """ Cloudflare API errors can contain a chain of errors""" |
| 48 | + |
| 49 | + return self.error_chain[ii] |
| 50 | + |
| 51 | + def __iter__(self): |
| 52 | + """ Cloudflare API errors can contain a chain of errors""" |
| 53 | + |
| 54 | + if self.error_chain == None: |
| 55 | + raise StopIteration |
| 56 | + for e in self.error_chain: |
| 57 | + yield e |
19 | 58 |
|
20 | | - return self.message |
| 59 | + def next(self): |
| 60 | + if self.error_chain == None: |
| 61 | + raise StopIteration() |
21 | 62 |
|
22 | 63 | class CloudFlareAPIError(CloudFlareError): |
23 | 64 | """ errors for Cloudflare API""" |
|
0 commit comments