dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Chart fixes

Browse Source
This commit is contained in:
Ruud
2014-05-06 15:51:09 +02:00
parent 1b6f010df2
commit 66a149590b
1 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
couchpotato/core/media/movie/providers/automation/imdb.py
+17 -12
View File
@@ -138,11 +138,11 @@ class IMDBAutomation(IMDBBase):
data = self.getHTMLData(url)
if data:
try:
html = BeautifulSoup(data)
if chart.get('type', 'html') == 'html':
result_div = html.find('table')
result_div = html.find('div', attrs = {'id': 'main'})
try:
if url in self.first_table:
@@ -177,26 +177,31 @@ class IMDBAutomation(IMDBBase):
for name in self.charts:
chart = self.charts[name].copy()
url = chart.get('url')
if self.conf('chart_display_%s' % name):
chart['list'] = []
data = self.getHTMLData(chart.get('url'))
data = self.getHTMLData(url)
if data:
html = BeautifulSoup(data)
try:
result_div = html.find('div', attrs = {'id': 'main'})
try:
if chart.get('url') in self.first_table:
table = result_div.find('table')
result_div = table if table else result_div
except:
pass
if chart.get('type', 'html') == 'html':
result_div = html.find('div', attrs = {'id': 'main'})
imdb_ids = getImdb(str(result_div), multiple = True)
try:
if url in self.first_table:
table = result_div.find('table')
result_div = table if table else result_div
except:
pass
imdb_ids = getImdb(str(result_div), multiple = True)
else:
imdb_ids = getImdb(str(data), multiple = True)
for imdb_id in imdb_ids[0:max_items]:
info = self.getInfo(imdb_id)
@@ -205,7 +210,7 @@ class IMDBAutomation(IMDBBase):
if self.shuttingDown():
break
except:
log.error('Failed loading IMDB chart results from %s: %s', (chart.get('url'), traceback.format_exc()))
log.error('Failed loading IMDB chart results from %s: %s', (url, traceback.format_exc()))
if chart['list']:
movie_lists.append(chart)