mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-17 06:45:05 -05:00
414 lines
15 KiB
Python
414 lines
15 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of Sick Beard.
|
|
#
|
|
# Sick Beard 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.
|
|
#
|
|
# Sick Beard 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 Sick Beard. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# Created on Sep 20, 2012
|
|
# @author: Dermot Buckley <dermot@buckley.ie>
|
|
# @copyright: Dermot Buckley
|
|
#
|
|
|
|
import time
|
|
import traceback
|
|
|
|
try:
|
|
import json
|
|
except ImportError:
|
|
from lib import simplejson as json
|
|
|
|
import sickbeard
|
|
|
|
from sickbeard import logger
|
|
from sickbeard import db
|
|
from sickbeard.exceptions import ex
|
|
from lib import requests
|
|
|
|
MAX_XEM_AGE_SECS = 86400 # 1 day
|
|
|
|
|
|
def get_scene_numbering(indexer_id, indexer, season, episode, fallback_to_xem=True):
|
|
"""
|
|
Returns a tuple, (season, episode), with the scene numbering (if there is one),
|
|
otherwise returns the xem numbering (if fallback_to_xem is set), otherwise
|
|
returns the TVDB and TVRAGE numbering.
|
|
(so the return values will always be set)
|
|
|
|
@param indexer_id: int
|
|
@param season: int
|
|
@param episode: int
|
|
@param fallback_to_xem: bool If set (the default), check xem for matches if there is no local scene numbering
|
|
@return: (int, int) a tuple with (season, episode)
|
|
"""
|
|
if indexer_id is None or season is None or episode is None:
|
|
return (season, episode)
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
result = find_scene_numbering(indexer_id, indexer, season, episode)
|
|
if result:
|
|
return result
|
|
else:
|
|
if fallback_to_xem:
|
|
xem_result = find_xem_numbering(indexer_id, indexer, season, episode)
|
|
if xem_result:
|
|
return xem_result
|
|
return (season, episode)
|
|
|
|
|
|
def find_scene_numbering(indexer_id, indexer, season, episode):
|
|
"""
|
|
Same as get_scene_numbering(), but returns None if scene numbering is not set
|
|
"""
|
|
if indexer_id is None or season is None or episode is None:
|
|
return (season, episode)
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
rows = myDB.select(
|
|
"SELECT scene_season, scene_episode FROM scene_numbering WHERE indexer = ? and indexer_id = ? and season = ? and episode = ?",
|
|
[indexer, indexer_id, season, episode])
|
|
if rows:
|
|
return (int(rows[0]["scene_season"]), int(rows[0]["scene_episode"]))
|
|
|
|
|
|
def get_indexer_numbering(indexer_id, indexer, sceneSeason, sceneEpisode, fallback_to_xem=True):
|
|
"""
|
|
Returns a tuple, (season, episode) with the TVDB and TVRAGE numbering for (sceneSeason, sceneEpisode)
|
|
(this works like the reverse of get_scene_numbering)
|
|
"""
|
|
if indexer_id is None or sceneSeason is None or sceneEpisode is None:
|
|
return (sceneSeason, sceneEpisode)
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
rows = myDB.select(
|
|
"SELECT season, episode FROM scene_numbering WHERE indexer = ? and indexer_id = ? and scene_season = ? and scene_episode = ?",
|
|
[indexer, indexer_id, sceneSeason, sceneEpisode])
|
|
if rows:
|
|
return (int(rows[0]["season"]), int(rows[0]["episode"]))
|
|
else:
|
|
if fallback_to_xem:
|
|
return get_indexer_numbering_for_xem(indexer_id, indexer, sceneSeason, sceneEpisode)
|
|
return (sceneSeason, sceneEpisode)
|
|
|
|
|
|
def get_scene_numbering_for_show(indexer_id, indexer):
|
|
"""
|
|
Returns a dict of (season, episode) : (sceneSeason, sceneEpisode) mappings
|
|
for an entire show. Both the keys and values of the dict are tuples.
|
|
Will be empty if there are no scene numbers set
|
|
"""
|
|
if indexer_id is None:
|
|
return {}
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
rows = myDB.select(
|
|
'SELECT season, episode, scene_season, scene_episode FROM scene_numbering WHERE indexer = ? and indexer_id = ? ORDER BY season, episode',
|
|
[indexer, indexer_id])
|
|
result = {}
|
|
for row in rows:
|
|
result[(int(row['season']), int(row['episode']))] = (int(row['scene_season']), int(row['scene_episode']))
|
|
|
|
return result
|
|
|
|
|
|
def set_scene_numbering(indexer_id, indexer, season, episode, sceneSeason=None, sceneEpisode=None):
|
|
"""
|
|
Set scene numbering for a season/episode.
|
|
To clear the scene numbering, leave both sceneSeason and sceneEpisode as None.
|
|
|
|
"""
|
|
if indexer_id is None or season is None or episode is None:
|
|
return
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
# sanity
|
|
#if sceneSeason == None: sceneSeason = season
|
|
#if sceneEpisode == None: sceneEpisode = episode
|
|
|
|
# delete any existing record first
|
|
myDB.action('DELETE FROM scene_numbering where indexer = ? and indexer_id = ? and season = ? and episode = ?',
|
|
[indexer, indexer_id, season, episode])
|
|
|
|
# now, if the new numbering is not the default, we save a new record
|
|
if sceneSeason is not None and sceneEpisode is not None:
|
|
myDB.action(
|
|
"INSERT INTO scene_numbering (indexer, indexer_id, season, episode, scene_season, scene_episode) VALUES (?,?,?,?,?,?)",
|
|
[indexer, indexer_id, season, episode, sceneSeason, sceneEpisode])
|
|
|
|
|
|
def find_xem_numbering(indexer_id, indexer, season, episode):
|
|
"""
|
|
Returns the scene numbering, as retrieved from xem.
|
|
Refreshes/Loads as needed.
|
|
|
|
@param indexer_id: int
|
|
@param season: int
|
|
@param episode: int
|
|
@return: (int, int) a tuple of scene_season, scene_episode, or None if there is no special mapping.
|
|
"""
|
|
if indexer_id is None or season is None or episode is None:
|
|
return None
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
if _xem_refresh_needed(indexer_id, indexer):
|
|
_xem_refresh(indexer_id, indexer)
|
|
|
|
cacheDB = db.DBConnection('cache.db')
|
|
myDB = db.DBConnection()
|
|
|
|
rows = cacheDB.select(
|
|
"SELECT scene_season, scene_episode FROM xem_numbering WHERE indexer = ? and indexer_id = ? and season = ? and episode = ?",
|
|
[indexer, indexer_id, season, episode])
|
|
|
|
scene_seasons = cacheDB.select(
|
|
"SELECT COUNT(DISTINCT scene_season) as count FROM xem_numbering WHERE indexer = ? and indexer_id = ?",
|
|
[indexer, indexer_id])
|
|
|
|
indexer_seasons = myDB.select(
|
|
"SELECT COUNT(DISTINCT season) as count FROM tv_episodes WHERE indexer = ? and showid = ?",
|
|
[indexer, indexer_id])
|
|
|
|
if rows:
|
|
return (int(rows[0]["scene_season"]), int(rows[0]["scene_episode"]))
|
|
elif int(scene_seasons[0]["count"]) > 0 and int(indexer_seasons[0]["count"]) > int(scene_seasons[0]["count"]):
|
|
return (0,0)
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_indexer_numbering_for_xem(indexer_id, indexer, sceneSeason, sceneEpisode):
|
|
"""
|
|
Reverse of find_xem_numbering: lookup a tvdb season and episode using scene numbering
|
|
|
|
@param indexer_id: int
|
|
@param sceneSeason: int
|
|
@param sceneEpisode: int
|
|
@return: (int, int) a tuple of (season, episode)
|
|
"""
|
|
if indexer_id is None or sceneSeason is None or sceneEpisode is None:
|
|
return None
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
if _xem_refresh_needed(indexer_id, indexer):
|
|
_xem_refresh(indexer_id, indexer)
|
|
cacheDB = db.DBConnection('cache.db')
|
|
rows = cacheDB.select(
|
|
"SELECT season, episode FROM xem_numbering WHERE indexer = ? and indexer_id = ? and scene_season = ? and scene_episode = ?",
|
|
[indexer, indexer_id, sceneSeason, sceneEpisode])
|
|
if rows:
|
|
return (int(rows[0]["season"]), int(rows[0]["episode"]))
|
|
else:
|
|
return (sceneSeason, sceneEpisode)
|
|
|
|
|
|
def _xem_refresh_needed(indexer_id, indexer):
|
|
"""
|
|
Is a refresh needed on a show?
|
|
|
|
@param indexer_id: int
|
|
@return: bool
|
|
"""
|
|
if indexer_id is None:
|
|
return False
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
cacheDB = db.DBConnection('cache.db')
|
|
rows = cacheDB.select("SELECT last_refreshed FROM xem_refresh WHERE indexer = ? and indexer_id = ?",
|
|
[indexer, indexer_id])
|
|
if rows:
|
|
return time.time() > (int(rows[0]['last_refreshed']) + MAX_XEM_AGE_SECS)
|
|
else:
|
|
return True
|
|
|
|
|
|
def _xem_refresh(indexer_id, indexer):
|
|
"""
|
|
Refresh data from xem for a tv show
|
|
|
|
@param indexer_id: int
|
|
"""
|
|
if indexer_id is None:
|
|
return
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
try:
|
|
logger.log(
|
|
u'Looking up XEM scene mapping for show %s on %s' % (indexer_id, sickbeard.indexerApi(indexer).name,),
|
|
logger.DEBUG)
|
|
data = requests.get("http://thexem.de/map/all?id=%s&origin=%s&destination=scene" % (
|
|
indexer_id, sickbeard.indexerApi(indexer).config['xem_origin'],), verify=False).json()
|
|
|
|
if data is None or data == '':
|
|
logger.log(u'No XEN data for show "%s on %s", trying TVTumbler' % (
|
|
indexer_id, sickbeard.indexerApi(indexer).name,), logger.MESSAGE)
|
|
data = requests.get("http://show-api.tvtumbler.com/api/thexem/all?id=%s&origin=%s&destination=scene" % (
|
|
indexer_id, sickbeard.indexerApi(indexer).config['xem_origin'],), verify=False).json()
|
|
if data is None or data == '':
|
|
logger.log(u'TVTumbler also failed for show "%s on %s". giving up.' % (indexer_id, indexer,),
|
|
logger.MESSAGE)
|
|
return None
|
|
|
|
result = data
|
|
ql = []
|
|
|
|
cacheDB = db.DBConnection('cache.db')
|
|
if result:
|
|
ql.append(["INSERT OR REPLACE INTO xem_refresh (indexer, indexer_id, last_refreshed) VALUES (?,?,?)",
|
|
[indexer, indexer_id, time.time()]])
|
|
if 'success' in result['result']:
|
|
ql.append(["DELETE FROM xem_numbering where indexer = ? and indexer_id = ?", [indexer, indexer_id]])
|
|
for entry in result['data']:
|
|
if 'scene' in entry:
|
|
ql.append([
|
|
"INSERT INTO xem_numbering (indexer, indexer_id, season, episode, scene_season, scene_episode) VALUES (?,?,?,?,?,?)",
|
|
[indexer, indexer_id, entry[sickbeard.indexerApi(indexer).config['xem_origin']]['season'],
|
|
entry[sickbeard.indexerApi(indexer).config['xem_origin']]['episode'],
|
|
entry['scene']['season'], entry['scene']['episode']]])
|
|
if 'scene_2' in entry: # for doubles
|
|
ql.append([
|
|
"INSERT INTO xem_numbering (indexer, indexer_id, season, episode, scene_season, scene_episode) VALUES (?,?,?,?,?,?)",
|
|
[indexer, indexer_id, entry[sickbeard.indexerApi(indexer).config['xem_origin']]['season'],
|
|
entry[sickbeard.indexerApi(indexer).config['xem_origin']]['episode'],
|
|
entry['scene_2']['season'], entry['scene_2']['episode']]])
|
|
else:
|
|
logger.log(u'Failed to get XEM scene data for show %s from %s because "%s"' % (
|
|
indexer_id, sickbeard.indexerApi(indexer).name, result['message']), logger.DEBUG)
|
|
else:
|
|
logger.log(u"Empty lookup result - no XEM data for show %s on %s" % (
|
|
indexer_id, sickbeard.indexerApi(indexer).name,), logger.DEBUG)
|
|
except Exception, e:
|
|
logger.log(u"Exception while refreshing XEM data for show " + str(indexer_id) + " on " + sickbeard.indexerApi(
|
|
indexer).name + ": " + ex(e), logger.WARNING)
|
|
logger.log(traceback.format_exc(), logger.DEBUG)
|
|
return None
|
|
|
|
if ql:
|
|
cacheDB.mass_action(ql)
|
|
|
|
def get_xem_numbering_for_show(indexer_id, indexer):
|
|
"""
|
|
Returns a dict of (season, episode) : (sceneSeason, sceneEpisode) mappings
|
|
for an entire show. Both the keys and values of the dict are tuples.
|
|
Will be empty if there are no scene numbers set in xem
|
|
"""
|
|
if indexer_id is None:
|
|
return {}
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
if _xem_refresh_needed(indexer_id, indexer):
|
|
_xem_refresh(indexer_id, indexer)
|
|
|
|
cacheDB = db.DBConnection('cache.db')
|
|
|
|
rows = cacheDB.select(
|
|
'SELECT season, episode, scene_season, scene_episode FROM xem_numbering WHERE indexer = ? and indexer_id = ? ORDER BY season, episode',
|
|
[indexer, indexer_id])
|
|
|
|
result = {}
|
|
for row in rows:
|
|
result[(int(row['season']), int(row['episode']))] = (int(row['scene_season']), int(row['scene_episode']))
|
|
|
|
return result
|
|
|
|
|
|
def get_xem_numbering_for_season(indexer_id, indexer, season):
|
|
"""
|
|
Returns a dict of (season, episode) : (sceneSeason, sceneEpisode) mappings
|
|
for an entire show. Both the keys and values of the dict are tuples.
|
|
Will be empty if there are no scene numbers set
|
|
"""
|
|
if indexer_id is None or season is None:
|
|
return {}
|
|
|
|
indexer_id = int(indexer_id)
|
|
indexer = int(indexer)
|
|
|
|
if _xem_refresh_needed(indexer_id, indexer):
|
|
_xem_refresh(indexer_id, indexer)
|
|
|
|
cacheDB = db.DBConnection('cache.db')
|
|
|
|
rows = cacheDB.select(
|
|
'SELECT season, scene_season FROM xem_numbering WHERE indexer = ? and indexer_id = ? AND season = ? ORDER BY season', [indexer, indexer_id, season])
|
|
|
|
result = {}
|
|
if rows:
|
|
for row in rows:
|
|
result.setdefault(int(row['season']), []).append(int(row['scene_season']))
|
|
else:
|
|
result.setdefault(int(season), []).append(int(season))
|
|
|
|
return result
|
|
|
|
|
|
def fix_scene_numbering():
|
|
ql = []
|
|
|
|
myDB = db.DBConnection()
|
|
|
|
sqlResults = myDB.select(
|
|
"SELECT showid, indexerid, indexer, episode_id, season, episode FROM tv_episodes WHERE scene_season = -1 OR scene_episode = -1")
|
|
|
|
for epResult in sqlResults:
|
|
|
|
indexerid = int(epResult["showid"])
|
|
indexer = int(epResult["indexer"])
|
|
season = int(epResult["season"])
|
|
episode = int(epResult["episode"])
|
|
|
|
logger.log(
|
|
u"Repairing any scene numbering issues for showid: " + str(epResult["showid"]) + u" season: " + str(
|
|
epResult["season"]) + u" episode: " + str(epResult["episode"]), logger.DEBUG)
|
|
|
|
scene_season, scene_episode = sickbeard.scene_numbering.get_scene_numbering(indexerid,
|
|
indexer,
|
|
season,
|
|
episode)
|
|
|
|
ql.append(
|
|
["UPDATE tv_episodes SET scene_season = ? WHERE indexerid = ?", [scene_season, epResult["indexerid"]]])
|
|
ql.append(
|
|
["UPDATE tv_episodes SET scene_episode = ? WHERE indexerid = ?", [scene_episode, epResult["indexerid"]]])
|
|
|
|
if ql:
|
|
myDB.mass_action(ql)
|