mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-06 01:15:05 -05:00
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
|
||
|
|
|