mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-22 10:43:38 +00:00
4855 lines
215 KiB
Python
4855 lines
215 KiB
Python
|
# Author: Dennis Lutter <lad1337@gmail.com>
|
||
|
# Author: Jonathon Saine <thezoggy@gmail.com>
|
||
|
#
|
||
|
# This file is part of SickGear.
|
||
|
#
|
||
|
# SickGear is free software: you can redistribute it and/or modify
|
||
|
# it under the terms of the GNU General Public License as published by
|
||
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
# (at your option) any later version.
|
||
|
#
|
||
|
# SickGear is distributed in the hope that it will be useful,
|
||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
# GNU General Public License for more details.
|
||
|
#
|
||
|
# You should have received a copy of the GNU General Public License
|
||
|
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
from __future__ import with_statement
|
||
|
|
||
|
# noinspection PyProtectedMember
|
||
|
from mimetypes import MimeTypes
|
||
|
from random import randint
|
||
|
|
||
|
import datetime
|
||
|
import glob
|
||
|
import copy
|
||
|
import os
|
||
|
import re
|
||
|
import time
|
||
|
import traceback
|
||
|
from . import webserve
|
||
|
|
||
|
# noinspection PyPep8Naming
|
||
|
import encodingKludge as ek
|
||
|
import exceptions_helper
|
||
|
from exceptions_helper import ex
|
||
|
from json_helper import is_orjson, json_dumps, JSON_INDENT, json_loads, JSONEncoder, ORJSON_OPTIONS
|
||
|
from tornado import gen
|
||
|
from tornado.concurrent import run_on_executor
|
||
|
from lib import subliminal
|
||
|
|
||
|
import sickgear
|
||
|
from . import classes, db, helpers, history, image_cache, logger, network_timezones, processTV, search_queue, ui
|
||
|
from .common import ARCHIVED, DOWNLOADED, FAILED, IGNORED, SKIPPED, SNATCHED, SNATCHED_ANY, SNATCHED_BEST, \
|
||
|
SNATCHED_PROPER, UNAIRED, UNKNOWN, WANTED, Quality, qualityPresetStrings, statusStrings
|
||
|
from .name_parser.parser import NameParser
|
||
|
from .helpers import starify
|
||
|
from .indexers import indexer_api, indexer_config
|
||
|
from .indexers.indexer_config import *
|
||
|
from lib.tvinfo_base.exceptions import *
|
||
|
from .scene_numbering import set_scene_numbering_helper
|
||
|
from .search_backlog import FORCED_BACKLOG
|
||
|
from .show_updater import clean_ignore_require_words
|
||
|
from .sgdatetime import SGDatetime
|
||
|
from .tv import TVEpisode, TVShow, TVidProdid
|
||
|
from .webserve import AddShows
|
||
|
import dateutil.parser
|
||
|
|
||
|
from _23 import decode_str, list_keys, unquote_plus
|
||
|
from six import integer_types, iteritems, iterkeys, PY2, string_types, text_type
|
||
|
|
||
|
# noinspection PyUnreachableCode
|
||
|
if False:
|
||
|
from typing import AnyStr, Dict, List, Tuple
|
||
|
|
||
|
|
||
|
dateFormat = "%Y-%m-%d"
|
||
|
dateTimeFormat = "%Y-%m-%d %H:%M"
|
||
|
timeFormat = '%A %I:%M %p'
|
||
|
|
||
|
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 access denied error
|
||
|
result_type_map = {RESULT_SUCCESS: "success",
|
||
|
RESULT_FAILURE: "failure",
|
||
|
RESULT_TIMEOUT: "timeout",
|
||
|
RESULT_ERROR: "error",
|
||
|
RESULT_FATAL: "fatal",
|
||
|
RESULT_DENIED: "denied",
|
||
|
}
|
||
|
# basically everything except RESULT_SUCCESS / success is bad
|
||
|
|
||
|
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,
|
||
|
'uhd4kweb': Quality.UHD4KWEB,
|
||
|
'unknown': Quality.UNKNOWN}
|
||
|
|
||
|
quality_map_inversed = {v: k for k, v in iteritems(quality_map)}
|
||
|
|
||
|
|
||
|
def api_log(obj, msg, level=logger.MESSAGE):
|
||
|
apikey_name = getattr(obj, 'apikey_name', '')
|
||
|
if apikey_name:
|
||
|
apikey_name = ' (%s)' % apikey_name
|
||
|
logger.log('%s%s' % ('API%s:: ' % apikey_name, msg), level)
|
||
|
|
||
|
|
||
|
class ApiServerLoading(webserve.BaseHandler):
|
||
|
@gen.coroutine
|
||
|
def get(self, route, *args, **kwargs):
|
||
|
self.finish(json_dumps({'error_msg': 'Server is loading'}))
|
||
|
|
||
|
post = get
|
||
|
|
||
|
|
||
|
if is_orjson:
|
||
|
|
||
|
def orjson_default(obj):
|
||
|
if isinstance(obj, set):
|
||
|
return list(obj)
|
||
|
elif isinstance(obj, TVEpisode):
|
||
|
return {'season': obj.season, 'episode': obj.episode}
|
||
|
elif isinstance(obj, TVShow):
|
||
|
return {'name': obj.name, 'indexer': obj.tvid, 'indexer_id': obj.prodid}
|
||
|
elif isinstance(obj, datetime.datetime):
|
||
|
return SGDatetime.sbfdatetime(obj, d_preset=dateFormat, t_preset='%H:%M %z')
|
||
|
elif isinstance(obj, datetime.date):
|
||
|
return SGDatetime.sbfdate(obj, d_preset=dateFormat)
|
||
|
raise TypeError
|
||
|
|
||
|
json_enc_kw = {'default': orjson_default, 'option': ORJSON_OPTIONS}
|
||
|
|
||
|
else:
|
||
|
|
||
|
class PythonObjectEncoder(JSONEncoder):
|
||
|
def default(self, obj):
|
||
|
if isinstance(obj, set):
|
||
|
return list(obj)
|
||
|
elif isinstance(obj, TVEpisode):
|
||
|
return {'season': obj.season, 'episode': obj.episode}
|
||
|
elif isinstance(obj, TVShow):
|
||
|
return {'name': obj.name, 'indexer': obj.tvid, 'indexer_id': obj.prodid}
|
||
|
elif isinstance(obj, datetime.datetime):
|
||
|
return SGDatetime.sbfdatetime(obj, d_preset=dateFormat, t_preset='%H:%M %z')
|
||
|
elif isinstance(obj, datetime.date):
|
||
|
return SGDatetime.sbfdate(obj, d_preset=dateFormat)
|
||
|
return JSONEncoder.default(self, obj)
|
||
|
|
||
|
json_enc_kw = {'cls': PythonObjectEncoder, 'indent': JSON_INDENT, 'sort_keys': True}
|
||
|
|
||
|
|
||
|
class Api(webserve.BaseHandler):
|
||
|
""" api class that returns json results """
|
||
|
version = 14 # use an int since float-point is unpredictable
|
||
|
|
||
|
def check_xsrf_cookie(self):
|
||
|
pass
|
||
|
|
||
|
def set_default_headers(self):
|
||
|
self.set_header('Cache-Control', 'no-store, no-cache, must-revalidate, max-age=0')
|
||
|
self.set_header('X-Robots-Tag', 'noindex, nofollow, noarchive, nocache, noodp, noydir, noimageindex, nosnippet')
|
||
|
if sickgear.SEND_SECURITY_HEADERS:
|
||
|
self.set_header('X-Frame-Options', 'SAMEORIGIN')
|
||
|
self.set_header('X-Application', 'SickGear')
|
||
|
self.set_header('X-API-Version', Api.version)
|
||
|
|
||
|
def prepare(self):
|
||
|
# Incorporate request JSON into arguments dictionary.
|
||
|
if self.request.body:
|
||
|
try:
|
||
|
json_data = {'payloadjson': json_loads(self.request.body)}
|
||
|
self.request.arguments.update(json_data)
|
||
|
except (BaseException, Exception):
|
||
|
raise ApiError('Unable to parse JSON.')
|
||
|
super(Api, self).prepare()
|
||
|
|
||
|
def post(self, route, *args, **kwargs):
|
||
|
return self.get(route, *args, **kwargs)
|
||
|
|
||
|
def _decode_params(self, kwargs):
|
||
|
for k, v in iteritems(kwargs):
|
||
|
if isinstance(v, list):
|
||
|
kwargs[k] = [decode_str(l) for l in v]
|
||
|
elif isinstance(v, dict):
|
||
|
kwargs[k] = {a: decode_str(b) for a, b in iteritems(v)}
|
||
|
else:
|
||
|
kwargs[k] = decode_str(v)
|
||
|
return kwargs
|
||
|
|
||
|
@gen.coroutine
|
||
|
def get(self, route, *args, **kwargs):
|
||
|
route = route.strip('/') or 'index'
|
||
|
|
||
|
kwargs = self._decode_params(self.request.arguments)
|
||
|
for arg, value in iteritems(kwargs):
|
||
|
if not isinstance(value, dict) and 1 == len(value):
|
||
|
kwargs[arg] = value[0]
|
||
|
|
||
|
args = args[1:]
|
||
|
|
||
|
self.apiKeys = sickgear.API_KEYS
|
||
|
access, accessMsg, args, kwargs = self._grand_access(self.apiKeys, route, args, kwargs)
|
||
|
|
||
|
# set the output callback
|
||
|
# default json
|
||
|
outputCallbackDict = {'default': self._out_as_json,
|
||
|
'image': lambda x: x['image'],
|
||
|
'raw': lambda x: x['raw']
|
||
|
}
|
||
|
|
||
|
# do we have access ?
|
||
|
if access:
|
||
|
api_log(self, accessMsg, logger.DEBUG)
|
||
|
else:
|
||
|
api_log(self, accessMsg, logger.WARNING)
|
||
|
result = yield self.async_call(outputCallbackDict['default'], (_responds(RESULT_DENIED, msg=accessMsg), ))
|
||
|
self.finish(result)
|
||
|
return
|
||
|
|
||
|
# 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
|
||
|
|
||
|
_call_dispatcher = profile(_call_dispatcher, immediate=True)
|
||
|
del kwargs["profile"]
|
||
|
|
||
|
# if debug was set call the "_call_dispatcher"
|
||
|
if 'debug' in kwargs:
|
||
|
outDict = _call_dispatcher(self, args,
|
||
|
kwargs) # this way we can debug the cherry.py traceback in the browser
|
||
|
del kwargs["debug"]
|
||
|
else: # if debug was not set we wrap the "call_dispatcher" in a try block to assure a json output
|
||
|
try:
|
||
|
outDict = yield self.async_call(_call_dispatcher, (self, args, kwargs))
|
||
|
except Exception as e: # real internal error
|
||
|
api_log(self, ex(e), logger.ERROR)
|
||
|
errorData = {"error_msg": ex(e),
|
||
|
"args": args,
|
||
|
"kwargs": kwargs}
|
||
|
outDict = _responds(RESULT_FATAL, errorData,
|
||
|
"SickGear encountered an internal error! Please report to the Devs")
|
||
|
|
||
|
if 'outputType' in outDict:
|
||
|
outputCallback = outputCallbackDict[outDict['outputType']]
|
||
|
else:
|
||
|
outputCallback = outputCallbackDict['default']
|
||
|
self.finish(outputCallback(outDict))
|
||
|
|
||
|
@run_on_executor
|
||
|
def async_call(self, function, ag):
|
||
|
try:
|
||
|
result = function(*ag)
|
||
|
return result
|
||
|
except Exception as e:
|
||
|
if PY2:
|
||
|
logger.log('traceback: %s' % traceback.format_exc(), logger.ERROR)
|
||
|
logger.log(ex(e), logger.ERROR)
|
||
|
raise e
|
||
|
|
||
|
def _out_as_json(self, dict):
|
||
|
self.set_header('Content-Type', 'application/json; charset=UTF-8')
|
||
|
try:
|
||
|
out = json_dumps(dict, **json_enc_kw)
|
||
|
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
|
||
|
|
||
|
except Exception as e: # if we fail to generate the output fake an error
|
||
|
api_log(self, traceback.format_exc(), logger.ERROR)
|
||
|
out = '{"result":"' + result_type_map[RESULT_ERROR] + '", "message": "error while composing output: "' + ex(
|
||
|
e) + '"}'
|
||
|
|
||
|
return out
|
||
|
|
||
|
def _grand_access(self, realKeys, apiKey, args, kwargs):
|
||
|
""" validate api key and log result """
|
||
|
remoteIp = self.request.remote_ip
|
||
|
self.apikey_name = ''
|
||
|
|
||
|
if not sickgear.USE_API:
|
||
|
msg = u'%s - SB API Disabled. ACCESS DENIED' % remoteIp
|
||
|
return False, msg, args, kwargs
|
||
|
if not apiKey:
|
||
|
msg = u'%s - gave NO API KEY. ACCESS DENIED' % remoteIp
|
||
|
return False, msg, args, kwargs
|
||
|
for realKey in realKeys:
|
||
|
if apiKey == realKey[1]:
|
||
|
self.apikey_name = realKey[0]
|
||
|
msg = u'%s - gave correct API KEY: %s. ACCESS GRANTED' % (remoteIp, realKey[0])
|
||
|
return True, msg, args, kwargs
|
||
|
msg = u'%s - gave WRONG API KEY %s. ACCESS DENIED' % (remoteIp, apiKey)
|
||
|
return False, msg, args, kwargs
|
||
|
|
||
|
|
||
|
def call_dispatcher(handler, args, kwargs):
|
||
|
""" 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
|
||
|
"""
|
||
|
cmds = None
|
||
|
if args:
|
||
|
cmds = args[0]
|
||
|
args = args[1:]
|
||
|
|
||
|
if "cmd" in kwargs:
|
||
|
cmds = kwargs["cmd"]
|
||
|
del kwargs["cmd"]
|
||
|
|
||
|
api_log(handler, u"cmd: '" + str(cmds) + "'", logger.DEBUG)
|
||
|
api_log(handler, u"all args: '" + str(args) + "'", logger.DEBUG)
|
||
|
api_log(handler, u"all kwargs: '" + str(kwargs) + "'", logger.DEBUG)
|
||
|
# logger.log(u"dateFormat: '" + str(dateFormat) + "'", logger.DEBUG)
|
||
|
|
||
|
|
||
|
outDict = {}
|
||
|
if None is not cmds:
|
||
|
cmds = cmds.split("|")
|
||
|
multiCmds = bool(1 < len(cmds))
|
||
|
for cmd in cmds:
|
||
|
curArgs, curKwargs = filter_params(cmd, args, kwargs)
|
||
|
cmdIndex = None
|
||
|
if 1 < len(cmd.split("_")): # was a index used for this cmd ?
|
||
|
cmd, cmdIndex = cmd.split("_") # this gives us the clear cmd and the index
|
||
|
|
||
|
api_log(handler, cmd + ": curKwargs " + str(curKwargs), logger.DEBUG)
|
||
|
if not (multiCmds and cmd in ('show.getposter', 'show.getbanner')): # skip these cmd while chaining
|
||
|
try:
|
||
|
if cmd in _functionMaper:
|
||
|
curOutDict = _functionMaper.get(cmd)(handler, curArgs,
|
||
|
curKwargs).run() # get the cmd class, init it and run()
|
||
|
elif _is_int(cmd):
|
||
|
curOutDict = TVDBShorthandWrapper(handler, curArgs, curKwargs, cmd).run()
|
||
|
else:
|
||
|
curOutDict = _responds(RESULT_ERROR, "No such cmd: '" + cmd + "'")
|
||
|
except ApiError as e: # Api errors that we raised, they are harmless
|
||
|
curOutDict = _responds(RESULT_ERROR, msg=ex(e))
|
||
|
else: # if someone chained one of the forbidden cmds they will get an error for this one cmd
|
||
|
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
|
||
|
if None is cmdIndex: # do we need a index dict for this cmd ?
|
||
|
outDict[cmd] = curOutDict
|
||
|
else:
|
||
|
if cmd not in outDict:
|
||
|
outDict[cmd] = {}
|
||
|
outDict[cmd][cmdIndex] = curOutDict
|
||
|
else:
|
||
|
outDict = curOutDict
|
||
|
|
||
|
if multiCmds: # if we had multiple cmds we have to wrap it in a response dict
|
||
|
outDict = _responds(RESULT_SUCCESS, outDict)
|
||
|
else: # index / no cmd given
|
||
|
outDict = CMD_SickBeard(handler, args, kwargs).run()
|
||
|
|
||
|
return outDict
|
||
|
|
||
|
|
||
|
def filter_params(cmd, args, kwargs):
|
||
|
# type: (AnyStr, List, Dict) -> Tuple[List, Dict]
|
||
|
""" 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.prodid=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.prodid=101501&show.seasonlist_2.prodid=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 0 == kwarg.find(cmd + "."):
|
||
|
cleanKey = kwarg.rpartition(".")[2]
|
||
|
curKwargs[cleanKey] = kwargs[kwarg].lower()
|
||
|
elif "." not in kwarg: # the kwarg was not namespaced therefore a "global"
|
||
|
curKwargs[kwarg] = kwargs[kwarg]
|
||
|
return curArgs, curKwargs
|
||
|
|
||
|
|
||
|
class ApiCall(object):
|
||
|
_help = {"desc": "No help message available. Please tell the devs that a help msg is missing for this cmd"}
|
||
|
|
||
|
def __init__(self,
|
||
|
handler,
|
||
|
args, # type: List
|
||
|
kwargs # type: Dict
|
||
|
):
|
||
|
|
||
|
# missing
|
||
|
try:
|
||
|
if self._missing:
|
||
|
self.run = self.return_missing
|
||
|
except AttributeError:
|
||
|
pass
|
||
|
# help
|
||
|
if 'help' in kwargs:
|
||
|
self.run = self.return_help
|
||
|
|
||
|
# RequestHandler
|
||
|
self.handler = handler
|
||
|
|
||
|
# old Sickbeard call
|
||
|
self._sickbeard_call = getattr(self, '_sickbeard_call', False)
|
||
|
if 'help' not in kwargs and self._sickbeard_call:
|
||
|
call_name = _functionMaper_reversed.get(self.__class__, '')
|
||
|
if 'sb' != call_name:
|
||
|
self.log('SickBeard API call "%s" should be replaced with SickGear API "%s" calls to get much '
|
||
|
'improved detail and functionality, contact your App developer and ask them to update '
|
||
|
'their code.' % (call_name, self._get_old_command()), logger.WARNING)
|
||
|
|
||
|
self._requiredParams = []
|
||
|
self._optionalParams = []
|
||
|
|
||
|
@property
|
||
|
def sickbeard_call(self):
|
||
|
if hasattr(self, '_sickbeard_call'):
|
||
|
return self._sickbeard_call
|
||
|
return False
|
||
|
|
||
|
@sickbeard_call.setter
|
||
|
def sickbeard_call(self, v):
|
||
|
self._sickbeard_call = v
|
||
|
|
||
|
def run(self):
|
||
|
# override with real output function in subclass
|
||
|
return {}
|
||
|
|
||
|
def log(self, msg, level=logger.MESSAGE):
|
||
|
api_log(self.handler, msg, level)
|
||
|
|
||
|
def _get_old_command(self, command_class=None):
|
||
|
c_class = command_class or self
|
||
|
new_call_name = None
|
||
|
help = getattr(c_class, '_help', None)
|
||
|
if getattr(c_class, '_sickbeard_call', False) or "SickGearCommand" in help:
|
||
|
call_name = _functionMaper_reversed.get(c_class.__class__, '')
|
||
|
new_call_name = 'sg.%s' % call_name.replace('sb.', '') if 'sb' != call_name else 'sg'
|
||
|
if new_call_name not in _functionMaper:
|
||
|
if isinstance(help, dict) and "SickGearCommand" in help \
|
||
|
and help['SickGearCommand'] in _functionMaper:
|
||
|
new_call_name = help['SickGearCommand']
|
||
|
else:
|
||
|
new_call_name = 'sg.*'
|
||
|
return new_call_name
|
||
|
|
||
|
def return_help(self):
|
||
|
try:
|
||
|
if self._requiredParams:
|
||
|
pass
|
||
|
except AttributeError:
|
||
|
self._requiredParams = []
|
||
|
try:
|
||
|
if self._optionalParams:
|
||
|
pass
|
||
|
except AttributeError:
|
||
|
self._optionalParams = []
|
||
|
|
||
|
for paramDict, param_type in [
|
||
|
(self._requiredParams, "requiredParameters"),
|
||
|
(self._optionalParams, "optionalParameters")
|
||
|
]:
|
||
|
|
||
|
if param_type in self._help:
|
||
|
for param_name in paramDict:
|
||
|
if param_name not in self._help[param_type]:
|
||
|
self._help[param_type][param_name] = {}
|
||
|
if paramDict[param_name]["allowedValues"]:
|
||
|
self._help[param_type][param_name]["allowedValues"] = paramDict[param_name]["allowedValues"]
|
||
|
else:
|
||
|
self._help[param_type][param_name]["allowedValues"] = "see desc"
|
||
|
self._help[param_type][param_name]["defaultValue"] = paramDict[param_name]["defaultValue"]
|
||
|
|
||
|
elif paramDict:
|
||
|
for param_name in paramDict:
|
||
|
self._help[param_type] = {}
|
||
|
self._help[param_type][param_name] = paramDict[param_name]
|
||
|
else:
|
||
|
self._help[param_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 1 == len(self._missing):
|
||
|
msg = "The required parameter: '" + self._missing[0] + "' was not set"
|
||
|
else:
|
||
|
msg = "The required parameters: '" + "','".join(self._missing) + "' where not set"
|
||
|
try:
|
||
|
remote_ip = self.handler.request.remote_ip
|
||
|
except (BaseException, Exception):
|
||
|
remote_ip = '"unknown ip"'
|
||
|
self.log("API call from host %s triggers :: %s: %s" %
|
||
|
(remote_ip, _functionMaper_reversed.get(self.__class__, ''), msg),
|
||
|
logger.ERROR)
|
||
|
return _responds(RESULT_ERROR, msg=msg)
|
||
|
|
||
|
def check_params(self,
|
||
|
args, # type: List
|
||
|
kwargs, # type: Dict
|
||
|
key,
|
||
|
default,
|
||
|
required,
|
||
|
type,
|
||
|
allowedValues,
|
||
|
sub_type=None
|
||
|
):
|
||
|
# TODO: explain this
|
||
|
""" function to check passed params for the shorthand wrapper
|
||
|
and to detect missing/required param
|
||
|
"""
|
||
|
# Fix for applications that send tvdbid instead of indexerid
|
||
|
if self.sickbeard_call and "indexerid" == key and "indexerid" not in kwargs:
|
||
|
key = "tvdbid"
|
||
|
|
||
|
missing = True
|
||
|
orgDefault = default
|
||
|
|
||
|
if "bool" == type:
|
||
|
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[key] = {"allowedValues": allowedValues, "defaultValue": orgDefault}
|
||
|
except AttributeError:
|
||
|
self._missing = []
|
||
|
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 = {key: {"allowedValues": allowedValues, "defaultValue": orgDefault}}
|
||
|
|
||
|
if default:
|
||
|
default = self._check_param_type(default, key, type, sub_type)
|
||
|
if "bool" == type:
|
||
|
type = []
|
||
|
self._check_param_value(default, key, allowedValues)
|
||
|
|
||
|
return default, args
|
||
|
|
||
|
def _check_param_type(self, value, name, type, sub_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 "int" == type:
|
||
|
if _is_int(value):
|
||
|
value = int(value)
|
||
|
else:
|
||
|
error = True
|
||
|
elif "bool" == type:
|
||
|
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 "list" == type:
|
||
|
if None is not sub_type:
|
||
|
if sub_type in integer_types:
|
||
|
if isinstance(value, integer_types):
|
||
|
value = [value]
|
||
|
elif isinstance(value, string_types):
|
||
|
if '|' in value:
|
||
|
li = [int(v) for v in value.split('|')]
|
||
|
if any([not isinstance(v, integer_types) for v in li]):
|
||
|
error = True
|
||
|
else:
|
||
|
value = li
|
||
|
else:
|
||
|
value = [int(value)]
|
||
|
else:
|
||
|
error = True
|
||
|
else:
|
||
|
li = value.split('|')
|
||
|
if any([sub_type is not type(v) for v in li]):
|
||
|
error = True
|
||
|
else:
|
||
|
value = li
|
||
|
else:
|
||
|
value = value.split("|")
|
||
|
elif "dict" == type:
|
||
|
if isinstance(value, dict):
|
||
|
value = value
|
||
|
else:
|
||
|
error = True
|
||
|
elif "string" == type:
|
||
|
pass
|
||
|
elif "ignore" == type:
|
||
|
pass
|
||
|
else:
|
||
|
self.log(u"Invalid param type set " + str(type) + " can not check or convert ignoring it",
|
||
|
logger.ERROR)
|
||
|
|
||
|
if error:
|
||
|
# this is a real ApiError !!
|
||
|
raise ApiError(
|
||
|
u"param: '" + str(name) + "' with given value: '" + str(value) + "' could not be parsed into '" + str(
|
||
|
type) + "'")
|
||
|
|
||
|
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 item not in allowedValues:
|
||
|
error = True
|
||
|
else:
|
||
|
if value not in allowedValues:
|
||
|
error = True
|
||
|
|
||
|
if error:
|
||
|
# this is kinda a ApiError but raising an error is the only way of quitting here
|
||
|
raise ApiError(u"param: '" + str(name) + "' with given value: '" + str(
|
||
|
value) + "' is out of allowed range '" + str(allowedValues) + "'")
|
||
|
|
||
|
|
||
|
class TVDBShorthandWrapper(ApiCall):
|
||
|
_help = {"desc": "this is an internal function wrapper. call the help command directly for more information"}
|
||
|
|
||
|
def __init__(self,
|
||
|
handler,
|
||
|
args, # type: List
|
||
|
kwargs, # type: Dict
|
||
|
sid # type: AnyStr
|
||
|
):
|
||
|
self.handler = handler
|
||
|
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
|
||
|
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" internal function wrapper """
|
||
|
args = (self.sid,) + self.origArgs
|
||
|
if self.e:
|
||
|
return CMD_Episode(self.handler, args, self.kwargs).run()
|
||
|
elif self.s:
|
||
|
return CMD_ShowSeasons(self.handler, args, self.kwargs).run()
|
||
|
else:
|
||
|
return CMD_Show(self.handler, args, self.kwargs).run()
|
||
|
|
||
|
|
||
|
# ###############################
|
||
|
# helper functions #
|
||
|
# ###############################
|
||
|
|
||
|
def _sizeof_fmt(num):
|
||
|
for x in ['bytes', 'KB', 'MB', 'GB', 'TB']:
|
||
|
if 1024.00 > num:
|
||
|
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 None is data:
|
||
|
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 1 != int(ordinal):
|
||
|
date = datetime.date.fromordinal(ordinal)
|
||
|
else:
|
||
|
return ""
|
||
|
return date.strftime(dateTimeFormat)
|
||
|
|
||
|
|
||
|
def _ordinal_to_dateForm(ordinal):
|
||
|
if 1 != int(ordinal):
|
||
|
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:
|
||
|
statusCodes += Quality.ARCHIVED
|
||
|
if "ignored" in statusStrings:
|
||
|
statusCodes.append(IGNORED)
|
||
|
if "unaired" in statusStrings:
|
||
|
statusCodes.append(UNAIRED)
|
||
|
return statusCodes
|
||
|
|
||
|
|
||
|
def _mapQuality(show_obj):
|
||
|
quality_map = _getQualityMap()
|
||
|
|
||
|
anyQualities = []
|
||
|
bestQualities = []
|
||
|
|
||
|
iqualityID, aqualityID = Quality.splitQuality(int(show_obj))
|
||
|
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_map_inversed
|
||
|
|
||
|
|
||
|
def _getRootDirs():
|
||
|
if "" == sickgear.ROOT_DIRS:
|
||
|
return {}
|
||
|
|
||
|
rootDir = {}
|
||
|
root_dirs = sickgear.ROOT_DIRS.split('|')
|
||
|
default_index = int(sickgear.ROOT_DIRS.split('|')[0])
|
||
|
|
||
|
rootDir["default_index"] = int(sickgear.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 = [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 (BaseException, Exception):
|
||
|
valid = 0
|
||
|
default = 0
|
||
|
if root_dir is default_dir:
|
||
|
default = 1
|
||
|
|
||
|
dir_list.append({'valid': valid, 'location': root_dir, 'default': default})
|
||
|
|
||
|
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 """
|
||
|
|
||
|
|
||
|
# -------------------------------------------------------------------------------------#
|
||
|
|
||
|
class CMD_ListCommands(ApiCall):
|
||
|
_help = {"desc": "list help of all commands"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get help information for all commands """
|
||
|
out = ''
|
||
|
table_sickgear_commands = ''
|
||
|
table_sickgear_commands = ''
|
||
|
for f, v in sorted(iteritems(_functionMaper), key=lambda x: (re.sub(r'^s[bg]\.', '', x[0], flags=re.I),
|
||
|
re.sub(r'^sg\.', '1', x[0], flags=re.I))):
|
||
|
if 'listcommands' == f:
|
||
|
continue
|
||
|
help = getattr(v, '_help', None)
|
||
|
is_old_command = isinstance(help, dict) and "SickGearCommand" in help
|
||
|
if is_old_command:
|
||
|
table_sickgear_commands += '<tr><td>%s</td>' % f
|
||
|
else:
|
||
|
table_sickgear_commands += '<tr><td>%s</td>' % f
|
||
|
color = ("", " sb")[is_old_command]
|
||
|
out += '<hr><h1 class="command%s">%s%s</h1>' \
|
||
|
% (color, f, ("",
|
||
|
" <span>(Sickbeard compatibility command)</span>"
|
||
|
)[is_old_command])
|
||
|
if isinstance(help, dict):
|
||
|
sg_cmd_new = self._get_old_command(command_class=v)
|
||
|
sg_cmd = ''
|
||
|
if sg_cmd_new:
|
||
|
sg_cmd = '<td>%s</td>' % sg_cmd_new
|
||
|
out += "<p class='use-sg-api'>for all features use SickGear API Command: <b>%s</b></p>" % sg_cmd_new
|
||
|
if "desc" in help:
|
||
|
if is_old_command:
|
||
|
table_sickgear_commands += '<td>%s</td>%s' % (help['desc'], sg_cmd)
|
||
|
else:
|
||
|
table_sickgear_commands += '<td>%s</td>' % help['desc']
|
||
|
out += help['desc']
|
||
|
|
||
|
table = ''
|
||
|
|
||
|
if "requiredParameters" in help and isinstance(help['requiredParameters'], dict):
|
||
|
for p, d in iteritems(help['requiredParameters']):
|
||
|
des = ''
|
||
|
if isinstance(d, dict) and 'desc' in d:
|
||
|
des = d.get('desc')
|
||
|
table += "<tr><td><span>%s <span class='parareq'>required</span></span></td>" \
|
||
|
"<td><p>%s</p></td></tr>" % (p, des)
|
||
|
|
||
|
if "optionalParameters" in help and isinstance(help['optionalParameters'], dict):
|
||
|
for p, d in iteritems(help['optionalParameters']):
|
||
|
des = ''
|
||
|
if isinstance(d, dict) and 'desc' in d:
|
||
|
des = d.get('desc')
|
||
|
table += "<tr><td><span>%s <span class='paraopt'>optional</span></span></td>" \
|
||
|
"<td><p>%s</p></td></tr>" % (p, des)
|
||
|
if table:
|
||
|
out += "<table class='sickbeardTable' cellspacing='0' border='1' cellpadding='0'><thead>" \
|
||
|
"<tr><th style='width: 25%'>Parameter</th><th>Description</th></tr>" \
|
||
|
"</thead><tbody>"
|
||
|
out += table
|
||
|
out += '</tbody></table>'
|
||
|
else:
|
||
|
if is_old_command:
|
||
|
table_sickgear_commands += '<td>%s</td><td></td>' % 'no description'
|
||
|
else:
|
||
|
table_sickgear_commands += '<td>%s</td>' % 'no description'
|
||
|
|
||
|
if is_old_command:
|
||
|
table_sickgear_commands += '</tr>'
|
||
|
else:
|
||
|
table_sickgear_commands += '</tr>'
|
||
|
|
||
|
if table_sickgear_commands:
|
||
|
out = "<h1>SickBeard Commands (compatibility):</h1>" \
|
||
|
"<table class='sickbeardTable' cellspacing='0' border='1' cellpadding='0'><thead>" \
|
||
|
"<tr><th style='width: 25%'>Command</th><th>Description</th>" \
|
||
|
"<th style='width: 25%'>Replacement SickGear Command</th></tr>" \
|
||
|
"</thead><tbody>" + table_sickgear_commands + '</tbody></table>' + out
|
||
|
|
||
|
if table_sickgear_commands:
|
||
|
out = "<h1>SickGear Commands:</h1>" \
|
||
|
"<table class='sickbeardTable' cellspacing='0' border='1' cellpadding='0'><thead>" \
|
||
|
"<tr><th style='width: 25%'>Command</th><th>Description</th></tr>" \
|
||
|
"</thead><tbody>" + table_sickgear_commands + '</tbody></table>' + out
|
||
|
|
||
|
return {'outputType': 'raw', 'raw': out}
|
||
|
|
||
|
|
||
|
class CMD_Help(ApiCall):
|
||
|
_help = {"desc": "get help information for a given subject/command",
|
||
|
"optionalParameters": {"subject": {"desc": "command - the top level command"}}}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.subject, args = self.check_params(args, kwargs, "subject", "help",
|
||
|
False, "string", iterkeys(_functionMaper))
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get help information for a given subject/command """
|
||
|
if self.subject in _functionMaper:
|
||
|
out = _responds(RESULT_SUCCESS, _functionMaper.get(self.subject)(None, (), {"help": 1}).run())
|
||
|
else:
|
||
|
out = _responds(RESULT_FAILURE, msg="No such cmd")
|
||
|
return out
|
||
|
|
||
|
|
||
|
class CMD_SickGearComingEpisodes(ApiCall):
|
||
|
_help = {"desc": "get the daily schedule",
|
||
|
"optionalParameters":
|
||
|
{"sort": {"desc": "change the sort order"},
|
||
|
"type": {"desc": "one or more of allowed values separated by |"},
|
||
|
"paused": {"desc": "0 to exclude paused shows, 1 to include them, "
|
||
|
"2 to only view paused, or omitted to use the SG default"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.sort, args = self.check_params(args, kwargs, "sort", "date", False, "string", ["date", "show", "network"])
|
||
|
self.type, args = self.check_params(args, kwargs, "type", "today|missed|soon|later", False, "list",
|
||
|
["missed", "later", "today", "soon"])
|
||
|
self.paused, args = self.check_params(args, kwargs, "paused", sickgear.EPISODE_VIEW_DISPLAY_PAUSED,
|
||
|
False, "int", [0, 1, 2])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the daily schedule """
|
||
|
sql_result, fanart, sorts, next_week_dt, today, next_week = webserve.MainHandler.get_daily_schedule()
|
||
|
sql_result.sort(key=sorts[(self.sort, 'time')['date' == self.sort]])
|
||
|
|
||
|
finalEpResults = {}
|
||
|
|
||
|
# add all requested types or all
|
||
|
for curType in self.type:
|
||
|
finalEpResults[curType] = []
|
||
|
|
||
|
for ep in sql_result:
|
||
|
"""
|
||
|
Missed: yesterday... (less than 1week)
|
||
|
Today: today
|
||
|
Soon: tomorrow till next week
|
||
|
Later: later than next week
|
||
|
"""
|
||
|
|
||
|
if not ((int(ep['paused']) and self.paused) or (not int(ep['paused']) and 2 != self.paused)):
|
||
|
continue
|
||
|
|
||
|
ep['airdate'] = int(ep["airdate"])
|
||
|
|
||
|
status = "soon"
|
||
|
if ep["airdate"] < today:
|
||
|
status = "missed"
|
||
|
elif ep["airdate"] >= next_week:
|
||
|
status = "later"
|
||
|
elif today <= ep["airdate"] < next_week:
|
||
|
if ep["airdate"] == today:
|
||
|
status = "today"
|
||
|
else:
|
||
|
status = "soon"
|
||
|
|
||
|
# skip unwanted
|
||
|
if None is not self.type and status not in self.type:
|
||
|
continue
|
||
|
|
||
|
if not ep["network"]:
|
||
|
ep["network"] = ""
|
||
|
|
||
|
ep["quality"] = _get_quality_string(ep["quality"])
|
||
|
# clean up Tvdb horrible airs field
|
||
|
ep['airs'] = str(ep['airs']).replace('am', ' AM').replace('pm', ' PM').replace(' ', ' ')
|
||
|
# start day of the week on 1 (monday)
|
||
|
ep['weekday'] = 1 + datetime.date.fromordinal(ep['airdate']).weekday()
|
||
|
ep['ep_name'] = ep['name']
|
||
|
ep['ep_plot'] = ep['description']
|
||
|
# add parsed_datetime to the dict
|
||
|
ep['local_datetime'] = SGDatetime.sbstrftime(
|
||
|
SGDatetime.convert_to_setting(ep['parsed_datetime'], force_local=True), dateTimeFormat)
|
||
|
ep['status_str'] = statusStrings[ep['status']]
|
||
|
ep['network'] = ep['episode_network'] or ep['network']
|
||
|
ep['timezone'] = ep['ep_timezone'] or ep['show_timezone'] or ep['timezone'] or (
|
||
|
ep['network'] and network_timezones.get_network_timezone(ep['network'], return_name=True)[1])
|
||
|
|
||
|
# remove all field we don't want for api response
|
||
|
for cur_f in list_keys(ep):
|
||
|
if cur_f not in [ # fields to preserve
|
||
|
'absolute_number', 'air_by_date', 'airdate', 'airs', 'archive_firstmatch',
|
||
|
'classification', 'data_network', 'data_show_name',
|
||
|
'ep_name', 'ep_plot', 'episode', 'episode_id', 'genre',
|
||
|
'imdb_id', 'imdb_url', 'indexer', 'indexer_id', 'indexerid',
|
||
|
'lang', 'local_datetime', 'network', 'overview', 'parsed_datetime', 'paused', 'prod_id',
|
||
|
'quality', 'runtime', 'scene', 'scene_absolute_number', 'scene_episode', 'scene_season',
|
||
|
'season', 'show_id', 'show_name', 'show_network', 'show_status', 'showid', 'startyear',
|
||
|
'status', 'status_str', 'tag', 'timezone', 'trakt_watched', 'tv_id', 'tvid_prodid',
|
||
|
'version', 'weekday'
|
||
|
]:
|
||
|
del ep[cur_f]
|
||
|
|
||
|
# Add tvdbid for backward compatibility
|
||
|
try:
|
||
|
show_obj = helpers.find_show_by_id({ep['tv_id']: ep['prod_id']})
|
||
|
ep['tvdbid'] = show_obj.ids.get(TVINFO_TVDB, {'id': 0})['id']
|
||
|
ep['ids'] = {k: v.get('id') for k, v in iteritems(show_obj.ids)}
|
||
|
except (BaseException, Exception):
|
||
|
ep['tvdbid'] = (None, ep['prod_id'])[TVINFO_TVDB == ep['tv_id']]
|
||
|
ep['ids'] = None
|
||
|
|
||
|
ep['airdate'] = SGDatetime.sbfdate(
|
||
|
datetime.date.fromordinal(ep['airdate']), d_preset=dateFormat)
|
||
|
ep['parsed_datetime'] = SGDatetime.sbfdatetime(ep['parsed_datetime'],
|
||
|
d_preset=dateFormat, t_preset='%H:%M %z')
|
||
|
|
||
|
# TODO: check if this obsolete
|
||
|
if status not in finalEpResults:
|
||
|
finalEpResults[status] = []
|
||
|
|
||
|
finalEpResults[status].append(ep)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, finalEpResults)
|
||
|
|
||
|
|
||
|
class CMD_ComingEpisodes(CMD_SickGearComingEpisodes):
|
||
|
_help = {"desc": "get the daily schedule",
|
||
|
"optionalParameters":
|
||
|
{"sort": {"desc": "change the sort order"},
|
||
|
"type": {"desc": "one or more of allowed values separated by |"},
|
||
|
"paused": {"desc": "0 to exclude paused shows, 1 to include them, or omitted to use the SG default"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.future"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearComingEpisodes.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearEpisode(ApiCall):
|
||
|
_help = {"desc": "get episode information",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
"optionalParameters": {"full_path": {
|
||
|
"desc": "show the full absolute path (if valid) instead of a relative path for the episode location"}},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get episode information """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT name, description, airdate, status, location, file_size, release_name, "
|
||
|
" subtitles, absolute_number,scene_season, scene_episode, scene_absolute_number"
|
||
|
" FROM tv_episodes"
|
||
|
" WHERE indexer = ? AND showid = ?"
|
||
|
" AND episode = ? AND season = ?",
|
||
|
[self.tvid, self.prodid, self.e, self.s])
|
||
|
if 1 != len(sql_result):
|
||
|
raise ApiError("Episode not found")
|
||
|
episode = sql_result[0]
|
||
|
# handle path options
|
||
|
# absolute vs relative vs broken
|
||
|
showPath = None
|
||
|
try:
|
||
|
showPath = show_obj.location
|
||
|
except exceptions_helper.ShowDirNotFoundException:
|
||
|
pass
|
||
|
|
||
|
if bool(self.fullPath) and showPath:
|
||
|
pass
|
||
|
elif not bool(self.fullPath) and showPath:
|
||
|
# using the length because lstrip removes too much
|
||
|
showPathLength = len(showPath) + 1 # the / or \ yeah not that nice i know
|
||
|
episode["location"] = episode["location"][showPathLength:]
|
||
|
elif not showPath: # show dir is broken ... episode path will be empty
|
||
|
episode["location"] = ""
|
||
|
# convert stuff to human form
|
||
|
timezone, episode['timezone'] = network_timezones.get_network_timezone(show_obj.network, return_name=True)
|
||
|
episode['airdate'] = SGDatetime.sbfdate(SGDatetime.convert_to_setting(
|
||
|
network_timezones.parse_date_time(int(episode['airdate']), show_obj.airs, timezone)), d_preset=dateFormat)
|
||
|
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_Episode(CMD_SickGearEpisode):
|
||
|
_help = {"desc": "get detailed episode info",
|
||
|
"requiredParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
"optionalParameters": {"full_path": {
|
||
|
"desc": "show the full absolute path (if valid) instead of a relative path for the episode location"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.episode",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearEpisode.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearEpisodeSearch(ApiCall):
|
||
|
_help = {"desc": "search for an episode. the response might take some time",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" search for an episode """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
# retrieve the episode object and fail if we can't get one
|
||
|
ep_obj = show_obj.get_episode(int(self.s), int(self.e))
|
||
|
if isinstance(ep_obj, str):
|
||
|
return _responds(RESULT_FAILURE, msg="Episode not found")
|
||
|
|
||
|
# make a queue item for it and put it on the queue
|
||
|
ep_queue_item = search_queue.ManualSearchQueueItem(show_obj, ep_obj)
|
||
|
sickgear.search_queue_scheduler.action.add_item(ep_queue_item)
|
||
|
|
||
|
# wait until the queue item tells us whether it worked or not
|
||
|
while None is ep_queue_item.success:
|
||
|
time.sleep(1)
|
||
|
|
||
|
# return the correct json value
|
||
|
if ep_queue_item.success:
|
||
|
status, quality = Quality.splitCompositeStatus(ep_obj.status)
|
||
|
# TODO: split quality and status?
|
||
|
return _responds(RESULT_SUCCESS, {"quality": _get_quality_string(quality)},
|
||
|
"Snatched (" + _get_quality_string(quality) + ")")
|
||
|
|
||
|
return _responds(RESULT_FAILURE, msg='Unable to find episode')
|
||
|
|
||
|
|
||
|
class CMD_EpisodeSearch(CMD_SickGearEpisodeSearch):
|
||
|
_help = {"desc": "search for an episode. the response might take some time",
|
||
|
"requiredParameters": {"tvdbid": {"desc": "thetvdb.com id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.episode.search",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearEpisodeSearch.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearEpisodeSetStatus(ApiCall):
|
||
|
_help = {"desc": "set status of an episode (or season if episode is not provided)",
|
||
|
"requiredParameters": {
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"status": {
|
||
|
"desc": "the status values: wanted, skipped, archived, ignored, failed, snatched, downloaded"}
|
||
|
},
|
||
|
"optionalParameters": {
|
||
|
"episode": {"desc": "the episode number"},
|
||
|
"force": {"desc": "should we replace existing (downloaded) episodes or not"},
|
||
|
"quality": {"desc": "set quality of episode(s), only for statuses: snatched, downloaded, archived"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.s, args = self.check_params(args, kwargs, "season", None, True, "int", [])
|
||
|
self.status, args = self.check_params(args, kwargs, "status", None, True, "string", [
|
||
|
"wanted", "skipped", "archived", "ignored", "failed", "snatched", "downloaded"])
|
||
|
# 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", [])
|
||
|
self.quality, args = self.check_params(args, kwargs, "quality", None, False, "string", [q for q in quality_map])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set status of an episode or a season (when no ep is provided) """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
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
|
||
|
else: # if we dont break out of the for loop we got here.
|
||
|
# 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!")
|
||
|
|
||
|
if None is not self.quality:
|
||
|
if self.status not in (SNATCHED, SNATCHED_BEST, SNATCHED_PROPER, DOWNLOADED, ARCHIVED):
|
||
|
return _responds(RESULT_FAILURE, msg="Can't set status %s together with quailty: %s" %
|
||
|
(statusStrings[self.status], self.quality))
|
||
|
self.quality = quality_map[self.quality]
|
||
|
|
||
|
# ep_obj_list = []
|
||
|
if self.e:
|
||
|
ep_obj = show_obj.get_episode(self.s, self.e)
|
||
|
if None is ep_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Episode not found")
|
||
|
ep_obj_list = [ep_obj]
|
||
|
else:
|
||
|
# get all episode numbers from self,season
|
||
|
ep_obj_list = show_obj.get_all_episodes(season=self.s)
|
||
|
|
||
|
def _epResult(result_code, ep, msg=""):
|
||
|
return {'season': ep.season, 'episode': ep.episode, 'status': _get_status_Strings(ep.status),
|
||
|
'result': result_type_map[result_code], 'message': msg}
|
||
|
|
||
|
ep_results = []
|
||
|
failure = False
|
||
|
start_backlog = False
|
||
|
segments = {}
|
||
|
|
||
|
sql_l = []
|
||
|
for ep_obj in ep_obj_list:
|
||
|
with ep_obj.lock:
|
||
|
if self.status == WANTED:
|
||
|
# figure out what episodes are wanted so we can backlog them
|
||
|
if ep_obj.season in segments:
|
||
|
segments[ep_obj.season].append(ep_obj)
|
||
|
else:
|
||
|
segments[ep_obj.season] = [ep_obj]
|
||
|
|
||
|
# don't let them mess up UNAIRED episodes
|
||
|
if ep_obj.status == UNAIRED:
|
||
|
if None is not self.e:
|
||
|
# setting the status of a UNAIRED is only considered a failure
|
||
|
# if we directly wanted this episode, but is ignored on a season request
|
||
|
ep_results.append(
|
||
|
_epResult(RESULT_FAILURE, ep_obj, "Refusing to change status because it is UNAIRED"))
|
||
|
failure = True
|
||
|
continue
|
||
|
|
||
|
# allow the user to force setting the status for an already downloaded episode
|
||
|
if ep_obj.status in Quality.DOWNLOADED and not self.force and None is self.quality:
|
||
|
ep_results.append(_epResult(RESULT_FAILURE, ep_obj,
|
||
|
"Refusing to change status because it is already marked as DOWNLOADED"))
|
||
|
failure = True
|
||
|
continue
|
||
|
|
||
|
if None is not self.quality:
|
||
|
ep_obj.status = Quality.compositeStatus(self.status, self.quality)
|
||
|
else:
|
||
|
ep_obj.status = self.status
|
||
|
result = ep_obj.get_sql()
|
||
|
if None is not result:
|
||
|
sql_l.append(result)
|
||
|
|
||
|
if self.status == WANTED:
|
||
|
start_backlog = True
|
||
|
ep_results.append(_epResult(RESULT_SUCCESS, ep_obj))
|
||
|
|
||
|
if 0 < len(sql_l):
|
||
|
my_db = db.DBConnection()
|
||
|
my_db.mass_action(sql_l)
|
||
|
|
||
|
extra_msg = ""
|
||
|
if start_backlog:
|
||
|
for season, segment in iteritems(segments): # type: int, List[TVEpisode]
|
||
|
backlog_queue_item = search_queue.BacklogQueueItem(show_obj, segment)
|
||
|
sickgear.search_queue_scheduler.action.add_item(backlog_queue_item)
|
||
|
|
||
|
self.log(u'Starting backlog for %s season %s because some episodes were set to WANTED' %
|
||
|
(show_obj.unique_name, season))
|
||
|
|
||
|
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)
|
||
|
|
||
|
|
||
|
class CMD_EpisodeSetStatus(CMD_SickGearEpisodeSetStatus):
|
||
|
_help = {"desc": "set status of an episode or season (when no ep is provided)",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"status": {"desc": "the status values: wanted, skipped, archived, ignored, failed"}
|
||
|
},
|
||
|
"optionalParameters": {"episode": {"desc": "the episode number"},
|
||
|
"force": {"desc": "should we replace existing (downloaded) episodes or not"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.episode.setstatus",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
CMD_SickGearEpisodeSetStatus.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearSubtitleSearch(ApiCall):
|
||
|
_help = {"desc": "search episode subtitles. the response might take some time",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" search episode subtitles """
|
||
|
if not sickgear.USE_SUBTITLES:
|
||
|
return _responds(RESULT_FAILURE, msg='Subtitle search is disabled in SickGear')
|
||
|
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
# retrieve the episode object and fail if we can't get one
|
||
|
ep_obj = show_obj.get_episode(int(self.s), int(self.e))
|
||
|
if isinstance(ep_obj, str):
|
||
|
return _responds(RESULT_FAILURE, msg="Episode not found")
|
||
|
|
||
|
# try do download subtitles for that episode
|
||
|
previous_subtitles = ep_obj.subtitles
|
||
|
|
||
|
try:
|
||
|
_ = ep_obj.download_subtitles()
|
||
|
except (BaseException, Exception):
|
||
|
return _responds(RESULT_FAILURE, msg='Unable to find subtitles')
|
||
|
|
||
|
# return the correct json value
|
||
|
if previous_subtitles != ep_obj.subtitles:
|
||
|
status = 'New subtitles downloaded: %s' % ' '.join([
|
||
|
"<img src='" + sickgear.WEB_ROOT + "/images/flags/" + subliminal.language.Language(
|
||
|
x).alpha2 + ".png' alt='" + subliminal.language.Language(x).name + "'/>" for x in
|
||
|
sorted(list(set(ep_obj.subtitles).difference(previous_subtitles)))])
|
||
|
response = _responds(RESULT_SUCCESS, msg='New subtitles found')
|
||
|
else:
|
||
|
status = 'No subtitles downloaded'
|
||
|
response = _responds(RESULT_FAILURE, msg='Unable to find subtitles')
|
||
|
|
||
|
ui.notifications.message('Subtitles Search', status)
|
||
|
|
||
|
return response
|
||
|
|
||
|
|
||
|
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"},
|
||
|
"season": {"desc": "the season number"},
|
||
|
"episode": {"desc": "the episode number"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.episode.subtitlesearch",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearExceptions(ApiCall):
|
||
|
_help = {"desc": "get scene exceptions for all or a given show",
|
||
|
"optionalParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, False, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, False, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get scene exceptions for all or a given show """
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
|
||
|
if None is self.prodid:
|
||
|
sql_result = my_db.select("SELECT s.indexer, se.show_name, se.indexer_id AS 'indexerid' "
|
||
|
"FROM scene_exceptions AS se INNER JOIN tv_shows as s "
|
||
|
"ON se.indexer_id == s.indexer_id")
|
||
|
scene_exceptions = {}
|
||
|
for cur_result in sql_result:
|
||
|
indexerid = cur_result["indexerid"]
|
||
|
indexer = cur_result["indexer"]
|
||
|
if self.sickbeard_call:
|
||
|
if indexerid not in scene_exceptions:
|
||
|
scene_exceptions[indexerid] = []
|
||
|
scene_exceptions[indexerid].append(cur_result["show_name"])
|
||
|
else:
|
||
|
if indexerid not in scene_exceptions.get(indexer, {}):
|
||
|
scene_exceptions.setdefault(indexer, {})[indexerid] = []
|
||
|
scene_exceptions.setdefault(indexer, {})[indexerid].append(cur_result["show_name"])
|
||
|
|
||
|
else:
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT indexer, show_name, indexer_id AS 'indexerid' FROM scene_exceptions "
|
||
|
"WHERE indexer = ? AND indexer_id = ?", [self.tvid, self.prodid])
|
||
|
scene_exceptions = []
|
||
|
for cur_result in sql_result:
|
||
|
scene_exceptions.append(cur_result["show_name"])
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, scene_exceptions)
|
||
|
|
||
|
|
||
|
class CMD_Exceptions(CMD_SickGearExceptions):
|
||
|
_help = {"desc": "get scene exceptions for all or a given show",
|
||
|
"optionalParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.exceptions",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearExceptions.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SetExceptions(ApiCall):
|
||
|
_help = {"desc": "set scene exceptions for a given show",
|
||
|
"requiredParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
"forseason": {"desc": "exception for season, -1 for all seasons"},
|
||
|
},
|
||
|
"optionalParameters": {"add": {"desc": "list of exceptions to add"},
|
||
|
"remove": {"desc": "list of exceptions to remove"}},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.forseason, args = self.check_params(args, kwargs, "forseason", None, True, "int", [])
|
||
|
# optional
|
||
|
self.add, args = self.check_params(args, kwargs, "add", None, False, "list", [])
|
||
|
self.remove, args = self.check_params(args, kwargs, "remove", None, False, "list", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
if not self.add and not self.remove:
|
||
|
return _responds(RESULT_FAILURE, 'No Exceptions provided to be add or removed.')
|
||
|
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, 'Could not find any show in db from indexer: %s with id: %s' %
|
||
|
(self.tvid, self.prodid))
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
sql_result = my_db.select("SELECT show_name, season, indexer, indexer_id AS 'indexerid'"
|
||
|
" FROM scene_exceptions"
|
||
|
" WHERE indexer = ? AND indexer_id = ?"
|
||
|
" AND season = ?",
|
||
|
[self.tvid, self.prodid, self.forseason])
|
||
|
|
||
|
cl = []
|
||
|
curexep = [(s['show_name'], s['season']) for s in sql_result]
|
||
|
add_list = []
|
||
|
remove_list = []
|
||
|
if self.remove:
|
||
|
for r in self.remove:
|
||
|
if (r, self.forseason) in curexep:
|
||
|
cl.append(['DELETE FROM scene_exceptions WHERE indexer = ? AND indexer_id = ? AND season = ? '
|
||
|
'AND show_name = ?', [self.tvid, self.prodid, self.forseason, r]])
|
||
|
try:
|
||
|
curexep.remove((r, self.forseason))
|
||
|
except ValueError:
|
||
|
pass
|
||
|
remove_list.append(r)
|
||
|
|
||
|
if self.add:
|
||
|
for a in self.add:
|
||
|
if (a, self.forseason) not in curexep:
|
||
|
cl.append(['INSERT INTO scene_exceptions (show_name, indexer, indexer_id, season) VALUES (?,?,?,?)',
|
||
|
[a, self.tvid, self.prodid, self.forseason]])
|
||
|
curexep.append((a, self.forseason))
|
||
|
add_list.append(a)
|
||
|
|
||
|
if cl:
|
||
|
my_db.mass_action(cl)
|
||
|
return _responds(RESULT_SUCCESS, data={'added': add_list, 'removed': remove_list, 'for season': self.forseason,
|
||
|
'current': [c[0] for c in curexep], 'indexer': self.tvid,
|
||
|
'indexerid': self.prodid},
|
||
|
msg='Exceptions changed.')
|
||
|
|
||
|
|
||
|
class CMD_SickGearHistory(ApiCall):
|
||
|
_help = {"desc": "get the sickgear downloaded/snatched history",
|
||
|
"optionalParameters": {"limit": {"desc": "limit returned results"},
|
||
|
"type": {"desc": "only show a specific type of results"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the sickgear downloaded/snatched history """
|
||
|
|
||
|
# typeCodes = []
|
||
|
type_filter = []
|
||
|
if "downloaded" == self.type:
|
||
|
type_filter = [DOWNLOADED, ARCHIVED, FAILED]
|
||
|
self.type = "Downloaded"
|
||
|
typeCodes = Quality.DOWNLOADED + Quality.ARCHIVED + Quality.FAILED
|
||
|
elif "snatched" == self.type:
|
||
|
type_filter = SNATCHED_ANY
|
||
|
self.type = "Snatched"
|
||
|
typeCodes = Quality.SNATCHED_ANY
|
||
|
else:
|
||
|
typeCodes = Quality.SNATCHED_ANY + Quality.DOWNLOADED + Quality.ARCHIVED + Quality.FAILED
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
|
||
|
ulimit = min(int(self.limit), 100)
|
||
|
if 0 == ulimit:
|
||
|
# noinspection SqlResolve
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT h.*, show_name, s.indexer FROM history h, tv_shows s WHERE h.hide = 0" +
|
||
|
" AND h.showid=s.indexer_id" +
|
||
|
("", " AND s.indexer=%s" % TVINFO_TVDB)[self.sickbeard_call] +
|
||
|
" AND action in (" + ','.join(['?'] * len(typeCodes)) + ") ORDER BY date DESC", typeCodes)
|
||
|
else:
|
||
|
# noinspection SqlResolve
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT h.*, show_name, s.indexer FROM history h, tv_shows s WHERE h.hide = 0" +
|
||
|
" AND h.showid=s.indexer_id" +
|
||
|
("", " AND s.indexer=%s" % TVINFO_TVDB)[self.sickbeard_call] +
|
||
|
" AND action in (" + ','.join(['?'] * len(typeCodes)) + ") ORDER BY date DESC LIMIT ?",
|
||
|
typeCodes + [ulimit])
|
||
|
|
||
|
results = []
|
||
|
np = NameParser(True, testing=True, indexer_lookup=False, try_scene_exceptions=False)
|
||
|
for cur_result in sql_result:
|
||
|
status, quality = Quality.splitCompositeStatus(int(cur_result["action"]))
|
||
|
if type_filter and status not in type_filter:
|
||
|
continue
|
||
|
status = _get_status_Strings(status)
|
||
|
cur_result["status"] = status
|
||
|
cur_result["quality"] = _get_quality_string(quality)
|
||
|
cur_result["date"] = _historyDate_to_dateTimeForm(str(cur_result["date"]))
|
||
|
del cur_result["action"]
|
||
|
_rename_element(cur_result, "showid", "indexerid")
|
||
|
cur_result["resource_path"] = os.path.dirname(cur_result["resource"])
|
||
|
cur_result["resource"] = os.path.basename(cur_result["resource"])
|
||
|
try:
|
||
|
p_r = np.parse(cur_result["resource"], cache_result=False)
|
||
|
except (BaseException, Exception):
|
||
|
p_r = None
|
||
|
cur_result['episode_numbers'] = (p_r and p_r.episode_numbers) or \
|
||
|
(cur_result['episode'] and [cur_result['episode']]) or []
|
||
|
# Add tvdbid for backward compatibility
|
||
|
cur_result['tvdbid'] = (None, cur_result['indexerid'])[TVINFO_TVDB == cur_result['indexer']]
|
||
|
results.append(cur_result)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, results)
|
||
|
|
||
|
|
||
|
class CMD_History(CMD_SickGearHistory):
|
||
|
_help = {"desc": "get the sickgear downloaded/snatched history",
|
||
|
"optionalParameters": {"limit": {"desc": "limit returned results"},
|
||
|
"type": {"desc": "only show a specific type of results"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.history",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearHistory.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearHistoryClear(ApiCall):
|
||
|
_help = {"desc": "clear the sickgear history"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" clear the sickgear history """
|
||
|
my_db = db.DBConnection()
|
||
|
my_db.action('UPDATE history SET hide = ? WHERE hide = 0', [1])
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, msg="History cleared")
|
||
|
|
||
|
|
||
|
class CMD_HistoryClear(CMD_SickGearHistoryClear):
|
||
|
_help = {"desc": "clear the sickgear history",
|
||
|
"SickGearCommand": "sg.history.clear"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearHistoryClear.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearHistoryTrim(ApiCall):
|
||
|
_help = {"desc": "trim the sickgear history by removing entries greater than 30 days old"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" trim the sickgear history """
|
||
|
my_db = db.DBConnection()
|
||
|
my_db.action("UPDATE history SET hide = ? WHERE date < " + str(
|
||
|
(datetime.datetime.now() - datetime.timedelta(days=30)).strftime(history.dateFormat)), [1])
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, msg="Removed history entries greater than 30 days old")
|
||
|
|
||
|
|
||
|
class CMD_HistoryTrim(CMD_SickGearHistoryTrim):
|
||
|
_help = {"desc": "trim the sickgear history by removing entries greater than 30 days old",
|
||
|
"SickGearCommand": "sg.history.trim"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearHistoryTrim.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearLogs(ApiCall):
|
||
|
_help = {"desc": "get log file entries",
|
||
|
"optionalParameters": {"min_level ": {
|
||
|
"desc": "the minimum level classification of log entries to return,"
|
||
|
" with each level inheriting log entries from the level above"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.min_level, args = self.check_params(args, kwargs, "min_level", "error", False, "string",
|
||
|
["error", "warning", "info", "debug"])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get log file entries """
|
||
|
# 10 = Debug / 20 = Info / 30 = Warning / 40 = Error
|
||
|
min_level = logger.reverseNames[str(self.min_level).upper()]
|
||
|
max_lines = 50
|
||
|
|
||
|
regex = r"^(\d\d\d\d)\-(\d\d)\-(\d\d)\s*(\d\d)\:(\d\d):(\d\d)\s*([A-Z]+)\s*(.+?)\s*\:\:\s*(.*)$"
|
||
|
|
||
|
final_data = []
|
||
|
normal_data = []
|
||
|
truncate = []
|
||
|
repeated = None
|
||
|
num_lines = 0
|
||
|
|
||
|
if os.path.isfile(logger.sb_log_instance.log_file_path):
|
||
|
auths = sickgear.GenericProvider.dedupe_auths(True)
|
||
|
rxc_auths = re.compile('(?i)%s' % '|'.join([(re.escape(_a)) for _a in auths]))
|
||
|
replacements = dict([(_a, starify(_a)) for _a in auths])
|
||
|
for cur_line in logger.sb_log_instance.reverse_readline(logger.sb_log_instance.log_file_path):
|
||
|
|
||
|
cur_line = decode_str(cur_line)
|
||
|
match = re.match(regex, cur_line)
|
||
|
|
||
|
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 = cur_line
|
||
|
continue
|
||
|
|
||
|
if 1 < len(truncate):
|
||
|
data = repeated.strip() + ' (... %s repeat lines)\n' % len(truncate)
|
||
|
if not final_data:
|
||
|
final_data = [data]
|
||
|
else:
|
||
|
final_data[-1] = data
|
||
|
|
||
|
truncate = [match.group(8) + match.group(9)]
|
||
|
|
||
|
# noinspection HttpUrlsUsage
|
||
|
if 'https://' in cur_line or 'http://' in cur_line:
|
||
|
for cur_change in rxc_auths.finditer(cur_line):
|
||
|
cur_line = '%s%s%s' % (cur_line[:cur_change.start()],
|
||
|
replacements[cur_line[cur_change.start():cur_change.end()]],
|
||
|
cur_line[cur_change.end():])
|
||
|
|
||
|
final_data.append(cur_line)
|
||
|
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
|
||
|
|
||
|
else:
|
||
|
if not any(normal_data) and not any([cur_line.strip()]):
|
||
|
continue
|
||
|
|
||
|
normal_data.append(re.sub(r'\r?\n', '<br />', cur_line.replace('<', '<').replace('>', '>')))
|
||
|
|
||
|
num_lines += 1
|
||
|
|
||
|
if num_lines >= max_lines:
|
||
|
break
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, final_data)
|
||
|
|
||
|
|
||
|
class CMD_Logs(CMD_SickGearLogs):
|
||
|
_help = {"desc": "get log file entries",
|
||
|
"optionalParameters": {"min_level ": {
|
||
|
"desc": "the minimum level classification of log entries to return,"
|
||
|
" with each level inheriting log entries from the level above"}},
|
||
|
"SickGearCommand": "sg.logs",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearLogs.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearPostProcess(ApiCall):
|
||
|
_help = {"desc": "process completed media files to a show location",
|
||
|
"optionalParameters": {"path": {"desc": "Path to process"},
|
||
|
"force_replace": {"desc": "Force already processed dir/files"},
|
||
|
"return_data": {"desc": "Return results for the process"},
|
||
|
"process_method": {"desc": "Symlink, hardlink, move, or copy file(s)"},
|
||
|
"is_priority": {"desc": "Replace file(s) even if existing at a higher quality"},
|
||
|
"type": {"desc": "Type of media process request this is, auto or manual"},
|
||
|
"failed": {"desc": "Mark as failed download"},
|
||
|
"client": {"desc": "String representing the calling client"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
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", False, "string", ["auto", "manual"])
|
||
|
self.failed, args = self.check_params(args, kwargs, "failed", 0, False, "bool", [])
|
||
|
self.client, args = self.check_params(args, kwargs, "client", None, False, "string", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" Starts the postprocess """
|
||
|
if not self.path and not sickgear.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 = sickgear.TV_DOWNLOAD_DIR
|
||
|
|
||
|
if not self.type:
|
||
|
self.type = 'manual'
|
||
|
|
||
|
data = processTV.processDir(self.path, process_method=self.process_method, force=self.force_replace,
|
||
|
force_replace=self.is_priority, failed=self.failed, pp_type=self.type,
|
||
|
client=self.client)
|
||
|
|
||
|
if not self.return_data:
|
||
|
data = ""
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, data=data, msg="Started postprocess for %s" % self.path)
|
||
|
|
||
|
|
||
|
class CMD_PostProcess(CMD_SickGearPostProcess):
|
||
|
_help = {"desc": "process completed media files to a show location",
|
||
|
"optionalParameters": {"path": {"desc": "Path to process"},
|
||
|
"force_replace": {"desc": "Force already processed dir/files"},
|
||
|
"return_data": {"desc": "Return results for the process"},
|
||
|
"process_method": {"desc": "Symlink, hardlink, move, or copy file(s)"},
|
||
|
"is_priority": {"desc": "Replace file(s) even if existing at a higher quality"},
|
||
|
"type": {"desc": "Type of media process request this is, auto or manual"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.postprocess",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
kwargs['failed'] = "0"
|
||
|
try:
|
||
|
if 'client' in kwargs:
|
||
|
del kwargs['client']
|
||
|
except (BaseException, Exception):
|
||
|
pass
|
||
|
CMD_SickGearPostProcess.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGear(ApiCall):
|
||
|
_help = {"desc": "get API information"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get API information """
|
||
|
data = {"sb_version": sickgear.BRANCH, "api_version": Api.version, "fork": "SickGear",
|
||
|
"api_commands": sorted([x for x in _functionMaper if 'listcommands' != x])}
|
||
|
return _responds(RESULT_SUCCESS, data)
|
||
|
|
||
|
|
||
|
class CMD_SickBeard(CMD_SickGear):
|
||
|
_help = {"desc": "get API information",
|
||
|
"SickGearCommand": "sg", }
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGear.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearAddRootDir(ApiCall):
|
||
|
_help = {"desc": "add a user configured parent directory",
|
||
|
"requiredParameters": {"location": {"desc": "the full path to root (parent) directory"}
|
||
|
},
|
||
|
"optionalParameters": {"default": {"desc": "make the location passed the default root (parent) directory"}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" add a user configured parent directory """
|
||
|
|
||
|
self.location = unquote_plus(self.location)
|
||
|
location_matched = 0
|
||
|
index = 0
|
||
|
|
||
|
# disallow 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 "" == sickgear.ROOT_DIRS:
|
||
|
self.default = 1
|
||
|
else:
|
||
|
root_dirs = sickgear.ROOT_DIRS.split('|')
|
||
|
index = int(sickgear.ROOT_DIRS.split('|')[0])
|
||
|
root_dirs.pop(0)
|
||
|
# clean up the list - replace %xx escapes by their single-character equivalent
|
||
|
root_dirs = [unquote_plus(x) for x in root_dirs]
|
||
|
for x in root_dirs:
|
||
|
if x == self.location:
|
||
|
location_matched = 1
|
||
|
if 1 == self.default:
|
||
|
index = root_dirs.index(self.location)
|
||
|
break
|
||
|
|
||
|
if 0 == location_matched:
|
||
|
if 1 == self.default:
|
||
|
root_dirs.insert(0, self.location)
|
||
|
else:
|
||
|
root_dirs.append(self.location)
|
||
|
|
||
|
root_dirs_new = [unquote_plus(x) for x in root_dirs]
|
||
|
root_dirs_new.insert(0, index)
|
||
|
root_dirs_new = '|'.join([text_type(x) for x in root_dirs_new])
|
||
|
|
||
|
sickgear.ROOT_DIRS = root_dirs_new
|
||
|
return _responds(RESULT_SUCCESS, _getRootDirs(), msg="Root directories updated")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardAddRootDir(CMD_SickGearAddRootDir):
|
||
|
_help = {"desc": "add a user configured parent directory ",
|
||
|
"requiredParameters": {"location": {"desc": "the full path to root (parent) directory"}
|
||
|
},
|
||
|
"optionalParameters": {"default": {"desc": "make the location passed the default root (parent) directory"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.addrootdir",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearAddRootDir.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearCheckScheduler(ApiCall):
|
||
|
_help = {"desc": "query the scheduler for event statuses, upcoming, running, and past"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" query the scheduler for event statuses, upcoming, running, and past """
|
||
|
my_db = db.DBConnection()
|
||
|
sql_result = my_db.select("SELECT last_backlog FROM info")
|
||
|
|
||
|
backlogPaused = sickgear.search_queue_scheduler.action.is_backlog_paused()
|
||
|
backlogRunning = sickgear.search_queue_scheduler.action.is_backlog_in_progress()
|
||
|
nextBacklog = sickgear.backlog_search_scheduler.next_run().strftime(dateFormat)
|
||
|
|
||
|
data = {"backlog_is_paused": int(backlogPaused), "backlog_is_running": int(backlogRunning),
|
||
|
"last_backlog": (0 < len(sql_result) and _ordinal_to_dateForm(sql_result[0]["last_backlog"])) or '',
|
||
|
"next_backlog": nextBacklog}
|
||
|
return _responds(RESULT_SUCCESS, data)
|
||
|
|
||
|
|
||
|
class CMD_SickBeardCheckScheduler(CMD_SickGearCheckScheduler):
|
||
|
_help = {"desc": "query the scheduler for event statuses, upcoming, running, and past",
|
||
|
"SickGearCommand": "sg.checkscheduler"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearCheckScheduler.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearDeleteRootDir(ApiCall):
|
||
|
_help = {"desc": "delete a user configured parent directory",
|
||
|
"requiredParameters": {"location": {"desc": "the full path to root (parent) directory"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.location, args = self.check_params(args, kwargs, "location", None, True, "string", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" delete a user configured parent directory """
|
||
|
if sickgear.ROOT_DIRS == "":
|
||
|
return _responds(RESULT_FAILURE, _getRootDirs(), msg="No root directories detected")
|
||
|
|
||
|
newIndex = 0
|
||
|
root_dirs_new = []
|
||
|
root_dirs = sickgear.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 = [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 = [unquote_plus(x) for x in root_dirs_new]
|
||
|
if 0 < len(root_dirs_new):
|
||
|
root_dirs_new.insert(0, newIndex)
|
||
|
root_dirs_new = "|".join([text_type(x) for x in root_dirs_new])
|
||
|
|
||
|
sickgear.ROOT_DIRS = root_dirs_new
|
||
|
# what if the root dir was not found?
|
||
|
return _responds(RESULT_SUCCESS, _getRootDirs(), msg="Root directory deleted")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardDeleteRootDir(CMD_SickGearDeleteRootDir):
|
||
|
_help = {"desc": "delete a user configured parent directory",
|
||
|
"requiredParameters": {"location": {"desc": "the full path to root (parent) directory"}},
|
||
|
"SickGearCommand": "sg.deleterootdir"
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearDeleteRootDir.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearForceSearch(ApiCall):
|
||
|
_help = {'desc': 'force the specified search type to run',
|
||
|
"requiredParameters": {"searchtype": {"desc": "type of search to be forced: recent, backlog, proper"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.searchtype, args = self.check_params(args, kwargs, "searchtype", "recent", True, "string",
|
||
|
["recent", "backlog", "proper"])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" force the specified search type to run """
|
||
|
result = None
|
||
|
if 'recent' == self.searchtype and not sickgear.search_queue_scheduler.action.is_recentsearch_in_progress() \
|
||
|
and not sickgear.recent_search_scheduler.action.amActive:
|
||
|
result = sickgear.recent_search_scheduler.forceRun()
|
||
|
elif 'backlog' == self.searchtype and not sickgear.search_queue_scheduler.action.is_backlog_in_progress() \
|
||
|
and not sickgear.backlog_search_scheduler.action.amActive:
|
||
|
sickgear.backlog_search_scheduler.force_search(force_type=FORCED_BACKLOG)
|
||
|
result = True
|
||
|
elif 'proper' == self.searchtype and not sickgear.search_queue_scheduler.action.is_propersearch_in_progress() \
|
||
|
and not sickgear.proper_finder_scheduler.action.amActive:
|
||
|
result = sickgear.proper_finder_scheduler.forceRun()
|
||
|
if result:
|
||
|
return _responds(RESULT_SUCCESS, msg='%s search successfully forced' % self.searchtype)
|
||
|
return _responds(RESULT_FAILURE,
|
||
|
msg='Can not force the %s search because it\'s already active' % self.searchtype)
|
||
|
|
||
|
|
||
|
class CMD_SickBeardForceSearch(CMD_SickGearForceSearch):
|
||
|
_help = {'desc': 'force the episode recent search',
|
||
|
"SickGearCommand": "sg.forcesearch", }
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['searchtype'] = 'recent'
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearForceSearch.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearSearchQueue(ApiCall):
|
||
|
_help = {'desc': 'get a list of the sickgear search queue states'}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get a list of the sickgear search queue states """
|
||
|
return _responds(RESULT_SUCCESS, sickgear.search_queue_scheduler.action.queue_length())
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetDefaults(ApiCall):
|
||
|
_help = {"desc": "get various sickgear default system values"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get various sickgear default system values """
|
||
|
|
||
|
anyQualities, bestQualities = _mapQuality(sickgear.QUALITY_DEFAULT)
|
||
|
|
||
|
data = {"status": statusStrings[sickgear.STATUS_DEFAULT].lower(),
|
||
|
"flatten_folders": int(sickgear.FLATTEN_FOLDERS_DEFAULT), "initial": anyQualities,
|
||
|
"archive": bestQualities, "future_show_paused": int(sickgear.EPISODE_VIEW_DISPLAY_PAUSED)}
|
||
|
return _responds(RESULT_SUCCESS, data)
|
||
|
|
||
|
|
||
|
class CMD_SickBeardGetDefaults(CMD_SickGearGetDefaults):
|
||
|
_help = {"desc": "get various sickgear default system values",
|
||
|
"SickGearCommand": "sg.getdefaults"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearGetDefaults.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetMessages(ApiCall):
|
||
|
_help = {"desc": "get list of ui notifications"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
messages = []
|
||
|
for cur_notification in ui.notifications.get_notifications(self.handler.request.remote_ip):
|
||
|
messages.append({"title": cur_notification.title,
|
||
|
"message": cur_notification.message,
|
||
|
"type": cur_notification.type})
|
||
|
return _responds(RESULT_SUCCESS, messages)
|
||
|
|
||
|
|
||
|
class CMD_SickBeardGetMessages(CMD_SickGearGetMessages):
|
||
|
_help = {"desc": "get list of ui notifications",
|
||
|
"SickGearCommand": "sg.getmessages"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearGetMessages.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetQualities(ApiCall):
|
||
|
_help = {"desc": "get globally available qualities"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
return _responds(RESULT_SUCCESS, quality_map)
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetIndexers(ApiCall):
|
||
|
_help = {"desc": "get tv info source list",
|
||
|
"optionalParameters": {"searchable-only": {"desc": "only return searchable sources"}}}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.searchable_only, args = self.check_params(args, kwargs, "searchable-only", False, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
result = {}
|
||
|
for i in indexer_config.tvinfo_config:
|
||
|
for d, v in iteritems(indexer_config.tvinfo_config[i]):
|
||
|
if self.searchable_only and (indexer_config.tvinfo_config[i].get('mapped_only') or
|
||
|
not indexer_config.tvinfo_config[i].get('active') or
|
||
|
indexer_config.tvinfo_config[i].get('defunct')):
|
||
|
continue
|
||
|
if d in ['id', 'name', 'show_url', 'mapped_only', 'main_url'] and \
|
||
|
isinstance(v, (string_types, tuple, dict, list, integer_types, float, bool)):
|
||
|
if 'mapped_only' == d:
|
||
|
key = 'searchable'
|
||
|
val = not v and indexer_config.tvinfo_config[i].get('active') \
|
||
|
and not indexer_config.tvinfo_config[i].get('defunct')
|
||
|
else:
|
||
|
key = d
|
||
|
if 'show_url' == d:
|
||
|
val = re.sub(r'%\d{,2}d', '{INDEXER-ID}', v, flags=re.I)
|
||
|
else:
|
||
|
val = v
|
||
|
result.setdefault(i, {}).update({key: val})
|
||
|
return _responds(RESULT_SUCCESS, result)
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetIndexerIcon(ApiCall):
|
||
|
_help = {"desc": "get tv info source icon",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().all_sources])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
# doesn't work
|
||
|
i = indexer_config.tvinfo_config.get(self.tvid)
|
||
|
if not i:
|
||
|
self.handler.set_status(404)
|
||
|
return _responds(RESULT_FAILURE, 'Icon not found')
|
||
|
img = i['icon']
|
||
|
image = ek.ek(os.path.join, sickgear.PROG_DIR, 'gui', 'slick', 'images', img)
|
||
|
if not ek.ek(os.path.isfile, image):
|
||
|
self.handler.set_status(404)
|
||
|
return _responds(RESULT_FAILURE, 'Icon not found')
|
||
|
return {'outputType': 'image', 'image': self.handler.get_image(image)}
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetNetworkIcon(ApiCall):
|
||
|
_help = {"desc": "get network icon",
|
||
|
"requiredParameters": {"network": {"desc": "name of network"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.network, args = self.check_params(args, kwargs, "network", None, True, "string", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
image = ek.ek(os.path.join, sickgear.PROG_DIR, 'gui', 'slick', 'images', 'network',
|
||
|
'%s.png' % self.network.lower())
|
||
|
if not ek.ek(os.path.isfile, image):
|
||
|
self.handler.set_status(404)
|
||
|
return _responds(RESULT_FAILURE, 'Icon not found')
|
||
|
return {'outputType': 'image', 'image': self.handler.get_image(image)}
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetqualityStrings(ApiCall):
|
||
|
_help = {"desc": "get human readable quality strings"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
return _responds(RESULT_SUCCESS, Quality.qualityStrings)
|
||
|
|
||
|
|
||
|
class CMD_SickGearGetRootDirs(ApiCall):
|
||
|
_help = {"desc": "get list of user configured parent directories"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get list of user configured parent directories """
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, _getRootDirs())
|
||
|
|
||
|
|
||
|
class CMD_SickBeardGetRootDirs(CMD_SickGearGetRootDirs):
|
||
|
_help = {"desc": "get list of user configured parent directories",
|
||
|
"SickGearCommand": "sg.getrootdirs"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearGetRootDirs.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearPauseBacklog(ApiCall):
|
||
|
_help = {"desc": "pause the backlog search",
|
||
|
"optionalParameters": {"pause": {"desc": "pause or unpause the global backlog"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.pause, args = self.check_params(args, kwargs, "pause", 0, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" pause the backlog search """
|
||
|
if self.pause:
|
||
|
sickgear.search_queue_scheduler.action.pause_backlog()
|
||
|
return _responds(RESULT_SUCCESS, msg="Backlog paused")
|
||
|
else:
|
||
|
sickgear.search_queue_scheduler.action.unpause_backlog()
|
||
|
return _responds(RESULT_SUCCESS, msg="Backlog unpaused")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardPauseBacklog(CMD_SickGearPauseBacklog):
|
||
|
_help = {"desc": "pause the backlog search",
|
||
|
"optionalParameters": {"pause": {"desc": "pause or unpause the global backlog"}},
|
||
|
"SickGearCommand": "sg.pausebacklog"
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearPauseBacklog.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearPing(ApiCall):
|
||
|
_help = {"desc": "check to see if sickgear is running", }
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" check to see if sickgear is running """
|
||
|
self.handler.set_header('Cache-Control', "max-age=0,no-cache,no-store")
|
||
|
if sickgear.started:
|
||
|
return _responds(RESULT_SUCCESS, {"pid": sickgear.PID}, "Pong")
|
||
|
else:
|
||
|
return _responds(RESULT_SUCCESS, msg="Pong")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardPing(CMD_SickGearPing):
|
||
|
_help = {"desc": "check to see if sickgear is running",
|
||
|
"SickGearCommand": "sg.ping"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearPing.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearRestart(ApiCall):
|
||
|
_help = {"desc": "restart sickgear"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" restart sickgear """
|
||
|
sickgear.restart(soft=False)
|
||
|
return _responds(RESULT_SUCCESS, msg="SickGear is restarting...")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardRestart(CMD_SickGearRestart):
|
||
|
_help = {"desc": "restart sickgear",
|
||
|
"SickGearCommand": "sg.restart"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearRestart.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearSearchIndexers(ApiCall):
|
||
|
_help = {"desc": "search for show on a tv info source with a given string and language",
|
||
|
"optionalParameters": {"name": {"desc": "name of the show you want to search for"},
|
||
|
"indexerid": {"desc": "thetvdb.com or tvrage.com unique id of a show"},
|
||
|
"lang": {"desc": "the 2 letter abbreviation lang id"},
|
||
|
"indexers": {"desc": "indexer to search, use -1 to search all indexers"}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
valid_languages = {
|
||
|
'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}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.name, args = self.check_params(args, kwargs, "name", None, False, "string", [])
|
||
|
self.prodid, 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())
|
||
|
self.indexers, args = self.check_params(args, kwargs, "indexers", -1, False, "list",
|
||
|
[-1] + [i for i in indexer_api.TVInfoAPI().search_sources], int)
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, False, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" search for show on a tv info source with a given string and language """
|
||
|
if 1 > len(self.indexers) and -1 in self.indexers:
|
||
|
raise ApiError('Mix of -1 (all Indexer) and specific Indexer not allowed')
|
||
|
|
||
|
all_indexer = 1 == len(self.indexers) and -1 == self.indexers[0]
|
||
|
lang_id = self.valid_languages['en']
|
||
|
|
||
|
def _parse_date(dt_str):
|
||
|
try:
|
||
|
return dateutil.parser.parse(dt_str)
|
||
|
except (BaseException, Exception):
|
||
|
return ''
|
||
|
|
||
|
if (self.name and not self.prodid) or (self.prodid and self.tvid):
|
||
|
results = []
|
||
|
indexertosearch = (self.indexers, [i for i in indexer_api.TVInfoAPI().sources if
|
||
|
indexer_api.TVInfoAPI(i).config.get('active') and
|
||
|
not indexer_api.TVInfoAPI(i).config.get('mapped_only') and
|
||
|
not indexer_api.TVInfoAPI(i).config.get('defunct')])[all_indexer]
|
||
|
search_kw = {}
|
||
|
search_name = None
|
||
|
if self.name:
|
||
|
search_kw['name'] = decode_str(self.name)
|
||
|
if self.prodid and self.tvid:
|
||
|
search_kw.update({'ids': {self.tvid: self.prodid}})
|
||
|
if self.tvid in indexertosearch:
|
||
|
indexertosearch = [self.tvid] + [i for i in indexertosearch if i != self.tvid]
|
||
|
|
||
|
for i in indexertosearch:
|
||
|
tvinfo_config = sickgear.TVInfoAPI(i).api_params.copy()
|
||
|
tvinfo_config['language'] = 'en'
|
||
|
tvinfo_config['custom_ui'] = classes.AllShowInfosNoFilterListUI
|
||
|
t = sickgear.TVInfoAPI(i).setup(**tvinfo_config)
|
||
|
|
||
|
try:
|
||
|
apiData = t.search_show(**search_kw)
|
||
|
except (BaseException, Exception) as e:
|
||
|
continue
|
||
|
|
||
|
if not apiData:
|
||
|
continue
|
||
|
|
||
|
for curSeries in (apiData or []):
|
||
|
if (int(curSeries['id']) == self.prodid and i == self.tvid) or \
|
||
|
any({self.tvid: self.prodid} == {k: v} for k, v in iteritems(curSeries['ids'])):
|
||
|
if not self.name and curSeries['ids'].get(self.tvid) == int(curSeries['id']):
|
||
|
search_kw['name'] = curSeries['seriesname']
|
||
|
search_name = curSeries['seriesname']
|
||
|
new_ids = {k: v for k, v in iteritems(curSeries['ids']) if v and k != self.tvid}
|
||
|
if new_ids:
|
||
|
if 'ids' not in search_kw:
|
||
|
search_kw['ids'] = {}
|
||
|
search_kw['ids'].update(new_ids)
|
||
|
|
||
|
s = {"indexerid": int(curSeries['id']),
|
||
|
"name": helpers.normalise_chars(curSeries['seriesname']),
|
||
|
"first_aired": isinstance(curSeries['firstaired'], string_types)
|
||
|
and SGDatetime.sbfdate(_parse_date(curSeries['firstaired']), d_preset=dateFormat) or '',
|
||
|
"indexer": i,
|
||
|
"aliases": curSeries.get('aliases') or [],
|
||
|
"network": curSeries.get('network') or '',
|
||
|
"overview": helpers.normalise_chars(curSeries.get('overview') or ''),
|
||
|
"language": curSeries.get('language') or '',
|
||
|
"genres": (curSeries.get('genres', '') or curSeries.get('genre', '') or '').replace('|', ', '),
|
||
|
"genre_list": curSeries.get('genre_list') or [],
|
||
|
"poster": curSeries.get('poster') or '',
|
||
|
"relevance": (curSeries['ids'].get(self.tvid) == int(curSeries['id']) and 100)
|
||
|
or AddShows.get_uw_ratio(self.name or search_name or '', curSeries['seriesname'],
|
||
|
curSeries.get('aliases') or [])}
|
||
|
if TVINFO_TVDB == i:
|
||
|
s["tvdbid"] = int(curSeries['id'])
|
||
|
else:
|
||
|
s["tvdbid"] = ''
|
||
|
results.append(s)
|
||
|
|
||
|
if not results:
|
||
|
return _responds(RESULT_FAILURE, msg="Did not get result from %s" %
|
||
|
', '.join([sickgear.TVInfoAPI(i).name for i in indexertosearch]))
|
||
|
|
||
|
results = sorted(results, key=lambda x: x['relevance'], reverse=True)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, {"results": results, "langid": lang_id, "lang": 'en'})
|
||
|
|
||
|
return _responds(RESULT_FAILURE, msg="Either indexerid or name is required")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardSearchIndexers(CMD_SickGearSearchIndexers):
|
||
|
_help = {"desc": "search for show on the tvdb with a given string and language",
|
||
|
"optionalParameters": {"name": {"desc": "name of the show you want to search for"},
|
||
|
"indexerid": {"desc": "thetvdb.com unique id of a show"},
|
||
|
"lang": {"desc": "the 2 letter abbreviation lang id"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.searchtv",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
kwargs['indexers'] = TVINFO_TVDB
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearSearchIndexers.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearSetDefaults(ApiCall):
|
||
|
_help = {"desc": "set various sickgear default system values",
|
||
|
"optionalParameters": {"initial": {"desc": "initial quality to use when adding shows"},
|
||
|
"archive": {"desc": "archive quality to use when adding shows"},
|
||
|
"flatten_folders": {"desc": "flatten show subfolders when adding shows"},
|
||
|
"status": {"desc": "status to change episodes to with missing media"},
|
||
|
"future_show_paused": {"desc": "show/hide paused shows on the daily schedule page"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.initial, args = self.check_params(args, kwargs, "initial", None, False, "list", [q for q in quality_map])
|
||
|
self.archive, args = self.check_params(args, kwargs, "archive", None, False, "list", [q for q in quality_map])
|
||
|
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", [])
|
||
|
self.status, args = self.check_params(args, kwargs, "status", None, False, "string",
|
||
|
["wanted", "skipped", "archived", "ignored"])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set various sickgear default system values """
|
||
|
|
||
|
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:
|
||
|
sickgear.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 because of the above
|
||
|
if self.status not 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")
|
||
|
sickgear.STATUS_DEFAULT = self.status
|
||
|
|
||
|
if None is not self.flatten_folders:
|
||
|
sickgear.FLATTEN_FOLDERS_DEFAULT = int(self.flatten_folders)
|
||
|
|
||
|
if None is not self.future_show_paused:
|
||
|
sickgear.EPISODE_VIEW_DISPLAY_PAUSED = int(self.future_show_paused)
|
||
|
|
||
|
sickgear.save_config()
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, msg="Saved defaults")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardSetDefaults(CMD_SickGearSetDefaults):
|
||
|
_help = {"desc": "set various sickgear default system values",
|
||
|
"optionalParameters": {"initial": {"desc": "initial quality to use when adding shows"},
|
||
|
"archive": {"desc": "archive quality to use when adding shows"},
|
||
|
"flatten_folders": {"desc": "flatten show subfolders when adding shows"},
|
||
|
"status": {"desc": "status to change episodes to with missing media"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.setdefaults",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearSetDefaults.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearSetSceneNumber(ApiCall):
|
||
|
_help = {"desc": "set scene numbers for a show",
|
||
|
"requiredParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"forSeason": {"desc": "season number of a show"},
|
||
|
"forEpisode": {"desc": "episode number of a show"},
|
||
|
"forAbsolute": {"desc": "absolute episode number of a show"},
|
||
|
"sceneSeason": {"desc": "scene season number of a show to set"},
|
||
|
"sceneEpisode": {"desc": "scene episode number of a show to set"},
|
||
|
"sceneAbsolute": {"desc": "scene absolute episode number of a show to set"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.tvid, args = self.check_params(
|
||
|
args, kwargs, "indexer", None, True, "int", [i for i in indexer_api.TVInfoAPI().sources])
|
||
|
# optional
|
||
|
self.forSeason, args = self.check_params(args, kwargs, "forSeason", None, False, "int", [])
|
||
|
self.forEpisode, args = self.check_params(args, kwargs, "forEpisode", None, False, "int", [])
|
||
|
self.forAbsolute, args = self.check_params(args, kwargs, "forAbsolute", None, False, "int", [])
|
||
|
self.sceneSeason, args = self.check_params(args, kwargs, "sceneSeason", None, False, "int", [])
|
||
|
self.sceneEpisode, args = self.check_params(args, kwargs, "sceneEpisode", None, False, "int", [])
|
||
|
self.sceneAbsolute, args = self.check_params(args, kwargs, "sceneAbsolute", None, False, "int", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" saving scene numbers """
|
||
|
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Can't find show")
|
||
|
if not show_obj.is_scene:
|
||
|
return _responds(RESULT_FAILURE, msg="Show scene numbering disabled")
|
||
|
|
||
|
result = set_scene_numbering_helper(self.tvid, self.prodid, self.forSeason, self.forEpisode,
|
||
|
self.forAbsolute, self.sceneSeason, self.sceneEpisode, self.sceneAbsolute)
|
||
|
|
||
|
if not result['success']:
|
||
|
return _responds(RESULT_FAILURE, result)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, result)
|
||
|
|
||
|
|
||
|
class CMD_SickGearActivateSceneNumber(ApiCall):
|
||
|
_help = {"desc": "de-/activate scene numbers",
|
||
|
"requiredParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
"activate": {"desc": "de-/activate scene numbering"}},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.activate, args = self.check_params(args, kwargs, "activate", None, True, "bool", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" de-/activate scene numbers """
|
||
|
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Can't find show")
|
||
|
|
||
|
show_obj.scene = int(self.activate)
|
||
|
show_obj.save_to_db()
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, data={'indexer': self.tvid, 'indexerid': self.prodid,
|
||
|
'show_name': show_obj.name, 'scenenumbering': show_obj.is_scene},
|
||
|
msg="Scene Numbering %sactivated" % ('de', '')[self.activate])
|
||
|
|
||
|
|
||
|
class CMD_SickGearShutdown(ApiCall):
|
||
|
_help = {"desc": "shutdown sickgear"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" shutdown sickgear """
|
||
|
sickgear.events.put(sickgear.events.SystemEvent.SHUTDOWN)
|
||
|
return _responds(RESULT_SUCCESS, msg="SickGear is shutting down...")
|
||
|
|
||
|
|
||
|
class CMD_SickBeardShutdown(CMD_SickGearShutdown):
|
||
|
_help = {"desc": "shutdown sickgear",
|
||
|
"SickGearCommand": "sg.shutdown",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShutdown.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearListIgnoreWords(ApiCall):
|
||
|
_help = {"desc": "get ignore word list (uses global list if both indexerid and indexer params are not set)",
|
||
|
"optionalParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, False, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, False, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get ignore word list """
|
||
|
if self.tvid and self.prodid:
|
||
|
my_db = db.DBConnection()
|
||
|
sql_result = my_db.select(
|
||
|
'SELECT show_name, rls_ignore_words, rls_global_exclude_ignore'
|
||
|
' FROM tv_shows'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[self.tvid, self.prodid])
|
||
|
if sql_result:
|
||
|
ignore_words = sql_result[0]['rls_ignore_words']
|
||
|
return_data = {'type': 'show', 'indexer': self.tvid, 'indexerid': self.prodid,
|
||
|
'show name': sql_result[0]['show_name'],
|
||
|
'global exclude ignore':
|
||
|
helpers.split_word_str(sql_result[0]['rls_global_exclude_ignore'])[0]}
|
||
|
return_type = '%s:' % sql_result[0]['show_name']
|
||
|
else:
|
||
|
return _responds(RESULT_FAILURE, msg='Show not found.')
|
||
|
elif (None is self.tvid) != (None is self.prodid):
|
||
|
return _responds(RESULT_FAILURE, msg='You must supply indexer + indexerid.')
|
||
|
else:
|
||
|
ignore_words = helpers.generate_word_str(sickgear.IGNORE_WORDS, sickgear.IGNORE_WORDS_REGEX)
|
||
|
return_data = {'type': 'global'}
|
||
|
return_type = 'Global'
|
||
|
|
||
|
return_data['use regex'] = ignore_words.startswith('regex:')
|
||
|
return_data['ignore words'] = [w.strip() for w in ignore_words.replace('regex:', '').split(',') if w.strip()]
|
||
|
return _responds(RESULT_SUCCESS, data=return_data, msg="%s ignore word list" % return_type)
|
||
|
|
||
|
|
||
|
class ApiSetWords(ApiCall):
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, False, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, False, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.add, args = self.check_params(args, kwargs, "add", None, False, "list", [])
|
||
|
self.remove, args = self.check_params(args, kwargs, "remove", None, False, "list", [])
|
||
|
self.add_exclude, args = self.check_params(args, kwargs, "add_exclude", None, False, "list", [])
|
||
|
self.remove_exclude, args = self.check_params(args, kwargs, "remove_exclude", None, False, "list", [])
|
||
|
self.regex, args = self.check_params(args, kwargs, "regex", None, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def _create_words(self, words, exclude_list=None):
|
||
|
exclude_list = exclude_list or []
|
||
|
word_list = {w.strip() for w in words.replace('regex:', '').split(',') if w.strip()}
|
||
|
|
||
|
use_regex = self.regex if None is not self.regex else words.startswith('regex:')
|
||
|
|
||
|
for a in self.add or []:
|
||
|
if a not in exclude_list:
|
||
|
word_list.add(a.strip())
|
||
|
for r in self.remove or []:
|
||
|
try:
|
||
|
word_list.remove(r)
|
||
|
except KeyError:
|
||
|
pass
|
||
|
|
||
|
return use_regex, word_list, \
|
||
|
('', '%s%s' % (('', 'regex:')[use_regex],
|
||
|
', '.join([w.strip() for w in word_list if w.strip()])))[0 < len(word_list)]
|
||
|
|
||
|
|
||
|
class CMD_SickGearSetIgnoreWords(ApiSetWords):
|
||
|
_help = {"desc": "set ignore word list (uses global list if both indexerid and indexer params are not set)",
|
||
|
"optionalParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
"add": {"desc": "add words to list"},
|
||
|
"remove": {"desc": "remove words from list"},
|
||
|
"add_exclude": {"desc": "add global exclude words"},
|
||
|
"remove_exclude": {"desc": "remove global exclude words"},
|
||
|
"regex": {"desc": "interpret ALL (including existing) ignore words as regex"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
super(CMD_SickGearSetIgnoreWords, self).__init__(handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set ignore word list """
|
||
|
if (not self.add and not self.remove and not self.add_exclude and not self.remove_exclude) or \
|
||
|
((self.add_exclude or self.remove_exclude) and not (self.tvid and self.prodid)):
|
||
|
return _responds(RESULT_FAILURE, msg=('No indexer, indexerid provided',
|
||
|
'No words to add/remove provided')[None is not self.tvid and
|
||
|
None is not self.prodid])
|
||
|
|
||
|
use_regex = None
|
||
|
return_type = ''
|
||
|
ignore_list = set()
|
||
|
|
||
|
if self.tvid and self.prodid:
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
return_data = {'type': 'show', 'indexer': self.tvid, 'indexerid': self.prodid,
|
||
|
'show name': show_obj.name}
|
||
|
|
||
|
if any([self.add, self.remove, self.add_exclude, self.remove_exclude]):
|
||
|
my_db = db.DBConnection()
|
||
|
sql_results = my_db.select(
|
||
|
'SELECT show_name, rls_ignore_words, rls_global_exclude_ignore'
|
||
|
' FROM tv_shows'
|
||
|
' WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid])
|
||
|
if sql_results:
|
||
|
ignore_words = sql_results[0]['rls_ignore_words']
|
||
|
ignore_list, use_regex = helpers.split_word_str(ignore_words)
|
||
|
exclude_ignore = helpers.split_word_str(sql_results[0]['rls_global_exclude_ignore'])[0]
|
||
|
exclude_ignore = {i for i in exclude_ignore if i in sickgear.IGNORE_WORDS}
|
||
|
return_type = '%s:' % sql_results[0]['show_name']
|
||
|
|
||
|
if self.add or self.remove:
|
||
|
use_regex, ignore_list, new_ignore_words = \
|
||
|
self._create_words(ignore_words, sickgear.IGNORE_WORDS)
|
||
|
my_db.action('UPDATE tv_shows'
|
||
|
' SET rls_ignore_words = ?'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[new_ignore_words, self.tvid, self.prodid])
|
||
|
show_obj.rls_ignore_words, show_obj.rls_ignore_words_regex = \
|
||
|
helpers.split_word_str(new_ignore_words)
|
||
|
|
||
|
if self.add_exclude or self.remove_exclude:
|
||
|
for a in self.add_exclude or []:
|
||
|
if a in sickgear.IGNORE_WORDS:
|
||
|
exclude_ignore.add(a)
|
||
|
for r in self.remove_exclude or []:
|
||
|
try:
|
||
|
exclude_ignore.remove(r)
|
||
|
except KeyError:
|
||
|
pass
|
||
|
|
||
|
my_db.action('UPDATE tv_shows SET rls_global_exclude_ignore = ?'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[helpers.generate_word_str(exclude_ignore), self.tvid, self.prodid])
|
||
|
show_obj.rls_global_exclude_ignore = copy.copy(exclude_ignore)
|
||
|
|
||
|
return_data['global exclude ignore'] = exclude_ignore
|
||
|
elif (None is self.tvid) != (None is self.prodid):
|
||
|
return _responds(RESULT_FAILURE, msg='You must supply indexer + indexerid.')
|
||
|
else:
|
||
|
ignore_words = helpers.generate_word_str(sickgear.IGNORE_WORDS, sickgear.IGNORE_WORDS_REGEX)
|
||
|
use_regex, ignore_list, new_ignore_words = self._create_words(ignore_words)
|
||
|
sickgear.IGNORE_WORDS, sickgear.IGNORE_WORDS_REGEX = helpers.split_word_str(new_ignore_words)
|
||
|
sickgear.save_config()
|
||
|
return_data = {'type': 'global'}
|
||
|
return_type = 'Global'
|
||
|
|
||
|
if None is not use_regex:
|
||
|
return_data['use regex'] = use_regex
|
||
|
elif None is not self.regex:
|
||
|
return_data['use regex'] = self.regex
|
||
|
return_data['ignore words'] = ignore_list
|
||
|
clean_ignore_require_words()
|
||
|
return _responds(RESULT_SUCCESS, data=return_data, msg="%s set ignore word list" % return_type)
|
||
|
|
||
|
|
||
|
class CMD_SickGearListRequireWords(ApiCall):
|
||
|
_help = {"desc": "get require word list (uses global list if both indexerid and indexer params are not set)",
|
||
|
"optionalParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, False, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, False, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
self.old_call = False
|
||
|
|
||
|
def run(self):
|
||
|
""" get require word list """
|
||
|
if self.tvid and self.prodid:
|
||
|
my_db = db.DBConnection()
|
||
|
sql_result = my_db.select(
|
||
|
'SELECT show_name, rls_require_words, rls_global_exclude_require'
|
||
|
' FROM tv_shows'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[self.tvid, self.prodid])
|
||
|
if sql_result:
|
||
|
require_words = sql_result[0]['rls_require_words']
|
||
|
return_data = {'type': 'show', 'indexer': self.tvid, 'indexerid': self.prodid,
|
||
|
'show name': sql_result[0]['show_name'],
|
||
|
'global exclude require':
|
||
|
helpers.split_word_str(sql_result[0]['rls_global_exclude_require'])[0]}
|
||
|
return_type = '%s:' % sql_result[0]['show_name']
|
||
|
else:
|
||
|
return _responds(RESULT_FAILURE, msg='Show not found.')
|
||
|
elif (None is self.tvid) != (None is self.prodid):
|
||
|
return _responds(RESULT_FAILURE, msg='You must supply indexer + indexerid.')
|
||
|
else:
|
||
|
require_words = helpers.generate_word_str(sickgear.REQUIRE_WORDS, sickgear.REQUIRE_WORDS_REGEX)
|
||
|
return_data = {'type': 'global'}
|
||
|
return_type = 'Global'
|
||
|
|
||
|
return_data['use regex'] = require_words.startswith('regex:')
|
||
|
return_data['require%s words' % ('', 'd')[self.old_call]] = [w.strip() for w in
|
||
|
require_words.replace('regex:', '').split(',')
|
||
|
if w.strip()]
|
||
|
return _responds(RESULT_SUCCESS, data=return_data, msg="%s require word list" % return_type)
|
||
|
|
||
|
|
||
|
class CMD_SickGearListRequireWords_old(CMD_SickGearListRequireWords):
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
CMD_SickGearListRequireWords.__init__(self, handler, args, kwargs)
|
||
|
self.old_call = True
|
||
|
|
||
|
|
||
|
class CMD_SickGearSetRequireWords(ApiSetWords):
|
||
|
_help = {"desc": "set require word list (uses global list if both indexerid and indexer params are not set)",
|
||
|
"optionalParameters": {"indexerid": {"desc": "unique id of a show"},
|
||
|
"indexer": {"desc": "indexer of a show"},
|
||
|
"add": {"desc": "add words to list"},
|
||
|
"remove": {"desc": "remove words from list"},
|
||
|
"add_exclude": {"desc": "add global exclude words"},
|
||
|
"remove_exclude": {"desc": "remove global exclude words"},
|
||
|
"regex": {"desc": "interpret ALL (including existing) ignore words as regex"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
super(CMD_SickGearSetRequireWords, self).__init__(handler, args, kwargs)
|
||
|
self.old_call = False
|
||
|
|
||
|
def run(self):
|
||
|
""" set require words """
|
||
|
if (not self.add and not self.remove and not self.add_exclude and not self.remove_exclude) or \
|
||
|
((self.add_exclude or self.remove_exclude) and not (self.tvid and self.prodid)):
|
||
|
return _responds(RESULT_FAILURE, msg=('No indexer, indexerid provided',
|
||
|
'No words to add/remove provided')[None is not self.tvid and
|
||
|
None is not self.prodid])
|
||
|
|
||
|
use_regex = None
|
||
|
return_type = ''
|
||
|
require_list = set()
|
||
|
|
||
|
if self.tvid and self.prodid:
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
return_data = {'type': 'show', 'indexer': self.tvid, 'indexerid': self.prodid,
|
||
|
'show name': show_obj.name}
|
||
|
|
||
|
if any([self.add, self.remove, self.add_exclude, self.remove_exclude]):
|
||
|
my_db = db.DBConnection()
|
||
|
sql_result = my_db.select(
|
||
|
'SELECT show_name, rls_require_words, rls_global_exclude_require'
|
||
|
' FROM tv_shows'
|
||
|
' WHERE indexer = ? AND indexer_id = ?', [self.tvid, self.prodid])
|
||
|
if sql_result:
|
||
|
require_words = sql_result[0]['rls_require_words']
|
||
|
require_list, use_regex = helpers.split_word_str(require_words)
|
||
|
exclude_require = helpers.split_word_str(sql_result[0]['rls_global_exclude_require'])[0]
|
||
|
exclude_require = {r for r in exclude_require if r in sickgear.REQUIRE_WORDS}
|
||
|
return_type = '%s:' % sql_result[0]['show_name']
|
||
|
|
||
|
if self.add or self.remove:
|
||
|
use_regex, require_list, new_require_words = \
|
||
|
self._create_words(require_words, sickgear.REQUIRE_WORDS)
|
||
|
my_db.action('UPDATE tv_shows'
|
||
|
' SET rls_require_words = ?'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[new_require_words, self.tvid, self.prodid])
|
||
|
show_obj.rls_require_words, show_obj.rls_require_words_regex = \
|
||
|
helpers.split_word_str(new_require_words)
|
||
|
|
||
|
if self.add_exclude or self.remove_exclude:
|
||
|
for a in self.add_exclude or []:
|
||
|
if a in sickgear.REQUIRE_WORDS:
|
||
|
exclude_require.add(a)
|
||
|
for r in self.remove_exclude or []:
|
||
|
try:
|
||
|
exclude_require.remove(r)
|
||
|
except KeyError:
|
||
|
pass
|
||
|
my_db.action(
|
||
|
'UPDATE tv_shows SET rls_global_exclude_require = ?'
|
||
|
' WHERE indexer = ? AND indexer_id = ?',
|
||
|
[helpers.generate_word_str(exclude_require), self.tvid, self.prodid])
|
||
|
show_obj.rls_global_exclude_require = copy.copy(exclude_require)
|
||
|
|
||
|
return_data['global exclude require'] = exclude_require
|
||
|
elif (None is self.tvid) != (None is self.prodid):
|
||
|
return _responds(RESULT_FAILURE, msg='You must supply indexer + indexerid.')
|
||
|
else:
|
||
|
require_words = helpers.generate_word_str(sickgear.REQUIRE_WORDS, sickgear.REQUIRE_WORDS_REGEX)
|
||
|
use_regex, require_list, new_require_words = self._create_words(require_words)
|
||
|
|
||
|
sickgear.REQUIRE_WORDS, sickgear.REQUIRE_WORDS_REGEX = helpers.split_word_str(new_require_words)
|
||
|
sickgear.save_config()
|
||
|
return_data = {'type': 'global'}
|
||
|
return_type = 'Global'
|
||
|
|
||
|
if None is not use_regex:
|
||
|
return_data['use regex'] = use_regex
|
||
|
elif None is not self.regex:
|
||
|
return_data['use regex'] = self.regex
|
||
|
return_data['require%s words' % ('', 'd')[self.old_call]] = require_list
|
||
|
clean_ignore_require_words()
|
||
|
return _responds(RESULT_SUCCESS, data=return_data,
|
||
|
msg="%s set %s" % (return_type, ('require word list', 'requried words')[self.old_call]))
|
||
|
|
||
|
|
||
|
class CMD_SickGearSetRequireWords_old(CMD_SickGearSetRequireWords):
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
CMD_SickGearSetRequireWords.__init__(self, handler, args, kwargs)
|
||
|
self.old_call = True
|
||
|
|
||
|
|
||
|
class CMD_SickGearUpdateWatchedState(ApiCall):
|
||
|
_help = {"desc": "Update db with details of media file(s) that are watched or unwatched",
|
||
|
"requiredParameters": {
|
||
|
"payloadjson": {
|
||
|
"desc": "a dict of dicts transmitted as JSON via POST request"},
|
||
|
}}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.payloadjson, args = self.check_params(args, kwargs, "payloadjson", None, True, "dict", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" Update db with details of media file that is watched or unwatched """
|
||
|
payload = self.payloadjson.copy()
|
||
|
|
||
|
webserve.MainHandler.update_watched_state(payload, as_json=False)
|
||
|
|
||
|
if not payload:
|
||
|
return _responds(RESULT_FAILURE, msg='Request made to SickGear with invalid payload')
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, payload)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShow(ApiCall):
|
||
|
_help = {"desc": "get show information",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"overview": {"desc": "include overview"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
self.overview, args = self.check_params(args, kwargs, "overview", False, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get show information """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
showDict = {"season_list": CMD_SickGearShowSeasonList(self.handler, (),
|
||
|
{'indexer': self.tvid, 'indexerid': self.prodid}
|
||
|
).run()['data'],
|
||
|
"cache": CMD_SickGearShowCache(self.handler, (), {'indexer': self.tvid,
|
||
|
'indexerid': self.prodid}).run()['data']}
|
||
|
|
||
|
genreList = []
|
||
|
if show_obj.genre:
|
||
|
genreListTmp = show_obj.genre.split("|")
|
||
|
for genre in genreListTmp:
|
||
|
if genre:
|
||
|
genreList.append(genre)
|
||
|
showDict["genre"] = genreList
|
||
|
showDict["quality"] = _get_quality_string(show_obj.quality)
|
||
|
|
||
|
anyQualities, bestQualities = _mapQuality(show_obj.quality)
|
||
|
showDict["quality_details"] = {"initial": anyQualities, "archive": bestQualities}
|
||
|
|
||
|
try:
|
||
|
showDict["location"] = show_obj.location
|
||
|
except exceptions_helper.ShowDirNotFoundException:
|
||
|
showDict["location"] = ""
|
||
|
|
||
|
showDict["language"] = show_obj.lang
|
||
|
showDict["show_name"] = show_obj.name
|
||
|
showDict["paused"] = show_obj.paused
|
||
|
showDict["subtitles"] = show_obj.subtitles
|
||
|
showDict["air_by_date"] = show_obj.air_by_date
|
||
|
showDict["flatten_folders"] = show_obj.flatten_folders
|
||
|
showDict["sports"] = show_obj.sports
|
||
|
showDict["anime"] = show_obj.anime
|
||
|
# clean up tvdb horrible airs field
|
||
|
showDict["airs"] = str(show_obj.airs).replace('am', ' AM').replace('pm', ' PM').replace(' ', ' ')
|
||
|
showDict["indexerid"] = self.prodid
|
||
|
showDict["tvrage_id"] = show_obj.ids.get(TVINFO_TVRAGE, {'id': 0})['id']
|
||
|
showDict['ids'] = {k: v.get('id') for k, v in iteritems(show_obj.ids)}
|
||
|
showDict["tvrage_name"] = show_obj.name
|
||
|
showDict["network"] = show_obj.network
|
||
|
if not showDict["network"]:
|
||
|
showDict["network"] = ""
|
||
|
showDict["status"] = show_obj.status
|
||
|
showDict["scenenumbering"] = show_obj.is_scene
|
||
|
showDict["upgrade_once"] = show_obj.upgrade_once
|
||
|
showDict["ignorewords"] = helpers.generate_word_str(show_obj.rls_ignore_words, show_obj.rls_ignore_words_regex)
|
||
|
showDict["global_exclude_ignore"] = helpers.generate_word_str(show_obj.rls_global_exclude_ignore)
|
||
|
showDict["requirewords"] = helpers.generate_word_str(show_obj.rls_require_words, show_obj.rls_require_words_regex)
|
||
|
showDict["global_exclude_require"] = helpers.generate_word_str(show_obj.rls_global_exclude_require)
|
||
|
if self.overview:
|
||
|
showDict["overview"] = show_obj.overview
|
||
|
showDict["prune"] = show_obj.prune
|
||
|
showDict["tag"] = show_obj.tag
|
||
|
showDict["imdb_id"] = show_obj.imdbid
|
||
|
showDict["classification"] = show_obj.classification
|
||
|
showDict["runtime"] = show_obj.runtime
|
||
|
showDict["startyear"] = show_obj.startyear
|
||
|
showDict["indexer"] = show_obj.tvid
|
||
|
timezone, showDict['timezone'] = network_timezones.get_network_timezone(showDict['network'], return_name=True)
|
||
|
|
||
|
if show_obj.next_episode():
|
||
|
dtEpisodeAirs = SGDatetime.convert_to_setting(
|
||
|
network_timezones.parse_date_time(show_obj.nextaired, showDict['airs'], timezone))
|
||
|
showDict['airs'] = SGDatetime.sbftime(dtEpisodeAirs, t_preset=timeFormat).lstrip('0').replace(' 0', ' ')
|
||
|
showDict['next_ep_airdate'] = SGDatetime.sbfdate(dtEpisodeAirs, d_preset=dateFormat)
|
||
|
else:
|
||
|
showDict['next_ep_airdate'] = ''
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, showDict)
|
||
|
|
||
|
|
||
|
class CMD_Show(CMD_SickGearShow):
|
||
|
_help = {"desc": "get thetvdb.com show information",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"overview": {"desc": "include overview"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShow.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowAddExisting(ApiCall):
|
||
|
_help = {"desc": "add a show to SickGear from an existing folder",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "production id of a show"},
|
||
|
"location": {"desc": "full path to the existing folder for the show"}
|
||
|
},
|
||
|
"optionalParameters": {"initial": {"desc": "initial quality for the show"},
|
||
|
"archive": {"desc": "archive quality for the show"},
|
||
|
"upgrade_once": {"desc": "upgrade only once"},
|
||
|
"pause": {"desc": "pause show search tasks to allow edits"},
|
||
|
"subtitles": {"desc": "allow search episode subtitle"},
|
||
|
"flatten_folders": {"desc": "flatten subfolders for the show"}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().search_sources])
|
||
|
self.location, args = self.check_params(args, kwargs, "location", None, True, "string", [])
|
||
|
# optional
|
||
|
self.initial, args = self.check_params(args, kwargs, "initial", None, False, "list", [q for q in quality_map])
|
||
|
self.archive, args = self.check_params(args, kwargs, "archive", None, False, "list", [q for q in quality_map])
|
||
|
self.upgradeonce, args = self.check_params(args, kwargs, "upgrade_once", False, False, "bool", [])
|
||
|
|
||
|
self.pause, args = self.check_params(args, kwargs, "pause", int(sickgear.PAUSE_DEFAULT), False, "int", [])
|
||
|
self.flatten_folders, args = self.check_params(args, kwargs, "flatten_folders",
|
||
|
str(sickgear.FLATTEN_FOLDERS_DEFAULT), False, "bool", [])
|
||
|
self.subtitles, args = self.check_params(args, kwargs, "subtitles", int(sickgear.USE_SUBTITLES), False, "int",
|
||
|
[])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" add a show to SickGear from an existing folder """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if show_obj:
|
||
|
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')
|
||
|
|
||
|
lINDEXER_API_PARMS = sickgear.TVInfoAPI(self.tvid).api_params.copy()
|
||
|
lINDEXER_API_PARMS['language'] = 'en'
|
||
|
lINDEXER_API_PARMS['custom_ui'] = classes.AllShowInfosNoFilterListUI
|
||
|
lINDEXER_API_PARMS['actors'] = False
|
||
|
|
||
|
t = sickgear.TVInfoAPI(self.tvid).setup(**lINDEXER_API_PARMS)
|
||
|
|
||
|
try:
|
||
|
myShow = t[int(self.prodid), False]
|
||
|
except BaseTVinfoError as e:
|
||
|
self.log(u"Unable to find show with id " + str(self.tvid), logger.WARNING)
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
indexerName = None
|
||
|
if None is myShow or getattr(t, 'show_not_found', False) or not myShow.data['seriesname']:
|
||
|
self.log(
|
||
|
"Found show with tvid %s prodid %s, however it contained no show name" % (self.tvid, self.prodid),
|
||
|
logger.DEBUG)
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
else:
|
||
|
indexerName = myShow.data['seriesname']
|
||
|
|
||
|
if not indexerName:
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
# use default quality as a failsafe
|
||
|
newQuality = int(sickgear.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)
|
||
|
|
||
|
sickgear.show_queue_scheduler.action.add_show(
|
||
|
int(self.tvid), int(self.prodid), self.location,
|
||
|
quality=newQuality, upgrade_once=self.upgradeonce,
|
||
|
paused=self.pause, default_status=SKIPPED, flatten_folders=int(self.flatten_folders)
|
||
|
)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, {"name": indexerName}, indexerName + " has been queued to be added")
|
||
|
|
||
|
|
||
|
class CMD_ShowAddExisting(CMD_SickGearShowAddExisting):
|
||
|
_help = {"desc": "add a show to SickGear from an existing folder",
|
||
|
"requiredParameters": {"tvdbid": {"desc": "thetvdb.com id"},
|
||
|
"location": {"desc": "full path to the existing folder for the show"}
|
||
|
},
|
||
|
"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"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.addexisting",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
# required
|
||
|
if 'tvdbid' in kwargs and 'indexerid' not in kwargs:
|
||
|
kwargs['indexerid'], args = self.check_params(args, kwargs, "tvdbid", None, True, "int", [])
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowAddExisting.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowAddNew(ApiCall):
|
||
|
_help = {"desc": "add a new show to sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of show"},
|
||
|
"indexerid": {"desc": "id of show"},
|
||
|
},
|
||
|
"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"},
|
||
|
"upgrade_once": {"desc": "upgrade only once"},
|
||
|
"flatten_folders": {"desc": "flatten subfolders for the show"},
|
||
|
"status": {"desc": "status of missing episodes"},
|
||
|
"subtitles": {"desc": "allow search episode subtitle"},
|
||
|
"anime": {"desc": "set show to anime"},
|
||
|
"scene": {"desc": "show searches episodes by scene numbering"}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().search_sources])
|
||
|
# optional
|
||
|
self.location, args = self.check_params(args, kwargs, "location", None, False, "string", [])
|
||
|
self.initial, args = self.check_params(args, kwargs, "initial", None, False, "list", [q for q in quality_map])
|
||
|
self.archive, args = self.check_params(args, kwargs, "archive", None, False, "list", [q for q in quality_map])
|
||
|
self.upgradeonce, args = self.check_params(args, kwargs, "upgrade_once", False, False, "bool", [])
|
||
|
self.pause, args = self.check_params(args, kwargs, "pause", int(sickgear.PAUSE_DEFAULT), False, "int", [])
|
||
|
self.status, args = self.check_params(args, kwargs, "status", None, False, "string",
|
||
|
["wanted", "skipped", "archived", "ignored"])
|
||
|
self.scene, args = self.check_params(args, kwargs, "scene", int(sickgear.SCENE_DEFAULT), False, "int", [])
|
||
|
self.subtitles, args = self.check_params(
|
||
|
args, kwargs, "subtitles", int(sickgear.USE_SUBTITLES), False, "int", [])
|
||
|
self.flatten_folders, args = self.check_params(args, kwargs, "flatten_folders",
|
||
|
str(sickgear.FLATTEN_FOLDERS_DEFAULT), False, "bool", [])
|
||
|
self.anime, args = self.check_params(args, kwargs, "anime", int(sickgear.ANIME_DEFAULT), False, "int", [])
|
||
|
self.lang = 'en'
|
||
|
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" add a show to SickGear from an existing folder """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="An existing indexerid already exists in database")
|
||
|
|
||
|
if not self.location:
|
||
|
if "" != sickgear.ROOT_DIRS:
|
||
|
root_dirs = sickgear.ROOT_DIRS.split('|')
|
||
|
root_dirs.pop(0)
|
||
|
default_index = int(sickgear.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")
|
||
|
|
||
|
# use default quality as a failsafe
|
||
|
newQuality = int(sickgear.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 = sickgear.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 self.status not 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
|
||
|
|
||
|
lINDEXER_API_PARMS = sickgear.TVInfoAPI(self.tvid).api_params.copy()
|
||
|
lINDEXER_API_PARMS['language'] = 'en'
|
||
|
lINDEXER_API_PARMS['custom_ui'] = classes.AllShowInfosNoFilterListUI
|
||
|
lINDEXER_API_PARMS['actors'] = False
|
||
|
|
||
|
t = sickgear.TVInfoAPI(self.tvid).setup(**lINDEXER_API_PARMS)
|
||
|
|
||
|
try:
|
||
|
myShow = t[int(self.prodid), False]
|
||
|
except BaseTVinfoError as e:
|
||
|
self.log(u"Unable to find show with id " + str(self.tvid), logger.WARNING)
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
indexerName = None
|
||
|
if None is myShow or getattr(t, 'show_not_found', False) or not myShow.data['seriesname']:
|
||
|
self.log(
|
||
|
"Found show with tvid %s prodid %s, however it contained no show name" % (self.tvid, self.prodid),
|
||
|
logger.DEBUG)
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
else:
|
||
|
indexerName = myShow.data['seriesname']
|
||
|
|
||
|
if not indexerName:
|
||
|
return _responds(RESULT_FAILURE, msg="Unable to retrieve information from indexer")
|
||
|
|
||
|
# moved the logic check to the end in an attempt to eliminate empty directory being created from previous errors
|
||
|
showPath = helpers.generate_show_dir_name(self.location, indexerName)
|
||
|
|
||
|
# don't create show dir if config says not to
|
||
|
if sickgear.ADD_SHOWS_WO_DIR:
|
||
|
self.log(u"Skipping initial creation of " + showPath + " due to config.ini setting")
|
||
|
else:
|
||
|
dir_exists = helpers.make_dir(showPath)
|
||
|
if not dir_exists:
|
||
|
self.log(u"Unable to create the folder " + showPath + ", can't add the show", logger.ERROR)
|
||
|
return _responds(RESULT_FAILURE, {"path": showPath},
|
||
|
"Unable to create the folder " + showPath + ", can't add the show")
|
||
|
else:
|
||
|
helpers.chmod_as_parent(showPath)
|
||
|
|
||
|
sickgear.show_queue_scheduler.action.add_show(
|
||
|
int(self.tvid), int(self.prodid), showPath,
|
||
|
quality=newQuality, upgrade_once=self.upgradeonce,
|
||
|
paused=self.pause, default_status=newStatus, scene=self.scene, subtitles=self.subtitles,
|
||
|
flatten_folders=int(self.flatten_folders), anime=self.anime,
|
||
|
new_show=True, lang=self.lang
|
||
|
)
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, {"name": indexerName}, indexerName + " has been queued to be added")
|
||
|
|
||
|
|
||
|
class CMD_ShowAddNew(CMD_SickGearShowAddNew):
|
||
|
_help = {"desc": "add a new show to sickgear",
|
||
|
"requiredParameters": {"tvdbid": {"desc": "thetvdb.com id"}
|
||
|
},
|
||
|
"optionalParameters": {"location": {"desc": "base path for where the show folder is to be created"},
|
||
|
"initial": {"desc": "initial quality for the show"},
|
||
|
"archive": {"desc": "archive quality for the show"},
|
||
|
"pause": {"desc": "pause show search tasks to allow edits"},
|
||
|
"status": {"desc": "status of missing episodes"},
|
||
|
"scene": {"desc": "show searches episodes by scene numbering"},
|
||
|
"subtitles": {"desc": "allow search episode subtitle"},
|
||
|
"flatten_folders": {"desc": "flatten subfolders for the show"},
|
||
|
"anime": {"desc": "set show to anime"},
|
||
|
"lang": {"desc": "the 2 letter lang abbreviation id"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.addnew",
|
||
|
}
|
||
|
|
||
|
valid_languages = {
|
||
|
'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}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
if 'tvdbid' in kwargs and 'indexerid' not in kwargs:
|
||
|
kwargs['indexerid'], args = self.check_params(args, kwargs, "tvdbid", None, True, "int", [])
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowAddNew.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowCache(ApiCall):
|
||
|
_help = {"desc": "check sickgear's cache to see if the banner or poster image for a show is valid",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" check sickgear's cache to see if the banner or poster image for a show is valid """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
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(show_obj.tvid, show_obj.prodid)):
|
||
|
has_poster = 1
|
||
|
if ek.ek(os.path.isfile, cache_obj.banner_path(show_obj.tvid, show_obj.prodid)):
|
||
|
has_banner = 1
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, {"poster": has_poster, "banner": has_banner})
|
||
|
|
||
|
|
||
|
class CMD_ShowCache(CMD_SickGearShowCache):
|
||
|
_help = {"desc": "check sickgear's cache to see if the banner or poster image for a show is valid",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.cache",
|
||
|
}
|
||
|
|
||
|
def __init__(self,
|
||
|
handler,
|
||
|
args, # type: List
|
||
|
kwargs # type: Dict
|
||
|
):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowCache.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowDelete(ApiCall):
|
||
|
_help = {"desc": "delete a show from sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"full": {"desc": "delete files/folder of show"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
self.full_delete, args = self.check_params(args, kwargs, "full", False, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" delete a show from sickgear """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
if sickgear.show_queue_scheduler.action.isBeingAdded(
|
||
|
show_obj) or sickgear.show_queue_scheduler.action.isBeingUpdated(show_obj):
|
||
|
return _responds(RESULT_FAILURE, msg="Show can not be deleted while being added or updated")
|
||
|
|
||
|
show_obj.delete_show(full=self.full_delete)
|
||
|
return _responds(RESULT_SUCCESS, msg='%s has been deleted' % show_obj.unique_name)
|
||
|
|
||
|
|
||
|
class CMD_ShowDelete(CMD_SickGearShowDelete):
|
||
|
_help = {"desc": "delete a show from sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.delete",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowDelete.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowGetQuality(ApiCall):
|
||
|
_help = {"desc": "get the quality setting for a show in sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the quality setting for a show in sickgear """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
anyQualities, bestQualities = _mapQuality(show_obj.quality)
|
||
|
|
||
|
data = {"initial": anyQualities, "archive": bestQualities}
|
||
|
|
||
|
if not self.sickbeard_call:
|
||
|
data['upgrade_once'] = show_obj.upgrade_once
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, data)
|
||
|
|
||
|
|
||
|
class CMD_ShowGetQuality(CMD_SickGearShowGetQuality):
|
||
|
_help = {"desc": "get the quality setting for a show in sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.getquality",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowGetQuality.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowGetPoster(ApiCall):
|
||
|
_help = {"desc": "get the poster stored for a show in sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the poster for a show in sickgear """
|
||
|
return {'outputType': 'image', 'image': self.handler.show_poster({self.tvid: self.prodid}, 'poster', True)}
|
||
|
|
||
|
|
||
|
class CMD_ShowGetPoster(CMD_SickGearShowGetPoster):
|
||
|
_help = {"desc": "get the poster stored for a show in sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.getposter",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowGetPoster.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowGetBanner(ApiCall):
|
||
|
_help = {"desc": "get the banner stored for a show in sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the banner for a show in sickgear """
|
||
|
return {'outputType': 'image', 'image': self.handler.show_poster({self.tvid: self.prodid}, 'banner', True)}
|
||
|
|
||
|
|
||
|
class CMD_ShowGetBanner(CMD_SickGearShowGetBanner):
|
||
|
_help = {"desc": "get the banner stored for a show in sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.getbanner",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowGetBanner.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowListFanart(ApiCall):
|
||
|
_help = {"desc": "get list of fanarts stored for a show",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get list of fanarts stored for a show """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
fanart = []
|
||
|
rating_names = {10: 'group', 20: 'favorite', 30: 'avoid'}
|
||
|
cache_obj = image_cache.ImageCache()
|
||
|
for img in ek.ek(glob.glob, cache_obj.fanart_path(
|
||
|
show_obj.tvid, show_obj.prodid).replace('fanart.jpg', '*')) or []:
|
||
|
match = re.search(r'(\d+(?:\.(\w*?(\d*)))?\.(?:\w{5,8}))\.fanart\.', img, re.I)
|
||
|
if match and match.group(1):
|
||
|
fanart += [(match.group(1), rating_names.get(sickgear.FANART_RATINGS.get(
|
||
|
str(TVidProdid({self.tvid: self.prodid})()), {}).get(match.group(1), ''), ''))]
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, fanart)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowRateFanart(ApiCall):
|
||
|
_help = {"desc": "set a fanart rating",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
"fanartname": {"desc": "fanart name form sg.show.listfanart"},
|
||
|
"rating": {"desc": "rate: unrate, group, favorite, avoid"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
self.fanartname, args = self.check_params(args, kwargs, "fanartname", None, True, "string", [])
|
||
|
self.rating, args = self.check_params(args, kwargs, "rating", None, True, "string",
|
||
|
['unrate', 'group', 'favorite', 'avoid'])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set a fanart rating """
|
||
|
cache_obj = image_cache.ImageCache()
|
||
|
fanartfile = cache_obj.fanart_path(self.tvid, self.prodid).replace('fanart.jpg',
|
||
|
'%s.fanart.jpg' % self.fanartname)
|
||
|
if not ek.ek(os.path.isfile, fanartfile):
|
||
|
return _responds(RESULT_FAILURE, msg='Unknown Fanart')
|
||
|
fan_ratings = {'unrate': 0, 'group': 10, 'favorite': 20, 'avoid': 30}
|
||
|
show_id = TVidProdid({self.tvid: self.prodid})()
|
||
|
if 'unrate' == self.rating and str(show_id) in sickgear.FANART_RATINGS \
|
||
|
and self.fanartname in sickgear.FANART_RATINGS[str(show_id)]:
|
||
|
del sickgear.FANART_RATINGS[str(show_id)][self.fanartname]
|
||
|
else:
|
||
|
sickgear.FANART_RATINGS.setdefault(str(show_id), {})[self.fanartname] = fan_ratings[self.rating]
|
||
|
sickgear.save_config()
|
||
|
return _responds(RESULT_SUCCESS, msg='Rated Fanart: %s = %s' % (self.fanartname, self.rating))
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowGetFanart(ApiCall):
|
||
|
_help = {"desc": "get the fanart stored for a show"
|
||
|
" X-Fanartname response header returns Fanart name or default for not found",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"fanartname": {"desc": "fanart name form sg.show.listfanart"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
self.fanartname, args = self.check_params(args, kwargs, "fanartname", None, False, "string", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the fanart stored for a show """
|
||
|
cache_obj = image_cache.ImageCache()
|
||
|
default_fanartfile = ek.ek(os.path.join, sickgear.PROG_DIR, 'gui', 'slick', 'images', 'trans.png')
|
||
|
fanartfile = default_fanartfile
|
||
|
used_fanart = 'default'
|
||
|
if self.fanartname:
|
||
|
fanartfile = cache_obj.fanart_path(self.tvid, self.prodid).replace('fanart.jpg',
|
||
|
'%s.fanart.jpg' % self.fanartname)
|
||
|
if not ek.ek(os.path.isfile, fanartfile):
|
||
|
fanartfile = default_fanartfile
|
||
|
used_fanart = self.fanartname
|
||
|
else:
|
||
|
fanart = []
|
||
|
for img in ek.ek(glob.glob, cache_obj.fanart_path(self.tvid, self.prodid).replace('fanart.jpg', '*')) or []:
|
||
|
if not ek.ek(os.path.isfile, img):
|
||
|
continue
|
||
|
match = re.search(r'(\d+(?:\.(\w*?(\d*)))?\.(?:\w{5,8}))\.fanart\.', img, re.I)
|
||
|
if match and match.group(1):
|
||
|
fanart += [(img, match.group(1), sickgear.FANART_RATINGS.get(
|
||
|
str(TVidProdid({self.tvid: self.prodid})()), {}).get(match.group(1), 0))]
|
||
|
if fanart:
|
||
|
fanartsorted = sorted([f for f in fanart if f[2] != 30], key=lambda x: x[2], reverse=True)
|
||
|
max_fa = max([f[2] for f in fanartsorted])
|
||
|
fanartsorted = [f for f in fanartsorted if f[2] == max_fa]
|
||
|
if fanartsorted:
|
||
|
random_fanart = randint(0, len(fanartsorted) - 1)
|
||
|
fanartfile = fanartsorted[random_fanart][0]
|
||
|
used_fanart = fanartsorted[random_fanart][1]
|
||
|
|
||
|
if fanartfile and ek.ek(os.path.isfile, fanartfile):
|
||
|
with ek.ek(open, fanartfile, 'rb') as f:
|
||
|
mime_type, encoding = MimeTypes().guess_type(fanartfile)
|
||
|
self.handler.set_header('X-Fanartname', used_fanart)
|
||
|
self.handler.set_header('Content-Type', mime_type)
|
||
|
return {'outputType': 'image', 'image': f.read()}
|
||
|
|
||
|
# we should never get here
|
||
|
return _responds(RESULT_FAILURE, msg='No Fanart found')
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowPause(ApiCall):
|
||
|
_help = {"desc": "set the paused state for a show",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"pause": {"desc": "the pause state to set"}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set the paused state for a show """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
if self.pause:
|
||
|
show_obj.paused = True
|
||
|
show_obj.save_to_db()
|
||
|
return _responds(RESULT_SUCCESS, msg='%s has been paused' % show_obj.unique_name)
|
||
|
else:
|
||
|
show_obj.paused = False
|
||
|
show_obj.save_to_db()
|
||
|
return _responds(RESULT_SUCCESS, msg='%s has been unpaused' % show_obj.unique_name)
|
||
|
|
||
|
# return _responds(RESULT_FAILURE, msg=str(show_obj.name) + " was unable to be paused")
|
||
|
|
||
|
|
||
|
class CMD_ShowPause(CMD_SickGearShowPause):
|
||
|
_help = {"desc": "set the paused state for a show",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"pause": {"desc": "the pause state to set"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.pause",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowPause.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowRefresh(ApiCall):
|
||
|
_help = {"desc": "refresh a show in sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"}, },
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" refresh a show in sickgear """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
try:
|
||
|
sickgear.show_queue_scheduler.action.refreshShow(show_obj)
|
||
|
return _responds(RESULT_SUCCESS, msg='%s has queued to be refreshed' % show_obj.unique_name)
|
||
|
except exceptions_helper.CantRefreshException as e:
|
||
|
# TODO: log the exception
|
||
|
return _responds(RESULT_FAILURE, msg='Unable to refresh %s. %s' % (show_obj.unique_name, ex(e)))
|
||
|
|
||
|
|
||
|
class CMD_ShowRefresh(CMD_SickGearShowRefresh):
|
||
|
_help = {"desc": "refresh a show in sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.refresh",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowRefresh.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowSeasonList(ApiCall):
|
||
|
_help = {"desc": "get a season list for a show",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"sort": {"desc": "change the sort order from descending to ascending"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
self.sort, args = self.check_params(args, kwargs, "sort", "desc", False, "string",
|
||
|
["asc", "desc"]) # "asc" and "desc" default and fallback is "desc"
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get a season list for a show """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
if "asc" == self.sort:
|
||
|
# noinspection SqlRedundantOrderingDirection
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT DISTINCT season"
|
||
|
" FROM tv_episodes"
|
||
|
" WHERE indexer = ? AND showid = ?"
|
||
|
" ORDER BY season ASC",
|
||
|
[self.tvid, self.prodid])
|
||
|
else:
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT DISTINCT season"
|
||
|
" FROM tv_episodes"
|
||
|
" WHERE indexer = ? AND showid = ?"
|
||
|
" ORDER BY season DESC",
|
||
|
[self.tvid, self.prodid])
|
||
|
seasonList = [] # a list with all season numbers
|
||
|
for cur_result in sql_result:
|
||
|
seasonList.append(int(cur_result["season"]))
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, seasonList)
|
||
|
|
||
|
|
||
|
class CMD_ShowSeasonList(CMD_SickGearShowSeasonList):
|
||
|
_help = {"desc": "get a season list for a show",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"sort": {"desc": "change the sort order from descending to ascending"}
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.seasonlist",
|
||
|
}
|
||
|
|
||
|
def __init__(self,
|
||
|
handler,
|
||
|
args, # type: List
|
||
|
kwargs # type: Dict
|
||
|
):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowSeasonList.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowSeasons(ApiCall):
|
||
|
_help = {"desc": "get episode list of all or a specific season",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"season": {"desc": "the season number"},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get episode list of all or a specific season """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
|
||
|
if None is self.season:
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT name, description, absolute_number, scene_absolute_number, episode,"
|
||
|
" scene_episode, scene_season, airdate, status, season"
|
||
|
" FROM tv_episodes"
|
||
|
" WHERE indexer = ? AND showid = ?",
|
||
|
[self.tvid, self.prodid])
|
||
|
seasons = {} # type: Dict[int, Dict]
|
||
|
for cur_result in sql_result:
|
||
|
status, quality = Quality.splitCompositeStatus(int(cur_result["status"]))
|
||
|
cur_result["status"] = _get_status_Strings(status)
|
||
|
cur_result["quality"] = _get_quality_string(quality)
|
||
|
timezone, cur_result['timezone'] = network_timezones.get_network_timezone(show_obj.network,
|
||
|
return_name=True)
|
||
|
dtEpisodeAirs = SGDatetime.convert_to_setting(
|
||
|
network_timezones.parse_date_time(cur_result['airdate'], show_obj.airs, timezone))
|
||
|
cur_result['airdate'] = SGDatetime.sbfdate(dtEpisodeAirs, d_preset=dateFormat)
|
||
|
cur_result['scene_episode'] = helpers.try_int(cur_result['scene_episode'])
|
||
|
cur_result['scene_season'] = helpers.try_int(cur_result['scene_season'])
|
||
|
cur_result['absolute_number'] = helpers.try_int(cur_result['absolute_number'])
|
||
|
cur_result['scene_absolute_number'] = helpers.try_int(cur_result['scene_absolute_number'])
|
||
|
curSeason = int(cur_result["season"])
|
||
|
curEpisode = int(cur_result["episode"])
|
||
|
del cur_result["season"]
|
||
|
del cur_result["episode"]
|
||
|
if curSeason not in seasons:
|
||
|
seasons[curSeason] = {}
|
||
|
seasons[curSeason][curEpisode] = cur_result
|
||
|
|
||
|
else:
|
||
|
sql_result = my_db.select(
|
||
|
"SELECT name, description, absolute_number, scene_absolute_number,"
|
||
|
" episode, scene_episode, scene_season, airdate, status"
|
||
|
" FROM tv_episodes"
|
||
|
" WHERE indexer = ? AND showid = ? AND season = ?",
|
||
|
[self.tvid, self.prodid, self.season])
|
||
|
if len(sql_result) == 0:
|
||
|
return _responds(RESULT_FAILURE, msg="Season not found")
|
||
|
seasons = {}
|
||
|
for cur_result in sql_result:
|
||
|
curEpisode = int(cur_result["episode"])
|
||
|
del cur_result["episode"]
|
||
|
status, quality = Quality.splitCompositeStatus(int(cur_result["status"]))
|
||
|
cur_result["status"] = _get_status_Strings(status)
|
||
|
cur_result["quality"] = _get_quality_string(quality)
|
||
|
timezone, cur_result['timezone'] = network_timezones.get_network_timezone(show_obj.network,
|
||
|
return_name=True)
|
||
|
dtEpisodeAirs = SGDatetime.convert_to_setting(
|
||
|
network_timezones.parse_date_time(cur_result['airdate'], show_obj.airs, timezone))
|
||
|
cur_result['airdate'] = SGDatetime.sbfdate(dtEpisodeAirs, d_preset=dateFormat)
|
||
|
cur_result['scene_episode'] = helpers.try_int(cur_result['scene_episode'])
|
||
|
cur_result['scene_season'] = helpers.try_int(cur_result['scene_season'])
|
||
|
cur_result['absolute_number'] = helpers.try_int(cur_result['absolute_number'])
|
||
|
cur_result['scene_absolute_number'] = helpers.try_int(cur_result['scene_absolute_number'])
|
||
|
if curEpisode not in seasons:
|
||
|
seasons[curEpisode] = {}
|
||
|
seasons[curEpisode] = cur_result
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, seasons)
|
||
|
|
||
|
|
||
|
class CMD_ShowSeasons(CMD_SickGearShowSeasons):
|
||
|
_help = {"desc": "get episode list of all or a specific season",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"optionalParameters": {"season": {"desc": "the season number"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.seasons",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowSeasons.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowSetQuality(ApiCall):
|
||
|
_help = {
|
||
|
"desc": "set desired quality of a show in sickgear"
|
||
|
" if neither initial or archive are provided then the config default quality will be used",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"}},
|
||
|
"optionalParameters": {"initial": {"desc": "initial quality for the show"},
|
||
|
"archive": {"desc": "archive quality for the show"},
|
||
|
"upgrade_once": {"desc": "upgrade only once"}}
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, 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 didn't use the new code everywhere.
|
||
|
# self.archive, args = self.check_params(
|
||
|
# args, kwargs, "archive", None, False, "list", _getQualityMap().values()[1:])
|
||
|
self.initial, args = self.check_params(args, kwargs, "initial", None, False, "list", [q for q in quality_map])
|
||
|
self.archive, args = self.check_params(args, kwargs, "archive", None, False, "list", [q for q in quality_map])
|
||
|
self.upgradeonce, args = self.check_params(args, kwargs, "upgrade_once", False, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" set the quality for a show in sickgear by taking in a deliminated
|
||
|
string of qualities, map to their value and combine for new values
|
||
|
"""
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
# use default quality as a failsafe
|
||
|
newQuality = int(sickgear.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)
|
||
|
show_obj.quality = newQuality
|
||
|
|
||
|
show_obj.upgrade_once = self.upgradeonce
|
||
|
|
||
|
show_obj.save_to_db()
|
||
|
|
||
|
return _responds(RESULT_SUCCESS,
|
||
|
msg='%s quality has been changed to %s' % (show_obj.unique_name,
|
||
|
_get_quality_string(show_obj.quality)))
|
||
|
|
||
|
|
||
|
class CMD_ShowSetQuality(CMD_SickGearShowSetQuality):
|
||
|
_help = {
|
||
|
"desc": "set desired quality of a show in sickgear"
|
||
|
" if neither initial or archive are provided then the config default quality will be used",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"}},
|
||
|
"optionalParameters": {"initial": {"desc": "initial quality for the show"},
|
||
|
"archive": {"desc": "archive quality for the show"}},
|
||
|
"SickGearCommand": "sg.show.setquality",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowSetQuality.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowStats(ApiCall):
|
||
|
_help = {"desc": "get episode statistics for a show",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get episode statistics for a given show """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
# show stats
|
||
|
episode_status_counts_total = {"total": 0}
|
||
|
for status in statusStrings.statusStrings:
|
||
|
if status in SNATCHED_ANY + [UNKNOWN, DOWNLOADED]:
|
||
|
continue
|
||
|
episode_status_counts_total[status] = 0
|
||
|
|
||
|
# add all the downloaded qualities
|
||
|
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 = {"total": 0}
|
||
|
for statusCode in Quality.SNATCHED_ANY:
|
||
|
status, quality = Quality.splitCompositeStatus(statusCode)
|
||
|
if quality in [Quality.NONE]:
|
||
|
continue
|
||
|
episode_qualities_counts_snatch[statusCode] = 0
|
||
|
|
||
|
my_db = db.DBConnection(row_type="dict")
|
||
|
sql_result = my_db.select("SELECT status, season FROM tv_episodes WHERE season != 0 AND showid = ? "
|
||
|
"AND indexer = ?",
|
||
|
[self.prodid, self.tvid])
|
||
|
# the main loop that goes through all episodes
|
||
|
for cur_result in sql_result:
|
||
|
status, quality = Quality.splitCompositeStatus(int(cur_result["status"]))
|
||
|
|
||
|
episode_status_counts_total["total"] += 1
|
||
|
|
||
|
if status in Quality.DOWNLOADED:
|
||
|
episode_qualities_counts_download["total"] += 1
|
||
|
episode_qualities_counts_download[int(cur_result["status"])] += 1
|
||
|
elif status in Quality.SNATCHED_ANY:
|
||
|
episode_qualities_counts_snatch["total"] += 1
|
||
|
episode_qualities_counts_snatch[int(cur_result["status"])] += 1
|
||
|
elif 0 == status: # we dont count NONE = 0 = N/A
|
||
|
pass
|
||
|
else:
|
||
|
episode_status_counts_total[status] += 1
|
||
|
|
||
|
# the outgoing container
|
||
|
episodes_stats = {"downloaded": {}}
|
||
|
# turning codes into strings
|
||
|
for statusCode in episode_qualities_counts_download:
|
||
|
if "total" == statusCode:
|
||
|
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"] = {}
|
||
|
# turning codes into strings
|
||
|
# and combining proper and normal
|
||
|
for statusCode in episode_qualities_counts_snatch:
|
||
|
if "total" == statusCode:
|
||
|
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:
|
||
|
if "total" == statusCode:
|
||
|
episodes_stats["total"] = episode_status_counts_total[statusCode]
|
||
|
continue
|
||
|
status, quality = Quality.splitCompositeStatus(int(statusCode))
|
||
|
statusString = statusStrings.statusStrings[statusCode].lower().replace(" ", "_").replace("(", "").replace(
|
||
|
")", "")
|
||
|
episodes_stats[statusString] = episode_status_counts_total[statusCode]
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, episodes_stats)
|
||
|
|
||
|
|
||
|
class CMD_ShowStats(CMD_SickGearShowStats):
|
||
|
_help = {"desc": "get episode statistics for a show",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.stats",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowStats.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowUpdate(ApiCall):
|
||
|
_help = {"desc": "update a show in sickgear",
|
||
|
"requiredParameters": {"indexer": {"desc": "indexer of a show"},
|
||
|
"indexerid": {"desc": "unique id of a show"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.tvid, args = self.check_params(args, kwargs, "indexer", None, True, "int",
|
||
|
[i for i in indexer_api.TVInfoAPI().search_sources])
|
||
|
self.prodid, args = self.check_params(args, kwargs, "indexerid", None, True, "int", [])
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" update a show in sickgear """
|
||
|
show_obj = helpers.find_show_by_id({self.tvid: self.prodid})
|
||
|
if not show_obj:
|
||
|
return _responds(RESULT_FAILURE, msg="Show not found")
|
||
|
|
||
|
try:
|
||
|
sickgear.show_queue_scheduler.action.updateShow(show_obj, True)
|
||
|
return _responds(RESULT_SUCCESS, msg='%s has queued to be updated' % show_obj.unique_name)
|
||
|
except exceptions_helper.CantUpdateException as e:
|
||
|
self.log(u'Unable to update %s. %s' % (show_obj.unique_name, ex(e)), logger.ERROR)
|
||
|
return _responds(RESULT_FAILURE, msg='Unable to update %s. %s' % (show_obj.unique_name, ex(e)))
|
||
|
|
||
|
|
||
|
class CMD_ShowUpdate(CMD_SickGearShowUpdate):
|
||
|
_help = {"desc": "update a show in sickgear",
|
||
|
"requiredParameters": {"indexerid": {"desc": "thetvdb.com id of a show"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.show.update",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
kwargs['indexer'] = TVINFO_TVDB
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowUpdate.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShows(ApiCall):
|
||
|
_help = {"desc": "get all shows in sickgear",
|
||
|
"optionalParameters": {"sort": {"desc": "sort the list of shows by show name instead of indexerid"},
|
||
|
"paused": {"desc": "only return shows that are set to paused"},
|
||
|
"overview": {"desc": "include overview"},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# 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", [])
|
||
|
self.overview, args = self.check_params(args, kwargs, "overview", False, False, "bool", [])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" display_is_int_multi( self.prodid ) shows in sickgear """
|
||
|
shows = {}
|
||
|
|
||
|
for cur_show_obj in sickgear.showList:
|
||
|
|
||
|
if self.sickbeard_call and TVINFO_TVDB != cur_show_obj.tvid:
|
||
|
continue
|
||
|
|
||
|
if None is not self.paused and bool(self.paused) != bool(cur_show_obj.paused):
|
||
|
continue
|
||
|
|
||
|
genreList = []
|
||
|
if cur_show_obj.genre:
|
||
|
genreListTmp = cur_show_obj.genre.split("|")
|
||
|
for genre in genreListTmp:
|
||
|
if genre:
|
||
|
genreList.append(genre)
|
||
|
anyQualities, bestQualities = _mapQuality(cur_show_obj.quality)
|
||
|
|
||
|
showDict = {
|
||
|
"paused": cur_show_obj.paused,
|
||
|
"quality": _get_quality_string(cur_show_obj.quality),
|
||
|
"language": cur_show_obj.lang,
|
||
|
"air_by_date": cur_show_obj.air_by_date,
|
||
|
"airs": cur_show_obj.airs,
|
||
|
"flatten_folders": cur_show_obj.flatten_folders,
|
||
|
"genre": genreList,
|
||
|
"location": cur_show_obj._location,
|
||
|
"quality_details": {"initial": anyQualities, "archive": bestQualities},
|
||
|
"sports": cur_show_obj.sports,
|
||
|
"anime": cur_show_obj.anime,
|
||
|
"indexerid": cur_show_obj.prodid,
|
||
|
"indexer": cur_show_obj.tvid,
|
||
|
"tvdbid": cur_show_obj.ids.get(TVINFO_TVDB, {'id': 0})['id'],
|
||
|
'ids': {k: v.get('id') for k, v in iteritems(cur_show_obj.ids)},
|
||
|
"tvrage_id": cur_show_obj.ids.get(TVINFO_TVRAGE, {'id': 0})['id'],
|
||
|
"tvrage_name": cur_show_obj.name,
|
||
|
"network": cur_show_obj.network,
|
||
|
"show_name": cur_show_obj.name,
|
||
|
"status": cur_show_obj.status,
|
||
|
"subtitles": cur_show_obj.subtitles,
|
||
|
"scenenumbering": cur_show_obj.is_scene,
|
||
|
"upgrade_once": cur_show_obj.upgrade_once,
|
||
|
"ignorewords": helpers.generate_word_str(cur_show_obj.rls_ignore_words, cur_show_obj.rls_ignore_words_regex),
|
||
|
"global_exclude_ignore": helpers.generate_word_str(cur_show_obj.rls_global_exclude_ignore),
|
||
|
"requirewords": helpers.generate_word_str(cur_show_obj.rls_require_words, cur_show_obj.rls_require_words_regex),
|
||
|
"global_exclude_require": helpers.generate_word_str(cur_show_obj.rls_global_exclude_require),
|
||
|
"prune": cur_show_obj.prune,
|
||
|
"tag": cur_show_obj.tag,
|
||
|
"imdb_id": cur_show_obj.imdbid,
|
||
|
"classification": cur_show_obj.classification,
|
||
|
"runtime": cur_show_obj.runtime,
|
||
|
"startyear": cur_show_obj.startyear,
|
||
|
}
|
||
|
|
||
|
if self.overview:
|
||
|
showDict["overview"] = cur_show_obj.overview
|
||
|
|
||
|
timezone, showDict['timezone'] = network_timezones.get_network_timezone(showDict['network'],
|
||
|
return_name=True)
|
||
|
|
||
|
if cur_show_obj.next_episode():
|
||
|
dtEpisodeAirs = SGDatetime.convert_to_setting(
|
||
|
network_timezones.parse_date_time(cur_show_obj.nextaired, cur_show_obj.airs, timezone))
|
||
|
showDict['next_ep_airdate'] = SGDatetime.sbfdate(dtEpisodeAirs, d_preset=dateFormat)
|
||
|
else:
|
||
|
showDict['next_ep_airdate'] = ''
|
||
|
|
||
|
showDict["cache"] = CMD_SickGearShowCache(self.handler, (), {"indexer": cur_show_obj.tvid,
|
||
|
"indexerid": cur_show_obj.prodid}).run()["data"]
|
||
|
if not showDict["network"]:
|
||
|
showDict["network"] = ""
|
||
|
if "name" == self.sort:
|
||
|
shows[cur_show_obj.name] = showDict
|
||
|
else:
|
||
|
if self.sickbeard_call:
|
||
|
shows[cur_show_obj.prodid] = showDict
|
||
|
else:
|
||
|
shows[cur_show_obj.tvid_prodid] = showDict
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, shows)
|
||
|
|
||
|
|
||
|
class CMD_Shows(CMD_SickGearShows):
|
||
|
_help = {"desc": "get all thetvdb.com shows in sickgear",
|
||
|
"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"},
|
||
|
"overview": {"desc": "include overview"},
|
||
|
},
|
||
|
"SickGearCommand": "sg.shows",
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShows.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowsBrowseTrakt(ApiCall):
|
||
|
_help = {"desc": "browse trakt shows in sickgear",
|
||
|
"requiredParameters": {"type": {"desc": "type to browse: anticipated, newshows, newseasons, popular, "
|
||
|
"trending, recommended, watchlist"},
|
||
|
},
|
||
|
"optionalParameters": {"account_id": {"desc": "account_id for recommended, watchlist - "
|
||
|
"see sg.listtraktaccounts"}},
|
||
|
}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
self.type, args = self.check_params(args, kwargs, "type", "anticipated", True, "string",
|
||
|
["anticipated", "newshows", "newseasons", "popular", "trending",
|
||
|
"recommended", "watchlist"])
|
||
|
# optional
|
||
|
self.account, args = self.check_params(args, kwargs, "account_id", None, False, "int",
|
||
|
[s for s in sickgear.TRAKT_ACCOUNTS])
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" browse trakt shows in sickgear """
|
||
|
urls = {'anticipated': 'shows/anticipated?limit=%s&' % 100,
|
||
|
'newshows': '/calendars/all/shows/new/%s/%s?' % (SGDatetime.sbfdate(
|
||
|
dt=datetime.datetime.now() + datetime.timedelta(days=-16), d_preset='%Y-%m-%d'), 32),
|
||
|
'newseasons': '/calendars/all/shows/premieres/%s/%s?' % (SGDatetime.sbfdate(
|
||
|
dt=datetime.datetime.now() + datetime.timedelta(days=-16), d_preset='%Y-%m-%d'), 32),
|
||
|
'popular': 'shows/popular?limit=%s&' % 100,
|
||
|
'trending': 'shows/trending?limit=%s&' % 100,
|
||
|
'recommended': 'recommendations/shows?limit=%s&' % 100,
|
||
|
}
|
||
|
kwargs = {}
|
||
|
if self.type in ('recommended', 'watchlist'):
|
||
|
if not self.account:
|
||
|
return _responds(RESULT_FAILURE, msg='Need Trakt account')
|
||
|
kwargs['send_oauth'] = self.account
|
||
|
urls['watchlist'] = 'users/%s/watchlist/shows?limit=%s&' \
|
||
|
% (sickgear.TRAKT_ACCOUNTS[self.account].slug, 100)
|
||
|
try:
|
||
|
data, oldest, newest = AddShows.get_trakt_data(urls[self.type], **kwargs)
|
||
|
except Exception as e:
|
||
|
return _responds(RESULT_FAILURE, msg=ex(e))
|
||
|
return _responds(RESULT_SUCCESS, data)
|
||
|
|
||
|
|
||
|
class CMD_SickGearListTraktAccounts(ApiCall):
|
||
|
_help = {"desc": "list Trakt accounts in sickgear"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" list Trakt accounts in sickgear """
|
||
|
accounts = [{'name': v.name, 'account_id': v.account_id} for a, v in iteritems(sickgear.TRAKT_ACCOUNTS)]
|
||
|
return _responds(RESULT_SUCCESS, accounts)
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowsForceUpdate(ApiCall):
|
||
|
_help = {"desc": "force the daily show update now"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" force the daily show update now """
|
||
|
if sickgear.show_queue_scheduler.action.isShowUpdateRunning() \
|
||
|
or sickgear.show_update_scheduler.action.amActive:
|
||
|
return _responds(RESULT_FAILURE, msg="show update already running.")
|
||
|
|
||
|
result = sickgear.show_update_scheduler.forceRun()
|
||
|
if result:
|
||
|
return _responds(RESULT_SUCCESS, msg="daily show update started")
|
||
|
return _responds(RESULT_FAILURE, msg="can't start show update currently")
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowsQueue(ApiCall):
|
||
|
_help = {"desc": "list the show update queue"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" list the show update queue """
|
||
|
return _responds(RESULT_SUCCESS, sickgear.show_queue_scheduler.action.queue_length())
|
||
|
|
||
|
|
||
|
class CMD_SickGearShowsStats(ApiCall):
|
||
|
_help = {"desc": "get the global shows and episode stats"}
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
ApiCall.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
def run(self):
|
||
|
""" get the global shows and episode stats """
|
||
|
stats = {}
|
||
|
|
||
|
indexer_limit = ('', ' AND indexer = %s' % TVINFO_TVDB)[self.sickbeard_call]
|
||
|
my_db = db.DBConnection()
|
||
|
today = str(datetime.date.today().toordinal())
|
||
|
|
||
|
stats["shows_total"] = (len(sickgear.showList),
|
||
|
len([cur_so for cur_so in sickgear.showList
|
||
|
if TVINFO_TVDB == cur_so.tvid]))[self.sickbeard_call]
|
||
|
|
||
|
stats["shows_active"] = len(
|
||
|
[cur_so for cur_so in sickgear.showList
|
||
|
if 0 == cur_so.paused and "Ended" != cur_so.status
|
||
|
and (not self.sickbeard_call
|
||
|
or TVINFO_TVDB == cur_so.tvid)])
|
||
|
|
||
|
stats["ep_downloaded"] = my_db.select("SELECT COUNT(*) FROM tv_episodes WHERE status IN (" + ",".join(
|
||
|
[str(status) for status in
|
||
|
Quality.DOWNLOADED + Quality.ARCHIVED]) + ") AND season != 0 and episode != 0 AND airdate <= " + today +
|
||
|
indexer_limit)[0][0]
|
||
|
|
||
|
stats["ep_total"] = my_db.select(
|
||
|
"SELECT COUNT(*) FROM tv_episodes WHERE season != 0 AND episode != 0 AND (airdate != 1 OR status IN (" +
|
||
|
",".join([str(status) for status in Quality.SNATCHED_ANY + Quality.DOWNLOADED + Quality.ARCHIVED]) +
|
||
|
")) AND airdate <= " + today + " AND status != " + str(IGNORED) + indexer_limit)[0][0]
|
||
|
|
||
|
return _responds(RESULT_SUCCESS, stats)
|
||
|
|
||
|
|
||
|
# WARNING: never define a cmd call string that contains a "_" (underscore)
|
||
|
class CMD_ShowsStats(CMD_SickGearShowsStats):
|
||
|
_help = {"desc": "get the global thetvdb.com shows and episode stats",
|
||
|
"SickGearCommand": "sg.shows.stats"}
|
||
|
# this is reserved for cmd indexes used while cmd chaining
|
||
|
|
||
|
def __init__(self, handler, args, kwargs):
|
||
|
# required
|
||
|
# optional
|
||
|
# super, missing, help
|
||
|
self.sickbeard_call = True
|
||
|
CMD_SickGearShowsStats.__init__(self, handler, args, kwargs)
|
||
|
|
||
|
# WARNING: never define a param name that contains a "." (dot)
|
||
|
# this is reserved for cmd namspaces used while cmd chaining
|
||
|
# 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,
|
||
|
"listcommands": CMD_ListCommands,
|
||
|
"future": CMD_ComingEpisodes,
|
||
|
"sg.future": CMD_SickGearComingEpisodes,
|
||
|
"episode": CMD_Episode,
|
||
|
"sg.episode": CMD_SickGearEpisode,
|
||
|
"episode.search": CMD_EpisodeSearch,
|
||
|
"sg.episode.search": CMD_SickGearEpisodeSearch,
|
||
|
"episode.setstatus": CMD_EpisodeSetStatus,
|
||
|
"sg.episode.setstatus": CMD_SickGearEpisodeSetStatus,
|
||
|
"episode.subtitlesearch": CMD_SubtitleSearch,
|
||
|
"sg.episode.subtitlesearch": CMD_SickGearSubtitleSearch,
|
||
|
"exceptions": CMD_Exceptions,
|
||
|
"sg.exceptions": CMD_SickGearExceptions,
|
||
|
'sg.setexceptions': CMD_SetExceptions,
|
||
|
"history": CMD_History,
|
||
|
"sg.history": CMD_SickGearHistory,
|
||
|
"history.clear": CMD_HistoryClear,
|
||
|
"sg.history.clear": CMD_SickGearHistoryClear,
|
||
|
"history.trim": CMD_HistoryTrim,
|
||
|
"sg.history.trim": CMD_SickGearHistoryTrim,
|
||
|
"logs": CMD_Logs,
|
||
|
"sg.logs": CMD_SickGearLogs,
|
||
|
"sb": CMD_SickBeard,
|
||
|
"sg": CMD_SickGear,
|
||
|
"postprocess": CMD_PostProcess,
|
||
|
"sg.postprocess": CMD_SickGearPostProcess,
|
||
|
"sb.addrootdir": CMD_SickBeardAddRootDir,
|
||
|
"sg.addrootdir": CMD_SickGearAddRootDir,
|
||
|
"sb.checkscheduler": CMD_SickBeardCheckScheduler,
|
||
|
"sg.checkscheduler": CMD_SickGearCheckScheduler,
|
||
|
"sb.deleterootdir": CMD_SickBeardDeleteRootDir,
|
||
|
"sg.deleterootdir": CMD_SickGearDeleteRootDir,
|
||
|
"sb.forcesearch": CMD_SickBeardForceSearch,
|
||
|
"sg.forcesearch": CMD_SickGearForceSearch,
|
||
|
"sg.searchqueue": CMD_SickGearSearchQueue,
|
||
|
"sb.getdefaults": CMD_SickBeardGetDefaults,
|
||
|
"sg.getdefaults": CMD_SickGearGetDefaults,
|
||
|
"sb.getmessages": CMD_SickBeardGetMessages,
|
||
|
"sg.getmessages": CMD_SickGearGetMessages,
|
||
|
"sg.getqualities": CMD_SickGearGetQualities,
|
||
|
"sg.getqualitystrings": CMD_SickGearGetqualityStrings,
|
||
|
"sb.getrootdirs": CMD_SickBeardGetRootDirs,
|
||
|
"sg.getrootdirs": CMD_SickGearGetRootDirs,
|
||
|
"sb.pausebacklog": CMD_SickBeardPauseBacklog,
|
||
|
"sg.pausebacklog": CMD_SickGearPauseBacklog,
|
||
|
"sb.ping": CMD_SickBeardPing,
|
||
|
"sg.ping": CMD_SickGearPing,
|
||
|
"sb.restart": CMD_SickBeardRestart,
|
||
|
"sg.restart": CMD_SickGearRestart,
|
||
|
"sb.searchtvdb": CMD_SickBeardSearchIndexers,
|
||
|
"sg.searchtv": CMD_SickGearSearchIndexers,
|
||
|
"sb.setdefaults": CMD_SickBeardSetDefaults,
|
||
|
"sg.setdefaults": CMD_SickGearSetDefaults,
|
||
|
"sg.setscenenumber": CMD_SickGearSetSceneNumber,
|
||
|
"sg.activatescenenumbering": CMD_SickGearActivateSceneNumber,
|
||
|
"sg.getindexers": CMD_SickGearGetIndexers,
|
||
|
"sg.getindexericon": CMD_SickGearGetIndexerIcon,
|
||
|
"sg.getnetworkicon": CMD_SickGearGetNetworkIcon,
|
||
|
"sb.shutdown": CMD_SickBeardShutdown,
|
||
|
"sg.shutdown": CMD_SickGearShutdown,
|
||
|
"sg.listignorewords": CMD_SickGearListIgnoreWords,
|
||
|
"sg.setignorewords": CMD_SickGearSetIgnoreWords,
|
||
|
"sg.listrequiredwords": CMD_SickGearListRequireWords_old, # old method name
|
||
|
"sg.setrequiredwords": CMD_SickGearSetRequireWords_old, # old method name
|
||
|
"sg.listrequirewords": CMD_SickGearListRequireWords,
|
||
|
"sg.setrequirewords": CMD_SickGearSetRequireWords,
|
||
|
"sg.updatewatchedstate": CMD_SickGearUpdateWatchedState,
|
||
|
"show": CMD_Show,
|
||
|
"sg.show": CMD_SickGearShow,
|
||
|
"show.addexisting": CMD_ShowAddExisting,
|
||
|
"sg.show.addexisting": CMD_SickGearShowAddExisting,
|
||
|
"show.addnew": CMD_ShowAddNew,
|
||
|
"sg.show.addnew": CMD_SickGearShowAddNew,
|
||
|
"show.cache": CMD_ShowCache,
|
||
|
"sg.show.cache": CMD_SickGearShowCache,
|
||
|
"show.delete": CMD_ShowDelete,
|
||
|
"sg.show.delete": CMD_SickGearShowDelete,
|
||
|
"show.getquality": CMD_ShowGetQuality,
|
||
|
"sg.show.getquality": CMD_SickGearShowGetQuality,
|
||
|
"show.getposter": CMD_ShowGetPoster,
|
||
|
"sg.show.getposter": CMD_SickGearShowGetPoster,
|
||
|
"show.getbanner": CMD_ShowGetBanner,
|
||
|
"sg.show.getbanner": CMD_SickGearShowGetBanner,
|
||
|
"sg.show.listfanart": CMD_SickGearShowListFanart,
|
||
|
"sg.show.ratefanart": CMD_SickGearShowRateFanart,
|
||
|
"sg.show.getfanart": CMD_SickGearShowGetFanart,
|
||
|
"show.pause": CMD_ShowPause,
|
||
|
"sg.show.pause": CMD_SickGearShowPause,
|
||
|
"show.refresh": CMD_ShowRefresh,
|
||
|
"sg.show.refresh": CMD_SickGearShowRefresh,
|
||
|
"show.seasonlist": CMD_ShowSeasonList,
|
||
|
"sg.show.seasonlist": CMD_SickGearShowSeasonList,
|
||
|
"show.seasons": CMD_ShowSeasons,
|
||
|
"sg.show.seasons": CMD_SickGearShowSeasons,
|
||
|
"show.setquality": CMD_ShowSetQuality,
|
||
|
"sg.show.setquality": CMD_SickGearShowSetQuality,
|
||
|
"show.stats": CMD_ShowStats,
|
||
|
"sg.show.stats": CMD_SickGearShowStats,
|
||
|
"show.update": CMD_ShowUpdate,
|
||
|
"sg.show.update": CMD_SickGearShowUpdate,
|
||
|
"shows": CMD_Shows,
|
||
|
"sg.shows": CMD_SickGearShows,
|
||
|
"sg.shows.browsetrakt": CMD_SickGearShowsBrowseTrakt,
|
||
|
"sg.listtraktaccounts": CMD_SickGearListTraktAccounts,
|
||
|
"shows.stats": CMD_ShowsStats,
|
||
|
"sg.shows.stats": CMD_SickGearShowsStats,
|
||
|
"sg.shows.forceupdate": CMD_SickGearShowsForceUpdate,
|
||
|
"sg.shows.queue": CMD_SickGearShowsQueue,
|
||
|
}
|
||
|
|
||
|
_functionMaper_reversed = {v: k for k, v in iteritems(_functionMaper)}
|