2014-03-10 01:18:05 -04:00
# Author: Nic Wolfe <nic@wolfeden.ca>
# URL: http://code.google.com/p/sickbeard/
#
2014-05-23 08:37:22 -04:00
# This file is part of SickRage.
2014-03-10 01:18:05 -04:00
#
2014-05-23 08:37:22 -04:00
# SickRage is free software: you can redistribute it and/or modify
2014-03-10 01:18:05 -04: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 08:37:22 -04:00
# SickRage is distributed in the hope that it will be useful,
2014-03-10 01:18:05 -04: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 08:37:22 -04:00
# along with SickRage. If not, see <http://www.gnu.org/licenses/>.
2014-03-10 01:18:05 -04:00
from __future__ import with_statement
import traceback
2014-05-11 08:49:07 -04:00
import threading
2014-03-10 01:18:05 -04:00
import sickbeard
from lib . imdb import _exceptions as imdb_exceptions
from sickbeard . common import SKIPPED , WANTED
from sickbeard . tv import TVShow
from sickbeard import exceptions , logger , ui , db
from sickbeard import generic_queue
from sickbeard import name_cache
from sickbeard . exceptions import ex
class ShowQueue ( generic_queue . GenericQueue ) :
def __init__ ( self ) :
generic_queue . GenericQueue . __init__ ( self )
self . queue_name = " SHOWQUEUE "
def _isInQueue ( self , show , actions ) :
2014-05-19 21:04:23 -04:00
return show in [ x . show for x in self . queue if x . action_id in actions ]
2014-03-10 01:18:05 -04:00
def _isBeingSomethinged ( self , show , actions ) :
2014-03-20 14:03:22 -04:00
return self . currentItem != None and show == self . currentItem . show and \
2014-03-25 01:57:24 -04:00
self . currentItem . action_id in actions
2014-03-10 01:18:05 -04:00
def isInUpdateQueue ( self , show ) :
return self . _isInQueue ( show , ( ShowQueueActions . UPDATE , ShowQueueActions . FORCEUPDATE ) )
def isInRefreshQueue ( self , show ) :
return self . _isInQueue ( show , ( ShowQueueActions . REFRESH , ) )
def isInRenameQueue ( self , show ) :
return self . _isInQueue ( show , ( ShowQueueActions . RENAME , ) )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def isInSubtitleQueue ( self , show ) :
return self . _isInQueue ( show , ( ShowQueueActions . SUBTITLE , ) )
def isBeingAdded ( self , show ) :
return self . _isBeingSomethinged ( show , ( ShowQueueActions . ADD , ) )
def isBeingUpdated ( self , show ) :
return self . _isBeingSomethinged ( show , ( ShowQueueActions . UPDATE , ShowQueueActions . FORCEUPDATE ) )
def isBeingRefreshed ( self , show ) :
return self . _isBeingSomethinged ( show , ( ShowQueueActions . REFRESH , ) )
def isBeingRenamed ( self , show ) :
return self . _isBeingSomethinged ( show , ( ShowQueueActions . RENAME , ) )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def isBeingSubtitled ( self , show ) :
return self . _isBeingSomethinged ( show , ( ShowQueueActions . SUBTITLE , ) )
def _getLoadingShowList ( self ) :
2014-05-19 21:04:23 -04:00
return [ x for x in self . queue + [ self . currentItem ] if x != None and x . isLoading ]
2014-03-10 01:18:05 -04:00
loadingShowList = property ( _getLoadingShowList )
def updateShow ( self , show , force = False ) :
if self . isBeingAdded ( show ) :
2014-03-25 01:57:24 -04:00
raise exceptions . CantUpdateException (
" Show is still being added, wait until it is finished before you update. " )
2014-03-10 01:18:05 -04:00
if self . isBeingUpdated ( show ) :
2014-03-25 01:57:24 -04:00
raise exceptions . CantUpdateException (
" This show is already being updated, can ' t update again until it ' s done. " )
2014-03-10 01:18:05 -04:00
if self . isInUpdateQueue ( show ) :
2014-03-25 01:57:24 -04:00
raise exceptions . CantUpdateException (
" This show is already being updated, can ' t update again until it ' s done. " )
2014-03-10 01:18:05 -04:00
if not force :
queueItemObj = QueueItemUpdate ( show )
else :
queueItemObj = QueueItemForceUpdate ( show )
self . add_item ( queueItemObj )
return queueItemObj
def refreshShow ( self , show , force = False ) :
if self . isBeingRefreshed ( show ) and not force :
raise exceptions . CantRefreshException ( " This show is already being refreshed, not refreshing again. " )
if ( self . isBeingUpdated ( show ) or self . isInUpdateQueue ( show ) ) and not force :
2014-03-25 01:57:24 -04:00
logger . log (
u " A refresh was attempted but there is already an update queued or in progress. Since updates do a refres at the end anyway I ' m skipping this request. " ,
logger . DEBUG )
2014-03-10 01:18:05 -04:00
return
queueItemObj = QueueItemRefresh ( show )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
self . add_item ( queueItemObj )
return queueItemObj
def renameShowEpisodes ( self , show , force = False ) :
queueItemObj = QueueItemRename ( show )
self . add_item ( queueItemObj )
return queueItemObj
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def downloadSubtitles ( self , show , force = False ) :
queueItemObj = QueueItemSubtitle ( show )
self . add_item ( queueItemObj )
return queueItemObj
2014-03-25 01:57:24 -04:00
def addShow ( self , indexer , indexer_id , showDir , default_status = None , quality = None , flatten_folders = None ,
2014-05-26 02:29:22 -04:00
subtitles = None , lang = " en " , anime = None ) :
2014-03-25 01:57:24 -04:00
queueItemObj = QueueItemAdd ( indexer , indexer_id , showDir , default_status , quality , flatten_folders , lang ,
2014-05-26 02:29:22 -04:00
subtitles , anime )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
self . add_item ( queueItemObj )
return queueItemObj
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
class ShowQueueActions :
2014-03-25 01:57:24 -04:00
REFRESH = 1
ADD = 2
UPDATE = 3
FORCEUPDATE = 4
RENAME = 5
SUBTITLE = 6
2014-03-10 01:18:05 -04:00
names = { REFRESH : ' Refresh ' ,
2014-03-25 01:57:24 -04:00
ADD : ' Add ' ,
UPDATE : ' Update ' ,
FORCEUPDATE : ' Force Update ' ,
RENAME : ' Rename ' ,
SUBTITLE : ' Subtitle ' ,
}
2014-03-10 01:18:05 -04:00
class ShowQueueItem ( generic_queue . QueueItem ) :
"""
Represents an item in the queue waiting to be executed
Can be either :
- show being added ( may or may not be associated with a show object )
- show being refreshed
- show being updated
- show being force updated
- show being subtitled
"""
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def __init__ ( self , action_id , show ) :
generic_queue . QueueItem . __init__ ( self , ShowQueueActions . names [ action_id ] , action_id )
self . show = show
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def isInQueue ( self ) :
2014-05-19 21:04:23 -04:00
return self in sickbeard . showQueueScheduler . action . queue + [ sickbeard . showQueueScheduler . action . currentItem ] #@UndefinedVariable
2014-03-10 01:18:05 -04:00
def _getName ( self ) :
return str ( self . show . indexerid )
def _isLoading ( self ) :
return False
show_name = property ( _getName )
isLoading = property ( _isLoading )
class QueueItemAdd ( ShowQueueItem ) :
2014-05-26 02:29:22 -04:00
def __init__ ( self , indexer , indexer_id , showDir , default_status , quality , flatten_folders , lang , subtitles , anime ) :
2014-03-10 01:18:05 -04:00
self . indexer = indexer
self . indexer_id = indexer_id
self . showDir = showDir
self . default_status = default_status
self . quality = quality
self . flatten_folders = flatten_folders
self . lang = lang
self . subtitles = subtitles
2014-05-26 02:29:22 -04:00
self . anime = anime
2014-03-10 01:18:05 -04:00
self . show = None
# this will initialize self.show to None
ShowQueueItem . __init__ ( self , ShowQueueActions . ADD , self . show )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
def _getName ( self ) :
"""
Returns the show name if there is a show object created , if not returns
the dir that the show is being added to .
"""
2014-03-20 14:03:22 -04:00
if self . show == None :
2014-03-10 01:18:05 -04:00
return self . showDir
return self . show . name
show_name = property ( _getName )
def _isLoading ( self ) :
"""
Returns True if we ' ve gotten far enough to have a show object, or False
if we still only know the folder name .
"""
2014-03-20 14:03:22 -04:00
if self . show == None :
2014-03-10 01:18:05 -04:00
return True
return False
isLoading = property ( _isLoading )
def execute ( self ) :
ShowQueueItem . execute ( self )
logger . log ( u " Starting to add show " + self . showDir )
2014-04-27 06:58:49 -04:00
# make sure the Indexer IDs are valid
2014-03-10 01:18:05 -04:00
try :
2014-03-18 10:41:46 -04:00
2014-03-26 15:28:46 -04:00
lINDEXER_API_PARMS = sickbeard . indexerApi ( self . indexer ) . api_params . copy ( )
2014-03-18 10:41:46 -04:00
if self . lang :
lINDEXER_API_PARMS [ ' language ' ] = self . lang
2014-05-01 02:26:13 -04:00
logger . log ( u " " + str ( sickbeard . indexerApi ( self . indexer ) . name ) + " : " + repr ( lINDEXER_API_PARMS ) )
2014-03-18 10:41:46 -04:00
2014-03-26 15:28:46 -04:00
t = sickbeard . indexerApi ( self . indexer ) . indexer ( * * lINDEXER_API_PARMS )
2014-03-18 10:41:46 -04:00
s = t [ self . indexer_id ]
2014-04-27 06:58:49 -04:00
# this usually only happens if they have an NFO in their show dir which gave us a Indexer ID that has no proper english version of the show
2014-03-18 10:41:46 -04:00
if getattr ( s , ' seriesname ' , None ) is None :
2014-05-01 02:26:13 -04:00
logger . log ( u " Show in " + self . showDir + " has no name on " + str ( sickbeard . indexerApi ( self . indexer ) . name ) + " , probably the wrong language used to search with. " , logger . ERROR )
2014-03-25 01:57:24 -04:00
ui . notifications . error ( " Unable to add show " ,
2014-05-01 02:26:13 -04:00
" Show in " + self . showDir + " has no name on " + str ( sickbeard . indexerApi (
self . indexer ) . name ) + " , probably the wrong language. Delete .nfo and add manually in the correct language. " )
2014-03-10 01:18:05 -04:00
self . _finishEarly ( )
return
2014-03-18 10:41:46 -04:00
# if the show has no episodes/seasons
if not s :
2014-05-01 02:26:13 -04:00
logger . log ( u " Show " + str ( s [ ' seriesname ' ] ) + " is on " + str ( sickbeard . indexerApi ( self . indexer ) . name ) + " but contains no season/episode data. " , logger . ERROR )
2014-03-25 01:57:24 -04:00
ui . notifications . error ( " Unable to add show " ,
2014-05-01 02:26:13 -04:00
" Show " + str ( s [ ' seriesname ' ] ) + " is on " + str ( sickbeard . indexerApi (
self . indexer ) . name ) + " but contains no season/episode data. " )
2014-03-18 10:41:46 -04:00
self . _finishEarly ( )
return
except Exception , e :
2014-05-01 02:26:13 -04:00
logger . log ( u " Unable to find show ID: " + str ( self . indexer_id ) + " on Indexer: " + str ( sickbeard . indexerApi ( self . indexer ) . name ) , logger . ERROR )
2014-03-25 01:57:24 -04:00
ui . notifications . error ( " Unable to add show " ,
2014-05-01 02:26:13 -04:00
" Unable to look up the show in " + self . showDir + " on " + str ( sickbeard . indexerApi (
self . indexer ) . name ) + " using ID " + str (
2014-03-25 01:57:24 -04:00
self . indexer_id ) + " , not using the NFO. Delete .nfo and try adding manually again. " )
2014-03-18 10:41:46 -04:00
self . _finishEarly ( )
return
2014-03-10 01:18:05 -04:00
2014-03-18 10:41:46 -04:00
try :
2014-03-10 01:18:05 -04:00
# clear the name cache
name_cache . clearCache ( )
newShow = TVShow ( self . indexer , self . indexer_id , self . lang )
newShow . loadFromIndexer ( )
self . show = newShow
# set up initial values
self . show . location = self . showDir
2014-03-20 14:03:22 -04:00
self . show . subtitles = self . subtitles if self . subtitles != None else sickbeard . SUBTITLES_DEFAULT
2014-03-10 01:18:05 -04:00
self . show . quality = self . quality if self . quality else sickbeard . QUALITY_DEFAULT
2014-03-20 14:03:22 -04:00
self . show . flatten_folders = self . flatten_folders if self . flatten_folders != None else sickbeard . FLATTEN_FOLDERS_DEFAULT
2014-05-30 06:01:49 -04:00
self . show . anime = self . anime if self . anime != None else sickbeard . ANIME_DEFAULT
2014-03-10 01:18:05 -04:00
self . show . paused = False
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
# be smartish about this
2014-04-28 05:15:29 -04:00
if self . show . genre and " talk show " in self . show . genre . lower ( ) :
2014-03-10 01:18:05 -04:00
self . show . air_by_date = 1
2014-05-23 01:02:49 -04:00
if self . show . genre and " documentary " in self . show . genre . lower ( ) :
self . show . air_by_date = 0
if self . show . classification and " sports " in self . show . classification . lower ( ) :
2014-04-28 05:15:29 -04:00
self . show . sports = 1
2014-05-30 06:01:49 -04:00
#if self.show.genre and "animation" in self.show.genre.lower():
# self.show.anime = 1
2014-03-10 01:18:05 -04:00
2014-03-25 01:57:24 -04:00
except sickbeard . indexer_exception , e :
logger . log (
u " Unable to add show due to an error with " + sickbeard . indexerApi ( self . indexer ) . name + " : " + ex ( e ) ,
logger . ERROR )
2014-03-10 01:18:05 -04:00
if self . show :
2014-03-25 01:57:24 -04:00
ui . notifications . error (
" Unable to add " + str ( self . show . name ) + " due to an error with " + sickbeard . indexerApi (
self . indexer ) . name + " " )
2014-03-10 01:18:05 -04:00
else :
2014-03-25 01:57:24 -04:00
ui . notifications . error (
" Unable to add show due to an error with " + sickbeard . indexerApi ( self . indexer ) . name + " " )
2014-03-10 01:18:05 -04:00
self . _finishEarly ( )
return
except exceptions . MultipleShowObjectsException :
logger . log ( u " The show in " + self . showDir + " is already in your show list, skipping " , logger . ERROR )
ui . notifications . error ( ' Show skipped ' , " The show in " + self . showDir + " is already in your show list " )
self . _finishEarly ( )
return
except Exception , e :
2014-03-25 01:57:24 -04:00
logger . log ( u " Error trying to add show: " + ex ( e ) , logger . ERROR )
2014-03-10 01:18:05 -04:00
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
self . _finishEarly ( )
raise
logger . log ( u " Retrieving show info from IMDb " , logger . DEBUG )
try :
self . show . loadIMDbInfo ( )
except imdb_exceptions . IMDbError , e :
#todo Insert UI notification
logger . log ( u " Something wrong on IMDb api: " + ex ( e ) , logger . WARNING )
except Exception , e :
logger . log ( u " Error loading IMDb info: " + ex ( e ) , logger . ERROR )
try :
self . show . saveToDB ( )
except Exception , e :
logger . log ( u " Error saving the show to the database: " + ex ( e ) , logger . ERROR )
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
self . _finishEarly ( )
raise
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
# add it to the show list
2014-03-25 01:57:24 -04:00
sickbeard . showList . append ( self . show )
2014-03-10 01:18:05 -04:00
try :
self . show . loadEpisodesFromIndexer ( )
except Exception , e :
2014-03-25 01:57:24 -04:00
logger . log (
u " Error with " + sickbeard . indexerApi ( self . show . indexer ) . name + " , not creating episode list: " + ex ( e ) ,
logger . ERROR )
2014-03-10 01:18:05 -04:00
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
2014-05-27 03:44:23 -04:00
# before we parse local files lets update exceptions
sickbeard . scene_exceptions . retrieve_exceptions ( )
2014-03-10 01:18:05 -04:00
try :
self . show . loadEpisodesFromDir ( )
except Exception , e :
logger . log ( u " Error searching dir for episodes: " + ex ( e ) , logger . ERROR )
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
# if they gave a custom status then change all the eps to it
if self . default_status != SKIPPED :
logger . log ( u " Setting all episodes to the specified default status: " + str ( self . default_status ) )
myDB = db . DBConnection ( )
2014-03-25 01:57:24 -04:00
myDB . action ( " UPDATE tv_episodes SET status = ? WHERE status = ? AND showid = ? AND season != 0 " ,
[ self . default_status , SKIPPED , self . show . indexerid ] )
2014-03-10 01:18:05 -04:00
# if they started with WANTED eps then run the backlog
if self . default_status == WANTED :
logger . log ( u " Launching backlog for this show since its episodes are WANTED " )
2014-03-25 01:57:24 -04:00
sickbeard . backlogSearchScheduler . action . searchBacklog ( [ self . show ] ) #@UndefinedVariable
2014-03-10 01:18:05 -04:00
2014-05-14 05:42:08 -04:00
self . show . writeMetadata ( )
self . show . updateMetadata ( )
2014-03-13 23:07:15 -04:00
self . show . populateCache ( )
2014-03-10 01:18:05 -04:00
self . show . flushEpisodes ( )
# if there are specific episodes that need to be added by trakt
sickbeard . traktWatchListCheckerSchedular . action . manageNewShow ( self . show )
2014-05-30 02:49:47 -04:00
# Load XEM data to DB for show
sickbeard . scene_numbering . xem_refresh ( self . show . indexerid , self . show . indexer , force = True )
2014-05-30 06:01:49 -04:00
# check if show has XEM mapping so we can determin if searches should go by scene numbering or indexer numbering.
if sickbeard . scene_numbering . get_xem_numbering_for_show ( self . show . indexerid , self . show . indexer ) :
self . show . scene = 1
2014-03-10 01:18:05 -04:00
self . finish ( )
def _finishEarly ( self ) :
2014-03-20 14:03:22 -04:00
if self . show != None :
2014-03-10 01:18:05 -04:00
self . show . deleteShow ( )
self . finish ( )
class QueueItemRefresh ( ShowQueueItem ) :
2014-05-30 01:48:02 -04:00
def __init__ ( self , show = None , force = False ) :
2014-03-10 01:18:05 -04:00
ShowQueueItem . __init__ ( self , ShowQueueActions . REFRESH , show )
# do refreshes first because they're quick
self . priority = generic_queue . QueuePriorities . HIGH
2014-05-30 01:48:02 -04:00
# force refresh certain items
self . force = force
2014-03-10 01:18:05 -04:00
def execute ( self ) :
ShowQueueItem . execute ( self )
logger . log ( u " Performing refresh on " + self . show . name )
self . show . refreshDir ( )
2014-05-14 00:14:54 -04:00
self . show . writeMetadata ( )
2014-05-30 01:48:02 -04:00
if self . force :
self . show . updateMetadata ( )
2014-03-10 01:18:05 -04:00
self . show . populateCache ( )
2014-05-30 02:49:47 -04:00
# Load XEM data to DB for show
sickbeard . scene_numbering . xem_refresh ( self . show . indexerid , self . show . indexer , force = self . force )
2014-03-10 01:18:05 -04:00
self . inProgress = False
class QueueItemRename ( ShowQueueItem ) :
def __init__ ( self , show = None ) :
ShowQueueItem . __init__ ( self , ShowQueueActions . RENAME , show )
def execute ( self ) :
ShowQueueItem . execute ( self )
logger . log ( u " Performing rename on " + self . show . name )
try :
show_loc = self . show . location
except exceptions . ShowDirNotFoundException :
logger . log ( u " Can ' t perform rename on " + self . show . name + " when the show dir is missing. " , logger . WARNING )
return
ep_obj_rename_list = [ ]
ep_obj_list = self . show . getAllEpisodes ( has_location = True )
for cur_ep_obj in ep_obj_list :
# Only want to rename if we have a location
if cur_ep_obj . location :
if cur_ep_obj . relatedEps :
# do we have one of multi-episodes in the rename list already
have_already = False
for cur_related_ep in cur_ep_obj . relatedEps + [ cur_ep_obj ] :
if cur_related_ep in ep_obj_rename_list :
have_already = True
break
if not have_already :
ep_obj_rename_list . append ( cur_ep_obj )
else :
ep_obj_rename_list . append ( cur_ep_obj )
for cur_ep_obj in ep_obj_rename_list :
cur_ep_obj . rename ( )
self . inProgress = False
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
class QueueItemSubtitle ( ShowQueueItem ) :
def __init__ ( self , show = None ) :
ShowQueueItem . __init__ ( self , ShowQueueActions . SUBTITLE , show )
def execute ( self ) :
ShowQueueItem . execute ( self )
2014-03-25 01:57:24 -04:00
logger . log ( u " Downloading subtitles for " + self . show . name )
2014-03-10 01:18:05 -04:00
self . show . downloadSubtitles ( )
self . inProgress = False
class QueueItemUpdate ( ShowQueueItem ) :
def __init__ ( self , show = None ) :
ShowQueueItem . __init__ ( self , ShowQueueActions . UPDATE , show )
self . force = False
def execute ( self ) :
ShowQueueItem . execute ( self )
logger . log ( u " Beginning update of " + self . show . name )
2014-03-25 01:57:24 -04:00
logger . log ( u " Retrieving show info from " + sickbeard . indexerApi ( self . show . indexer ) . name + " " , logger . DEBUG )
2014-03-10 01:18:05 -04:00
try :
2014-03-25 01:57:24 -04:00
self . show . loadFromIndexer ( cache = not self . force )
except sickbeard . indexer_error , e :
logger . log ( u " Unable to contact " + sickbeard . indexerApi ( self . show . indexer ) . name + " , aborting: " + ex ( e ) ,
logger . WARNING )
2014-03-10 01:18:05 -04:00
return
2014-03-25 01:57:24 -04:00
except sickbeard . indexer_attributenotfound , e :
logger . log ( u " Data retrieved from " + sickbeard . indexerApi (
self . show . indexer ) . name + " was incomplete, aborting: " + ex ( e ) , logger . ERROR )
2014-03-10 01:18:05 -04:00
return
logger . log ( u " Retrieving show info from IMDb " , logger . DEBUG )
try :
self . show . loadIMDbInfo ( )
except imdb_exceptions . IMDbError , e :
logger . log ( u " Something wrong on IMDb api: " + ex ( e ) , logger . WARNING )
except Exception , e :
logger . log ( u " Error loading IMDb info: " + ex ( e ) , logger . ERROR )
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
try :
self . show . saveToDB ( )
except Exception , e :
logger . log ( u " Error saving the episode to the database: " + ex ( e ) , logger . ERROR )
logger . log ( traceback . format_exc ( ) , logger . DEBUG )
2014-03-25 01:57:24 -04:00
2014-03-10 01:18:05 -04:00
# get episode list from DB
logger . log ( u " Loading all episodes from the database " , logger . DEBUG )
DBEpList = self . show . loadEpisodesFromDB ( )
# get episode list from TVDB
2014-03-25 01:57:24 -04:00
logger . log ( u " Loading all episodes from " + sickbeard . indexerApi ( self . show . indexer ) . name + " " , logger . DEBUG )
2014-03-10 01:18:05 -04:00
try :
IndexerEpList = self . show . loadEpisodesFromIndexer ( cache = not self . force )
2014-03-25 01:57:24 -04:00
except sickbeard . indexer_exception , e :
logger . log ( u " Unable to get info from " + sickbeard . indexerApi (
self . show . indexer ) . name + " , the show info will not be refreshed: " + ex ( e ) , logger . ERROR )
2014-03-10 01:18:05 -04:00
IndexerEpList = None
2014-03-20 14:03:22 -04:00
if IndexerEpList == None :
2014-03-25 01:57:24 -04:00
logger . log ( u " No data returned from " + sickbeard . indexerApi (
self . show . indexer ) . name + " , unable to update this show " , logger . ERROR )
2014-03-10 01:18:05 -04:00
else :
# for each ep we found on TVDB delete it from the DB list
for curSeason in IndexerEpList :
for curEpisode in IndexerEpList [ curSeason ] :
2014-03-25 01:57:24 -04:00
logger . log ( u " Removing " + str ( curSeason ) + " x " + str ( curEpisode ) + " from the DB list " ,
logger . DEBUG )
2014-03-10 01:18:05 -04:00
if curSeason in DBEpList and curEpisode in DBEpList [ curSeason ] :
del DBEpList [ curSeason ] [ curEpisode ]
# for the remaining episodes in the DB list just delete them from the DB
for curSeason in DBEpList :
for curEpisode in DBEpList [ curSeason ] :
2014-03-25 01:57:24 -04:00
logger . log ( u " Permanently deleting episode " + str ( curSeason ) + " x " + str (
curEpisode ) + " from the database " , logger . MESSAGE )
2014-03-10 01:18:05 -04:00
curEp = self . show . getEpisode ( curSeason , curEpisode )
try :
curEp . deleteEpisode ( )
except exceptions . EpisodeDeletedException :
pass
2014-05-30 01:48:02 -04:00
sickbeard . showQueueScheduler . action . refreshShow ( self . show , self . force )
2014-03-10 01:18:05 -04:00
class QueueItemForceUpdate ( QueueItemUpdate ) :
def __init__ ( self , show = None ) :
ShowQueueItem . __init__ ( self , ShowQueueActions . FORCEUPDATE , show )
self . force = True