2014-03-10 05:18:05 +00:00
# Author: Dennis Lutter <lad1337@gmail.com>
# Author: Jonathon Saine <thezoggy@gmail.com>
# URL: http://code.google.com/p/sickbeard/
#
2014-11-12 16:43:14 +00:00
# This file is part of SickGear.
2014-03-10 05:18:05 +00:00
#
2014-11-12 16:43:14 +00:00
# SickGear 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-11-12 16:43:14 +00:00
# SickGear 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-11-12 16:43:14 +00:00
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
2014-03-10 05:18:05 +00:00
from __future__ import with_statement
import os
import time
import urllib
import datetime
import re
import traceback
import sickbeard
import webserve
2014-06-11 08:34:28 +00:00
2014-03-10 05:18:05 +00:00
from sickbeard import db , logger , exceptions , history , ui , helpers
from sickbeard import encodingKludge as ek
from sickbeard import search_queue
2014-07-27 10:59:21 +00:00
from sickbeard import image_cache
from sickbeard import classes
2014-09-30 23:13:04 +00:00
from sickbeard import processTV
2014-11-02 03:55:29 +00:00
from sickbeard import network_timezones , sbdatetime
2014-07-27 10:59:21 +00:00
from sickbeard . exceptions import ex
2017-11-27 19:35:20 +00:00
from sickbeard . common import SNATCHED , SNATCHED_ANY , SNATCHED_PROPER , SNATCHED_BEST , DOWNLOADED , SKIPPED , UNAIRED , IGNORED , ARCHIVED , WANTED , UNKNOWN
2014-11-27 03:30:00 +00:00
from sickbeard . helpers import remove_article
2014-03-10 05:18:05 +00:00
from common import Quality , qualityPresetStrings , statusStrings
2016-09-04 20:00:44 +00:00
from sickbeard . indexers . indexer_config import *
2015-02-06 11:39:10 +00:00
from sickbeard . webserve import MainHandler
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
try :
import json
except ImportError :
from lib import simplejson as json
from lib import subliminal
dateFormat = " % Y- % m- %d "
dateTimeFormat = " % Y- % m- %d % H: % M "
2014-11-02 03:55:29 +00:00
timeFormat = ' % A % I: % M % p '
2014-03-10 05:18:05 +00:00
2014-03-25 05:57:24 +00:00
RESULT_SUCCESS = 10 # only use inside the run methods
RESULT_FAILURE = 20 # only use inside the run methods
RESULT_TIMEOUT = 30 # not used yet :(
RESULT_ERROR = 40 # only use outside of the run methods !
RESULT_FATAL = 50 # only use in Api.default() ! this is the "we encountered an internal error" error
RESULT_DENIED = 60 # only use in Api.default() ! this is the acces denied error
2014-03-10 05:18:05 +00:00
result_type_map = { RESULT_SUCCESS : " success " ,
2014-03-25 05:57:24 +00:00
RESULT_FAILURE : " failure " ,
RESULT_TIMEOUT : " timeout " ,
RESULT_ERROR : " error " ,
RESULT_FATAL : " fatal " ,
RESULT_DENIED : " denied " ,
}
2014-03-10 05:18:05 +00:00
# basically everything except RESULT_SUCCESS / success is bad
2015-02-06 11:39:10 +00:00
class Api ( webserve . BaseHandler ) :
2014-03-10 05:18:05 +00:00
""" api class that returns json results """
2014-03-25 05:57:24 +00:00
version = 4 # use an int since float-point is unpredictible
2014-03-10 05:18:05 +00:00
intent = 4
2015-02-06 11:39:10 +00:00
def set_default_headers ( self ) :
self . set_header ( ' Cache-Control ' , ' no-store, no-cache, must-revalidate, max-age=0 ' )
2015-08-07 23:53:30 +00:00
self . set_header ( ' X-Robots-Tag ' , ' noindex, nofollow, noarchive, nocache, noodp, noydir, noimageindex, nosnippet ' )
2015-02-06 11:39:10 +00:00
def get ( self , route , * args , * * kwargs ) :
route = route . strip ( ' / ' ) or ' index '
kwargs = self . request . arguments
for arg , value in kwargs . items ( ) :
if len ( value ) == 1 :
kwargs [ arg ] = value [ 0 ]
args = args [ 1 : ]
2014-03-10 05:18:05 +00:00
self . apiKey = sickbeard . API_KEY
2015-02-06 11:39:10 +00:00
access , accessMsg , args , kwargs = self . _grand_access ( self . apiKey , route , args , kwargs )
2014-03-10 05:18:05 +00:00
# set the output callback
# default json
outputCallbackDict = { ' default ' : self . _out_as_json ,
' image ' : lambda x : x [ ' image ' ] ,
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
# do we have acces ?
if access :
logger . log ( accessMsg , logger . DEBUG )
else :
logger . log ( accessMsg , logger . WARNING )
2014-03-20 18:03:22 +00:00
return outputCallbackDict [ ' default ' ] ( _responds ( RESULT_DENIED , msg = accessMsg ) )
2014-03-10 05:18:05 +00:00
# set the original call_dispatcher as the local _call_dispatcher
_call_dispatcher = call_dispatcher
# if profile was set wrap "_call_dispatcher" in the profile function
if ' profile ' in kwargs :
from lib . profilehooks import profile
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
_call_dispatcher = profile ( _call_dispatcher , immediate = True )
del kwargs [ " profile " ]
# if debug was set call the "_call_dispatcher"
if ' debug ' in kwargs :
2014-07-12 01:38:22 +00:00
outDict = _call_dispatcher ( self , args ,
kwargs ) # this way we can debug the cherry.py traceback in the browser
2014-03-10 05:18:05 +00:00
del kwargs [ " debug " ]
2014-03-25 05:57:24 +00:00
else : # if debug was not set we wrap the "call_dispatcher" in a try block to assure a json output
2014-03-10 05:18:05 +00:00
try :
2014-06-19 15:47:28 +00:00
outDict = _call_dispatcher ( self , args , kwargs )
2015-06-08 12:47:01 +00:00
except Exception as e : # real internal error oohhh nooo :(
2014-03-10 05:18:05 +00:00
logger . log ( u " API :: " + ex ( e ) , logger . ERROR )
errorData = { " error_msg " : ex ( e ) ,
" args " : args ,
" kwargs " : kwargs }
2014-03-25 05:57:24 +00:00
outDict = _responds ( RESULT_FATAL , errorData ,
2014-11-12 16:43:14 +00:00
" SickGear encountered an internal error! Please report to the Devs " )
2014-03-10 05:18:05 +00:00
if ' outputType ' in outDict :
outputCallback = outputCallbackDict [ outDict [ ' outputType ' ] ]
else :
outputCallback = outputCallbackDict [ ' default ' ]
2015-02-06 11:39:10 +00:00
self . finish ( outputCallback ( outDict ) )
def _out_as_json ( self , dict ) :
2016-01-12 02:07:41 +00:00
self . set_header ( ' Content-Type ' , ' application/json; charset=UTF-8 ' )
2015-02-06 11:39:10 +00:00
try :
out = json . dumps ( dict , indent = self . intent , sort_keys = True )
2016-01-12 02:07:41 +00:00
callback = self . get_query_argument ( ' callback ' , None ) or self . get_query_argument ( ' jsonp ' , None )
if None is not callback :
out = ' %s ( %s ); ' % ( callback , out ) # wrap with JSONP call if requested
2015-02-06 11:39:10 +00:00
2015-06-08 12:47:01 +00:00
except Exception as e : # if we fail to generate the output fake an error
2017-06-04 16:19:22 +00:00
logger . log ( u ' API :: ' + traceback . format_exc ( ) , logger . ERROR )
2015-02-06 11:39:10 +00:00
out = ' { " result " : " ' + result_type_map [ RESULT_ERROR ] + ' " , " message " : " error while composing output: " ' + ex (
e ) + ' " } '
2016-01-12 02:07:41 +00:00
return out
2015-02-06 11:39:10 +00:00
def _grand_access ( self , realKey , apiKey , args , kwargs ) :
""" validate api key and log result """
remoteIp = self . request . remote_ip
if not sickbeard . USE_API :
msg = u ' API :: ' + remoteIp + ' - SB API Disabled. ACCESS DENIED '
return False , msg , args , kwargs
elif apiKey == realKey :
msg = u ' API :: ' + remoteIp + ' - gave correct API KEY. ACCESS GRANTED '
return True , msg , args , kwargs
elif not apiKey :
msg = u ' API :: ' + remoteIp + ' - gave NO API KEY. ACCESS DENIED '
return False , msg , args , kwargs
else :
msg = u ' API :: ' + remoteIp + ' - gave WRONG API KEY ' + apiKey + ' . ACCESS DENIED '
return False , msg , args , kwargs
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def call_dispatcher ( handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
""" calls the appropriate CMD class
looks for a cmd in args and kwargs
or calls the TVDBShorthandWrapper when the first args element is a number
or returns an error that there is no such cmd
"""
logger . log ( u " API :: all args: ' " + str ( args ) + " ' " , logger . DEBUG )
logger . log ( u " API :: all kwargs: ' " + str ( kwargs ) + " ' " , logger . DEBUG )
2014-07-12 01:38:22 +00:00
# logger.log(u"API :: dateFormat: '" + str(dateFormat) + "'", logger.DEBUG)
2014-03-10 05:18:05 +00:00
cmds = None
if args :
cmds = args [ 0 ]
args = args [ 1 : ]
if " cmd " in kwargs :
cmds = kwargs [ " cmd " ]
del kwargs [ " cmd " ]
outDict = { }
2014-03-20 18:03:22 +00:00
if cmds != None :
2014-03-10 05:18:05 +00:00
cmds = cmds . split ( " | " )
multiCmds = bool ( len ( cmds ) > 1 )
for cmd in cmds :
curArgs , curKwargs = filter_params ( cmd , args , kwargs )
cmdIndex = None
2014-03-25 05:57:24 +00:00
if len ( cmd . split ( " _ " ) ) > 1 : # was a index used for this cmd ?
cmd , cmdIndex = cmd . split ( " _ " ) # this gives us the clear cmd and the index
2014-03-10 05:18:05 +00:00
logger . log ( u " API :: " + cmd + " : curKwargs " + str ( curKwargs ) , logger . DEBUG )
2014-03-25 05:57:24 +00:00
if not ( multiCmds and cmd in ( ' show.getposter ' , ' show.getbanner ' ) ) : # skip these cmd while chaining
2014-03-10 05:18:05 +00:00
try :
if cmd in _functionMaper :
2014-06-19 15:47:28 +00:00
curOutDict = _functionMaper . get ( cmd ) ( handler , curArgs ,
2014-03-25 05:57:24 +00:00
curKwargs ) . run ( ) # get the cmd class, init it and run()
2014-03-10 05:18:05 +00:00
elif _is_int ( cmd ) :
2014-06-19 15:47:28 +00:00
curOutDict = TVDBShorthandWrapper ( handler , curArgs , curKwargs , cmd ) . run ( )
2014-03-10 05:18:05 +00:00
else :
curOutDict = _responds ( RESULT_ERROR , " No such cmd: ' " + cmd + " ' " )
2015-06-08 12:47:01 +00:00
except ApiError as e : # Api errors that we raised, they are harmless
2014-03-10 05:18:05 +00:00
curOutDict = _responds ( RESULT_ERROR , msg = ex ( e ) )
2014-03-25 05:57:24 +00:00
else : # if someone chained one of the forbiden cmds they will get an error for this one cmd
2014-03-10 05:18:05 +00:00
curOutDict = _responds ( RESULT_ERROR , msg = " The cmd ' " + cmd + " ' is not supported while chaining " )
if multiCmds :
# note: if multiple same cmds are issued but one has not an index defined it will override all others
# or the other way around, this depends on the order of the cmds
# this is not a bug
2014-03-25 05:57:24 +00:00
if cmdIndex is None : # do we need a index dict for this cmd ?
2014-03-10 05:18:05 +00:00
outDict [ cmd ] = curOutDict
else :
if not cmd in outDict :
outDict [ cmd ] = { }
outDict [ cmd ] [ cmdIndex ] = curOutDict
else :
outDict = curOutDict
2014-03-25 05:57:24 +00:00
if multiCmds : # if we had multiple cmds we have to wrap it in a response dict
2014-03-10 05:18:05 +00:00
outDict = _responds ( RESULT_SUCCESS , outDict )
2014-03-25 05:57:24 +00:00
else : # index / no cmd given
2014-06-19 15:47:28 +00:00
outDict = CMD_SickBeard ( handler , args , kwargs ) . run ( )
2014-03-10 05:18:05 +00:00
return outDict
def filter_params ( cmd , args , kwargs ) :
""" return only params kwargs that are for cmd
and rename them to a clean version ( remove " <cmd>_ " )
args are shared across all cmds
all args and kwarks are lowerd
cmd are separated by " | " e . g . & cmd = shows | future
kwargs are namespaced with " . " e . g . show . indexerid = 101501
if a karg has no namespace asing it anyways ( global )
full e . g .
/ api ? apikey = 1234 & cmd = show . seasonlist_asd | show . seasonlist_2 & show . seasonlist_asd . indexerid = 101501 & show . seasonlist_2 . indexerid = 79488 & sort = asc
two calls of show . seasonlist
one has the index " asd " the other one " 2 "
the " indexerid " kwargs / params have the indexed cmd as a namspace
and the kwarg / param " sort " is a used as a global
"""
curArgs = [ ]
for arg in args :
curArgs . append ( arg . lower ( ) )
curArgs = tuple ( curArgs )
curKwargs = { }
for kwarg in kwargs :
if kwarg . find ( cmd + " . " ) == 0 :
cleanKey = kwarg . rpartition ( " . " ) [ 2 ]
curKwargs [ cleanKey ] = kwargs [ kwarg ] . lower ( )
2014-03-25 05:57:24 +00:00
elif not " . " in kwarg : # the kwarg was not namespaced therefore a "global"
2014-03-10 05:18:05 +00:00
curKwargs [ kwarg ] = kwargs [ kwarg ]
return curArgs , curKwargs
2014-06-19 15:47:28 +00:00
class ApiCall ( object ) :
2014-03-10 05:18:05 +00:00
_help = { " desc " : " No help message available. Please tell the devs that a help msg is missing for this cmd " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-06-18 17:06:06 +00:00
2014-03-10 05:18:05 +00:00
# missing
try :
if self . _missing :
self . run = self . return_missing
except AttributeError :
pass
# help
if ' help ' in kwargs :
self . run = self . return_help
2014-06-19 15:47:28 +00:00
# RequestHandler
self . handler = handler
2014-03-10 05:18:05 +00:00
def run ( self ) :
# override with real output function in subclass
return { }
def return_help ( self ) :
try :
if self . _requiredParams :
pass
except AttributeError :
self . _requiredParams = [ ]
try :
if self . _optionalParams :
pass
except AttributeError :
self . _optionalParams = [ ]
for paramDict , type in [ ( self . _requiredParams , " requiredParameters " ) ,
2014-03-25 05:57:24 +00:00
( self . _optionalParams , " optionalParameters " ) ] :
2014-03-10 05:18:05 +00:00
if type in self . _help :
for paramName in paramDict :
if not paramName in self . _help [ type ] :
self . _help [ type ] [ paramName ] = { }
if paramDict [ paramName ] [ " allowedValues " ] :
self . _help [ type ] [ paramName ] [ " allowedValues " ] = paramDict [ paramName ] [ " allowedValues " ]
else :
self . _help [ type ] [ paramName ] [ " allowedValues " ] = " see desc "
self . _help [ type ] [ paramName ] [ " defaultValue " ] = paramDict [ paramName ] [ " defaultValue " ]
elif paramDict :
for paramName in paramDict :
self . _help [ type ] = { }
self . _help [ type ] [ paramName ] = paramDict [ paramName ]
else :
self . _help [ type ] = { }
msg = " No description available "
if " desc " in self . _help :
msg = self . _help [ " desc " ]
del self . _help [ " desc " ]
return _responds ( RESULT_SUCCESS , self . _help , msg )
def return_missing ( self ) :
if len ( self . _missing ) == 1 :
msg = " The required parameter: ' " + self . _missing [ 0 ] + " ' was not set "
else :
msg = " The required parameters: ' " + " ' , ' " . join ( self . _missing ) + " ' where not set "
return _responds ( RESULT_ERROR , msg = msg )
def check_params ( self , args , kwargs , key , default , required , type , allowedValues ) :
# TODO: explain this
""" function to check passed params for the shorthand wrapper
and to detect missing / required param
"""
2014-05-02 11:01:42 +00:00
# Fix for applications that send tvdbid instead of indexerid
2014-04-28 13:59:01 +00:00
if key == " indexerid " and " indexerid " not in kwargs :
key = " tvdbid "
2014-05-02 11:01:42 +00:00
2014-03-10 05:18:05 +00:00
missing = True
orgDefault = default
if type == " bool " :
allowedValues = [ 0 , 1 ]
if args :
default = args [ 0 ]
missing = False
args = args [ 1 : ]
if kwargs . get ( key ) :
default = kwargs . get ( key )
missing = False
if required :
try :
self . _missing
self . _requiredParams . append ( key )
except AttributeError :
self . _missing = [ ]
self . _requiredParams = { }
self . _requiredParams [ key ] = { " allowedValues " : allowedValues ,
" defaultValue " : orgDefault }
if missing and key not in self . _missing :
self . _missing . append ( key )
else :
try :
self . _optionalParams [ key ] = { " allowedValues " : allowedValues ,
" defaultValue " : orgDefault }
except AttributeError :
self . _optionalParams = { }
self . _optionalParams [ key ] = { " allowedValues " : allowedValues ,
" defaultValue " : orgDefault }
if default :
default = self . _check_param_type ( default , key , type )
if type == " bool " :
type = [ ]
self . _check_param_value ( default , key , allowedValues )
return default , args
def _check_param_type ( self , value , name , type ) :
""" checks if value can be converted / parsed to type
will raise an error on failure
or will convert it to type and return new converted value
can check for :
- int : will be converted into int
- bool : will be converted to False / True
- list : will always return a list
- string : will do nothing for now
- ignore : will ignore it , just like " string "
"""
error = False
if type == " int " :
if _is_int ( value ) :
value = int ( value )
else :
error = True
elif type == " bool " :
if value in ( " 0 " , " 1 " ) :
value = bool ( int ( value ) )
elif value in ( " true " , " True " , " TRUE " ) :
value = True
elif value in ( " false " , " False " , " FALSE " ) :
value = False
else :
error = True
elif type == " list " :
value = value . split ( " | " )
elif type == " string " :
pass
elif type == " ignore " :
pass
else :
2014-03-25 05:57:24 +00:00
logger . log ( u " API :: Invalid param type set " + str ( type ) + " can not check or convert ignoring it " ,
logger . ERROR )
2014-03-10 05:18:05 +00:00
if error :
# this is a real ApiError !!
2014-03-25 05:57:24 +00:00
raise ApiError (
u " param: ' " + str ( name ) + " ' with given value: ' " + str ( value ) + " ' could not be parsed into ' " + str (
type ) + " ' " )
2014-03-10 05:18:05 +00:00
return value
def _check_param_value ( self , value , name , allowedValues ) :
""" will check if value (or all values in it ) are in allowed values
will raise an exception if value is " out of range "
if bool ( allowedValue ) == False a check is not performed and all values are excepted
"""
if allowedValues :
error = False
if isinstance ( value , list ) :
for item in value :
if not item in allowedValues :
error = True
else :
if not value in allowedValues :
error = True
if error :
# this is kinda a ApiError but raising an error is the only way of quitting here
2014-03-25 05:57:24 +00:00
raise ApiError ( u " param: ' " + str ( name ) + " ' with given value: ' " + str (
value ) + " ' is out of allowed range ' " + str ( allowedValues ) + " ' " )
2014-03-10 05:18:05 +00:00
class TVDBShorthandWrapper ( ApiCall ) :
_help = { " desc " : " this is an internal function wrapper. call the help command directly for more information " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs , sid ) :
self . handler = handler
2014-03-10 05:18:05 +00:00
self . origArgs = args
self . kwargs = kwargs
self . sid = sid
self . s , args = self . check_params ( args , kwargs , " s " , None , False , " ignore " , [ ] )
self . e , args = self . check_params ( args , kwargs , " e " , None , False , " ignore " , [ ] )
self . args = args
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" internal function wrapper """
args = ( self . sid , ) + self . origArgs
if self . e :
2014-06-19 15:47:28 +00:00
return CMD_Episode ( self . handler , args , self . kwargs ) . run ( )
2014-03-10 05:18:05 +00:00
elif self . s :
2014-06-19 15:47:28 +00:00
return CMD_ShowSeasons ( self . handler , args , self . kwargs ) . run ( )
2014-03-10 05:18:05 +00:00
else :
2014-06-19 15:47:28 +00:00
return CMD_Show ( self . handler , args , self . kwargs ) . run ( )
2014-03-10 05:18:05 +00:00
2014-07-12 01:38:22 +00:00
# ###############################
# helper functions #
# ###############################
2014-03-10 05:18:05 +00:00
def _sizeof_fmt ( num ) :
for x in [ ' bytes ' , ' KB ' , ' MB ' , ' GB ' , ' TB ' ] :
if num < 1024.00 :
return " %3.2f %s " % ( num , x )
num / = 1024.00
def _is_int ( data ) :
try :
int ( data )
except ( TypeError , ValueError , OverflowError ) :
return False
else :
return True
def _rename_element ( dict , oldKey , newKey ) :
try :
dict [ newKey ] = dict [ oldKey ]
del dict [ oldKey ]
except ( ValueError , TypeError , NameError ) :
pass
return dict
def _responds ( result_type , data = None , msg = " " ) :
"""
result is a string of given " type " ( success / failure / timeout / error )
message is a human readable string , can be empty
data is either a dict or a array , can be a empty dict or empty array
"""
if data is None :
data = { }
return { " result " : result_type_map [ result_type ] ,
" message " : msg ,
" data " : data }
def _get_quality_string ( q ) :
qualityString = " Custom "
if q in qualityPresetStrings :
qualityString = qualityPresetStrings [ q ]
elif q in Quality . qualityStrings :
qualityString = Quality . qualityStrings [ q ]
return qualityString
def _get_status_Strings ( s ) :
return statusStrings [ s ]
def _ordinal_to_dateTimeForm ( ordinal ) :
# workaround for episodes with no airdate
if int ( ordinal ) != 1 :
date = datetime . date . fromordinal ( ordinal )
else :
return " "
return date . strftime ( dateTimeFormat )
def _ordinal_to_dateForm ( ordinal ) :
if int ( ordinal ) != 1 :
date = datetime . date . fromordinal ( ordinal )
else :
return " "
return date . strftime ( dateFormat )
def _historyDate_to_dateTimeForm ( timeString ) :
date = datetime . datetime . strptime ( timeString , history . dateFormat )
return date . strftime ( dateTimeFormat )
def _replace_statusStrings_with_statusCodes ( statusStrings ) :
statusCodes = [ ]
if " snatched " in statusStrings :
statusCodes + = Quality . SNATCHED
if " downloaded " in statusStrings :
statusCodes + = Quality . DOWNLOADED
if " skipped " in statusStrings :
statusCodes . append ( SKIPPED )
if " wanted " in statusStrings :
statusCodes . append ( WANTED )
if " archived " in statusStrings :
2017-11-27 19:35:20 +00:00
statusCodes + = Quality . ARCHIVED
2014-03-10 05:18:05 +00:00
if " ignored " in statusStrings :
statusCodes . append ( IGNORED )
if " unaired " in statusStrings :
statusCodes . append ( UNAIRED )
return statusCodes
def _mapQuality ( showObj ) :
quality_map = _getQualityMap ( )
anyQualities = [ ]
bestQualities = [ ]
iqualityID , aqualityID = Quality . splitQuality ( int ( showObj ) )
if iqualityID :
for quality in iqualityID :
anyQualities . append ( quality_map [ quality ] )
if aqualityID :
for quality in aqualityID :
bestQualities . append ( quality_map [ quality ] )
return anyQualities , bestQualities
def _getQualityMap ( ) :
return { Quality . SDTV : ' sdtv ' ,
Quality . SDDVD : ' sddvd ' ,
Quality . HDTV : ' hdtv ' ,
Quality . RAWHDTV : ' rawhdtv ' ,
Quality . FULLHDTV : ' fullhdtv ' ,
Quality . HDWEBDL : ' hdwebdl ' ,
Quality . FULLHDWEBDL : ' fullhdwebdl ' ,
Quality . HDBLURAY : ' hdbluray ' ,
Quality . FULLHDBLURAY : ' fullhdbluray ' ,
Quality . UNKNOWN : ' unknown ' }
def _getRootDirs ( ) :
if sickbeard . ROOT_DIRS == " " :
return { }
rootDir = { }
root_dirs = sickbeard . ROOT_DIRS . split ( ' | ' )
default_index = int ( sickbeard . ROOT_DIRS . split ( ' | ' ) [ 0 ] )
rootDir [ " default_index " ] = int ( sickbeard . ROOT_DIRS . split ( ' | ' ) [ 0 ] )
# remove default_index value from list (this fixes the offset)
root_dirs . pop ( 0 )
if len ( root_dirs ) < default_index :
return { }
# clean up the list - replace %xx escapes by their single-character equivalent
root_dirs = [ urllib . unquote_plus ( x ) for x in root_dirs ]
default_dir = root_dirs [ default_index ]
dir_list = [ ]
for root_dir in root_dirs :
valid = 1
try :
ek . ek ( os . listdir , root_dir )
except :
valid = 0
default = 0
if root_dir is default_dir :
default = 1
curDir = { }
curDir [ ' valid ' ] = valid
curDir [ ' location ' ] = root_dir
curDir [ ' default ' ] = default
dir_list . append ( curDir )
return dir_list
class ApiError ( Exception ) :
" Generic API error "
class IntParseError ( Exception ) :
" A value could not be parsed into a int. But should be parsable to a int "
2014-03-25 05:57:24 +00:00
2014-07-12 01:38:22 +00:00
# -------------------------------------------------------------------------------------#
2014-03-10 05:18:05 +00:00
class CMD_Help ( ApiCall ) :
_help = { " desc " : " display help information for a given subject/command " ,
" optionalParameters " : { " subject " : { " desc " : " command - the top level command " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . subject , args = self . check_params ( args , kwargs , " subject " , " help " , False , " string " , _functionMaper . keys ( ) )
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display help information for a given subject/command """
if self . subject in _functionMaper :
out = _responds ( RESULT_SUCCESS , _functionMaper . get ( self . subject ) ( ( ) , { " help " : 1 } ) . run ( ) )
else :
out = _responds ( RESULT_FAILURE , msg = " No such cmd " )
return out
class CMD_ComingEpisodes ( ApiCall ) :
_help = { " desc " : " display the coming episodes " ,
" optionalParameters " : { " sort " : { " desc " : " change the sort order " } ,
2014-03-25 05:57:24 +00:00
" type " : { " desc " : " one or more of allowedValues separated by | " } ,
" paused " : {
2014-07-12 01:38:22 +00:00
" desc " : " 0 to exclude paused shows, 1 to include them, or omitted to use the SB default " } ,
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . sort , args = self . check_params ( args , kwargs , " sort " , " date " , False , " string " , [ " date " , " show " , " network " ] )
2014-03-25 05:57:24 +00:00
self . type , args = self . check_params ( args , kwargs , " type " , " today|missed|soon|later " , False , " list " ,
[ " missed " , " later " , " today " , " soon " ] )
2014-12-16 14:14:54 +00:00
self . paused , args = self . check_params ( args , kwargs , " paused " , sickbeard . EPISODE_VIEW_DISPLAY_PAUSED , False , " int " ,
2014-03-25 05:57:24 +00:00
[ 0 , 1 ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display the coming episodes """
2014-11-25 22:12:00 +00:00
today_dt = datetime . date . today ( )
today = today_dt . toordinal ( )
yesterday_dt = today_dt - datetime . timedelta ( days = 1 )
yesterday = yesterday_dt . toordinal ( )
2014-11-24 12:05:29 +00:00
tomorrow = ( datetime . date . today ( ) + datetime . timedelta ( days = 1 ) ) . toordinal ( )
2014-11-25 22:12:00 +00:00
next_week_dt = ( datetime . date . today ( ) + datetime . timedelta ( days = 7 ) )
next_week = ( next_week_dt + datetime . timedelta ( days = 1 ) ) . toordinal ( )
2014-12-16 14:14:54 +00:00
recently = ( yesterday_dt - datetime . timedelta ( days = sickbeard . EPISODE_VIEW_MISSED_RANGE ) ) . toordinal ( )
2014-03-10 05:18:05 +00:00
done_show_list = [ ]
2017-11-27 19:35:20 +00:00
qualList = Quality . SNATCHED + Quality . DOWNLOADED + Quality . ARCHIVED + [ IGNORED ]
2014-03-10 05:18:05 +00:00
2014-11-24 12:05:29 +00:00
myDB = db . DBConnection ( )
2014-03-25 05:57:24 +00:00
sql_results = myDB . select (
2014-11-24 12:05:29 +00:00
" SELECT airdate, airs, episode, name AS ' ep_name ' , description AS ' ep_plot ' , network, season, showid AS ' indexerid ' , show_name, tv_shows.quality AS quality, tv_shows.status AS ' show_status ' , tv_shows.paused AS ' paused ' FROM tv_episodes, tv_shows WHERE season != 0 AND airdate >= ? AND airdate <= ? AND tv_shows.indexer_id = tv_episodes.showid AND tv_episodes.status NOT IN ( " + ' , ' . join (
[ ' ? ' ] * len ( qualList ) ) + " ) " , [ yesterday , next_week ] + qualList )
2014-03-10 05:18:05 +00:00
for cur_result in sql_results :
done_show_list . append ( int ( cur_result [ " indexerid " ] ) )
2014-03-25 05:57:24 +00:00
more_sql_results = myDB . select (
" SELECT airdate, airs, episode, name AS ' ep_name ' , description AS ' ep_plot ' , network, season, showid AS ' indexerid ' , show_name, tv_shows.quality AS quality, tv_shows.status AS ' show_status ' , tv_shows.paused AS ' paused ' FROM tv_episodes outer_eps, tv_shows WHERE season != 0 AND showid NOT IN ( " + ' , ' . join (
[ ' ? ' ] * len (
done_show_list ) ) + " ) AND tv_shows.indexer_id = outer_eps.showid AND airdate = (SELECT airdate FROM tv_episodes inner_eps WHERE inner_eps.season != 0 AND inner_eps.showid = outer_eps.showid AND inner_eps.airdate >= ? ORDER BY inner_eps.airdate ASC LIMIT 1) AND outer_eps.status NOT IN ( " + ' , ' . join (
[ ' ? ' ] * len ( Quality . DOWNLOADED + Quality . SNATCHED ) ) + " ) " ,
done_show_list + [ next_week ] + Quality . DOWNLOADED + Quality . SNATCHED )
2014-03-10 05:18:05 +00:00
sql_results + = more_sql_results
2014-03-25 05:57:24 +00:00
more_sql_results = myDB . select (
2014-11-24 12:05:29 +00:00
" SELECT airdate, airs, episode, name AS ' ep_name ' , description AS ' ep_plot ' , network, season, showid AS ' indexerid ' , show_name, tv_shows.quality AS quality, tv_shows.status AS ' show_status ' , tv_shows.paused AS ' paused ' FROM tv_episodes, tv_shows WHERE season != 0 AND tv_shows.indexer_id = tv_episodes.showid AND airdate <= ? AND airdate >= ? AND tv_episodes.status = ? AND tv_episodes.status NOT IN ( " + ' , ' . join (
[ ' ? ' ] * len ( qualList ) ) + " ) " , [ tomorrow , recently , WANTED ] + qualList )
2014-03-10 05:18:05 +00:00
sql_results + = more_sql_results
2014-11-24 12:05:29 +00:00
sql_results = list ( set ( sql_results ) )
# make a dict out of the sql results
sql_results = [ dict ( row ) for row in sql_results ]
2016-02-02 12:22:58 +00:00
2014-11-25 22:12:00 +00:00
# multi dimension sort
2014-03-10 05:18:05 +00:00
sorts = {
2014-11-25 22:12:00 +00:00
' date ' : ( lambda a , b : cmp (
2014-12-16 14:14:54 +00:00
( a [ ' parsed_datetime ' ] , a [ ' data_show_name ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' parsed_datetime ' ] , b [ ' data_show_name ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) ) ,
2014-11-25 22:12:00 +00:00
' network ' : ( lambda a , b : cmp (
2014-12-16 14:14:54 +00:00
( a [ ' data_network ' ] , a [ ' parsed_datetime ' ] , a [ ' data_show_name ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' data_network ' ] , b [ ' parsed_datetime ' ] , b [ ' data_show_name ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) ) ,
' show ' : ( lambda a , b : cmp (
( a [ ' data_show_name ' ] , a [ ' parsed_datetime ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' data_show_name ' ] , b [ ' parsed_datetime ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) )
2014-03-10 05:18:05 +00:00
}
2015-01-21 16:07:24 +00:00
def value_maybe_article ( value = None ) :
if None is value :
return ' '
2014-12-16 14:14:54 +00:00
return ( remove_article ( value . lower ( ) ) , value . lower ( ) ) [ sickbeard . SORT_ARTICLE ]
2014-11-25 22:12:00 +00:00
# add parsed_datetime to the dict
2014-11-24 12:05:29 +00:00
for index , item in enumerate ( sql_results ) :
sql_results [ index ] [ ' parsed_datetime ' ] = network_timezones . parse_date_time ( item [ ' airdate ' ] , item [ ' airs ' ] , item [ ' network ' ] )
2014-12-16 14:14:54 +00:00
sql_results [ index ] [ ' data_show_name ' ] = value_maybe_article ( item [ ' show_name ' ] )
sql_results [ index ] [ ' data_network ' ] = value_maybe_article ( item [ ' network ' ] )
2014-03-10 05:18:05 +00:00
sql_results . sort ( sorts [ self . sort ] )
2014-12-16 14:14:54 +00:00
2014-03-10 05:18:05 +00:00
finalEpResults = { }
# add all requested types or all
for curType in self . type :
finalEpResults [ curType ] = [ ]
for ep in sql_results :
"""
Missed : yesterday . . . ( less than 1 week )
Today : today
Soon : tomorrow till next week
Later : later than next week
"""
if ep [ " paused " ] and not self . paused :
continue
2014-11-24 12:05:29 +00:00
ep [ ' airdate ' ] = int ( ep [ " airdate " ] )
2014-11-02 03:55:29 +00:00
2014-03-10 05:18:05 +00:00
status = " soon "
if ep [ " airdate " ] < today :
status = " missed "
elif ep [ " airdate " ] > = next_week :
status = " later "
2014-03-20 18:03:22 +00:00
elif ep [ " airdate " ] > = today and ep [ " airdate " ] < next_week :
2014-03-10 05:18:05 +00:00
if ep [ " airdate " ] == today :
status = " today "
else :
status = " soon "
# skip unwanted
2014-11-02 03:55:29 +00:00
if self . type is not None and not status in self . type :
2014-03-10 05:18:05 +00:00
continue
if not ep [ " network " ] :
ep [ " network " ] = " "
2014-11-02 03:55:29 +00:00
2014-03-10 05:18:05 +00:00
ep [ " quality " ] = _get_quality_string ( ep [ " quality " ] )
# clean up tvdb horrible airs field
2014-11-24 12:05:29 +00:00
ep [ ' airs ' ] = str ( ep [ ' airs ' ] ) . replace ( ' am ' , ' AM ' ) . replace ( ' pm ' , ' PM ' ) . replace ( ' ' , ' ' )
2014-03-10 05:18:05 +00:00
# start day of the week on 1 (monday)
2014-11-24 12:05:29 +00:00
ep [ ' weekday ' ] = 1 + datetime . date . fromordinal ( ep [ ' airdate ' ] ) . weekday ( )
2014-05-02 11:01:42 +00:00
# Add tvdbid for backward compability
2014-04-28 15:56:40 +00:00
ep [ " tvdbid " ] = ep [ ' indexerid ' ]
2014-11-24 12:05:29 +00:00
ep [ ' airdate ' ] = sbdatetime . sbdatetime . sbfdate ( datetime . date . fromordinal ( ep [ ' airdate ' ] ) , d_preset = dateFormat )
ep [ ' parsed_datetime ' ] = sbdatetime . sbdatetime . sbfdatetime ( ep [ ' parsed_datetime ' ] , d_preset = dateFormat , t_preset = ' % H: % M % z ' )
2014-03-10 05:18:05 +00:00
# TODO: check if this obsolete
if not status in finalEpResults :
finalEpResults [ status ] = [ ]
finalEpResults [ status ] . append ( ep )
2014-06-21 22:46:59 +00:00
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , finalEpResults )
class CMD_Episode ( ApiCall ) :
_help = { " desc " : " display detailed info about an episode " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
" season " : { " desc " : " the season number " } ,
" episode " : { " desc " : " the episode number " }
} ,
" optionalParameters " : { " full_path " : {
2014-07-12 01:38:22 +00:00
" desc " : " show the full absolute path (if valid) instead of a relative path for the episode location " }
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
self . s , args = self . check_params ( args , kwargs , " season " , None , True , " int " , [ ] )
self . e , args = self . check_params ( args , kwargs , " episode " , None , True , " int " , [ ] )
# optional
self . fullPath , args = self . check_params ( args , kwargs , " full_path " , 0 , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display detailed info about an episode """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
myDB = db . DBConnection ( row_type = " dict " )
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select (
" SELECT name, description, airdate, status, location, file_size, release_name, subtitles FROM tv_episodes WHERE showid = ? AND episode = ? AND season = ? " ,
[ self . indexerid , self . e , self . s ] )
2014-03-10 05:18:05 +00:00
if not len ( sqlResults ) == 1 :
raise ApiError ( " Episode not found " )
episode = sqlResults [ 0 ]
# handle path options
# absolute vs relative vs broken
showPath = None
try :
showPath = showObj . location
except sickbeard . exceptions . ShowDirNotFoundException :
pass
if bool ( self . fullPath ) == True and showPath :
pass
elif bool ( self . fullPath ) == False and showPath :
# using the length because lstrip removes to much
2014-03-25 05:57:24 +00:00
showPathLength = len ( showPath ) + 1 # the / or \ yeah not that nice i know
2014-03-10 05:18:05 +00:00
episode [ " location " ] = episode [ " location " ] [ showPathLength : ]
2014-03-25 05:57:24 +00:00
elif not showPath : # show dir is broken ... episode path will be empty
2014-03-10 05:18:05 +00:00
episode [ " location " ] = " "
# convert stuff to human form
2014-11-05 11:33:05 +00:00
episode [ ' airdate ' ] = sbdatetime . sbdatetime . sbfdate ( sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( int ( episode [ ' airdate ' ] ) , showObj . airs , showObj . network ) ) , d_preset = dateFormat )
2014-03-10 05:18:05 +00:00
status , quality = Quality . splitCompositeStatus ( int ( episode [ " status " ] ) )
episode [ " status " ] = _get_status_Strings ( status )
episode [ " quality " ] = _get_quality_string ( quality )
episode [ " file_size_human " ] = _sizeof_fmt ( episode [ " file_size " ] )
return _responds ( RESULT_SUCCESS , episode )
class CMD_EpisodeSearch ( ApiCall ) :
_help = { " desc " : " search for an episode. the response might take some time " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
" season " : { " desc " : " the season number " } ,
" episode " : { " desc " : " the episode number " }
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
self . s , args = self . check_params ( args , kwargs , " season " , None , True , " int " , [ ] )
self . e , args = self . check_params ( args , kwargs , " episode " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" search for an episode """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
# retrieve the episode object and fail if we can't get one
epObj = showObj . getEpisode ( int ( self . s ) , int ( self . e ) )
if isinstance ( epObj , str ) :
return _responds ( RESULT_FAILURE , msg = " Episode not found " )
# make a queue item for it and put it on the queue
2014-05-23 17:05:44 +00:00
ep_queue_item = search_queue . ManualSearchQueueItem ( showObj , epObj )
2014-03-25 05:57:24 +00:00
sickbeard . searchQueueScheduler . action . add_item ( ep_queue_item ) #@UndefinedVariable
2014-03-10 05:18:05 +00:00
# wait until the queue item tells us whether it worked or not
2014-03-25 05:57:24 +00:00
while ep_queue_item . success == None : #@UndefinedVariable
2014-03-10 05:18:05 +00:00
time . sleep ( 1 )
# return the correct json value
if ep_queue_item . success :
2014-03-25 05:57:24 +00:00
status , quality = Quality . splitCompositeStatus ( epObj . status ) #@UnusedVariable
2014-03-10 05:18:05 +00:00
# TODO: split quality and status?
2014-03-25 05:57:24 +00:00
return _responds ( RESULT_SUCCESS , { " quality " : _get_quality_string ( quality ) } ,
" Snatched ( " + _get_quality_string ( quality ) + " ) " )
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = ' Unable to find episode ' )
class CMD_EpisodeSetStatus ( ApiCall ) :
_help = { " desc " : " set status of an episode or season (when no ep is provided) " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
" season " : { " desc " : " the season number " } ,
" status " : { " desc " : " the status values: wanted, skipped, archived, ignored, failed " }
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " episode " : { " desc " : " the episode number " } ,
" force " : { " desc " : " should we replace existing (downloaded) episodes or not " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
self . s , args = self . check_params ( args , kwargs , " season " , None , True , " int " , [ ] )
2014-03-25 05:57:24 +00:00
self . status , args = self . check_params ( args , kwargs , " status " , None , True , " string " ,
[ " wanted " , " skipped " , " archived " , " ignored " , " failed " ] )
2014-03-10 05:18:05 +00:00
# optional
self . e , args = self . check_params ( args , kwargs , " episode " , None , False , " int " , [ ] )
self . force , args = self . check_params ( args , kwargs , " force " , 0 , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" set status of an episode or a season (when no ep is provided) """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
# convert the string status to a int
for status in statusStrings . statusStrings :
if str ( statusStrings [ status ] ) . lower ( ) == str ( self . status ) . lower ( ) :
self . status = status
break
2014-03-25 05:57:24 +00:00
else : # if we dont break out of the for loop we got here.
2014-03-10 05:18:05 +00:00
# the allowed values has at least one item that could not be matched against the internal status strings
raise ApiError ( " The status string could not be matched to a status. Report to Devs! " )
ep_list = [ ]
if self . e :
epObj = showObj . getEpisode ( self . s , self . e )
2014-03-20 18:03:22 +00:00
if epObj == None :
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Episode not found " )
ep_list = [ epObj ]
else :
# get all episode numbers frome self,season
ep_list = showObj . getAllEpisodes ( season = self . s )
def _epResult ( result_code , ep , msg = " " ) :
2014-03-25 05:57:24 +00:00
return { ' season ' : ep . season , ' episode ' : ep . episode , ' status ' : _get_status_Strings ( ep . status ) ,
' result ' : result_type_map [ result_code ] , ' message ' : msg }
2014-03-10 05:18:05 +00:00
ep_results = [ ]
failure = False
start_backlog = False
2014-09-07 04:36:23 +00:00
segments = { }
2014-05-30 10:01:49 +00:00
sql_l = [ ]
2014-03-10 05:18:05 +00:00
for epObj in ep_list :
with epObj . lock :
2014-07-24 04:44:11 +00:00
if self . status == WANTED :
# figure out what episodes are wanted so we can backlog them
2014-11-20 16:37:58 +00:00
if epObj . season in segments :
2014-09-07 04:36:23 +00:00
segments [ epObj . season ] . append ( epObj )
2014-07-24 04:44:11 +00:00
else :
2014-09-07 04:36:23 +00:00
segments [ epObj . season ] = [ epObj ]
2014-07-24 04:44:11 +00:00
2014-03-10 05:18:05 +00:00
# don't let them mess up UNAIRED episodes
if epObj . status == UNAIRED :
2014-03-25 05:57:24 +00:00
if self . e != None : # setting the status of a unaired is only considert a failure if we directly wanted this episode, but is ignored on a season request
ep_results . append (
_epResult ( RESULT_FAILURE , epObj , " Refusing to change status because it is UNAIRED " ) )
2014-03-10 05:18:05 +00:00
failure = True
continue
# allow the user to force setting the status for an already downloaded episode
if epObj . status in Quality . DOWNLOADED and not self . force :
2014-03-25 05:57:24 +00:00
ep_results . append ( _epResult ( RESULT_FAILURE , epObj ,
" Refusing to change status because it is already marked as DOWNLOADED " ) )
2014-03-10 05:18:05 +00:00
failure = True
continue
epObj . status = self . status
2015-03-29 11:20:29 +00:00
result = epObj . get_sql ( )
if None is not result :
sql_l . append ( result )
2014-03-10 05:18:05 +00:00
if self . status == WANTED :
start_backlog = True
ep_results . append ( _epResult ( RESULT_SUCCESS , epObj ) )
2015-03-29 11:20:29 +00:00
if 0 < len ( sql_l ) :
2014-05-30 10:01:49 +00:00
myDB = db . DBConnection ( )
myDB . mass_action ( sql_l )
2014-06-30 15:57:32 +00:00
2014-03-10 05:18:05 +00:00
extra_msg = " "
if start_backlog :
2014-09-07 04:36:23 +00:00
for season , segment in segments . items ( ) :
cur_backlog_queue_item = search_queue . BacklogQueueItem ( showObj , segment )
sickbeard . searchQueueScheduler . action . add_item ( cur_backlog_queue_item ) #@UndefinedVariable
2014-07-11 06:44:36 +00:00
logger . log ( u " API :: Starting backlog for " + showObj . name + " season " + str (
season ) + " because some episodes were set to WANTED " )
2014-09-07 04:36:23 +00:00
2014-03-10 05:18:05 +00:00
extra_msg = " Backlog started "
if failure :
return _responds ( RESULT_FAILURE , ep_results , ' Failed to set all or some status. Check data. ' + extra_msg )
else :
return _responds ( RESULT_SUCCESS , msg = ' All status set successfully. ' + extra_msg )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
class CMD_SubtitleSearch ( ApiCall ) :
_help = { " desc " : " search episode subtitles. the response might take some time " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
" season " : { " desc " : " the season number " } ,
" episode " : { " desc " : " the episode number " }
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
self . s , args = self . check_params ( args , kwargs , " season " , None , True , " int " , [ ] )
self . e , args = self . check_params ( args , kwargs , " episode " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" search episode subtitles """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
# retrieve the episode object and fail if we can't get one
epObj = showObj . getEpisode ( int ( self . s ) , int ( self . e ) )
if isinstance ( epObj , str ) :
return _responds ( RESULT_FAILURE , msg = " Episode not found " )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
# try do download subtitles for that episode
previous_subtitles = epObj . subtitles
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
try :
subtitles = epObj . downloadSubtitles ( )
except :
return _responds ( RESULT_FAILURE , msg = ' Unable to find subtitles ' )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
# return the correct json value
if previous_subtitles != epObj . subtitles :
2014-03-25 05:57:24 +00:00
status = ' New subtitles downloaded: %s ' % ' ' . join ( [
" <img src= ' " + sickbeard . WEB_ROOT + " /images/flags/ " + subliminal . language . Language (
x ) . alpha2 + " .png ' alt= ' " + subliminal . language . Language ( x ) . name + " ' /> " for x in
sorted ( list ( set ( epObj . subtitles ) . difference ( previous_subtitles ) ) ) ] )
2014-03-10 05:18:05 +00:00
response = _responds ( RESULT_SUCCESS , msg = ' New subtitles found ' )
else :
status = ' No subtitles downloaded '
response = _responds ( RESULT_FAILURE , msg = ' Unable to find subtitles ' )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
ui . notifications . message ( ' Subtitles Search ' , status )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
return response
class CMD_Exceptions ( ApiCall ) :
_help = { " desc " : " display scene exceptions for all or a given show " ,
" optionalParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , False , " int " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display scene exceptions for all or a given show """
2016-09-04 20:00:44 +00:00
myDB = db . DBConnection ( row_type = " dict " )
2014-03-10 05:18:05 +00:00
2014-03-20 18:03:22 +00:00
if self . indexerid == None :
2014-03-10 05:18:05 +00:00
sqlResults = myDB . select ( " SELECT show_name, indexer_id AS ' indexerid ' FROM scene_exceptions " )
scene_exceptions = { }
for row in sqlResults :
indexerid = row [ " indexerid " ]
if not indexerid in scene_exceptions :
scene_exceptions [ indexerid ] = [ ]
scene_exceptions [ indexerid ] . append ( row [ " show_name " ] )
else :
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select (
" SELECT show_name, indexer_id AS ' indexerid ' FROM scene_exceptions WHERE indexer_id = ? " ,
[ self . indexerid ] )
2014-03-10 05:18:05 +00:00
scene_exceptions = [ ]
for row in sqlResults :
scene_exceptions . append ( row [ " show_name " ] )
return _responds ( RESULT_SUCCESS , scene_exceptions )
class CMD_History ( ApiCall ) :
_help = { " desc " : " display sickbeard downloaded/snatched history " ,
" optionalParameters " : { " limit " : { " desc " : " limit returned results " } ,
" type " : { " desc " : " only show a specific type of results " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . limit , args = self . check_params ( args , kwargs , " limit " , 100 , False , " int " , [ ] )
self . type , args = self . check_params ( args , kwargs , " type " , None , False , " string " , [ " downloaded " , " snatched " ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display sickbeard downloaded/snatched history """
typeCodes = [ ]
if self . type == " downloaded " :
self . type = " Downloaded "
typeCodes = Quality . DOWNLOADED
elif self . type == " snatched " :
self . type = " Snatched "
typeCodes = Quality . SNATCHED
else :
typeCodes = Quality . SNATCHED + Quality . DOWNLOADED
myDB = db . DBConnection ( row_type = " dict " )
ulimit = min ( int ( self . limit ) , 100 )
if ulimit == 0 :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select (
" SELECT h.*, show_name FROM history h, tv_shows s WHERE h.showid=s.indexer_id AND action in ( " + ' , ' . join (
[ ' ? ' ] * len ( typeCodes ) ) + " ) ORDER BY date DESC " , typeCodes )
2014-03-10 05:18:05 +00:00
else :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select (
" SELECT h.*, show_name FROM history h, tv_shows s WHERE h.showid=s.indexer_id AND action in ( " + ' , ' . join (
[ ' ? ' ] * len ( typeCodes ) ) + " ) ORDER BY date DESC LIMIT ? " , typeCodes + [ ulimit ] )
2014-03-10 05:18:05 +00:00
results = [ ]
for row in sqlResults :
status , quality = Quality . splitCompositeStatus ( int ( row [ " action " ] ) )
status = _get_status_Strings ( status )
if self . type and not status == self . type :
continue
row [ " status " ] = status
row [ " quality " ] = _get_quality_string ( quality )
row [ " date " ] = _historyDate_to_dateTimeForm ( str ( row [ " date " ] ) )
del row [ " action " ]
_rename_element ( row , " showid " , " indexerid " )
row [ " resource_path " ] = os . path . dirname ( row [ " resource " ] )
row [ " resource " ] = os . path . basename ( row [ " resource " ] )
2014-05-02 11:01:42 +00:00
# Add tvdbid for backward compability
2014-04-29 07:37:27 +00:00
row [ ' tvdbid ' ] = row [ ' indexerid ' ]
2014-03-10 05:18:05 +00:00
results . append ( row )
return _responds ( RESULT_SUCCESS , results )
class CMD_HistoryClear ( ApiCall ) :
_help = { " desc " : " clear sickbeard ' s history " ,
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" clear sickbeard ' s history """
myDB = db . DBConnection ( )
myDB . action ( " DELETE FROM history WHERE 1=1 " )
return _responds ( RESULT_SUCCESS , msg = " History cleared " )
class CMD_HistoryTrim ( ApiCall ) :
_help = { " desc " : " trim sickbeard ' s history by removing entries greater than 30 days old "
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" trim sickbeard ' s history """
myDB = db . DBConnection ( )
2014-03-25 05:57:24 +00:00
myDB . action ( " DELETE FROM history WHERE date < " + str (
( datetime . datetime . today ( ) - datetime . timedelta ( days = 30 ) ) . strftime ( history . dateFormat ) ) )
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = " Removed history entries greater than 30 days old " )
class CMD_Logs ( ApiCall ) :
_help = { " desc " : " view sickbeard ' s log " ,
2014-03-25 05:57:24 +00:00
" optionalParameters " : { " min_level " : {
2014-07-12 01:38:22 +00:00
" desc " : " the minimum level classification of log entries to show, with each level inherting its above level " } }
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
2014-03-25 05:57:24 +00:00
self . min_level , args = self . check_params ( args , kwargs , " min_level " , " error " , False , " string " ,
[ " error " , " warning " , " info " , " debug " ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
""" view sickgear ' s log """
2014-03-10 05:18:05 +00:00
# 10 = Debug / 20 = Info / 30 = Warning / 40 = Error
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
min_level = logger . reverseNames [ str ( self . min_level ) . upper ( ) ]
max_lines = 50
2014-03-10 05:18:05 +00:00
regex = " ^( \ d \ d \ d \ d) \ -( \ d \ d) \ -( \ d \ d) \ s*( \ d \ d) \ :( \ d \ d):( \ d \ d) \ s*([A-Z]+) \ s*(.+?) \ s* \ : \ : \ s*(.*)$ "
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
final_data = [ ]
normal_data = [ ]
truncate = [ ]
repeated = None
num_lines = 0
2014-03-10 05:18:05 +00:00
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
if os . path . isfile ( logger . sb_log_instance . log_file_path ) :
for x in logger . sb_log_instance . reverse_readline ( logger . sb_log_instance . log_file_path ) :
x = x . decode ( ' utf-8 ' )
match = re . match ( regex , x )
if match :
level = match . group ( 7 )
if level not in logger . reverseNames :
normal_data = [ ]
continue
if logger . reverseNames [ level ] > = min_level :
if truncate and not normal_data and truncate [ 0 ] == match . group ( 8 ) + match . group ( 9 ) :
truncate + = [ match . group ( 8 ) + match . group ( 9 ) ]
repeated = x
continue
if 1 < len ( truncate ) :
final_data [ - 1 ] = repeated . strip ( ) + ' (... %s repeat lines) \n ' % len ( truncate )
truncate = [ match . group ( 8 ) + match . group ( 9 ) ]
final_data . append ( x )
if any ( normal_data ) :
final_data + = [ ' %02s ) %s ' % ( n + 1 , x ) for n , x in enumerate ( normal_data [ : : - 1 ] ) ] + \
[ ' <br /> ' ]
num_lines + = len ( normal_data )
normal_data = [ ]
else :
normal_data = [ ]
continue
2014-03-10 05:18:05 +00:00
else :
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
if not any ( normal_data ) and not any ( [ x . strip ( ) ] ) :
continue
2014-03-10 05:18:05 +00:00
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
normal_data . append ( re . sub ( r ' \ r? \ n ' , ' <br /> ' , x . replace ( ' < ' , ' < ' ) . replace ( ' > ' , ' > ' ) ) )
2014-03-10 05:18:05 +00:00
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
num_lines + = 1
2014-03-10 05:18:05 +00:00
Add fanart to Episodes View, Display Show, Edit Show, and Media Renamer page.
Add "Maximum fanart image files per show to cache" to config General/Interface.
Add populate images when the daily show updater is run with a default maximum 3 images per show.
Change force full update in a show will replace existing images with new.
Add fanart livepanel to lower right of Episodes View and Display Show page.
Add highlight panel red until button is clicked a few times.
Add flick through multiple background images on Episodes View and Display Show page.
Add persistent move poster image to right hand side or hide on Display Show page (multi-click the eye).
Add persistent translucency of background images on Episodes View and Display Show page.
Add persistent fanart rating to avoid art completely, random display, random from a group, or display fave always.
Add persistent views of the show detail on Display Show page.
Add persistent views on Episodes View.
Add persistent button to collapse and expand card images on Episode View/Layout daybyday.
Add non persistent "Open gear" and "Full fanart" image views to Episodes View and Display Show page.
Add "smart" selection of fanart image to display on Episode view.
Change insert [!] and change text shade of ended shows in drop down show list on Display Show page.
Change button graphic for next and previous show of show list on Display Show page.
Add logic to hide some livepanel buttons until artwork becomes available or in other circumstances.
Add "(Ended)" where appropriate to show title on Display Show page.
Add links to fanart.tv where appropriate on Display Show page.
Change use tense for label "Airs" or "Aired" depending on if show ended.
Change display "No files" instead of "0 files" and "Upgrade once" instead of "End upgrade on first match".
Add persistent button to newest season to "Show all" episodes.
Add persistent button to all shown seasons to "Hide most" episodes.
Add button to older seasons to toggle "Show Season n" or "Show Specials" with "Hide..." episodes.
Add season level status counts next to each season header on display show page
Add sorting to season table headers on display show page
Add filename and size to quality badge on display show page, removed its redundant "downloaded" text
Remove redundant "Add show" buttons
Change combine the NFO and TBN columns into a single Meta column
Change reduce screen estate used by episode numbers columns
Change improve clarity of text on Add Show page.
Add "Reset fanart ratings" to show Edit/Other tab.
Add fanart usage to show Edit/Other tab.
Add fanart keys guide to show Edit/Other tab.
Change add placeholder tip to "Alternative release name(s)" on show Edit.
Change add placeholder tip to search box on shows Search.
Change hide Anime tips on show Edit when selecting its mutually exclusive options.
Change label "End upgrade on first match" to "Upgrade once" on show Edit.
Change improve performance rendering displayShow.
Add total episodes to start of show description (excludes specials if those are hidden).
Add "Add show" actions i.e. "Search", "Trakt cards", "IMDb cards", and "Anime" to Shows menu.
Add "Import (existing)" action to Tools menu.
Change SD quality from red to dark green, 2160p UHD 4K is red.
Change relocate the functions of Logs & Errors to the right side Tools menu -> View Log File.
Add warning indicator to the Tools menu in different colour depending on error count (green through red).
Change View Log error item output from reversed to natural order.
Change View Log add a typeface and some colour to improve readability.
Change View Log/Errors only display "Clear Errors" button when there are errors to clear.
Change improve performance of View Log File.
2016-02-28 23:43:40 +00:00
if num_lines > = max_lines :
break
return _responds ( RESULT_SUCCESS , final_data )
2014-03-10 05:18:05 +00:00
2014-09-30 23:13:04 +00:00
class CMD_PostProcess ( ApiCall ) :
_help = { " desc " : " Manual postprocess TV Download Dir " ,
" optionalParameters " : { " path " : { " desc " : " Post process this folder " } ,
" force_replace " : { " desc " : " Force already Post Processed Dir/Files " } ,
" return_data " : { " desc " : " Returns result for the process " } ,
" process_method " : { " desc " : " Symlink, hardlink, move or copy the file " } ,
" is_priority " : { " desc " : " Replace the file even if it exists in a higher quality) " } ,
" type " : { " desc " : " What type of postprocess request is this, auto of manual " }
}
}
def __init__ ( self , handler , args , kwargs ) :
# required
# optional
self . path , args = self . check_params ( args , kwargs , " path " , None , False , " string " , [ ] )
self . force_replace , args = self . check_params ( args , kwargs , " force_replace " , 0 , False , " bool " , [ ] )
self . return_data , args = self . check_params ( args , kwargs , " return_data " , 0 , False , " bool " , [ ] )
self . process_method , args = self . check_params ( args , kwargs , " process_method " , False , False , " string " , [ " copy " , " symlink " , " hardlink " , " move " ] )
self . is_priority , args = self . check_params ( args , kwargs , " is_priority " , 0 , False , " bool " , [ ] )
self . type , args = self . check_params ( args , kwargs , " type " , " auto " , None , " string " , [ " auto " , " manual " ] )
# super, missing, help
ApiCall . __init__ ( self , handler , args , kwargs )
def run ( self ) :
""" Starts the postprocess """
if not self . path and not sickbeard . TV_DOWNLOAD_DIR :
return _responds ( RESULT_FAILURE , msg = " You need to provide a path or set TV Download Dir " )
if not self . path :
self . path = sickbeard . TV_DOWNLOAD_DIR
if not self . type :
self . type = ' manual '
2015-05-08 02:46:54 +00:00
data = processTV . processDir ( self . path , process_method = self . process_method , force = self . force_replace , force_replace = self . is_priority , failed = False , type = self . type )
2014-09-30 23:13:04 +00:00
if not self . return_data :
data = " "
return _responds ( RESULT_SUCCESS , data = data , msg = " Started postprocess for %s " % self . path )
2014-03-10 05:18:05 +00:00
class CMD_SickBeard ( ApiCall ) :
_help = { " desc " : " display misc sickbeard related information " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display misc sickbeard related information """
2014-07-29 00:11:16 +00:00
data = { " sb_version " : sickbeard . BRANCH , " api_version " : Api . version ,
2014-03-25 05:57:24 +00:00
" api_commands " : sorted ( _functionMaper . keys ( ) ) }
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , data )
class CMD_SickBeardAddRootDir ( ApiCall ) :
_help = { " desc " : " add a sickbeard user ' s parent directory " ,
" requiredParameters " : { " location " : { " desc " : " the full path to root (parent) directory " }
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " default " : { " desc " : " make the location passed the default root (parent) directory " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . location , args = self . check_params ( args , kwargs , " location " , None , True , " string " , [ ] )
# optional
self . default , args = self . check_params ( args , kwargs , " default " , 0 , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" add a parent directory to sickbeard ' s config """
self . location = urllib . unquote_plus ( self . location )
location_matched = 0
2014-06-15 21:45:09 +00:00
index = 0
2014-03-10 05:18:05 +00:00
# dissallow adding/setting an invalid dir
if not ek . ek ( os . path . isdir , self . location ) :
return _responds ( RESULT_FAILURE , msg = " Location is invalid " )
root_dirs = [ ]
if sickbeard . ROOT_DIRS == " " :
self . default = 1
else :
root_dirs = sickbeard . ROOT_DIRS . split ( ' | ' )
index = int ( sickbeard . ROOT_DIRS . split ( ' | ' ) [ 0 ] )
root_dirs . pop ( 0 )
# clean up the list - replace %xx escapes by their single-character equivalent
root_dirs = [ urllib . unquote_plus ( x ) for x in root_dirs ]
for x in root_dirs :
2014-03-25 05:57:24 +00:00
if ( x == self . location ) :
2014-03-10 05:18:05 +00:00
location_matched = 1
if ( self . default == 1 ) :
index = root_dirs . index ( self . location )
break
2014-03-25 05:57:24 +00:00
if ( location_matched == 0 ) :
2014-03-10 05:18:05 +00:00
if ( self . default == 1 ) :
root_dirs . insert ( 0 , self . location )
else :
root_dirs . append ( self . location )
root_dirs_new = [ urllib . unquote_plus ( x ) for x in root_dirs ]
root_dirs_new . insert ( 0 , index )
root_dirs_new = ' | ' . join ( unicode ( x ) for x in root_dirs_new )
sickbeard . ROOT_DIRS = root_dirs_new
return _responds ( RESULT_SUCCESS , _getRootDirs ( ) , msg = " Root directories updated " )
class CMD_SickBeardCheckScheduler ( ApiCall ) :
_help = { " desc " : " query the scheduler " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" query the scheduler """
myDB = db . DBConnection ( )
sqlResults = myDB . select ( " SELECT last_backlog FROM info " )
2014-03-25 05:57:24 +00:00
backlogPaused = sickbeard . searchQueueScheduler . action . is_backlog_paused ( ) #@UndefinedVariable
backlogRunning = sickbeard . searchQueueScheduler . action . is_backlog_in_progress ( ) #@UndefinedVariable
2016-09-04 20:00:44 +00:00
nextBacklog = sickbeard . backlogSearchScheduler . next_run ( ) . strftime ( dateFormat ) . decode ( sickbeard . SYS_ENCODING )
2014-03-10 05:18:05 +00:00
2014-03-25 05:57:24 +00:00
data = { " backlog_is_paused " : int ( backlogPaused ) , " backlog_is_running " : int ( backlogRunning ) ,
" last_backlog " : _ordinal_to_dateForm ( sqlResults [ 0 ] [ " last_backlog " ] ) ,
2014-05-11 12:49:07 +00:00
" next_backlog " : nextBacklog }
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , data )
class CMD_SickBeardDeleteRootDir ( ApiCall ) :
_help = { " desc " : " delete a sickbeard user ' s parent directory " ,
2014-03-25 05:57:24 +00:00
" requiredParameters " : { " location " : { " desc " : " the full path to root (parent) directory " } }
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . location , args = self . check_params ( args , kwargs , " location " , None , True , " string " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" delete a parent directory from sickbeard ' s config """
if sickbeard . ROOT_DIRS == " " :
return _responds ( RESULT_FAILURE , _getRootDirs ( ) , msg = " No root directories detected " )
2014-06-15 21:45:09 +00:00
newIndex = 0
2014-03-10 05:18:05 +00:00
root_dirs_new = [ ]
root_dirs = sickbeard . ROOT_DIRS . split ( ' | ' )
index = int ( root_dirs [ 0 ] )
root_dirs . pop ( 0 )
# clean up the list - replace %xx escapes by their single-character equivalent
root_dirs = [ urllib . unquote_plus ( x ) for x in root_dirs ]
old_root_dir = root_dirs [ index ]
for curRootDir in root_dirs :
if not curRootDir == self . location :
root_dirs_new . append ( curRootDir )
else :
newIndex = 0
for curIndex , curNewRootDir in enumerate ( root_dirs_new ) :
if curNewRootDir is old_root_dir :
newIndex = curIndex
break
root_dirs_new = [ urllib . unquote_plus ( x ) for x in root_dirs_new ]
if len ( root_dirs_new ) > 0 :
root_dirs_new . insert ( 0 , newIndex )
root_dirs_new = " | " . join ( unicode ( x ) for x in root_dirs_new )
sickbeard . ROOT_DIRS = root_dirs_new
# what if the root dir was not found?
return _responds ( RESULT_SUCCESS , _getRootDirs ( ) , msg = " Root directory deleted " )
2016-02-02 12:22:58 +00:00
class CMD_SickBeardForceSearch ( ApiCall ) :
_help = { ' desc ' : ' force the episode recent search early ' }
def __init__ ( self , handler , args , kwargs ) :
# required
# optional
# super, missing, help
ApiCall . __init__ ( self , handler , args , kwargs )
def run ( self ) :
""" force the episode search early """
# Searching all providers for any needed episodes
result = sickbeard . recentSearchScheduler . forceRun ( )
if result :
return _responds ( RESULT_SUCCESS , msg = ' Episode recent search successfully forced ' )
return _responds ( RESULT_FAILURE , msg = ' Can not force the episode recent search because it \' s already active ' )
2014-03-10 05:18:05 +00:00
class CMD_SickBeardGetDefaults ( ApiCall ) :
_help = { " desc " : " get sickbeard user defaults " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" get sickbeard user defaults """
anyQualities , bestQualities = _mapQuality ( sickbeard . QUALITY_DEFAULT )
2014-03-25 05:57:24 +00:00
data = { " status " : statusStrings [ sickbeard . STATUS_DEFAULT ] . lower ( ) ,
" flatten_folders " : int ( sickbeard . FLATTEN_FOLDERS_DEFAULT ) , " initial " : anyQualities ,
2014-12-16 14:14:54 +00:00
" archive " : bestQualities , " future_show_paused " : int ( sickbeard . EPISODE_VIEW_DISPLAY_PAUSED ) }
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , data )
class CMD_SickBeardGetMessages ( ApiCall ) :
_help = { " desc " : " get all messages " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
messages = [ ]
2014-06-30 03:58:34 +00:00
for cur_notification in ui . notifications . get_notifications ( self . handler . request . remote_ip ) :
2014-03-10 05:18:05 +00:00
messages . append ( { " title " : cur_notification . title ,
2014-03-25 05:57:24 +00:00
" message " : cur_notification . message ,
" type " : cur_notification . type } )
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , messages )
class CMD_SickBeardGetRootDirs ( ApiCall ) :
_help = { " desc " : " get sickbeard user parent directories " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" get the parent directories defined in sickbeard ' s config """
return _responds ( RESULT_SUCCESS , _getRootDirs ( ) )
class CMD_SickBeardPauseBacklog ( ApiCall ) :
_help = { " desc " : " pause the backlog search " ,
2014-03-25 05:57:24 +00:00
" optionalParameters " : { " pause " : { " desc " : " pause or unpause the global backlog " } }
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . pause , args = self . check_params ( args , kwargs , " pause " , 0 , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" pause the backlog search """
2014-05-02 11:33:06 +00:00
if self . pause :
2014-03-25 05:57:24 +00:00
sickbeard . searchQueueScheduler . action . pause_backlog ( ) #@UndefinedVariable
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = " Backlog paused " )
else :
2014-03-25 05:57:24 +00:00
sickbeard . searchQueueScheduler . action . unpause_backlog ( ) #@UndefinedVariable
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = " Backlog unpaused " )
class CMD_SickBeardPing ( ApiCall ) :
_help = { " desc " : " check to see if sickbeard is running " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" check to see if sickbeard is running """
2014-06-19 15:47:28 +00:00
self . handler . set_header ( ' Cache-Control ' , " max-age=0,no-cache,no-store " )
2014-03-10 05:18:05 +00:00
if sickbeard . started :
return _responds ( RESULT_SUCCESS , { " pid " : sickbeard . PID } , " Pong " )
else :
return _responds ( RESULT_SUCCESS , msg = " Pong " )
class CMD_SickBeardRestart ( ApiCall ) :
_help = { " desc " : " restart sickbeard " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" restart sickbeard """
2014-07-08 22:17:34 +00:00
sickbeard . events . put ( sickbeard . events . SystemEvent . RESTART )
2014-11-12 16:43:14 +00:00
return _responds ( RESULT_SUCCESS , msg = " SickGear is restarting... " )
2014-03-10 05:18:05 +00:00
2014-04-24 03:22:21 +00:00
class CMD_SickBeardSearchIndexers ( ApiCall ) :
_help = { " desc " : " search for show on the indexers with a given string and language " ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " name " : { " desc " : " name of the show you want to search for " } ,
2014-07-27 10:59:21 +00:00
" indexerid " : { " desc " : " thetvdb.com or tvrage.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
" lang " : { " desc " : " the 2 letter abbreviation lang id " }
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
valid_languages = {
2014-03-25 05:57:24 +00:00
' el ' : 20 , ' en ' : 7 , ' zh ' : 27 , ' it ' : 15 , ' cs ' : 28 , ' es ' : 16 , ' ru ' : 22 ,
' nl ' : 13 , ' pt ' : 26 , ' no ' : 9 , ' tr ' : 21 , ' pl ' : 18 , ' fr ' : 17 , ' hr ' : 31 ,
' de ' : 14 , ' da ' : 10 , ' fi ' : 11 , ' hu ' : 19 , ' ja ' : 25 , ' he ' : 24 , ' ko ' : 32 ,
' sv ' : 8 , ' sl ' : 30 }
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . name , args = self . check_params ( args , kwargs , " name " , None , False , " string " , [ ] )
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , False , " int " , [ ] )
self . lang , args = self . check_params ( args , kwargs , " lang " , " en " , False , " string " , self . valid_languages . keys ( ) )
2014-11-06 15:22:10 +00:00
self . indexer , args = self . check_params ( args , kwargs , " indexer " , 1 , False , " int " , [ ] )
2014-03-12 05:28:30 +00:00
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" search for show at tvdb with a given string and language """
2014-03-25 05:57:24 +00:00
if self . name and not self . indexerid : # only name was given
2014-07-27 10:59:21 +00:00
lINDEXER_API_PARMS = sickbeard . indexerApi ( self . indexer ) . api_params . copy ( )
lINDEXER_API_PARMS [ ' language ' ] = self . lang
lINDEXER_API_PARMS [ ' custom_ui ' ] = classes . AllShowsListUI
t = sickbeard . indexerApi ( self . indexer ) . indexer ( * * lINDEXER_API_PARMS )
2014-03-10 05:18:05 +00:00
2014-07-27 10:59:21 +00:00
apiData = None
try :
apiData = t [ str ( self . name ) . encode ( ) ]
2015-06-08 12:47:01 +00:00
except Exception as e :
2014-07-27 10:59:21 +00:00
pass
if not apiData :
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Did not get result from tvdb " )
2014-07-27 10:59:21 +00:00
results = [ ]
for curSeries in apiData :
2014-11-06 15:22:10 +00:00
results . append ( { " indexerid " : int ( curSeries [ ' id ' ] ) ,
" tvdbid " : int ( curSeries [ ' id ' ] ) ,
" name " : curSeries [ ' seriesname ' ] ,
" first_aired " : curSeries [ ' firstaired ' ] ,
" indexer " : self . indexer } )
2014-07-27 10:59:21 +00:00
lang_id = self . valid_languages [ self . lang ]
return _responds ( RESULT_SUCCESS , { " results " : results , " langid " : lang_id } )
2014-03-10 05:18:05 +00:00
elif self . indexerid :
2014-03-26 19:28:46 +00:00
lINDEXER_API_PARMS = sickbeard . indexerApi ( self . indexer ) . api_params . copy ( )
2014-03-10 05:18:05 +00:00
lang_id = self . valid_languages [ self . lang ]
if self . lang and not self . lang == ' en ' :
lINDEXER_API_PARMS [ ' language ' ] = self . lang
2014-03-26 19:28:46 +00:00
lINDEXER_API_PARMS [ ' actors ' ] = False
t = sickbeard . indexerApi ( self . indexer ) . indexer ( * * lINDEXER_API_PARMS )
2014-03-10 05:18:05 +00:00
try :
myShow = t [ int ( self . indexerid ) ]
2014-03-25 05:57:24 +00:00
except ( sickbeard . indexer_shownotfound , sickbeard . indexer_error ) :
2014-03-10 05:18:05 +00:00
logger . log ( u " API :: Unable to find show with id " + str ( self . indexerid ) , logger . WARNING )
return _responds ( RESULT_SUCCESS , { " results " : [ ] , " langid " : lang_id } )
if not myShow . data [ ' seriesname ' ] :
2014-03-25 05:57:24 +00:00
logger . log (
u " API :: Found show with indexerid " + str ( self . indexerid ) + " , however it contained no show name " ,
logger . DEBUG )
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Show contains no name, invalid result " )
showOut = [ { " indexerid " : self . indexerid ,
2014-07-12 01:38:22 +00:00
" tvdbid " : self . indexerid ,
2014-03-25 05:57:24 +00:00
" name " : unicode ( myShow . data [ ' seriesname ' ] ) ,
" first_aired " : myShow . data [ ' firstaired ' ] } ]
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , { " results " : showOut , " langid " : lang_id } )
else :
return _responds ( RESULT_FAILURE , msg = " Either indexerid or name is required " )
class CMD_SickBeardSetDefaults ( ApiCall ) :
_help = { " desc " : " set sickbeard user defaults " ,
" optionalParameters " : { " initial " : { " desc " : " initial quality for the show " } ,
" archive " : { " desc " : " archive quality for the show " } ,
" flatten_folders " : { " desc " : " flatten subfolders within the show directory " } ,
" status " : { " desc " : " status of missing episodes " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
2014-03-25 05:57:24 +00:00
self . initial , args = self . check_params ( args , kwargs , " initial " , None , False , " list " ,
[ " sdtv " , " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " ,
" fullhdwebdl " , " hdbluray " , " fullhdbluray " , " unknown " ] )
self . archive , args = self . check_params ( args , kwargs , " archive " , None , False , " list " ,
[ " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " , " fullhdwebdl " ,
" hdbluray " , " fullhdbluray " ] )
2014-03-10 05:18:05 +00:00
self . future_show_paused , args = self . check_params ( args , kwargs , " future_show_paused " , None , False , " bool " , [ ] )
self . flatten_folders , args = self . check_params ( args , kwargs , " flatten_folders " , None , False , " bool " , [ ] )
2014-03-25 05:57:24 +00:00
self . status , args = self . check_params ( args , kwargs , " status " , None , False , " string " ,
[ " wanted " , " skipped " , " archived " , " ignored " ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" set sickbeard user defaults """
quality_map = { ' sdtv ' : Quality . SDTV ,
' sddvd ' : Quality . SDDVD ,
' hdtv ' : Quality . HDTV ,
' rawhdtv ' : Quality . RAWHDTV ,
' fullhdtv ' : Quality . FULLHDTV ,
' hdwebdl ' : Quality . HDWEBDL ,
' fullhdwebdl ' : Quality . FULLHDWEBDL ,
' hdbluray ' : Quality . HDBLURAY ,
' fullhdbluray ' : Quality . FULLHDBLURAY ,
' unknown ' : Quality . UNKNOWN }
iqualityID = [ ]
aqualityID = [ ]
if self . initial :
for quality in self . initial :
iqualityID . append ( quality_map [ quality ] )
if self . archive :
for quality in self . archive :
aqualityID . append ( quality_map [ quality ] )
if iqualityID or aqualityID :
sickbeard . QUALITY_DEFAULT = Quality . combineQualities ( iqualityID , aqualityID )
if self . status :
# convert the string status to a int
for status in statusStrings . statusStrings :
if statusStrings [ status ] . lower ( ) == str ( self . status ) . lower ( ) :
self . status = status
break
# this should be obsolete bcause of the above
if not self . status in statusStrings . statusStrings :
raise ApiError ( " Invalid Status " )
#only allow the status options we want
if int ( self . status ) not in ( 3 , 5 , 6 , 7 ) :
raise ApiError ( " Status Prohibited " )
sickbeard . STATUS_DEFAULT = self . status
2014-03-20 18:03:22 +00:00
if self . flatten_folders != None :
2014-03-10 05:18:05 +00:00
sickbeard . FLATTEN_FOLDERS_DEFAULT = int ( self . flatten_folders )
2014-03-20 18:03:22 +00:00
if self . future_show_paused != None :
2014-12-16 14:14:54 +00:00
sickbeard . EPISODE_VIEW_DISPLAY_PAUSED = int ( self . future_show_paused )
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = " Saved defaults " )
class CMD_SickBeardShutdown ( ApiCall ) :
_help = { " desc " : " shutdown sickbeard " }
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" shutdown sickbeard """
2014-07-08 22:17:34 +00:00
sickbeard . events . put ( sickbeard . events . SystemEvent . SHUTDOWN )
2014-11-12 16:43:14 +00:00
return _responds ( RESULT_SUCCESS , msg = " SickGear is shutting down... " )
2014-03-10 05:18:05 +00:00
class CMD_Show ( ApiCall ) :
_help = { " desc " : " display information for a given show " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display information for a given show """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
showDict = { }
2014-06-19 22:46:41 +00:00
showDict [ " season_list " ] = CMD_ShowSeasonList ( self . handler , ( ) , { " indexerid " : self . indexerid } ) . run ( ) [ " data " ]
showDict [ " cache " ] = CMD_ShowCache ( self . handler , ( ) , { " indexerid " : self . indexerid } ) . run ( ) [ " data " ]
2014-03-10 05:18:05 +00:00
genreList = [ ]
if showObj . genre :
genreListTmp = showObj . genre . split ( " | " )
for genre in genreListTmp :
if genre :
genreList . append ( genre )
showDict [ " genre " ] = genreList
showDict [ " quality " ] = _get_quality_string ( showObj . quality )
anyQualities , bestQualities = _mapQuality ( showObj . quality )
showDict [ " quality_details " ] = { " initial " : anyQualities , " archive " : bestQualities }
try :
showDict [ " location " ] = showObj . location
except sickbeard . exceptions . ShowDirNotFoundException :
showDict [ " location " ] = " "
showDict [ " language " ] = showObj . lang
showDict [ " show_name " ] = showObj . name
showDict [ " paused " ] = showObj . paused
2014-10-30 21:24:08 +00:00
showDict [ " subtitles " ] = showObj . subtitles
2014-03-10 05:18:05 +00:00
showDict [ " air_by_date " ] = showObj . air_by_date
showDict [ " flatten_folders " ] = showObj . flatten_folders
2014-07-12 01:38:22 +00:00
showDict [ " sports " ] = showObj . sports
showDict [ " anime " ] = showObj . anime
2014-03-10 05:18:05 +00:00
#clean up tvdb horrible airs field
showDict [ " airs " ] = str ( showObj . airs ) . replace ( ' am ' , ' AM ' ) . replace ( ' pm ' , ' PM ' ) . replace ( ' ' , ' ' )
2014-08-26 14:44:15 +00:00
showDict [ " indexerid " ] = self . indexerid
2016-09-04 20:00:44 +00:00
showDict [ " tvrage_id " ] = showObj . ids . get ( INDEXER_TVRAGE , { ' id ' : 0 } ) [ ' id ' ]
2014-07-12 01:38:22 +00:00
showDict [ " tvrage_name " ] = showObj . name
2014-03-10 05:18:05 +00:00
showDict [ " network " ] = showObj . network
if not showDict [ " network " ] :
showDict [ " network " ] = " "
showDict [ " status " ] = showObj . status
2014-11-02 03:55:29 +00:00
if showObj . nextaired :
2014-11-05 11:33:05 +00:00
dtEpisodeAirs = sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( showObj . nextaired , showDict [ ' airs ' ] , showDict [ ' network ' ] ) )
2014-11-02 03:55:29 +00:00
showDict [ ' airs ' ] = sbdatetime . sbdatetime . sbftime ( dtEpisodeAirs , t_preset = timeFormat ) . lstrip ( ' 0 ' ) . replace ( ' 0 ' , ' ' )
showDict [ ' next_ep_airdate ' ] = sbdatetime . sbdatetime . sbfdate ( dtEpisodeAirs , d_preset = dateFormat )
2014-07-09 06:07:26 +00:00
else :
2014-11-02 03:55:29 +00:00
showDict [ ' next_ep_airdate ' ] = ' '
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , showDict )
class CMD_ShowAddExisting ( ApiCall ) :
_help = { " desc " : " add a show in sickbeard with an existing folder " ,
2014-04-25 23:14:24 +00:00
" requiredParameters " : { " tvdbid or tvrageid " : { " desc " : " thetvdb.com or tvrage.com id " } ,
2014-03-10 05:18:05 +00:00
" location " : { " desc " : " full path to the existing folder for the show " }
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " initial " : { " desc " : " initial quality for the show " } ,
" archive " : { " desc " : " archive quality for the show " } ,
" flatten_folders " : { " desc " : " flatten subfolders for the show " } ,
" subtitles " : { " desc " : " allow search episode subtitle " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-04-25 23:14:24 +00:00
if " tvdbid " in args or " tvdbid " in kwargs :
_INDEXER_INT = 1
_INDEXER = " tvdbid "
elif " tvrageid " in args or " tvrageid " in kwargs :
_INDEXER_INT = 2
_INDEXER = " tvrageid "
2014-04-25 23:39:53 +00:00
else :
_INDEXER_INT = None
2014-06-19 15:47:28 +00:00
_INDEXER = None
2014-03-10 05:18:05 +00:00
# required
2014-04-25 23:14:24 +00:00
self . indexerid , args = self . check_params ( args , kwargs , _INDEXER , None , True , " int " , [ ] )
self . indexer = _INDEXER_INT
2014-03-10 05:18:05 +00:00
self . location , args = self . check_params ( args , kwargs , " location " , None , True , " string " , [ ] )
# optional
2014-03-25 05:57:24 +00:00
self . initial , args = self . check_params ( args , kwargs , " initial " , None , False , " list " ,
[ " sdtv " , " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " ,
" fullhdwebdl " , " hdbluray " , " fullhdbluray " , " unknown " ] )
self . archive , args = self . check_params ( args , kwargs , " archive " , None , False , " list " ,
[ " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " , " fullhdwebdl " ,
" hdbluray " , " fullhdbluray " ] )
self . flatten_folders , args = self . check_params ( args , kwargs , " flatten_folders " ,
str ( sickbeard . FLATTEN_FOLDERS_DEFAULT ) , False , " bool " , [ ] )
self . subtitles , args = self . check_params ( args , kwargs , " subtitles " , int ( sickbeard . USE_SUBTITLES ) , False , " int " ,
[ ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" add a show in sickbeard with an existing folder """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if showObj :
return _responds ( RESULT_FAILURE , msg = " An existing indexerid already exists in the database " )
if not ek . ek ( os . path . isdir , self . location ) :
return _responds ( RESULT_FAILURE , msg = ' Not a valid location ' )
2014-04-24 03:22:21 +00:00
indexerName = None
2014-07-12 01:38:22 +00:00
indexerResult = CMD_SickBeardSearchIndexers ( self . handler , [ ] ,
{ " indexerid " : self . indexerid , " indexer " : self . indexer } ) . run ( )
2014-03-10 05:18:05 +00:00
2014-04-24 03:22:21 +00:00
if indexerResult [ ' result ' ] == result_type_map [ RESULT_SUCCESS ] :
if not indexerResult [ ' data ' ] [ ' results ' ] :
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Empty results returned, check indexerid and try again " )
2014-04-24 03:22:21 +00:00
if len ( indexerResult [ ' data ' ] [ ' results ' ] ) == 1 and ' name ' in indexerResult [ ' data ' ] [ ' results ' ] [ 0 ] :
indexerName = indexerResult [ ' data ' ] [ ' results ' ] [ 0 ] [ ' name ' ]
2014-03-10 05:18:05 +00:00
2014-04-24 03:22:21 +00:00
if not indexerName :
return _responds ( RESULT_FAILURE , msg = " Unable to retrieve information from indexer " )
2014-03-10 05:18:05 +00:00
quality_map = { ' sdtv ' : Quality . SDTV ,
' sddvd ' : Quality . SDDVD ,
' hdtv ' : Quality . HDTV ,
' rawhdtv ' : Quality . RAWHDTV ,
' fullhdtv ' : Quality . FULLHDTV ,
' hdwebdl ' : Quality . HDWEBDL ,
' fullhdwebdl ' : Quality . FULLHDWEBDL ,
' hdbluray ' : Quality . HDBLURAY ,
' fullhdbluray ' : Quality . FULLHDBLURAY ,
' unknown ' : Quality . UNKNOWN }
#use default quality as a failsafe
newQuality = int ( sickbeard . QUALITY_DEFAULT )
iqualityID = [ ]
aqualityID = [ ]
if self . initial :
for quality in self . initial :
iqualityID . append ( quality_map [ quality ] )
if self . archive :
for quality in self . archive :
aqualityID . append ( quality_map [ quality ] )
if iqualityID or aqualityID :
newQuality = Quality . combineQualities ( iqualityID , aqualityID )
2014-07-11 17:42:42 +00:00
sickbeard . showQueueScheduler . action . addShow ( int ( self . indexer ) , int ( self . indexerid ) , self . location , SKIPPED ,
newQuality , int ( self . flatten_folders ) )
2014-04-24 03:22:21 +00:00
return _responds ( RESULT_SUCCESS , { " name " : indexerName } , indexerName + " has been queued to be added " )
2014-03-10 05:18:05 +00:00
class CMD_ShowAddNew ( ApiCall ) :
_help = { " desc " : " add a new show to sickbeard " ,
2014-04-25 23:14:24 +00:00
" requiredParameters " : { " tvdbid or tvrageid " : { " desc " : " thetvdb.com or tvrage.com id " }
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " initial " : { " desc " : " initial quality for the show " } ,
" location " : { " desc " : " base path for where the show folder is to be created " } ,
" archive " : { " desc " : " archive quality for the show " } ,
" flatten_folders " : { " desc " : " flatten subfolders for the show " } ,
" status " : { " desc " : " status of missing episodes " } ,
" lang " : { " desc " : " the 2 letter lang abbreviation id " } ,
2014-07-11 17:45:45 +00:00
" subtitles " : { " desc " : " allow search episode subtitle " } ,
" anime " : { " desc " : " set show to anime " } ,
" scene " : { " desc " : " show searches episodes by scene numbering " }
2014-03-10 05:18:05 +00:00
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
valid_languages = {
2014-03-25 05:57:24 +00:00
' el ' : 20 , ' en ' : 7 , ' zh ' : 27 , ' it ' : 15 , ' cs ' : 28 , ' es ' : 16 , ' ru ' : 22 ,
' nl ' : 13 , ' pt ' : 26 , ' no ' : 9 , ' tr ' : 21 , ' pl ' : 18 , ' fr ' : 17 , ' hr ' : 31 ,
' de ' : 14 , ' da ' : 10 , ' fi ' : 11 , ' hu ' : 19 , ' ja ' : 25 , ' he ' : 24 , ' ko ' : 32 ,
' sv ' : 8 , ' sl ' : 30 }
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-04-25 23:14:24 +00:00
if " tvdbid " in args or " tvdbid " in kwargs :
_INDEXER_INT = 1
_INDEXER = " tvdbid "
elif " tvrageid " in args or " tvrageid " in kwargs :
_INDEXER_INT = 2
_INDEXER = " tvrageid "
2014-04-25 23:39:53 +00:00
else :
_INDEXER_INT = None
2014-06-19 15:47:28 +00:00
_INDEXER = None
2014-03-10 05:18:05 +00:00
# required
2014-04-25 23:14:24 +00:00
self . indexerid , args = self . check_params ( args , kwargs , _INDEXER , None , True , " int " , [ ] )
self . indexer = _INDEXER_INT
2014-03-10 05:18:05 +00:00
# optional
self . location , args = self . check_params ( args , kwargs , " location " , None , False , " string " , [ ] )
2014-03-25 05:57:24 +00:00
self . initial , args = self . check_params ( args , kwargs , " initial " , None , False , " list " ,
[ " sdtv " , " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " ,
" fullhdwebdl " , " hdbluray " , " fullhdbluray " , " unknown " ] )
self . archive , args = self . check_params ( args , kwargs , " archive " , None , False , " list " ,
[ " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " , " fullhdwebdl " ,
" hdbluray " , " fullhdbluray " ] )
self . flatten_folders , args = self . check_params ( args , kwargs , " flatten_folders " ,
str ( sickbeard . FLATTEN_FOLDERS_DEFAULT ) , False , " bool " , [ ] )
self . status , args = self . check_params ( args , kwargs , " status " , None , False , " string " ,
[ " wanted " , " skipped " , " archived " , " ignored " ] )
2014-03-10 05:18:05 +00:00
self . lang , args = self . check_params ( args , kwargs , " lang " , " en " , False , " string " , self . valid_languages . keys ( ) )
2014-03-25 05:57:24 +00:00
self . subtitles , args = self . check_params ( args , kwargs , " subtitles " , int ( sickbeard . USE_SUBTITLES ) , False , " int " ,
[ ] )
2014-07-11 17:42:42 +00:00
self . anime , args = self . check_params ( args , kwargs , " anime " , int ( sickbeard . ANIME_DEFAULT ) , False , " int " ,
[ ] )
self . scene , args = self . check_params ( args , kwargs , " scene " , int ( sickbeard . SCENE_DEFAULT ) , False , " int " ,
[ ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" add a show in sickbeard with an existing folder """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if showObj :
return _responds ( RESULT_FAILURE , msg = " An existing indexerid already exists in database " )
if not self . location :
if sickbeard . ROOT_DIRS != " " :
root_dirs = sickbeard . ROOT_DIRS . split ( ' | ' )
root_dirs . pop ( 0 )
default_index = int ( sickbeard . ROOT_DIRS . split ( ' | ' ) [ 0 ] )
self . location = root_dirs [ default_index ]
else :
return _responds ( RESULT_FAILURE , msg = " Root directory is not set, please provide a location " )
if not ek . ek ( os . path . isdir , self . location ) :
return _responds ( RESULT_FAILURE , msg = " ' " + self . location + " ' is not a valid location " )
quality_map = { ' sdtv ' : Quality . SDTV ,
' sddvd ' : Quality . SDDVD ,
' hdtv ' : Quality . HDTV ,
' rawhdtv ' : Quality . RAWHDTV ,
' fullhdtv ' : Quality . FULLHDTV ,
' hdwebdl ' : Quality . HDWEBDL ,
' fullhdwebdl ' : Quality . FULLHDWEBDL ,
' hdbluray ' : Quality . HDBLURAY ,
' fullhdbluray ' : Quality . FULLHDBLURAY ,
' unknown ' : Quality . UNKNOWN }
# use default quality as a failsafe
newQuality = int ( sickbeard . QUALITY_DEFAULT )
iqualityID = [ ]
aqualityID = [ ]
if self . initial :
for quality in self . initial :
iqualityID . append ( quality_map [ quality ] )
if self . archive :
for quality in self . archive :
aqualityID . append ( quality_map [ quality ] )
if iqualityID or aqualityID :
newQuality = Quality . combineQualities ( iqualityID , aqualityID )
# use default status as a failsafe
newStatus = sickbeard . STATUS_DEFAULT
if self . status :
# convert the string status to a int
for status in statusStrings . statusStrings :
if statusStrings [ status ] . lower ( ) == str ( self . status ) . lower ( ) :
self . status = status
break
#TODO: check if obsolete
if not self . status in statusStrings . statusStrings :
raise ApiError ( " Invalid Status " )
# only allow the status options we want
if int ( self . status ) not in ( 3 , 5 , 6 , 7 ) :
return _responds ( RESULT_FAILURE , msg = " Status prohibited " )
newStatus = self . status
2014-04-24 03:22:21 +00:00
indexerName = None
2014-07-12 01:38:22 +00:00
indexerResult = CMD_SickBeardSearchIndexers ( self . handler , [ ] ,
{ " indexerid " : self . indexerid , " indexer " : self . indexer } ) . run ( )
2014-03-10 05:18:05 +00:00
2014-04-24 03:22:21 +00:00
if indexerResult [ ' result ' ] == result_type_map [ RESULT_SUCCESS ] :
if not indexerResult [ ' data ' ] [ ' results ' ] :
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Empty results returned, check indexerid and try again " )
2014-04-24 03:22:21 +00:00
if len ( indexerResult [ ' data ' ] [ ' results ' ] ) == 1 and ' name ' in indexerResult [ ' data ' ] [ ' results ' ] [ 0 ] :
indexerName = indexerResult [ ' data ' ] [ ' results ' ] [ 0 ] [ ' name ' ]
2014-03-10 05:18:05 +00:00
2014-04-24 03:22:21 +00:00
if not indexerName :
return _responds ( RESULT_FAILURE , msg = " Unable to retrieve information from indexer " )
2014-03-10 05:18:05 +00:00
# moved the logic check to the end in an attempt to eliminate empty directory being created from previous errors
2014-04-24 03:22:21 +00:00
showPath = ek . ek ( os . path . join , self . location , helpers . sanitizeFileName ( indexerName ) )
2014-03-10 05:18:05 +00:00
# don't create show dir if config says not to
if sickbeard . ADD_SHOWS_WO_DIR :
logger . log ( u " Skipping initial creation of " + showPath + " due to config.ini setting " )
else :
dir_exists = helpers . makeDir ( showPath )
if not dir_exists :
logger . log ( u " API :: Unable to create the folder " + showPath + " , can ' t add the show " , logger . ERROR )
2014-03-25 05:57:24 +00:00
return _responds ( RESULT_FAILURE , { " path " : showPath } ,
" Unable to create the folder " + showPath + " , can ' t add the show " )
2014-03-10 05:18:05 +00:00
else :
helpers . chmodAsParent ( showPath )
2014-07-12 01:38:22 +00:00
sickbeard . showQueueScheduler . action . addShow ( int ( self . indexer ) , int ( self . indexerid ) , showPath , newStatus ,
newQuality ,
2014-07-11 17:42:42 +00:00
int ( self . flatten_folders ) , self . lang , self . subtitles , self . anime ,
2017-11-18 09:21:57 +00:00
self . scene , new_show = True ) # @UndefinedVariable
2014-07-11 17:42:42 +00:00
2014-04-24 03:22:21 +00:00
return _responds ( RESULT_SUCCESS , { " name " : indexerName } , indexerName + " has been queued to be added " )
2014-03-10 05:18:05 +00:00
class CMD_ShowCache ( ApiCall ) :
_help = { " desc " : " check sickbeard ' s cache to see if the banner or poster image for a show is valid " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" check sickbeard ' s cache to see if the banner or poster image for a show is valid """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
#TODO: catch if cache dir is missing/invalid.. so it doesn't break show/show.cache
#return {"poster": 0, "banner": 0}
cache_obj = image_cache . ImageCache ( )
has_poster = 0
has_banner = 0
if ek . ek ( os . path . isfile , cache_obj . poster_path ( showObj . indexerid ) ) :
has_poster = 1
if ek . ek ( os . path . isfile , cache_obj . banner_path ( showObj . indexerid ) ) :
has_banner = 1
return _responds ( RESULT_SUCCESS , { " poster " : has_poster , " banner " : has_banner } )
class CMD_ShowDelete ( ApiCall ) :
_help = { " desc " : " delete a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" delete a show in sickbeard """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
2014-03-25 05:57:24 +00:00
if sickbeard . showQueueScheduler . action . isBeingAdded (
showObj ) or sickbeard . showQueueScheduler . action . isBeingUpdated ( showObj ) : #@UndefinedVariable
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_FAILURE , msg = " Show can not be deleted while being added or updated " )
showObj . deleteShow ( )
return _responds ( RESULT_SUCCESS , msg = str ( showObj . name ) + " has been deleted " )
class CMD_ShowGetQuality ( ApiCall ) :
_help = { " desc " : " get quality setting for a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" get quality setting for a show in sickbeard """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
anyQualities , bestQualities = _mapQuality ( showObj . quality )
return _responds ( RESULT_SUCCESS , { " initial " : anyQualities , " archive " : bestQualities } )
class CMD_ShowGetPoster ( ApiCall ) :
_help = { " desc " : " get the poster stored for a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" get the poster for a show in sickbeard """
2015-02-06 11:39:10 +00:00
return { ' outputType ' : ' image ' , ' image ' : self . handler . showPoster ( self . indexerid , ' poster ' , True ) }
2014-03-10 05:18:05 +00:00
class CMD_ShowGetBanner ( ApiCall ) :
_help = { " desc " : " get the banner stored for a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" get the banner for a show in sickbeard """
2015-02-06 11:39:10 +00:00
return { ' outputType ' : ' image ' , ' image ' : self . handler . showPoster ( self . indexerid , ' banner ' , True ) }
2014-03-10 05:18:05 +00:00
class CMD_ShowPause ( ApiCall ) :
_help = { " desc " : " set a show ' s paused state in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " pause " : { " desc " : " set the pause state of the show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
self . pause , args = self . check_params ( args , kwargs , " pause " , 0 , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" set a show ' s paused state in sickbeard """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
2014-05-02 11:33:06 +00:00
if self . pause :
2014-03-10 05:18:05 +00:00
showObj . paused = 1
return _responds ( RESULT_SUCCESS , msg = str ( showObj . name ) + " has been paused " )
else :
showObj . paused = 0
return _responds ( RESULT_SUCCESS , msg = str ( showObj . name ) + " has been unpaused " )
2014-03-20 18:03:22 +00:00
return _responds ( RESULT_FAILURE , msg = str ( showObj . name ) + " was unable to be paused " )
2014-03-10 05:18:05 +00:00
class CMD_ShowRefresh ( ApiCall ) :
_help = { " desc " : " refresh a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" refresh a show in sickbeard """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
try :
2014-03-25 05:57:24 +00:00
sickbeard . showQueueScheduler . action . refreshShow ( showObj ) #@UndefinedVariable
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = str ( showObj . name ) + " has queued to be refreshed " )
except exceptions . CantRefreshException :
# TODO: log the excption
return _responds ( RESULT_FAILURE , msg = " Unable to refresh " + str ( showObj . name ) )
class CMD_ShowSeasonList ( ApiCall ) :
_help = { " desc " : " display the season list for a given show " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " sort " : { " desc " : " change the sort order from descending to ascending " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
2014-03-25 05:57:24 +00:00
self . sort , args = self . check_params ( args , kwargs , " sort " , " desc " , False , " string " ,
[ " asc " , " desc " ] ) # "asc" and "desc" default and fallback is "desc"
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display the season list for a given show """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
myDB = db . DBConnection ( row_type = " dict " )
if self . sort == " asc " :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select ( " SELECT DISTINCT season FROM tv_episodes WHERE showid = ? ORDER BY season ASC " ,
[ self . indexerid ] )
2014-03-10 05:18:05 +00:00
else :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select ( " SELECT DISTINCT season FROM tv_episodes WHERE showid = ? ORDER BY season DESC " ,
[ self . indexerid ] )
seasonList = [ ] # a list with all season numbers
2014-03-10 05:18:05 +00:00
for row in sqlResults :
seasonList . append ( int ( row [ " season " ] ) )
return _responds ( RESULT_SUCCESS , seasonList )
class CMD_ShowSeasons ( ApiCall ) :
_help = { " desc " : " display a listing of episodes for all or a given season " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
2014-03-25 05:57:24 +00:00
} ,
2014-03-10 05:18:05 +00:00
" optionalParameters " : { " season " : { " desc " : " the season number " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
self . season , args = self . check_params ( args , kwargs , " season " , None , False , " int " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display a listing of episodes for all or a given show """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
myDB = db . DBConnection ( row_type = " dict " )
2014-03-20 18:03:22 +00:00
if self . season == None :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select ( " SELECT name, episode, airdate, status, season FROM tv_episodes WHERE showid = ? " ,
[ self . indexerid ] )
2014-03-10 05:18:05 +00:00
seasons = { }
for row in sqlResults :
status , quality = Quality . splitCompositeStatus ( int ( row [ " status " ] ) )
row [ " status " ] = _get_status_Strings ( status )
row [ " quality " ] = _get_quality_string ( quality )
2014-11-05 11:33:05 +00:00
dtEpisodeAirs = sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( row [ ' airdate ' ] , showObj . airs , showObj . network ) )
2014-11-02 03:55:29 +00:00
row [ ' airdate ' ] = sbdatetime . sbdatetime . sbfdate ( dtEpisodeAirs , d_preset = dateFormat )
2014-03-10 05:18:05 +00:00
curSeason = int ( row [ " season " ] )
curEpisode = int ( row [ " episode " ] )
del row [ " season " ]
del row [ " episode " ]
if not curSeason in seasons :
seasons [ curSeason ] = { }
seasons [ curSeason ] [ curEpisode ] = row
else :
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select (
" SELECT name, episode, airdate, status FROM tv_episodes WHERE showid = ? AND season = ? " ,
[ self . indexerid , self . season ] )
2014-03-10 05:18:05 +00:00
if len ( sqlResults ) is 0 :
return _responds ( RESULT_FAILURE , msg = " Season not found " )
seasons = { }
for row in sqlResults :
curEpisode = int ( row [ " episode " ] )
del row [ " episode " ]
status , quality = Quality . splitCompositeStatus ( int ( row [ " status " ] ) )
row [ " status " ] = _get_status_Strings ( status )
row [ " quality " ] = _get_quality_string ( quality )
2014-11-05 11:33:05 +00:00
dtEpisodeAirs = sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( row [ ' airdate ' ] , showObj . airs , showObj . network ) )
2014-11-02 03:55:29 +00:00
row [ ' airdate ' ] = sbdatetime . sbdatetime . sbfdate ( dtEpisodeAirs , d_preset = dateFormat )
2014-03-10 05:18:05 +00:00
if not curEpisode in seasons :
seasons [ curEpisode ] = { }
seasons [ curEpisode ] = row
return _responds ( RESULT_SUCCESS , seasons )
class CMD_ShowSetQuality ( ApiCall ) :
2014-03-25 05:57:24 +00:00
_help = {
2014-07-12 01:38:22 +00:00
" desc " : " set desired quality of a show in sickbeard. if neither initial or archive are provided then the config default quality will be used " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " }
} ,
" optionalParameters " : { " initial " : { " desc " : " initial quality for the show " } ,
" archive " : { " desc " : " archive quality for the show " }
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# this for whatever reason removes hdbluray not sdtv... which is just wrong. reverting to previous code.. plus we didnt use the new code everywhere.
# self.archive, args = self.check_params(args, kwargs, "archive", None, False, "list", _getQualityMap().values()[1:])
2014-03-25 05:57:24 +00:00
self . initial , args = self . check_params ( args , kwargs , " initial " , None , False , " list " ,
[ " sdtv " , " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " ,
" fullhdwebdl " , " hdbluray " , " fullhdbluray " , " unknown " ] )
self . archive , args = self . check_params ( args , kwargs , " archive " , None , False , " list " ,
[ " sddvd " , " hdtv " , " rawhdtv " , " fullhdtv " , " hdwebdl " , " fullhdwebdl " ,
" hdbluray " , " fullhdbluray " ] )
2014-03-10 05:18:05 +00:00
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" set the quality for a show in sickbeard by taking in a deliminated
string of qualities , map to their value and combine for new values
"""
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
quality_map = { ' sdtv ' : Quality . SDTV ,
' sddvd ' : Quality . SDDVD ,
' hdtv ' : Quality . HDTV ,
' rawhdtv ' : Quality . RAWHDTV ,
' fullhdtv ' : Quality . FULLHDTV ,
' hdwebdl ' : Quality . HDWEBDL ,
' fullhdwebdl ' : Quality . FULLHDWEBDL ,
' hdbluray ' : Quality . HDBLURAY ,
' fullhdbluray ' : Quality . FULLHDBLURAY ,
' unknown ' : Quality . UNKNOWN }
#use default quality as a failsafe
newQuality = int ( sickbeard . QUALITY_DEFAULT )
iqualityID = [ ]
aqualityID = [ ]
if self . initial :
for quality in self . initial :
iqualityID . append ( quality_map [ quality ] )
if self . archive :
for quality in self . archive :
aqualityID . append ( quality_map [ quality ] )
if iqualityID or aqualityID :
newQuality = Quality . combineQualities ( iqualityID , aqualityID )
showObj . quality = newQuality
2014-03-25 05:57:24 +00:00
return _responds ( RESULT_SUCCESS ,
msg = showObj . name + " quality has been changed to " + _get_quality_string ( showObj . quality ) )
2014-03-10 05:18:05 +00:00
class CMD_ShowStats ( ApiCall ) :
_help = { " desc " : " display episode statistics for a given show " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display episode statistics for a given show """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
# show stats
episode_status_counts_total = { }
episode_status_counts_total [ " total " ] = 0
for status in statusStrings . statusStrings . keys ( ) :
2017-11-27 19:35:20 +00:00
if status in SNATCHED_ANY + [ UNKNOWN , DOWNLOADED ] :
2014-03-10 05:18:05 +00:00
continue
episode_status_counts_total [ status ] = 0
# add all the downloaded qualities
episode_qualities_counts_download = { }
episode_qualities_counts_download [ " total " ] = 0
for statusCode in Quality . DOWNLOADED :
status , quality = Quality . splitCompositeStatus ( statusCode )
if quality in [ Quality . NONE ] :
continue
episode_qualities_counts_download [ statusCode ] = 0
# add all snatched qualities
episode_qualities_counts_snatch = { }
episode_qualities_counts_snatch [ " total " ] = 0
2017-11-27 19:35:20 +00:00
for statusCode in Quality . SNATCHED_ANY :
2014-03-10 05:18:05 +00:00
status , quality = Quality . splitCompositeStatus ( statusCode )
if quality in [ Quality . NONE ] :
continue
episode_qualities_counts_snatch [ statusCode ] = 0
myDB = db . DBConnection ( row_type = " dict " )
2014-03-25 05:57:24 +00:00
sqlResults = myDB . select ( " SELECT status, season FROM tv_episodes WHERE season != 0 AND showid = ? " ,
[ self . indexerid ] )
2014-03-10 05:18:05 +00:00
# the main loop that goes through all episodes
for row in sqlResults :
status , quality = Quality . splitCompositeStatus ( int ( row [ " status " ] ) )
episode_status_counts_total [ " total " ] + = 1
if status in Quality . DOWNLOADED :
episode_qualities_counts_download [ " total " ] + = 1
episode_qualities_counts_download [ int ( row [ " status " ] ) ] + = 1
2017-11-27 19:35:20 +00:00
elif status in Quality . SNATCHED_ANY :
2014-03-10 05:18:05 +00:00
episode_qualities_counts_snatch [ " total " ] + = 1
episode_qualities_counts_snatch [ int ( row [ " status " ] ) ] + = 1
2014-03-25 05:57:24 +00:00
elif status == 0 : # we dont count NONE = 0 = N/A
2014-03-10 05:18:05 +00:00
pass
else :
episode_status_counts_total [ status ] + = 1
# the outgoing container
episodes_stats = { }
episodes_stats [ " downloaded " ] = { }
# truning codes into strings
for statusCode in episode_qualities_counts_download :
2014-04-02 14:23:17 +00:00
if statusCode == " total " :
2014-03-10 05:18:05 +00:00
episodes_stats [ " downloaded " ] [ " total " ] = episode_qualities_counts_download [ statusCode ]
continue
status , quality = Quality . splitCompositeStatus ( int ( statusCode ) )
statusString = Quality . qualityStrings [ quality ] . lower ( ) . replace ( " " , " _ " ) . replace ( " ( " , " " ) . replace ( " ) " , " " )
episodes_stats [ " downloaded " ] [ statusString ] = episode_qualities_counts_download [ statusCode ]
episodes_stats [ " snatched " ] = { }
# truning codes into strings
# and combining proper and normal
for statusCode in episode_qualities_counts_snatch :
2014-04-02 14:23:17 +00:00
if statusCode == " total " :
2014-03-10 05:18:05 +00:00
episodes_stats [ " snatched " ] [ " total " ] = episode_qualities_counts_snatch [ statusCode ]
continue
status , quality = Quality . splitCompositeStatus ( int ( statusCode ) )
statusString = Quality . qualityStrings [ quality ] . lower ( ) . replace ( " " , " _ " ) . replace ( " ( " , " " ) . replace ( " ) " , " " )
if Quality . qualityStrings [ quality ] in episodes_stats [ " snatched " ] :
episodes_stats [ " snatched " ] [ statusString ] + = episode_qualities_counts_snatch [ statusCode ]
else :
episodes_stats [ " snatched " ] [ statusString ] = episode_qualities_counts_snatch [ statusCode ]
#episodes_stats["total"] = {}
for statusCode in episode_status_counts_total :
2014-04-02 14:23:17 +00:00
if statusCode == " total " :
2014-03-10 05:18:05 +00:00
episodes_stats [ " total " ] = episode_status_counts_total [ statusCode ]
continue
status , quality = Quality . splitCompositeStatus ( int ( statusCode ) )
2014-03-25 05:57:24 +00:00
statusString = statusStrings . statusStrings [ statusCode ] . lower ( ) . replace ( " " , " _ " ) . replace ( " ( " , " " ) . replace (
" ) " , " " )
2014-03-10 05:18:05 +00:00
episodes_stats [ statusString ] = episode_status_counts_total [ statusCode ]
return _responds ( RESULT_SUCCESS , episodes_stats )
class CMD_ShowUpdate ( ApiCall ) :
_help = { " desc " : " update a show in sickbeard " ,
" requiredParameters " : { " indexerid " : { " desc " : " thetvdb.com unique id of a show " } ,
}
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
self . indexerid , args = self . check_params ( args , kwargs , " indexerid " , None , True , " int " , [ ] )
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" update a show in sickbeard """
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( self . indexerid ) )
if not showObj :
return _responds ( RESULT_FAILURE , msg = " Show not found " )
try :
2014-03-25 05:57:24 +00:00
sickbeard . showQueueScheduler . action . updateShow ( showObj , True ) #@UndefinedVariable
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , msg = str ( showObj . name ) + " has queued to be updated " )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2014-03-10 05:18:05 +00:00
logger . log ( u " API:: Unable to update " + str ( showObj . name ) + " . " + str ( ex ( e ) ) , logger . ERROR )
return _responds ( RESULT_FAILURE , msg = " Unable to update " + str ( showObj . name ) )
class CMD_Shows ( ApiCall ) :
_help = { " desc " : " display all shows in sickbeard " ,
" optionalParameters " : { " sort " : { " desc " : " sort the list of shows by show name instead of indexerid " } ,
" paused " : { " desc " : " only show the shows that are set to paused " } ,
2014-03-25 05:57:24 +00:00
} ,
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
self . sort , args = self . check_params ( args , kwargs , " sort " , " id " , False , " string " , [ " id " , " name " ] )
self . paused , args = self . check_params ( args , kwargs , " paused " , None , False , " bool " , [ ] )
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display_is_int_multi( self.indexerid )shows in sickbeard """
shows = { }
for curShow in sickbeard . showList :
2014-03-20 18:03:22 +00:00
if self . paused != None and bool ( self . paused ) != bool ( curShow . paused ) :
2014-03-10 05:18:05 +00:00
continue
2014-07-12 01:38:22 +00:00
showDict = {
" paused " : curShow . paused ,
" quality " : _get_quality_string ( curShow . quality ) ,
" language " : curShow . lang ,
" air_by_date " : curShow . air_by_date ,
" sports " : curShow . sports ,
" anime " : curShow . anime ,
" indexerid " : curShow . indexerid ,
2016-09-04 20:00:44 +00:00
" tvdbid " : curShow . ids . get ( INDEXER_TVDB , { ' id ' : 0 } ) [ ' id ' ] ,
" tvrage_id " : curShow . ids . get ( INDEXER_TVRAGE , { ' id ' : 0 } ) [ ' id ' ] ,
2014-07-12 01:38:22 +00:00
" tvrage_name " : curShow . name ,
" network " : curShow . network ,
" show_name " : curShow . name ,
" status " : curShow . status ,
2014-10-30 21:24:08 +00:00
" subtitles " : curShow . subtitles ,
2014-07-12 01:38:22 +00:00
}
2014-11-02 03:55:29 +00:00
if curShow . nextaired :
2014-11-05 11:33:05 +00:00
dtEpisodeAirs = sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( curShow . nextaired , curShow . airs , showDict [ ' network ' ] ) )
2014-11-02 03:55:29 +00:00
showDict [ ' next_ep_airdate ' ] = sbdatetime . sbdatetime . sbfdate ( dtEpisodeAirs , d_preset = dateFormat )
else :
showDict [ ' next_ep_airdate ' ] = ' '
2014-06-19 22:46:41 +00:00
showDict [ " cache " ] = CMD_ShowCache ( self . handler , ( ) , { " indexerid " : curShow . indexerid } ) . run ( ) [ " data " ]
2014-03-10 05:18:05 +00:00
if not showDict [ " network " ] :
showDict [ " network " ] = " "
if self . sort == " name " :
shows [ curShow . name ] = showDict
else :
shows [ curShow . indexerid ] = showDict
2014-11-02 03:55:29 +00:00
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , shows )
class CMD_ShowsStats ( ApiCall ) :
_help = { " desc " : " display the global shows and episode stats "
2014-03-25 05:57:24 +00:00
}
2014-03-10 05:18:05 +00:00
2014-06-19 15:47:28 +00:00
def __init__ ( self , handler , args , kwargs ) :
2014-03-10 05:18:05 +00:00
# required
# optional
# super, missing, help
2014-06-19 15:47:28 +00:00
ApiCall . __init__ ( self , handler , args , kwargs )
2014-03-10 05:18:05 +00:00
def run ( self ) :
""" display the global shows and episode stats """
stats = { }
myDB = db . DBConnection ( )
today = str ( datetime . date . today ( ) . toordinal ( ) )
stats [ " shows_total " ] = len ( sickbeard . showList )
2014-03-25 05:57:24 +00:00
stats [ " shows_active " ] = len (
[ show for show in sickbeard . showList if show . paused == 0 and show . status != " Ended " ] )
stats [ " ep_downloaded " ] = myDB . select ( " SELECT COUNT(*) FROM tv_episodes WHERE status IN ( " + " , " . join (
[ str ( show ) for show in
2017-11-27 19:35:20 +00:00
Quality . DOWNLOADED + Quality . ARCHIVED ] ) + " ) AND season != 0 and episode != 0 AND airdate <= " + today + " " ) [ 0 ] [
2014-03-25 05:57:24 +00:00
0 ]
stats [ " ep_total " ] = myDB . select (
" SELECT COUNT(*) FROM tv_episodes WHERE season != 0 AND episode != 0 AND (airdate != 1 OR status IN ( " + " , " . join (
2017-11-27 19:35:20 +00:00
[ str ( show ) for show in Quality . SNATCHED_ANY + Quality . DOWNLOADED + Quality . ARCHIVED ] ) +
" )) AND airdate <= " + today + " AND status != " + str ( IGNORED ) + " " ) [ 0 ] [ 0 ]
2014-07-12 01:38:22 +00:00
2014-03-10 05:18:05 +00:00
return _responds ( RESULT_SUCCESS , stats )
# WARNING: never define a cmd call string that contains a "_" (underscore)
# this is reserved for cmd indexes used while cmd chaining
# WARNING: never define a param name that contains a "." (dot)
# this is reserved for cmd namspaces used while cmd chaining
_functionMaper = { " help " : CMD_Help ,
" future " : CMD_ComingEpisodes ,
" episode " : CMD_Episode ,
" episode.search " : CMD_EpisodeSearch ,
" episode.setstatus " : CMD_EpisodeSetStatus ,
" episode.subtitlesearch " : CMD_SubtitleSearch ,
" exceptions " : CMD_Exceptions ,
" history " : CMD_History ,
" history.clear " : CMD_HistoryClear ,
" history.trim " : CMD_HistoryTrim ,
" logs " : CMD_Logs ,
" sb " : CMD_SickBeard ,
2014-09-30 23:13:04 +00:00
" postprocess " : CMD_PostProcess ,
2014-03-10 05:18:05 +00:00
" sb.addrootdir " : CMD_SickBeardAddRootDir ,
" sb.checkscheduler " : CMD_SickBeardCheckScheduler ,
" sb.deleterootdir " : CMD_SickBeardDeleteRootDir ,
2016-02-02 12:22:58 +00:00
" sb.forcesearch " : CMD_SickBeardForceSearch ,
2014-03-10 05:18:05 +00:00
" sb.getdefaults " : CMD_SickBeardGetDefaults ,
" sb.getmessages " : CMD_SickBeardGetMessages ,
" sb.getrootdirs " : CMD_SickBeardGetRootDirs ,
" sb.pausebacklog " : CMD_SickBeardPauseBacklog ,
" sb.ping " : CMD_SickBeardPing ,
" sb.restart " : CMD_SickBeardRestart ,
2014-04-24 03:22:21 +00:00
" sb.searchtvdb " : CMD_SickBeardSearchIndexers ,
2014-03-10 05:18:05 +00:00
" sb.setdefaults " : CMD_SickBeardSetDefaults ,
" sb.shutdown " : CMD_SickBeardShutdown ,
" show " : CMD_Show ,
" show.addexisting " : CMD_ShowAddExisting ,
" show.addnew " : CMD_ShowAddNew ,
" show.cache " : CMD_ShowCache ,
" show.delete " : CMD_ShowDelete ,
" show.getquality " : CMD_ShowGetQuality ,
" show.getposter " : CMD_ShowGetPoster ,
" show.getbanner " : CMD_ShowGetBanner ,
" show.pause " : CMD_ShowPause ,
" show.refresh " : CMD_ShowRefresh ,
" show.seasonlist " : CMD_ShowSeasonList ,
" show.seasons " : CMD_ShowSeasons ,
" show.setquality " : CMD_ShowSetQuality ,
" show.stats " : CMD_ShowStats ,
" show.update " : CMD_ShowUpdate ,
" shows " : CMD_Shows ,
2014-03-25 05:57:24 +00:00
" shows.stats " : CMD_ShowsStats
}