mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 21:53:37 +00:00
87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
|
import requests
|
||
|
|
import re
|
||
|
|
import lib.fanart as fanart
|
||
|
|
from sickbeard.bs4_parser import BS4Parser
|
||
|
|
from .errors import ResponseFanartError
|
||
|
|
|
||
|
|
|
||
|
|
class Request(object):
|
||
|
|
def __init__(self, apikey, tvdb_id, ws=fanart.WS.TV):
|
||
|
|
self._apikey = apikey
|
||
|
|
self._tvdb_id = tvdb_id
|
||
|
|
self._ws = ws
|
||
|
|
self._response = None
|
||
|
|
self._web_url = 'https://fanart.tv/series/%s'
|
||
|
|
self._assets_url = 'https://assets.fanart.tv'
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return fanart.BASEURL % (self._ws, self._tvdb_id, self._apikey)
|
||
|
|
|
||
|
|
def response(self):
|
||
|
|
|
||
|
|
try:
|
||
|
|
response = requests.get(str(self))
|
||
|
|
rjson = response.json()
|
||
|
|
image_type = u'showbackground'
|
||
|
|
rhtml = self.scrape_web(image_type)
|
||
|
|
if not isinstance(rjson, dict) and 0 == len(rhtml[image_type]):
|
||
|
|
raise Exception(response.text)
|
||
|
|
|
||
|
|
if 0 < len(rhtml[image_type]):
|
||
|
|
items = {image_type: []}
|
||
|
|
for item1 in rhtml[image_type]:
|
||
|
|
use_item = True
|
||
|
|
for k, item2 in enumerate(rjson[image_type] or []):
|
||
|
|
if '00' == item2['lang']: # adjust api data of no language to a default
|
||
|
|
rjson[image_type][k]['lang'] = u'en'
|
||
|
|
if item1['id'] == item2['id']:
|
||
|
|
use_item = False
|
||
|
|
break
|
||
|
|
if use_item:
|
||
|
|
items[image_type] += [item1]
|
||
|
|
rjson[image_type] += items[image_type]
|
||
|
|
return rjson
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
raise ResponseFanartError(str(e))
|
||
|
|
|
||
|
|
def scrape_web(self, image_type):
|
||
|
|
try:
|
||
|
|
data = requests.get(self._web_url % self._tvdb_id)
|
||
|
|
if not data:
|
||
|
|
return
|
||
|
|
|
||
|
|
with BS4Parser(data.text, features=['html5lib', 'permissive']) as html:
|
||
|
|
ul_item = html.find('ul', attrs={'class': image_type})
|
||
|
|
if ul_item:
|
||
|
|
li_items = ul_item('li')
|
||
|
|
if li_items:
|
||
|
|
image_urls = {image_type: []}
|
||
|
|
for li_item in li_items:
|
||
|
|
image_id = None
|
||
|
|
item = li_item.find('a', attrs={'class': 'download'}).get('href')
|
||
|
|
if item:
|
||
|
|
match = re.search(r'image=(\d+)', item, re.I)
|
||
|
|
if match:
|
||
|
|
image_id = u'%s' % match.group(1)
|
||
|
|
|
||
|
|
item = li_item.find('a', attrs={'rel': image_type}).get('href')
|
||
|
|
image_url = (u'%s%s' % (self._assets_url, item), None)[None is item]
|
||
|
|
|
||
|
|
item = li_item.find('div', attrs={'class': 'votes'}).get_text()
|
||
|
|
image_likes = (item, 0)[None is item]
|
||
|
|
|
||
|
|
item = li_item.find('div', attrs={'class': 'metrics'}).get_text()
|
||
|
|
image_lang = u'None found'
|
||
|
|
if item:
|
||
|
|
match = re.search(r'Language:\s*(\w+)', item, re.I)
|
||
|
|
if match:
|
||
|
|
image_lang = u'%s' % (match.group(1)[0:2:].lower(), 'en')['None' == match.group(1)]
|
||
|
|
|
||
|
|
if not (None is image_id or None is image_url):
|
||
|
|
image_urls[image_type].append({u'id': image_id, u'url': image_url, u'likes': image_likes, u'lang': image_lang})
|
||
|
|
|
||
|
|
return image_urls
|
||
|
|
except Exception, e:
|
||
|
|
pass
|