|
| 1 | +import requests |
| 2 | +from .currency import CurrencyService |
| 3 | +from .domain import DomainService |
| 4 | +from .email import EmailService |
| 5 | +from .ip import IPService |
| 6 | +from .webpage import WebpageService |
| 7 | + |
| 8 | + |
| 9 | +class _Interceptor(requests.auth.AuthBase): |
| 10 | + def __init__(self, key): |
| 11 | + self.key = key |
| 12 | + |
| 13 | + def __call__(self, r): |
| 14 | + r.headers['Authorization'] = 'Bearer ' + self.key |
| 15 | + return r |
| 16 | + |
| 17 | + |
| 18 | +class Client(): |
| 19 | + def __init__(self, key): |
| 20 | + self.key = key |
| 21 | + self.interceptor = _Interceptor(key) |
| 22 | + |
| 23 | + def _request(self, method, url, params=None, files=None, data=None): |
| 24 | + r = requests.request(method, url, auth=self.interceptor, |
| 25 | + params=params, files=files, data=data) |
| 26 | + data = r.json() |
| 27 | + if self._is_error(r): |
| 28 | + raise LabStackError(data['code'], data['message']) |
| 29 | + return data |
| 30 | + |
| 31 | + def _is_error(self, r): |
| 32 | + return not 200 <= r.status_code < 300 |
| 33 | + |
| 34 | + def currency(self): |
| 35 | + return CurrencyService(self) |
| 36 | + |
| 37 | + def domain(self): |
| 38 | + return DomainService(self) |
| 39 | + |
| 40 | + def email(self): |
| 41 | + return EmailService(self) |
| 42 | + |
| 43 | + def ip(self): |
| 44 | + return IPService(self) |
| 45 | + |
| 46 | + def webpage(self): |
| 47 | + return WebpageService(self) |
| 48 | + |
| 49 | + |
| 50 | +class LabStackError(Exception): |
| 51 | + def __init__(self, statusCode, code, message): |
| 52 | + self.statusCode = statusCode |
| 53 | + self.code = code |
| 54 | + self.message = message |
| 55 | + |
| 56 | + def __str__(self): |
| 57 | + return self.message |
0 commit comments