mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-25 09:18:50 -05:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import requests
|
|
import fanart
|
|
from fanart.errors import RequestFanartError, ResponseFanartError
|
|
|
|
|
|
class Request(object):
|
|
def __init__(self, apikey, id, ws, type=None, sort=None, limit=None):
|
|
self._apikey = apikey
|
|
self._id = id
|
|
self._ws = ws
|
|
self._type = type or fanart.TYPE.ALL
|
|
self._sort = sort or fanart.SORT.POPULAR
|
|
self._limit = limit or fanart.LIMIT.ALL
|
|
self.validate()
|
|
self._response = None
|
|
|
|
def validate(self):
|
|
for attribute_name in ('ws', 'type', 'sort', 'limit'):
|
|
attribute = getattr(self, '_' + attribute_name)
|
|
choices = getattr(fanart, attribute_name.upper() + '_LIST')
|
|
if attribute not in choices:
|
|
raise RequestFanartError('Not allowed {0}: {1} [{2}]'.format(attribute_name, attribute, ', '.join(choices)))
|
|
|
|
def __str__(self):
|
|
return fanart.BASEURL % (self._ws, self._id, self._apikey)
|
|
|
|
def response(self):
|
|
try:
|
|
response = requests.get(str(self))
|
|
rjson = response.json()
|
|
if not isinstance(rjson, dict):
|
|
raise Exception(response.text)
|
|
return rjson
|
|
except Exception as e:
|
|
raise ResponseFanartError(str(e))
|