2014-04-22 08:02:43 +00:00
# Author: Nic Wolfe <nic@wolfeden.ca>
# URL: http://code.google.com/p/sickbeard/
2014-03-10 05:18:05 +00:00
#
2014-05-23 12:37:22 +00:00
# This file is part of SickRage.
2014-03-10 05:18:05 +00:00
#
2014-05-23 12:37:22 +00:00
# SickRage is free software: you can redistribute it and/or modify
2014-03-10 05:18:05 +00:00
# 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.
#
2014-05-23 12:37:22 +00:00
# SickRage is distributed in the hope that it will be useful,
2014-03-10 05:18:05 +00:00
# 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
2014-05-23 12:37:22 +00:00
# along with SickRage. If not, see <http://www.gnu.org/licenses/>.
2014-03-10 05:18:05 +00:00
#
# Created on Sep 20, 2012
# @author: Dermot Buckley <dermot@buckley.ie>
# @copyright: Dermot Buckley
#
import time
import traceback
2014-05-28 21:13:29 +00:00
import sickbeard
from lib . tmdb_api import TMDB
2014-03-10 05:18:05 +00:00
try :
import json
except ImportError :
from lib import simplejson as json
2014-03-20 05:33:34 +00:00
import sickbeard
2014-03-10 05:18:05 +00:00
from sickbeard import logger
from sickbeard import db
from sickbeard . exceptions import ex
from lib import requests
2014-03-25 05:57:24 +00:00
MAX_XEM_AGE_SECS = 86400 # 1 day
2014-05-30 05:48:02 +00:00
def get_scene_numbering ( indexer_id , indexer , season , episode , fallback_to_xem = True ) :
2014-03-10 05:18:05 +00:00
"""
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
2014-03-11 02:32:02 +00:00
returns the TVDB and TVRAGE numbering .
2014-03-10 05:18:05 +00:00
( 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 :
2014-05-30 05:48:02 +00:00
return ( season , episode )
2014-03-20 05:33:34 +00:00
2014-05-30 05:48:02 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( indexer_id ) )
if not showObj . is_scene :
return ( season , episode )
2014-05-03 09:58:04 +00:00
2014-05-30 05:48:02 +00:00
result = find_scene_numbering ( int ( indexer_id ) , int ( indexer ) , season , episode )
2014-03-10 05:18:05 +00:00
if result :
return result
else :
if fallback_to_xem :
2014-05-30 05:48:02 +00:00
xem_result = find_xem_numbering ( int ( indexer_id ) , int ( indexer ) , season , episode )
2014-03-10 05:18:05 +00:00
if xem_result :
return xem_result
2014-05-30 05:48:02 +00:00
return ( season , episode )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def find_scene_numbering ( indexer_id , indexer , season , episode ) :
2014-03-10 05:18:05 +00:00
"""
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 :
2014-05-30 05:48:02 +00:00
return ( season , episode )
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-03-10 05:18:05 +00:00
myDB = db . DBConnection ( )
2014-03-20 05:33:34 +00:00
2014-03-25 05:57:24 +00:00
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT scene_season, scene_episode FROM scene_numbering WHERE indexer = ? and indexer_id = ? and season = ? and episode = ? and (scene_season or scene_episode) != 0 " ,
2014-03-25 05:57:24 +00:00
[ indexer , indexer_id , season , episode ] )
2014-03-10 05:18:05 +00:00
if rows :
2014-05-30 05:48:02 +00:00
return ( int ( rows [ 0 ] [ " scene_season " ] ) , int ( rows [ 0 ] [ " scene_episode " ] ) )
2014-03-21 08:34:00 +00:00
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def get_scene_absolute_numbering ( indexer_id , indexer , absolute_number , 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 absolute_number is None :
return absolute_number
indexer_id = int ( indexer_id )
indexer = int ( indexer )
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , indexer_id )
if not showObj . is_scene :
return absolute_number
result = find_scene_absolute_numbering ( indexer_id , indexer , absolute_number )
if result :
return result
else :
if fallback_to_xem :
xem_result = find_xem_absolute_numbering ( indexer_id , indexer , absolute_number )
if xem_result :
return xem_result
return absolute_number
def find_scene_absolute_numbering ( indexer_id , indexer , absolute_number ) :
"""
Same as get_scene_numbering ( ) , but returns None if scene numbering is not set
"""
if indexer_id is None or absolute_number is None :
return absolute_number
indexer_id = int ( indexer_id )
indexer = int ( indexer )
myDB = db . DBConnection ( )
rows = myDB . select (
" SELECT scene_absolute_number FROM scene_numbering WHERE indexer = ? and indexer_id = ? and absolute_number = ? and scene_absolute_number != 0 " ,
[ indexer , indexer_id , absolute_number ] )
if rows :
return int ( rows [ 0 ] [ " scene_absolute_number " ] )
def get_indexer_numbering ( indexer_id , indexer , sceneSeason , sceneEpisode , fallback_to_xem = True ) :
2014-03-10 05:18:05 +00:00
"""
2014-03-11 02:32:02 +00:00
Returns a tuple , ( season , episode ) with the TVDB and TVRAGE numbering for ( sceneSeason , sceneEpisode )
2014-03-10 05:18:05 +00:00
( 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 )
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-03-10 05:18:05 +00:00
myDB = db . DBConnection ( )
2014-03-25 05:57:24 +00:00
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT season, episode FROM scene_numbering WHERE indexer = ? and indexer_id = ? and scene_season = ? and scene_episode = ? " ,
2014-03-25 05:57:24 +00:00
[ indexer , indexer_id , sceneSeason , sceneEpisode ] )
2014-03-10 05:18:05 +00:00
if rows :
2014-05-30 05:48:02 +00:00
return ( int ( rows [ 0 ] [ " season " ] ) , int ( rows [ 0 ] [ " episode " ] ) )
2014-03-10 05:18:05 +00:00
else :
if fallback_to_xem :
2014-05-30 05:48:02 +00:00
return get_indexer_numbering_for_xem ( indexer_id , indexer , sceneSeason , sceneEpisode )
return ( sceneSeason , sceneEpisode )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def get_indexer_absolute_numbering ( indexer_id , indexer , sceneAbsoluteNumber , fallback_to_xem = True ) :
2014-05-28 21:13:29 +00:00
"""
Returns a tuple , ( season , episode , absolute_number ) with the TVDB and TVRAGE numbering for ( sceneAbsoluteNumber )
( this works like the reverse of get_absolute_numbering )
"""
if indexer_id is None or sceneAbsoluteNumber is None :
2014-05-30 05:48:02 +00:00
return sceneAbsoluteNumber
2014-05-28 21:13:29 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
myDB = db . DBConnection ( )
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT absolute_number FROM scene_numbering WHERE indexer = ? and indexer_id = ? and scene_absolute_number = ? " ,
2014-05-28 21:13:29 +00:00
[ indexer , indexer_id , sceneAbsoluteNumber ] )
if rows :
2014-05-30 05:48:02 +00:00
return int ( rows [ 0 ] [ " absolute_number " ] )
2014-05-28 21:13:29 +00:00
else :
if fallback_to_xem :
2014-05-30 05:48:02 +00:00
return get_indexer_absolute_numbering_for_xem ( indexer_id , indexer , sceneAbsoluteNumber )
return sceneAbsoluteNumber
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def set_scene_numbering ( indexer_id , indexer , season = None , episode = None , absolute_number = None , sceneSeason = None , sceneEpisode = None , sceneAbsolute = None ) :
2014-03-10 05:18:05 +00:00
"""
2014-05-30 05:48:02 +00:00
Set scene numbering for a season / episode .
To clear the scene numbering , leave both sceneSeason and sceneEpisode as None .
2014-03-10 05:18:05 +00:00
"""
if indexer_id is None :
2014-05-30 05:48:02 +00:00
return
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-03-10 05:18:05 +00:00
myDB = db . DBConnection ( )
2014-03-25 05:57:24 +00:00
2014-06-06 00:53:34 +00:00
if season and episode :
2014-05-30 05:48:02 +00:00
myDB . action (
" INSERT OR IGNORE INTO scene_numbering (indexer, indexer_id, season, episode) VALUES (?,?,?,?) " ,
[ indexer , indexer_id , season , episode ] )
2014-05-26 06:29:22 +00:00
2014-05-30 05:48:02 +00:00
myDB . action (
" UPDATE scene_numbering SET scene_season = ?, scene_episode = ? WHERE indexer = ? and indexer_id = ? and season = ? and episode = ? " ,
[ sceneSeason , sceneEpisode , indexer , indexer_id , season , episode ] )
2014-06-06 00:53:34 +00:00
elif absolute_number :
2014-05-30 05:48:02 +00:00
myDB . action (
" INSERT OR IGNORE INTO scene_numbering (indexer, indexer_id, absolute_number) VALUES (?,?,?) " ,
[ indexer , indexer_id , absolute_number ] )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
myDB . action (
" UPDATE scene_numbering SET scene_absolute_number = ? WHERE indexer = ? and indexer_id = ? and absolute_number = ? " ,
[ sceneAbsolute , indexer , indexer_id , absolute_number ] )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def find_xem_numbering ( indexer_id , indexer , season , episode ) :
2014-03-10 05:18:05 +00:00
"""
2014-05-30 05:48:02 +00:00
Returns the scene numbering , as retrieved from xem .
Refreshes / Loads as needed .
2014-03-10 05:18:05 +00:00
2014-05-30 05:48:02 +00:00
@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 .
2014-03-10 05:18:05 +00:00
"""
if indexer_id is None or season is None or episode is None :
2014-05-30 05:48:02 +00:00
return ( season , episode )
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
myDB = db . DBConnection ( )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
rows = myDB . select (
" SELECT scene_season, scene_episode FROM tv_episodes WHERE indexer = ? and showid = ? and season = ? and episode = ? and (scene_season or scene_episode) != 0 " ,
[ indexer , indexer_id , season , episode ] )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
if rows :
return ( int ( rows [ 0 ] [ " scene_season " ] ) , int ( rows [ 0 ] [ " scene_episode " ] ) )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def find_xem_absolute_numbering ( indexer_id , indexer , absolute_number ) :
2014-03-10 05:18:05 +00:00
"""
Returns the scene numbering , as retrieved from xem .
Refreshes / Loads as needed .
2014-05-30 05:48:02 +00:00
2014-03-10 05:18:05 +00:00
@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 .
"""
2014-05-30 05:48:02 +00:00
if indexer_id is None or absolute_number is None :
return absolute_number
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-03-25 05:57:24 +00:00
2014-05-28 21:13:29 +00:00
myDB = db . DBConnection ( )
2014-05-11 22:24:02 +00:00
2014-05-28 21:13:29 +00:00
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT scene_absolute_number FROM tv_episodes WHERE indexer = ? and showid = ? and absolute_number = ? and scene_absolute_number != 0 " ,
[ indexer , indexer_id , absolute_number ] )
2014-05-11 22:24:02 +00:00
2014-03-10 05:18:05 +00:00
if rows :
2014-05-30 05:48:02 +00:00
return int ( rows [ 0 ] [ " scene_absolute_number " ] )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
def get_indexer_numbering_for_xem ( indexer_id , indexer , sceneSeason , sceneEpisode ) :
2014-03-10 05:18:05 +00:00
"""
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 :
2014-05-30 05:48:02 +00:00
return ( sceneSeason , sceneEpisode )
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-05-28 21:13:29 +00:00
myDB = db . DBConnection ( )
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT season, episode FROM tv_episodes WHERE indexer = ? and showid = ? and scene_season = ? and scene_episode = ? " ,
2014-03-25 05:57:24 +00:00
[ indexer , indexer_id , sceneSeason , sceneEpisode ] )
2014-03-10 05:18:05 +00:00
if rows :
2014-05-30 05:48:02 +00:00
return ( int ( rows [ 0 ] [ " season " ] ) , int ( rows [ 0 ] [ " episode " ] ) )
2014-03-25 05:57:24 +00:00
2014-05-30 05:48:02 +00:00
return ( sceneSeason , sceneEpisode )
2014-05-28 21:13:29 +00:00
2014-05-30 05:48:02 +00:00
def get_indexer_absolute_numbering_for_xem ( indexer_id , indexer , sceneAbsoluteNumber ) :
2014-05-28 21:13:29 +00:00
"""
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 sceneAbsoluteNumber is None :
2014-05-30 05:48:02 +00:00
return sceneAbsoluteNumber
2014-05-28 21:13:29 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-05-28 21:13:29 +00:00
myDB = db . DBConnection ( )
rows = myDB . select (
2014-05-30 05:48:02 +00:00
" SELECT absolute_number FROM tv_episodes WHERE indexer = ? and showid = ? and scene_absolute_number = ? " ,
2014-05-28 21:13:29 +00:00
[ indexer , indexer_id , sceneAbsoluteNumber ] )
if rows :
2014-05-30 05:48:02 +00:00
return int ( rows [ 0 ] [ " absolute_number " ] )
return sceneAbsoluteNumber
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 = ? and (scene_season or scene_episode) != 0 ORDER BY season, episode ' ,
[ indexer , indexer_id ] )
result = { }
for row in rows :
season = int ( row [ ' season ' ] )
episode = int ( row [ ' episode ' ] )
scene_season = int ( row [ ' scene_season ' ] )
scene_episode = int ( row [ ' scene_episode ' ] )
result [ ( season , episode ) ] = ( scene_season , scene_episode )
return result
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 )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-05-30 05:48:02 +00:00
myDB = db . DBConnection ( )
rows = myDB . select (
' SELECT season, episode, scene_season, scene_episode FROM tv_episodes WHERE indexer = ? and showid = ? and (scene_season or scene_episode) != 0 ORDER BY season, episode ' ,
[ indexer , indexer_id ] )
result = { }
for row in rows :
season = int ( row [ ' season ' ] )
episode = int ( row [ ' episode ' ] )
scene_season = int ( row [ ' scene_season ' ] )
scene_episode = int ( row [ ' scene_episode ' ] )
result [ ( season , episode ) ] = ( scene_season , scene_episode )
return result
def get_scene_absolute_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 absolute_number, scene_absolute_number FROM scene_numbering WHERE indexer = ? and indexer_id = ? and scene_absolute_number != 0 ORDER BY absolute_number ' ,
[ indexer , indexer_id ] )
result = { }
for row in rows :
absolute_number = int ( row [ ' absolute_number ' ] )
scene_absolute_number = int ( row [ ' scene_absolute_number ' ] )
result [ absolute_number ] = scene_absolute_number
return result
def get_xem_absolute_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 )
2014-05-30 06:49:47 +00:00
xem_refresh ( indexer_id , indexer )
2014-05-30 05:48:02 +00:00
myDB = db . DBConnection ( )
result = { }
rows = myDB . select (
' SELECT absolute_number, scene_absolute_number FROM tv_episodes WHERE indexer = ? and showid = ? and scene_absolute_number != 0 ORDER BY absolute_number ' ,
[ indexer , indexer_id ] )
for row in rows :
absolute_number = int ( row [ ' absolute_number ' ] )
scene_absolute_number = int ( row [ ' scene_absolute_number ' ] )
result [ absolute_number ] = scene_absolute_number
return result
2014-03-25 05:57:24 +00:00
2014-05-30 06:49:47 +00:00
def xem_refresh ( indexer_id , indexer , force = False ) :
2014-03-10 05:18:05 +00:00
"""
2014-05-30 06:49:47 +00:00
Refresh data from xem for a tv show
2014-03-10 05:18:05 +00:00
@param indexer_id : int
"""
if indexer_id is None :
2014-05-30 06:49:47 +00:00
return
2014-03-20 05:33:34 +00:00
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-28 21:13:29 +00:00
myDB = db . DBConnection ( )
2014-05-30 06:49:47 +00:00
2014-05-28 21:13:29 +00:00
rows = myDB . select ( " SELECT last_refreshed FROM xem_refresh WHERE indexer = ? and indexer_id = ? " ,
2014-05-30 05:48:02 +00:00
[ indexer , indexer_id ] )
2014-03-10 05:18:05 +00:00
if rows :
2014-05-30 06:49:47 +00:00
refresh = time . time ( ) > ( int ( rows [ 0 ] [ ' last_refreshed ' ] ) + MAX_XEM_AGE_SECS )
2014-03-10 05:18:05 +00:00
else :
2014-05-30 06:49:47 +00:00
refresh = True
if refresh or force :
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 " % (
2014-05-03 09:58:04 +00:00
indexer_id , sickbeard . indexerApi ( indexer ) . config [ ' xem_origin ' ] , ) , verify = False ) . json ( )
2014-05-30 06:49:47 +00:00
2014-03-10 05:18:05 +00:00
if data is None or data == ' ' :
2014-05-30 06:49:47 +00:00
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 = [ ]
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 ' ] :
for entry in result [ ' data ' ] :
if ' scene ' in entry :
ql . append ( [
" UPDATE tv_episodes SET scene_season = ?, scene_episode = ?, scene_absolute_number = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ entry [ ' scene ' ] [ ' season ' ] ,
entry [ ' scene ' ] [ ' episode ' ] ,
entry [ ' scene ' ] [ ' absolute ' ] ,
indexer_id ,
entry [ sickbeard . indexerApi ( indexer ) . config [ ' xem_origin ' ] ] [ ' season ' ] ,
entry [ sickbeard . indexerApi ( indexer ) . config [ ' xem_origin ' ] ] [ ' episode ' ]
] ] )
if ' scene_2 ' in entry : # for doubles
ql . append ( [
" UPDATE tv_episodes SET scene_season = ?, scene_episode = ?, scene_absolute_number = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ entry [ ' scene_2 ' ] [ ' season ' ] ,
entry [ ' scene_2 ' ] [ ' episode ' ] ,
entry [ ' scene_2 ' ] [ ' absolute ' ] ,
indexer_id ,
entry [ sickbeard . indexerApi ( indexer ) . config [ ' xem_origin ' ] ] [ ' season ' ] ,
entry [ sickbeard . indexerApi ( indexer ) . config [ ' xem_origin ' ] ] [ ' 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 )
2014-03-10 05:18:05 +00:00
else :
2014-05-30 06:49:47 +00:00
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 :
myDB . mass_action ( ql )
# fix xem scene numbering issues
# fix_xem_numbering(indexer_id, indexer)
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
2014-05-28 21:13:29 +00:00
def fix_xem_numbering ( indexer_id , indexer ) :
2014-03-10 05:18:05 +00:00
"""
Returns a dict of ( season , episode ) : ( sceneSeason , sceneEpisode ) mappings
for an entire show . Both the keys and values of the dict are tuples .
2014-05-28 21:13:29 +00:00
Will be empty if there are no scene numbers set in xem
2014-03-10 05:18:05 +00:00
"""
2014-05-28 21:13:29 +00:00
if indexer_id is None :
2014-03-10 05:18:05 +00:00
return { }
2014-05-03 09:58:04 +00:00
indexer_id = int ( indexer_id )
indexer = int ( indexer )
2014-05-28 21:13:29 +00:00
# query = [{
2014-05-30 05:48:02 +00:00
# "name": self.show.name,
2014-05-28 21:13:29 +00:00
# "seasons": [{
# "episodes": [{
# "episode_number": None,
# "name": None
# }],
# "season_number": None,
# }],
# "/tv/tv_program/number_of_seasons": [],
# "/tv/tv_program/number_of_episodes": [],
# "/tv/tv_program/thetvdb_id": [],
# "/tv/tv_program/tvrage_id": [],
# "type": "/tv/tv_program",
# }]
#
#
# url = 'https://www.googleapis.com/freebase/v1/mqlread'
# api_key = "AIzaSyCCHNp4dhVHxJYzbLiCE4y4a1rgTnX4fDE"
# params = {
# 'query': json.dumps(query),
# 'key': api_key
# }
#
#
# def get_from_api(url, params=None):
# """Build request and return results
# """
# import xmltodict
#
# response = requests.get(url, params=params)
# if response.status_code == 200:
# try:
# return response.json()
# except ValueError:
# return xmltodict.parse(response.text)['Data']
#
# # Get query results
# tmp = get_from_api(url, params=params)['result']
2014-03-10 05:18:05 +00:00
2014-05-28 21:13:29 +00:00
myDB = db . DBConnection ( )
2014-03-10 05:18:05 +00:00
2014-05-28 21:13:29 +00:00
rows = myDB . select (
' SELECT season, episode, absolute_number, scene_season, scene_episode, scene_absolute_number FROM tv_episodes WHERE indexer = ? and showid = ? ' ,
[ indexer , indexer_id ] )
2014-05-03 22:29:00 +00:00
2014-05-28 21:13:29 +00:00
last_absolute_number = None
last_scene_season = None
last_scene_episode = None
last_scene_absolute_number = None
2014-05-03 22:29:00 +00:00
2014-05-28 21:13:29 +00:00
update_absolute_number = False
update_scene_season = False
update_scene_episode = False
update_scene_absolute_number = False
2014-05-03 22:29:00 +00:00
2014-05-28 21:13:29 +00:00
logger . log (
u ' Fixing any XEM scene mapping issues for show %s on %s ' % ( indexer_id , sickbeard . indexerApi ( indexer ) . name , ) ,
logger . DEBUG )
2014-05-03 22:29:00 +00:00
2014-05-28 21:13:29 +00:00
ql = [ ]
for row in rows :
season = int ( row [ ' season ' ] )
episode = int ( row [ ' episode ' ] )
2014-05-03 22:29:00 +00:00
2014-05-28 21:13:29 +00:00
if not int ( row [ ' scene_season ' ] ) and last_scene_season :
scene_season = last_scene_season + 1
update_scene_season = True
else :
scene_season = int ( row [ ' scene_season ' ] )
if last_scene_season and scene_season < last_scene_season :
scene_season = last_scene_season + 1
update_scene_season = True
if not int ( row [ ' scene_episode ' ] ) and last_scene_episode :
scene_episode = last_scene_episode + 1
update_scene_episode = True
else :
scene_episode = int ( row [ ' scene_episode ' ] )
if last_scene_episode and scene_episode < last_scene_episode :
scene_episode = last_scene_episode + 1
update_scene_episode = True
# check for unset values and correct them
if not int ( row [ ' absolute_number ' ] ) and last_absolute_number :
absolute_number = last_absolute_number + 1
update_absolute_number = True
else :
absolute_number = int ( row [ ' absolute_number ' ] )
if last_absolute_number and absolute_number < last_absolute_number :
absolute_number = last_absolute_number + 1
update_absolute_number = True
if not int ( row [ ' scene_absolute_number ' ] ) and last_scene_absolute_number :
scene_absolute_number = last_scene_absolute_number + 1
update_scene_absolute_number = True
else :
scene_absolute_number = int ( row [ ' scene_absolute_number ' ] )
if last_scene_absolute_number and scene_absolute_number < last_scene_absolute_number :
scene_absolute_number = last_scene_absolute_number + 1
update_scene_absolute_number = True
# store values for lookup on next iteration
last_absolute_number = absolute_number
last_scene_season = scene_season
last_scene_episode = scene_episode
last_scene_absolute_number = scene_absolute_number
if update_absolute_number :
ql . append ( [
" UPDATE tv_episodes SET absolute_number = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ absolute_number ,
indexer_id ,
season ,
episode
] ] )
update_absolute_number = False
if update_scene_season :
ql . append ( [
" UPDATE tv_episodes SET scene_season = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ scene_season ,
indexer_id ,
season ,
episode
] ] )
update_scene_season = False
if update_scene_episode :
ql . append ( [
" UPDATE tv_episodes SET scene_episode = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ scene_episode ,
indexer_id ,
season ,
episode
] ] )
update_scene_episode = False
if update_scene_absolute_number :
ql . append ( [
" UPDATE tv_episodes SET scene_absolute_number = ? WHERE showid = ? AND season = ? AND episode = ? " ,
[ scene_absolute_number ,
indexer_id ,
season ,
episode
] ] )
update_scene_absolute_number = False
2014-05-03 22:29:00 +00:00
2014-05-07 07:50:49 +00:00
if ql :
2014-05-28 21:13:29 +00:00
myDB . mass_action ( ql )