mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-18 07:15:13 -05:00
7165df7405
Used regex checks with javascript popup, for checking if valid imdb csv watchlist. Added test/test_tvdb_api.py for testing the tvdb api for retrieving shows by IMDB id. TODO: Help for assisting users in getting the correct IMDB url. maybe, make interval configurable First changes for imdb watchlist checker Thrown the couch code overboard and created my own version of an imdb watchlist scraper Added IMDB thread and test values with an IMDB csv url. These values should be replaced by front end UI settings. - Removed some tracelogs when unable to download watchlist - Added settings for enabling and setting of the watchlist csv url TODO: - create UI for setting from frontend - Implement frontend and backend checks for validating the csv url Default interval is 1 hour.
231 lines
8.5 KiB
Python
231 lines
8.5 KiB
Python
# Author: KontiSR
|
|
# URL: https://github.com/echel0n/SickRage
|
|
#
|
|
# This file is part of SickRage.
|
|
#
|
|
# SickRage is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickRage is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with SickRage. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import urllib2, urllib
|
|
import shutil
|
|
import urlparse
|
|
import os, datetime
|
|
import requests
|
|
import cookielib
|
|
import re
|
|
from urllib2 import HTTPError, URLError
|
|
|
|
import sickbeard
|
|
from sickbeard import encodingKludge as ek
|
|
from sickbeard import logger
|
|
from sickbeard import helpers
|
|
from sickbeard import search_queue
|
|
from sickbeard.common import SKIPPED, WANTED
|
|
from lib.tvdb_api.tvdb_api import *
|
|
|
|
class ImdbBase():
|
|
def _download(self, baseurl, querystring=""):
|
|
fullurl = baseurl + urllib.urlencode(querystring)
|
|
|
|
req = urllib2.Request(fullurl)
|
|
try:
|
|
response = urllib2.urlopen(req)
|
|
except HTTPError as e:
|
|
logger.log('Could not download IMDB watchlist', logger.DEBUG)
|
|
#print 'Error code: ', e.code
|
|
return False
|
|
except URLError as e:
|
|
logger.log('Could not download IMDB watchlist', logger.DEBUG)
|
|
#print 'Reason: ', e.reason
|
|
return False
|
|
|
|
redirurl = response.geturl()
|
|
htmlResponse = response.read()
|
|
|
|
validHtml = True#BeautifulSoup(htmlResponse, 'html.parser')
|
|
if validHtml:
|
|
return htmlResponse
|
|
|
|
return False
|
|
|
|
|
|
class IMDB(ImdbBase):
|
|
listOfImdbIds = []
|
|
def __init__(self):
|
|
self.listOfImdbIds = []
|
|
|
|
def run(self, force=False):
|
|
try:
|
|
# add shows from trakt.tv watchlist
|
|
if sickbeard.USE_IMDBWATCHLIST:
|
|
self.listOfImdbIds = [] # its about to all get re-added
|
|
self.checkWatchlist() # Check the | separated watchlists (csv) urls
|
|
if len(self.listOfImdbIds):
|
|
self.updateShowsInDb() # Update the db with possible new shows
|
|
|
|
except Exception:
|
|
logger.log(traceback.format_exc(), logger.DEBUG)
|
|
|
|
def _getTTs(self, html):
|
|
nrAddedTTs = 0
|
|
### Get the tt's (shows) from the ajax html. E.a. [ tt1958961|imdb|8.1|8.1|list, tt1958961|imdb|8.1|8.1|list ]
|
|
if not html:
|
|
return False
|
|
|
|
parsedshows = re.findall("(tt[0-9]+)\\|imdb\\|([.0-9]+)", html)
|
|
if not parsedshows:
|
|
return False
|
|
|
|
for show in parsedshows:
|
|
if show[0] not in [x['imdbid'] for x in self.listOfImdbIds]:
|
|
self.listOfImdbIds.append({"imdbid" : show[0], "score" : show[1]})
|
|
nrAddedTTs += 1
|
|
|
|
if nrAddedTTs > 0:
|
|
return nrAddedTTs
|
|
|
|
return False
|
|
|
|
def checkWatchlist(self):
|
|
|
|
### Get imdbListId from the csv url's
|
|
AjaxUrls = self._getImdbAjaxUrls(sickbeard.IMDB_WATCHLISTCSV)
|
|
|
|
### Get imdbUserId from the csv url's
|
|
for url in AjaxUrls:
|
|
getImdbHtml = self._download(url)
|
|
nrAdded = self._getTTs(getImdbHtml)
|
|
|
|
if self.listOfImdbIds:
|
|
return self.listOfImdbIds
|
|
|
|
return False
|
|
|
|
'''
|
|
Tries to use the csvUrls as a comma separated list of imdb csv urls,
|
|
to retrieve a userid and listid for each of the csv url.
|
|
For each csv url an Ajax url is created. Thats used to get the list of Tvshows.
|
|
'''
|
|
def _getImdbAjaxUrls(self, csvUrls):
|
|
ajaxUrls = []
|
|
ajaxUrlBase = u"http://www.imdb.com/list/_ajax/list_filter?"
|
|
|
|
reUserId = re.compile(".*(ur[0-9]+)")
|
|
reListId = re.compile(".*(ls[0-9]+)")
|
|
|
|
#if "|" in csvUrls:
|
|
#print "Multiple Watchlists detected"
|
|
csvurl = csvUrls.split("|")
|
|
for url in csvurl:
|
|
userIdMatch = reUserId.match(url)
|
|
listIdMatch = reListId.match(url)
|
|
|
|
if userIdMatch and listIdMatch:
|
|
query = {"list_id" : listIdMatch.groups()[0],
|
|
"list_class" : "WATCHLIST",
|
|
"view" : "compact",
|
|
"list_type" : "Titles",
|
|
"filter" : '{"title_type":["tv_series"]}',
|
|
"sort_field" : "created",
|
|
"sort_direction" : "desc",
|
|
"user_id" : userIdMatch.groups()[0] }
|
|
ajaxUrls.append(ajaxUrlBase + urllib.urlencode(query))
|
|
if ajaxUrls:
|
|
return ajaxUrls
|
|
|
|
return False
|
|
|
|
def updateShowsInDb(self):
|
|
nrOfaddedShows = 0
|
|
# Get list with thetvdb and imdbIds from DB (tt1234324)
|
|
|
|
|
|
# Get thetvdb indexer_id, showname from tvdb using the IMDB id. ttxxxxx
|
|
# Use "[{listOfImdbIds}]" for updating the db, if the show isn't in it
|
|
tvdb_instance = Tvdb(cache = True, useZip = True)
|
|
for watchlistShow in self.listOfImdbIds:
|
|
if watchlistShow['imdbid'] not in [x.imdbid for x in sickbeard.showList ]:
|
|
TvdbShow = tvdb_instance.search('',imdbid=watchlistShow['imdbid'])
|
|
if TvdbShow:
|
|
self._addDefaultShow(1, TvdbShow['id'], TvdbShow['seriesname'], False)
|
|
nrOfaddedShows += 1
|
|
|
|
return nrOfaddedShows if nrOfaddedShows > 0 else False
|
|
|
|
return False
|
|
|
|
|
|
def _addDefaultShow(self, indexer, indexer_id, name, status):
|
|
"""
|
|
Adds a new show with the default settings
|
|
"""
|
|
if not helpers.findCertainShow(sickbeard.showList, int(indexer_id)):
|
|
logger.log(u"Adding show " + str(indexer_id))
|
|
root_dirs = sickbeard.ROOT_DIRS.split('|')
|
|
|
|
try:
|
|
location = root_dirs[int(root_dirs[0]) + 1]
|
|
except:
|
|
location = None
|
|
|
|
if location:
|
|
showPath = ek.ek(os.path.join, location, helpers.sanitizeFileName(name))
|
|
dir_exists = helpers.makeDir(showPath)
|
|
if not dir_exists:
|
|
logger.log(u"Unable to create the folder " + showPath + ", can't add the show", logger.ERROR)
|
|
return
|
|
else:
|
|
helpers.chmodAsParent(showPath)
|
|
|
|
sickbeard.showQueueScheduler.action.addShow(int(indexer), int(indexer_id), showPath, status,
|
|
int(sickbeard.QUALITY_DEFAULT),
|
|
int(sickbeard.FLATTEN_FOLDERS_DEFAULT),
|
|
paused=False, anime = False)
|
|
else:
|
|
logger.log(u"There was an error creating the show, no root directory setting found", logger.ERROR)
|
|
return
|
|
|
|
# imdbWatchlistTv = "http://www.imdb.com/user/%s/watchlist?ref_=wl_ref_typ&sort=list_order,asc&mode=simple&page=%s&title_type=tvSeries"
|
|
# imdbWatchlistTv2 = "http://www.imdb.com/list/export?list_id=ls009966268&author_id=ur35235230&ref_=wl_exp"
|
|
# imdbUserId = "ur5968686"
|
|
# imdbListId = "ls005547625"
|
|
# imdbWlPage = "1"
|
|
# ajaxUrlBase = u"http://www.imdb.com/list/_ajax/list_filter?"
|
|
# ajaxUrlQueryString = u"list_id=%s&list_class=WATCHLIST&view=compact&list_type=Titles&filter={\"title_type\":[\"tv_series\"]}&sort_field=created&sort_direction=desc&user_id=%s" % (imdbListId, imdbUserId)
|
|
#
|
|
# query = {"list_id" : imdbListId,
|
|
# "list_class" : "WATCHLIST",
|
|
# "view" : "compact",
|
|
# "list_type" : "Titles",
|
|
# "filter" : '{"title_type":["tv_series"]}',
|
|
# "sort_field" : "created",
|
|
# "sort_direction" : "desc",
|
|
# "user_id" : imdbUserId }
|
|
#
|
|
# imdbwatchlistcsv = "http://www.imdb.com/list/export?list_id=ls005547625&author_id=ur5968686&ref_=wl_exp"
|
|
# imdbWatchListTvFullURL = ajaxUrlBase + urllib.urlencode(query)
|
|
# # /download("%s%s" % (baseurl, searchurl), "test.csv")
|
|
#
|
|
# IMDBobj = IMDB()
|
|
#
|
|
# #Test one csv
|
|
# imdbIds = IMDBobj.checkWatchlist(imdbwatchlistcsv)
|
|
# print IMDBobj.listOfImdbIds
|
|
#
|
|
# # Test two csv's
|
|
# imdbIds = IMDBobj.checkWatchlist(imdbwatchlistcsv + "|" + imdbWatchlistTv2)
|
|
# print IMDBobj.listOfImdbIds
|
|
#
|
|
# print imdbIds
|
|
|