2014-03-10 05:18:05 +00:00
# Author: Nic Wolfe <nic@wolfeden.ca>
# URL: http://code.google.com/p/sickbeard/
#
2014-11-12 16:43:14 +00:00
# This file is part of SickGear.
2014-03-10 05:18:05 +00:00
#
2014-11-12 16:43:14 +00:00
# SickGear is free software: you can redistribute it and/or modify
2014-03-10 05:18:05 +00:00
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
2014-11-12 16:43:14 +00:00
# SickGear is distributed in the hope that it will be useful,
2014-03-10 05:18:05 +00:00
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
2014-11-12 16:43:14 +00:00
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
2014-03-10 05:18:05 +00:00
from __future__ import with_statement
2014-07-27 10:59:21 +00:00
2014-07-06 14:45:01 +00:00
import os
2014-03-10 05:18:05 +00:00
import time
import urllib
import re
import datetime
import random
2015-02-23 02:19:25 +00:00
import traceback
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
from mimetypes import MimeTypes
from Cheetah . Template import Template
2015-06-13 14:52:46 +00:00
from six import iteritems
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
import sickbeard
from sickbeard import config , sab , clients , history , notifiers , processTV , ui , logger , helpers , exceptions , classes , \
db , search_queue , image_cache , naming , scene_exceptions , subtitles , network_timezones , sbdatetime
from sickbeard import encodingKludge as ek
2014-03-10 05:18:05 +00:00
from sickbeard . providers import newznab , rsstorrent
2014-07-28 01:46:15 +00:00
from sickbeard . common import Quality , Overview , statusStrings , qualityPresetStrings , cpu_presets
2015-03-29 08:29:40 +00:00
from sickbeard . common import SNATCHED , UNAIRED , IGNORED , ARCHIVED , WANTED , FAILED , SKIPPED
2014-03-10 05:18:05 +00:00
from sickbeard . common import SD , HD720p , HD1080p
from sickbeard . exceptions import ex
2015-02-24 13:37:27 +00:00
from sickbeard . helpers import remove_article , starify
2014-03-10 05:18:05 +00:00
from sickbeard . scene_exceptions import get_scene_exceptions
2014-03-25 05:57:24 +00:00
from sickbeard . scene_numbering import get_scene_numbering , set_scene_numbering , get_scene_numbering_for_show , \
2014-05-30 05:48:02 +00:00
get_xem_numbering_for_show , get_scene_absolute_numbering_for_show , get_xem_absolute_numbering_for_show , \
get_scene_absolute_numbering
2015-03-16 16:38:38 +00:00
from sickbeard . name_cache import buildNameCache
2015-02-23 02:19:25 +00:00
from sickbeard . browser import foldersAtPath
2015-01-25 11:58:49 +00:00
from sickbeard . blackandwhitelist import BlackAndWhiteList , short_group_names
2015-05-04 19:14:29 +00:00
from sickbeard . searchBacklog import FULL_BACKLOG , LIMITED_BACKLOG
2015-02-23 02:19:25 +00:00
from tornado import gen
from tornado . web import RequestHandler , authenticated
from lib import adba
from lib import subliminal
2014-03-10 05:18:05 +00:00
from lib . dateutil import tz
2014-05-30 05:48:02 +00:00
from lib . unrar2 import RarFile
2014-12-16 14:14:54 +00:00
from lib . trakt import TraktCall
2014-03-10 05:18:05 +00:00
try :
import json
except ImportError :
from lib import simplejson as json
2014-05-20 20:51:54 +00:00
2015-02-23 02:19:25 +00:00
class PageTemplate ( Template ) :
def __init__ ( self , headers , * args , * * KWs ) :
KWs [ ' file ' ] = os . path . join ( sickbeard . PROG_DIR , ' gui/ ' + sickbeard . GUI_NAME + ' /interfaces/default/ ' ,
KWs [ ' file ' ] )
super ( PageTemplate , self ) . __init__ ( * args , * * KWs )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
self . sbRoot = sickbeard . WEB_ROOT
self . sbHttpPort = sickbeard . WEB_PORT
self . sbHttpsPort = sickbeard . WEB_PORT
self . sbHttpsEnabled = sickbeard . ENABLE_HTTPS
self . sbHandleReverseProxy = sickbeard . HANDLE_REVERSE_PROXY
self . sbThemeName = sickbeard . THEME_NAME
2014-07-01 08:49:12 +00:00
2015-02-23 02:19:25 +00:00
if headers [ ' Host ' ] [ 0 ] == ' [ ' :
self . sbHost = re . match ( ' ^ \ [.* \ ] ' , headers [ ' Host ' ] , re . X | re . M | re . S ) . group ( 0 )
else :
self . sbHost = re . match ( ' ^[^:]+ ' , headers [ ' Host ' ] , re . X | re . M | re . S ) . group ( 0 )
2014-06-30 01:54:41 +00:00
2015-02-23 02:19:25 +00:00
if ' X-Forwarded-Host ' in headers :
self . sbHost = headers [ ' X-Forwarded-Host ' ]
if ' X-Forwarded-Port ' in headers :
sbHttpPort = headers [ ' X-Forwarded-Port ' ]
self . sbHttpsPort = sbHttpPort
if ' X-Forwarded-Proto ' in headers :
self . sbHttpsEnabled = True if headers [ ' X-Forwarded-Proto ' ] == ' https ' else False
2014-06-30 01:54:41 +00:00
2015-02-23 02:19:25 +00:00
logPageTitle = ' Logs & Errors '
if len ( classes . ErrorViewer . errors ) :
logPageTitle + = ' ( ' + str ( len ( classes . ErrorViewer . errors ) ) + ' ) '
self . logPageTitle = logPageTitle
self . sbPID = str ( sickbeard . PID )
self . menu = [
{ ' title ' : ' Home ' , ' key ' : ' home ' } ,
{ ' title ' : ' Episodes ' , ' key ' : ' episodeView ' } ,
{ ' title ' : ' History ' , ' key ' : ' history ' } ,
{ ' title ' : ' Manage ' , ' key ' : ' manage ' } ,
{ ' title ' : ' Config ' , ' key ' : ' config ' } ,
{ ' title ' : logPageTitle , ' key ' : ' errorlogs ' } ,
]
2014-06-30 01:54:41 +00:00
2015-02-23 02:19:25 +00:00
def compile ( self , * args , * * kwargs ) :
if not os . path . exists ( os . path . join ( sickbeard . CACHE_DIR , ' cheetah ' ) ) :
os . mkdir ( os . path . join ( sickbeard . CACHE_DIR , ' cheetah ' ) )
2014-07-01 08:49:12 +00:00
2015-02-23 02:19:25 +00:00
kwargs [ ' cacheModuleFilesForTracebacks ' ] = True
kwargs [ ' cacheDirForModuleFiles ' ] = os . path . join ( sickbeard . CACHE_DIR , ' cheetah ' )
return super ( PageTemplate , self ) . compile ( * args , * * kwargs )
2014-07-06 14:45:01 +00:00
2014-07-27 14:39:33 +00:00
2015-02-06 11:39:10 +00:00
class BaseHandler ( RequestHandler ) :
def set_default_headers ( self ) :
self . set_header ( ' Cache-Control ' , ' no-store, no-cache, must-revalidate, max-age=0 ' )
def redirect ( self , url , permanent = False , status = None ) :
if not url . startswith ( sickbeard . WEB_ROOT ) :
url = sickbeard . WEB_ROOT + url
super ( BaseHandler , self ) . redirect ( url , permanent , status )
def get_current_user ( self , * args , * * kwargs ) :
if sickbeard . WEB_USERNAME or sickbeard . WEB_PASSWORD :
2015-02-26 02:04:55 +00:00
return self . get_secure_cookie ( ' sickgear-session ' )
2015-02-06 11:39:10 +00:00
else :
return True
def showPoster ( self , show = None , which = None , api = None ) :
# Redirect initial poster/banner thumb to default images
if which [ 0 : 6 ] == ' poster ' :
default_image_name = ' poster.png '
else :
default_image_name = ' banner.png '
static_image_path = os . path . join ( ' /images ' , default_image_name )
if show and sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) ) :
cache_obj = image_cache . ImageCache ( )
image_file_name = None
if which == ' poster ' :
image_file_name = cache_obj . poster_path ( show )
if which == ' poster_thumb ' :
image_file_name = cache_obj . poster_thumb_path ( show )
if which == ' banner ' :
image_file_name = cache_obj . banner_path ( show )
if which == ' banner_thumb ' :
image_file_name = cache_obj . banner_thumb_path ( show )
if ek . ek ( os . path . isfile , image_file_name ) :
static_image_path = image_file_name
if api :
mime_type , encoding = MimeTypes ( ) . guess_type ( static_image_path )
self . set_header ( ' Content-Type ' , mime_type )
2015-06-14 03:49:23 +00:00
with open ( static_image_path , ' rb ' ) as img :
2015-02-06 11:39:10 +00:00
return img . read ( )
else :
static_image_path = os . path . normpath ( static_image_path . replace ( sickbeard . CACHE_DIR , ' /cache ' ) )
static_image_path = static_image_path . replace ( ' \\ ' , ' / ' )
2015-02-23 02:19:25 +00:00
self . redirect ( static_image_path )
2015-02-06 11:39:10 +00:00
class LoginHandler ( BaseHandler ) :
def get ( self , * args , * * kwargs ) :
if self . get_current_user ( ) :
self . redirect ( self . get_argument ( ' next ' , ' /home/ ' ) )
else :
t = PageTemplate ( headers = self . request . headers , file = ' login.tmpl ' )
t . resp = self . get_argument ( ' resp ' , ' ' )
self . set_status ( 401 )
self . finish ( t . respond ( ) )
def post ( self , * args , * * kwargs ) :
username = sickbeard . WEB_USERNAME
password = sickbeard . WEB_PASSWORD
2015-03-01 03:01:43 +00:00
if ( self . get_argument ( ' username ' ) == username ) and ( self . get_argument ( ' password ' ) == password ) :
2015-02-06 11:39:10 +00:00
remember_me = int ( self . get_argument ( ' remember_me ' , default = 0 ) or 0 )
2015-02-26 02:04:55 +00:00
self . set_secure_cookie ( ' sickgear-session ' , sickbeard . COOKIE_SECRET , expires_days = 30 if remember_me > 0 else None )
2015-02-06 11:39:10 +00:00
self . redirect ( self . get_argument ( ' next ' , ' /home/ ' ) )
else :
next_arg = ' &next= ' + self . get_argument ( ' next ' , ' /home/ ' )
self . redirect ( ' /login?resp=authfailed ' + next_arg )
2015-02-23 02:19:25 +00:00
class LogoutHandler ( BaseHandler ) :
def get ( self , * args , * * kwargs ) :
2015-02-28 04:23:46 +00:00
self . clear_cookie ( ' sickgear-session ' )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /login/ ' )
class CalendarHandler ( BaseHandler ) :
def get ( self , * args , * * kwargs ) :
if sickbeard . CALENDAR_UNPROTECTED or self . get_current_user ( ) :
self . write ( self . calendar ( ) )
else :
self . set_status ( 401 )
self . write ( ' User authentication required ' )
def calendar ( self , * args , * * kwargs ) :
""" iCalendar (iCal) - Standard RFC 5545 <http://tools.ietf.org/html/rfc5546>
Works with iCloud , Google Calendar and Outlook .
Provides a subscribeable URL for iCal subscriptions """
logger . log ( u ' Receiving iCal request from %s ' % self . request . remote_ip )
# Limit dates
past_date = ( datetime . date . today ( ) + datetime . timedelta ( weeks = - 52 ) ) . toordinal ( )
future_date = ( datetime . date . today ( ) + datetime . timedelta ( weeks = 52 ) ) . toordinal ( )
utc = tz . gettz ( ' GMT ' )
# Get all the shows that are not paused and are currently on air
myDB = db . DBConnection ( )
show_list = myDB . select (
' SELECT show_name, indexer_id, network, airs, runtime FROM tv_shows WHERE ( status = " Continuing " OR status = " Returning Series " ) AND paused != " 1 " ' )
nl = ' \\ n \\ n '
crlf = ' \r \n '
# Create iCal header
appname = ' SickGear '
ical = ' BEGIN:VCALENDAR %s VERSION:2.0 %s X-WR-CALNAME: %s %s X-WR-CALDESC: %s %s PRODID:// %s Upcoming Episodes// %s ' \
% ( crlf , crlf , appname , crlf , appname , crlf , appname , crlf )
for show in show_list :
# Get all episodes of this show airing between today and next month
episode_list = myDB . select (
' SELECT indexerid, name, season, episode, description, airdate FROM tv_episodes WHERE airdate >= ? AND airdate < ? AND showid = ? ' ,
( past_date , future_date , int ( show [ ' indexer_id ' ] ) ) )
for episode in episode_list :
air_date_time = network_timezones . parse_date_time ( episode [ ' airdate ' ] , show [ ' airs ' ] ,
show [ ' network ' ] ) . astimezone ( utc )
air_date_time_end = air_date_time + datetime . timedelta (
minutes = helpers . tryInt ( show [ ' runtime ' ] , 60 ) )
# Create event for episode
ical + = ' BEGIN:VEVENT %s ' % crlf \
+ ' DTSTART: %s T %s Z %s ' % ( air_date_time . strftime ( ' % Y % m %d ' ) , air_date_time . strftime ( ' % H % M % S ' ) , crlf ) \
+ ' DTEND: %s T %s Z %s ' % ( air_date_time_end . strftime ( ' % Y % m %d ' ) , air_date_time_end . strftime ( ' % H % M % S ' ) , crlf ) \
+ ' SUMMARY: %s - %s x %s - %s %s ' % ( show [ ' show_name ' ] , str ( episode [ ' season ' ] ) , str ( episode [ ' episode ' ] ) , episode [ ' name ' ] , crlf ) \
+ ' UID: %s - %s - %s -E %s S %s %s ' % ( appname , str ( datetime . date . today ( ) . isoformat ( ) ) , show [ ' show_name ' ] . replace ( ' ' , ' - ' ) , str ( episode [ ' episode ' ] ) , str ( episode [ ' season ' ] ) , crlf ) \
+ ' DESCRIPTION: %s on %s ' % ( ( show [ ' airs ' ] or ' (Unknown airs) ' ) , ( show [ ' network ' ] or ' Unknown network ' ) ) \
+ ( ' ' if not episode [ ' description ' ] else ' %s %s ' % ( nl , episode [ ' description ' ] . splitlines ( ) [ 0 ] ) ) \
+ ' %s END:VEVENT %s ' % ( crlf , crlf )
# Ending the iCal
return ical + ' END:VCALENDAR '
class IsAliveHandler ( BaseHandler ) :
def get ( self , * args , * * kwargs ) :
kwargs = self . request . arguments
if ' callback ' in kwargs and ' _ ' in kwargs :
callback , _ = kwargs [ ' callback ' ] [ 0 ] , kwargs [ ' _ ' ]
else :
return ' Error: Unsupported Request. Send jsonp request with callback variable in the query string. '
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
self . set_header ( ' Content-Type ' , ' text/javascript ' )
self . set_header ( ' Access-Control-Allow-Origin ' , ' * ' )
self . set_header ( ' Access-Control-Allow-Headers ' , ' x-requested-with ' )
if sickbeard . started :
results = callback + ' ( ' + json . dumps (
{ ' msg ' : str ( sickbeard . PID ) } ) + ' ); '
else :
results = callback + ' ( ' + json . dumps ( { ' msg ' : ' nope ' } ) + ' ); '
self . write ( results )
2015-02-06 11:39:10 +00:00
class WebHandler ( BaseHandler ) :
def page_not_found ( self ) :
t = PageTemplate ( headers = self . request . headers , file = ' 404.tmpl ' )
2015-02-23 02:19:25 +00:00
return t . respond ( )
2015-02-06 11:39:10 +00:00
@authenticated
@gen.coroutine
def get ( self , route , * args , * * kwargs ) :
route = route . strip ( ' / ' ) or ' index '
try :
method = getattr ( self , route )
except :
self . finish ( self . page_not_found ( ) )
else :
kwargss = self . request . arguments
for arg , value in kwargss . items ( ) :
if len ( value ) == 1 :
kwargss [ arg ] = value [ 0 ]
2015-02-23 02:19:25 +00:00
result = method ( * * kwargss )
if result :
self . finish ( result )
2015-02-06 11:39:10 +00:00
post = get
class MainHandler ( WebHandler ) :
def index ( self ) :
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/ ' )
2015-02-06 11:39:10 +00:00
2014-06-18 12:56:26 +00:00
def http_error_401_handler ( self ) :
""" Custom handler for 401 error """
return r ''' <!DOCTYPE html>
< html >
< head >
< title > % s < / title >
< / head >
< body >
< br / >
< font color = " #0000FF " > Error % s : You need to provide a valid username and password . < / font >
< / body >
< / html >
''' % ( ' Access denied ' , 401)
def write_error ( self , status_code , * * kwargs ) :
2014-06-29 10:05:33 +00:00
if status_code == 401 :
2014-06-30 02:52:04 +00:00
self . finish ( self . http_error_401_handler ( ) )
2014-06-29 10:05:33 +00:00
elif status_code == 404 :
2014-07-11 19:33:49 +00:00
self . redirect ( sickbeard . WEB_ROOT + ' /home/ ' )
2015-02-23 02:19:25 +00:00
elif self . settings . get ( ' debug ' ) and ' exc_info ' in kwargs :
exc_info = kwargs [ ' exc_info ' ]
trace_info = ' ' . join ( [ ' %s <br/> ' % line for line in traceback . format_exception ( * exc_info ) ] )
request_info = ' ' . join ( [ ' <strong> %s </strong>: %s <br/> ' % ( k , self . request . __dict__ [ k ] ) for k in
2014-07-06 14:45:01 +00:00
self . request . __dict__ . keys ( ) ] )
error = exc_info [ 1 ]
self . set_header ( ' Content-Type ' , ' text/html ' )
2015-02-23 02:19:25 +00:00
self . finish ( ''' <html>
2014-07-06 00:57:43 +00:00
< title > % s < / title >
< body >
< h2 > Error < / h2 >
< p > % s < / p >
< h2 > Traceback < / h2 >
< p > % s < / p >
< h2 > Request Info < / h2 >
< p > % s < / p >
< / body >
2015-02-23 02:19:25 +00:00
< / html > ''' % (error, error,
2014-07-06 00:57:43 +00:00
trace_info , request_info ) )
2014-06-18 12:56:26 +00:00
2014-06-11 08:34:28 +00:00
def robots_txt ( self , * args , * * kwargs ) :
""" Keep web crawlers out """
self . set_header ( ' Content-Type ' , ' text/plain ' )
2015-02-23 02:19:25 +00:00
return ' User-agent: * \n Disallow: / '
2014-06-11 08:34:28 +00:00
def setHomeLayout ( self , layout ) :
2014-10-09 00:00:36 +00:00
if layout not in ( ' poster ' , ' small ' , ' banner ' , ' simple ' ) :
2014-06-11 08:34:28 +00:00
layout = ' poster '
sickbeard . HOME_LAYOUT = layout
2015-02-28 13:33:37 +00:00
self . redirect ( ' /home/showlistView/ ' )
2014-06-11 08:34:28 +00:00
2014-10-26 05:42:45 +00:00
def setPosterSortBy ( self , sort ) :
2015-03-27 19:25:34 +00:00
if sort not in ( ' name ' , ' date ' , ' network ' , ' progress ' ) :
2014-10-26 05:42:45 +00:00
sort = ' name '
sickbeard . POSTER_SORTBY = sort
sickbeard . save_config ( )
def setPosterSortDir ( self , direction ) :
sickbeard . POSTER_SORTDIR = int ( direction )
sickbeard . save_config ( )
2014-06-11 08:34:28 +00:00
def setHistoryLayout ( self , layout ) :
if layout not in ( ' compact ' , ' detailed ' ) :
layout = ' detailed '
sickbeard . HISTORY_LAYOUT = layout
2015-02-23 02:19:25 +00:00
self . redirect ( ' /history/ ' )
2014-06-11 08:34:28 +00:00
def toggleDisplayShowSpecials ( self , show ) :
sickbeard . DISPLAY_SHOW_SPECIALS = not sickbeard . DISPLAY_SHOW_SPECIALS
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + show )
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
def setEpisodeViewLayout ( self , layout ) :
if layout not in ( ' poster ' , ' banner ' , ' list ' , ' daybyday ' ) :
2014-06-11 08:34:28 +00:00
layout = ' banner '
2014-12-16 14:14:54 +00:00
if ' daybyday ' == layout :
sickbeard . EPISODE_VIEW_SORT = ' time '
2014-09-15 17:43:49 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . EPISODE_VIEW_LAYOUT = layout
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . save_config ( )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /episodeView/ ' )
2014-12-16 14:14:54 +00:00
def toggleEpisodeViewDisplayPaused ( self , * args , * * kwargs ) :
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . EPISODE_VIEW_DISPLAY_PAUSED = not sickbeard . EPISODE_VIEW_DISPLAY_PAUSED
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . save_config ( )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /episodeView/ ' )
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
def setEpisodeViewSort ( self , sort , redir = 1 ) :
if sort not in ( ' time ' , ' network ' , ' show ' ) :
sort = ' time '
2014-11-06 04:55:43 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . EPISODE_VIEW_SORT = sort
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
sickbeard . save_config ( )
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
if int ( redir ) :
2015-02-23 02:19:25 +00:00
self . redirect ( ' /episodeView/ ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def episodeView ( self , layout = ' None ' ) :
2014-12-16 14:14:54 +00:00
""" display the episodes """
2014-11-25 22:12:00 +00:00
today_dt = datetime . date . today ( )
#today = today_dt.toordinal()
yesterday_dt = today_dt - datetime . timedelta ( days = 1 )
yesterday = yesterday_dt . toordinal ( )
2014-11-24 12:05:29 +00:00
tomorrow = ( datetime . date . today ( ) + datetime . timedelta ( days = 1 ) ) . toordinal ( )
2014-11-25 22:12:00 +00:00
next_week_dt = ( datetime . date . today ( ) + datetime . timedelta ( days = 7 ) )
next_week = ( next_week_dt + datetime . timedelta ( days = 1 ) ) . toordinal ( )
2015-02-05 20:22:50 +00:00
recently = ( yesterday_dt - datetime . timedelta ( days = sickbeard . EPISODE_VIEW_MISSED_RANGE ) ) . toordinal ( )
2014-06-11 08:34:28 +00:00
done_show_list = [ ]
2015-02-05 20:22:50 +00:00
qualities = Quality . DOWNLOADED + Quality . SNATCHED + [ ARCHIVED , IGNORED ]
2014-06-11 08:34:28 +00:00
2014-06-21 22:46:59 +00:00
myDB = db . DBConnection ( )
sql_results = myDB . select (
2015-02-05 20:22:50 +00:00
' SELECT *, tv_shows.status as show_status FROM tv_episodes, tv_shows WHERE season != 0 AND airdate >= ? AND airdate <= ? AND tv_shows.indexer_id = tv_episodes.showid AND tv_episodes.status NOT IN ( %s ) '
% ' , ' . join ( [ ' ? ' ] * len ( qualities ) ) ,
[ yesterday , next_week ] + qualities )
2014-06-11 08:34:28 +00:00
2014-06-21 22:46:59 +00:00
for cur_result in sql_results :
2015-02-05 20:22:50 +00:00
done_show_list . append ( int ( cur_result [ ' showid ' ] ) )
sql_results + = myDB . select (
' SELECT *, tv_shows.status as show_status FROM tv_episodes outer_eps, tv_shows WHERE season != 0 AND showid NOT IN ( %s ) '
% ' , ' . join ( [ ' ? ' ] * len ( done_show_list ) )
+ ' AND tv_shows.indexer_id = outer_eps.showid AND airdate = (SELECT airdate FROM tv_episodes inner_eps WHERE inner_eps.season != 0 AND inner_eps.showid = outer_eps.showid AND inner_eps.airdate >= ? ORDER BY inner_eps.airdate ASC LIMIT 1) AND outer_eps.status NOT IN ( %s ) '
% ' , ' . join ( [ ' ? ' ] * len ( Quality . DOWNLOADED + Quality . SNATCHED ) ) ,
done_show_list + [ next_week ] + Quality . DOWNLOADED + Quality . SNATCHED )
sql_results + = myDB . select (
' SELECT *, tv_shows.status as show_status FROM tv_episodes, tv_shows WHERE season != 0 AND tv_shows.indexer_id = tv_episodes.showid AND airdate <= ? AND airdate >= ? AND tv_episodes.status = ? AND tv_episodes.status NOT IN ( %s ) '
% ' , ' . join ( [ ' ? ' ] * len ( qualities ) ) ,
[ tomorrow , recently , WANTED ] + qualities )
2014-06-11 08:34:28 +00:00
2014-11-24 00:55:00 +00:00
sql_results = list ( set ( sql_results ) )
2014-12-16 14:14:54 +00:00
# make a dict out of the sql results
sql_results = [ dict ( row ) for row in sql_results ]
2014-11-25 22:12:00 +00:00
# multi dimension sort
2014-06-11 08:34:28 +00:00
sorts = {
2014-11-25 22:12:00 +00:00
' network ' : ( lambda a , b : cmp (
2014-12-16 14:14:54 +00:00
( a [ ' data_network ' ] , a [ ' localtime ' ] , a [ ' data_show_name ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' data_network ' ] , b [ ' localtime ' ] , b [ ' data_show_name ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) ) ,
' show ' : ( lambda a , b : cmp (
( a [ ' data_show_name ' ] , a [ ' localtime ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' data_show_name ' ] , b [ ' localtime ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) ) ,
' time ' : ( lambda a , b : cmp (
( a [ ' localtime ' ] , a [ ' data_show_name ' ] , a [ ' season ' ] , a [ ' episode ' ] ) ,
( b [ ' localtime ' ] , b [ ' data_show_name ' ] , b [ ' season ' ] , b [ ' episode ' ] ) ) )
2014-06-11 08:34:28 +00:00
}
2015-01-21 16:07:24 +00:00
def value_maybe_article ( value = None ) :
if None is value :
return ' '
2014-12-16 14:14:54 +00:00
return ( remove_article ( value . lower ( ) ) , value . lower ( ) ) [ sickbeard . SORT_ARTICLE ]
2014-06-11 08:34:28 +00:00
# add localtime to the dict
for index , item in enumerate ( sql_results ) :
2015-02-06 11:39:10 +00:00
sql_results [ index ] [ ' localtime ' ] = sbdatetime . sbdatetime . convert_to_setting ( network_timezones . parse_date_time ( item [ ' airdate ' ] ,
2014-11-05 11:33:05 +00:00
item [ ' airs ' ] , item [ ' network ' ] ) )
2014-12-16 14:14:54 +00:00
sql_results [ index ] [ ' data_show_name ' ] = value_maybe_article ( item [ ' show_name ' ] )
sql_results [ index ] [ ' data_network ' ] = value_maybe_article ( item [ ' network ' ] )
2014-06-11 08:34:28 +00:00
2014-12-16 14:14:54 +00:00
sql_results . sort ( sorts [ sickbeard . EPISODE_VIEW_SORT ] )
2014-06-11 08:34:28 +00:00
2015-02-05 20:22:50 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' episodeView.tmpl ' )
2014-11-25 22:12:00 +00:00
t . next_week = datetime . datetime . combine ( next_week_dt , datetime . time ( tzinfo = network_timezones . sb_timezone ) )
t . today = datetime . datetime . now ( network_timezones . sb_timezone )
2014-06-11 08:34:28 +00:00
t . sql_results = sql_results
# Allow local overriding of layout parameter
2014-12-16 14:14:54 +00:00
if layout and layout in ( ' banner ' , ' daybyday ' , ' list ' , ' poster ' ) :
2014-06-11 08:34:28 +00:00
t . layout = layout
else :
2014-12-16 14:14:54 +00:00
t . layout = sickbeard . EPISODE_VIEW_LAYOUT
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-06-11 08:34:28 +00:00
2015-02-06 11:39:10 +00:00
def _genericMessage ( self , subject , message ) :
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' genericMessage.tmpl ' )
t . submenu = self . HomeMenu ( )
2015-02-06 11:39:10 +00:00
t . subject = subject
t . message = message
2015-02-23 02:19:25 +00:00
return t . respond ( )
2015-02-06 11:39:10 +00:00
2015-02-23 02:19:25 +00:00
class Home ( MainHandler ) :
def HomeMenu ( self ) :
return [
{ ' title ' : ' Add Shows ' , ' path ' : ' home/addShows/ ' , } ,
{ ' title ' : ' Manual Post-Processing ' , ' path ' : ' home/postprocess/ ' } ,
{ ' title ' : ' Update XBMC ' , ' path ' : ' home/updateXBMC/ ' , ' requires ' : self . haveXBMC } ,
2015-02-25 12:33:40 +00:00
{ ' title ' : ' Update Kodi ' , ' path ' : ' home/updateKODI/ ' , ' requires ' : self . haveKODI } ,
2015-02-23 02:19:25 +00:00
{ ' title ' : ' Update Plex ' , ' path ' : ' home/updatePLEX/ ' , ' requires ' : self . havePLEX } ,
{ ' title ' : ' Manage Torrents ' , ' path ' : ' manage/manageTorrents ' , ' requires ' : self . haveTORRENT } ,
{ ' title ' : ' Restart ' , ' path ' : ' home/restart/?pid= ' + str ( sickbeard . PID ) , ' confirm ' : True } ,
{ ' title ' : ' Shutdown ' , ' path ' : ' home/shutdown/?pid= ' + str ( sickbeard . PID ) , ' confirm ' : True } ,
]
2015-02-06 11:39:10 +00:00
2015-02-23 02:19:25 +00:00
@staticmethod
def haveXBMC ( ) :
return sickbeard . USE_XBMC and sickbeard . XBMC_UPDATE_LIBRARY
2014-06-11 08:34:28 +00:00
2015-02-25 12:33:40 +00:00
@staticmethod
def haveKODI ( ) :
return sickbeard . USE_KODI and sickbeard . KODI_UPDATE_LIBRARY
2015-02-23 02:19:25 +00:00
@staticmethod
def havePLEX ( ) :
return sickbeard . USE_PLEX and sickbeard . PLEX_UPDATE_LIBRARY
2014-06-07 12:36:50 +00:00
2015-02-23 02:19:25 +00:00
@staticmethod
def haveTORRENT ( ) :
if sickbeard . USE_TORRENTS and sickbeard . TORRENT_METHOD != ' blackhole ' \
and ( sickbeard . ENABLE_HTTPS and sickbeard . TORRENT_HOST [ : 5 ] == ' https '
or not sickbeard . ENABLE_HTTPS and sickbeard . TORRENT_HOST [ : 5 ] == ' http: ' ) :
return True
else :
return False
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
@staticmethod
def _getEpisode ( show , season = None , episode = None , absolute = None ) :
if show is None :
return ' Invalid show parameters '
2015-02-13 04:05:42 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2015-02-13 04:05:42 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return ' Invalid show paramaters '
2015-02-13 04:05:42 +00:00
2015-02-23 02:19:25 +00:00
if absolute :
epObj = showObj . getEpisode ( absolute_number = int ( absolute ) )
elif season and episode :
epObj = showObj . getEpisode ( int ( season ) , int ( episode ) )
else :
return ' Invalid paramaters '
2014-06-21 22:46:59 +00:00
2015-02-23 02:19:25 +00:00
if epObj is None :
return " Episode couldn ' t be retrieved "
2014-06-21 22:46:59 +00:00
2015-02-23 02:19:25 +00:00
return epObj
2014-06-21 22:46:59 +00:00
2015-02-23 02:19:25 +00:00
def index ( self , * args , * * kwargs ) :
2015-02-25 00:34:26 +00:00
if ' episodes ' == sickbeard . DEFAULT_HOME :
self . redirect ( ' /episodeView/ ' )
elif ' history ' == sickbeard . DEFAULT_HOME :
self . redirect ( ' /history/ ' )
else :
self . redirect ( ' /home/showlistView/ ' )
2014-06-11 08:34:28 +00:00
2015-02-25 00:34:26 +00:00
def showlistView ( self ) :
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' home.tmpl ' )
2015-04-07 03:10:50 +00:00
t . showlists = [ ]
index = 0
if sickbeard . SHOWLIST_TAGVIEW == ' custom ' :
for name in sickbeard . SHOW_TAGS :
results = filter ( lambda x : x . tag == name , sickbeard . showList )
if results :
t . showlists . append ( [ ' container %s ' % index , name , results ] )
index + = 1
elif sickbeard . SHOWLIST_TAGVIEW == ' anime ' :
show_results = filter ( lambda x : not x . anime , sickbeard . showList )
anime_results = filter ( lambda x : x . anime , sickbeard . showList )
if show_results :
t . showlists . append ( [ ' container %s ' % index , ' Show List ' , show_results ] )
index + = 1
if anime_results :
t . showlists . append ( [ ' container %s ' % index , ' Anime List ' , anime_results ] )
2015-06-05 13:10:17 +00:00
if 0 == len ( t . showlists ) :
t . showlists . append ( [ ' container0 ' , ' Show List ' , sickbeard . showList ] )
2014-06-11 08:34:28 +00:00
2015-05-03 22:52:03 +00:00
if ' simple ' != sickbeard . HOME_LAYOUT :
t . network_images = { }
networks = { }
images_path = ek . ek ( os . path . join , sickbeard . PROG_DIR , ' gui ' , ' slick ' , ' images ' , ' network ' )
for item in sickbeard . showList :
network_name = ' nonetwork ' if None is item . network else item . network . replace ( u ' \u00C9 ' , ' e ' ) . lower ( )
if network_name not in networks :
filename = u ' %s .png ' % network_name
if not ek . ek ( os . path . isfile , ek . ek ( os . path . join , images_path , filename ) ) :
filename = u ' %s .png ' % re . sub ( r ' (?m)(.*) \ s+ \ ( \ w {2} \ )$ ' , r ' \ 1 ' , network_name )
if not ek . ek ( os . path . isfile , ek . ek ( os . path . join , images_path , filename ) ) :
filename = u ' nonetwork.png '
networks . setdefault ( network_name , filename )
t . network_images . setdefault ( item . indexerid , networks [ network_name ] )
2015-02-23 02:19:25 +00:00
t . submenu = self . HomeMenu ( )
2015-03-29 08:29:40 +00:00
t . layout = sickbeard . HOME_LAYOUT
# Get all show snatched / downloaded / next air date stats
myDB = db . DBConnection ( )
today = datetime . date . today ( ) . toordinal ( )
status_quality = ' , ' . join ( [ str ( x ) for x in Quality . SNATCHED + Quality . SNATCHED_PROPER ] )
status_download = ' , ' . join ( [ str ( x ) for x in Quality . DOWNLOADED + [ ARCHIVED ] ] )
status_total = ' %s , %s , %s ' % ( SKIPPED , WANTED , FAILED )
sql_statement = ' SELECT showid, '
sql_statement + = ' (SELECT COUNT(*) FROM tv_episodes WHERE showid=tv_eps.showid AND season > 0 AND episode > 0 AND airdate > 1 AND status IN ( %s )) AS ep_snatched, '
sql_statement + = ' (SELECT COUNT(*) FROM tv_episodes WHERE showid=tv_eps.showid AND season > 0 AND episode > 0 AND airdate > 1 AND status IN ( %s )) AS ep_downloaded, '
sql_statement + = ' (SELECT COUNT(*) FROM tv_episodes WHERE showid=tv_eps.showid AND season > 0 AND episode > 0 AND airdate > 1 AND ((airdate <= %s AND (status IN ( %s ))) OR (status IN ( %s )) OR (status IN ( %s )))) AS ep_total, '
sql_statement + = ' (SELECT airdate FROM tv_episodes WHERE showid=tv_eps.showid AND airdate >= %s AND (status = %s OR status = %s ) ORDER BY airdate ASC LIMIT 1) AS ep_airs_next '
sql_statement + = ' FROM tv_episodes tv_eps GROUP BY showid '
sql_result = myDB . select ( sql_statement % ( status_quality , status_download , today , status_total , status_quality , status_download , today , UNAIRED , WANTED ) )
t . show_stat = { }
for cur_result in sql_result :
t . show_stat [ cur_result [ ' showid ' ] ] = cur_result
2014-06-30 01:54:41 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2015-02-06 11:39:10 +00:00
2015-02-23 02:19:25 +00:00
def testSABnzbd ( self , host = None , username = None , password = None , apikey = None ) :
2015-02-06 11:39:10 +00:00
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2015-02-23 02:19:25 +00:00
host = config . clean_url ( host )
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . SAB_PASSWORD
if None is not apikey and starify ( apikey , True ) :
apikey = sickbeard . SAB_APIKEY
2015-02-23 02:19:25 +00:00
connection , accesMsg = sab . getSabAccesMethod ( host , username , password , apikey )
if connection :
authed , authMsg = sab . testAuthentication ( host , username , password , apikey ) # @UnusedVariable
if authed :
return ' Success. Connected and authenticated '
else :
return " Authentication failed. SABnzbd expects ' " + accesMsg + " ' as authentication method "
2015-02-06 11:39:10 +00:00
else :
2015-02-23 02:19:25 +00:00
return ' Unable to connect to host '
2015-02-06 11:39:10 +00:00
2015-02-23 02:19:25 +00:00
def testTorrent ( self , torrent_method = None , host = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_url ( host )
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . TORRENT_PASSWORD
2014-07-01 08:49:12 +00:00
2015-02-23 02:19:25 +00:00
client = clients . getClientIstance ( torrent_method )
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
connection , accesMsg = client ( host , username , password ) . testAuthentication ( )
2014-05-13 09:58:27 +00:00
2015-02-23 02:19:25 +00:00
return accesMsg
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
def testGrowl ( self , host = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host , default_port = 23053 )
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . GROWL_PASSWORD
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = notifiers . growl_notifier . test_notify ( host , password )
if password is None or password == ' ' :
pw_append = ' '
else :
pw_append = ' with password: ' + password
2014-07-06 14:45:01 +00:00
2015-02-23 02:19:25 +00:00
if result :
return ' Registered and Tested growl successfully ' + urllib . unquote_plus ( host ) + pw_append
else :
return ' Registration and Testing of growl failed ' + urllib . unquote_plus ( host ) + pw_append
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testProwl ( self , prowl_api = None , prowl_priority = 0 ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-07-08 03:27:24 +00:00
2015-02-24 13:37:27 +00:00
if None is not prowl_api and starify ( prowl_api , True ) :
prowl_api = sickbeard . PROWL_API
2015-02-23 02:19:25 +00:00
result = notifiers . prowl_notifier . test_notify ( prowl_api , prowl_priority )
if result :
return ' Test prowl notice sent successfully '
else :
return ' Test prowl notice failed '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testBoxcar2 ( self , accesstoken = None , sound = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not accesstoken and starify ( accesstoken , True ) :
accesstoken = sickbeard . BOXCAR2_ACCESSTOKEN
2015-02-23 02:19:25 +00:00
result = notifiers . boxcar2_notifier . test_notify ( accesstoken , sound )
if result :
return ' Boxcar2 notification succeeded. Check your Boxcar2 clients to make sure it worked '
else :
return ' Error sending Boxcar2 notification '
2014-03-10 05:18:05 +00:00
2015-02-25 11:30:52 +00:00
def testPushover ( self , userKey = None , apiKey = None , priority = None , device = None , sound = None ) :
2015-02-23 02:19:25 +00:00
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not userKey and starify ( userKey , True ) :
userKey = sickbeard . PUSHOVER_USERKEY
if None is not apiKey and starify ( apiKey , True ) :
apiKey = sickbeard . PUSHOVER_APIKEY
2015-02-25 11:30:52 +00:00
result = notifiers . pushover_notifier . test_notify ( userKey , apiKey , priority , device , sound )
2015-02-23 02:19:25 +00:00
if result :
return ' Pushover notification succeeded. Check your Pushover clients to make sure it worked '
else :
return ' Error sending Pushover notification '
2014-07-06 14:45:01 +00:00
2015-02-25 11:30:52 +00:00
def getPushoverDevices ( self , userKey = None , apiKey = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
if None is not userKey and starify ( userKey , True ) :
userKey = sickbeard . PUSHOVER_USERKEY
if None is not apiKey and starify ( apiKey , True ) :
apiKey = sickbeard . PUSHOVER_APIKEY
result = notifiers . pushover_notifier . get_devices ( userKey , apiKey )
if result :
return result
else :
return " {} "
2015-02-23 02:19:25 +00:00
def twitterStep1 ( self , * args , * * kwargs ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return notifiers . twitter_notifier . _get_authorization ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def twitterStep2 ( self , key ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = notifiers . twitter_notifier . _get_credentials ( key )
logger . log ( u ' result: ' + str ( result ) )
if result :
return ' Key verification successful '
else :
return ' Unable to verify key '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testTwitter ( self , * args , * * kwargs ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = notifiers . twitter_notifier . test_notify ( )
if result :
return ' Tweet successful, check your twitter to make sure it worked '
else :
return ' Error sending tweet '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testXBMC ( self , host = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-06-17 19:51:23 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_hosts ( host )
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . XBMC_PASSWORD
2015-02-23 02:19:25 +00:00
finalResult = ' '
for curHost in [ x . strip ( ) for x in host . split ( ' , ' ) ] :
curResult = notifiers . xbmc_notifier . test_notify ( urllib . unquote_plus ( curHost ) , username , password )
if len ( curResult . split ( ' : ' ) ) > 2 and ' OK ' in curResult . split ( ' : ' ) [ 2 ] :
finalResult + = ' Test XBMC notice sent successfully to ' + urllib . unquote_plus ( curHost )
else :
finalResult + = ' Test XBMC notice failed to ' + urllib . unquote_plus ( curHost )
finalResult + = " <br /> \n "
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return finalResult
2014-03-10 05:18:05 +00:00
2015-02-25 12:33:40 +00:00
def testKODI ( self , host = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
host = config . clean_hosts ( host )
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . KODI_PASSWORD
finalResult = ' '
for curHost in [ x . strip ( ) for x in host . split ( ' , ' ) ] :
curResult = notifiers . kodi_notifier . test_notify ( urllib . unquote_plus ( curHost ) , username , password )
if len ( curResult . split ( ' : ' ) ) > 2 and ' OK ' in curResult . split ( ' : ' ) [ 2 ] :
finalResult + = ' Test Kodi notice sent successfully to ' + urllib . unquote_plus ( curHost )
else :
finalResult + = ' Test Kodi notice failed to ' + urllib . unquote_plus ( curHost )
finalResult + = ' <br /> \n '
return finalResult
2015-02-23 02:19:25 +00:00
def testPMC ( self , host = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-25 05:57:24 +00:00
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . PLEX_PASSWORD
2015-02-23 02:19:25 +00:00
finalResult = ' '
for curHost in [ x . strip ( ) for x in host . split ( ' , ' ) ] :
curResult = notifiers . plex_notifier . test_notify_pmc ( urllib . unquote_plus ( curHost ) , username , password )
if len ( curResult . split ( ' : ' ) ) > 2 and ' OK ' in curResult . split ( ' : ' ) [ 2 ] :
finalResult + = ' Successful test notice sent to Plex client ... ' + urllib . unquote_plus ( curHost )
else :
finalResult + = ' Test failed for Plex client ... ' + urllib . unquote_plus ( curHost )
finalResult + = ' <br /> ' + ' \n '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Tested Plex client(s): ' , urllib . unquote_plus ( host . replace ( ' , ' , ' , ' ) ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return finalResult
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testPMS ( self , host = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-25 05:57:24 +00:00
2015-02-24 13:37:27 +00:00
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . PLEX_PASSWORD
2015-02-23 02:19:25 +00:00
finalResult = ' '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
curResult = notifiers . plex_notifier . test_notify_pms ( urllib . unquote_plus ( host ) , username , password )
if None is curResult :
finalResult + = ' Successful test of Plex server(s) ... ' + urllib . unquote_plus ( host . replace ( ' , ' , ' , ' ) )
else :
finalResult + = ' Test failed for Plex server(s) ... ' + urllib . unquote_plus ( curResult . replace ( ' , ' , ' , ' ) )
finalResult + = ' <br /> ' + ' \n '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Tested Plex Media Server host(s): ' , urllib . unquote_plus ( host . replace ( ' , ' , ' , ' ) ) )
2014-07-27 17:58:19 +00:00
2015-02-23 02:19:25 +00:00
return finalResult
2014-07-27 17:58:19 +00:00
2015-02-23 02:19:25 +00:00
def testLibnotify ( self , * args , * * kwargs ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-05-15 22:06:13 +00:00
2015-02-23 02:19:25 +00:00
if notifiers . libnotify_notifier . test_notify ( ) :
return ' Tried sending desktop notification via libnotify '
else :
return notifiers . libnotify . diagnose ( )
2014-05-15 22:06:13 +00:00
2015-02-23 02:19:25 +00:00
def testNMJ ( self , host = None , database = None , mount = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-05-15 04:16:46 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host )
result = notifiers . nmj_notifier . test_notify ( urllib . unquote_plus ( host ) , database , mount )
2014-05-15 04:16:46 +00:00
if result :
2015-02-23 02:19:25 +00:00
return ' Successfully started the scan update '
else :
return ' Test failed to start the scan update '
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def settingsNMJ ( self , host = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-05-19 17:40:25 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host )
result = notifiers . nmj_notifier . notify_settings ( urllib . unquote_plus ( host ) )
2014-05-19 17:40:25 +00:00
if result :
2015-02-23 02:19:25 +00:00
return ' { " message " : " Got settings from %(host)s " , " database " : " %(database)s " , " mount " : " %(mount)s " } ' % {
" host " : host , " database " : sickbeard . NMJ_DATABASE , " mount " : sickbeard . NMJ_MOUNT }
else :
return ' { " message " : " Failed! Make sure your Popcorn is on and NMJ is running. (see Log & Errors -> Debug for detailed info) " , " database " : " " , " mount " : " " } '
2014-05-15 21:43:45 +00:00
2015-02-23 02:19:25 +00:00
def testNMJv2 ( self , host = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host )
result = notifiers . nmjv2_notifier . test_notify ( urllib . unquote_plus ( host ) )
if result :
return ' Test notice sent successfully to ' + urllib . unquote_plus ( host )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
return ' Test notice failed to ' + urllib . unquote_plus ( host )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def settingsNMJv2 ( self , host = None , dbloc = None , instance = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host )
result = notifiers . nmjv2_notifier . notify_settings ( urllib . unquote_plus ( host ) , dbloc , instance )
if result :
return ' { " message " : " NMJ Database found at: %(host)s " , " database " : " %(database)s " } ' % { " host " : host ,
" database " : sickbeard . NMJv2_DATABASE }
else :
return ' { " message " : " Unable to find NMJ Database at location: %(dbloc)s . Is the right location selected and PCH running? " , " database " : " " } ' % {
" dbloc " : dbloc }
2014-06-17 19:51:23 +00:00
2015-02-23 02:19:25 +00:00
def testTrakt ( self , api = None , username = None , password = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-06-11 08:34:28 +00:00
2015-02-24 13:37:27 +00:00
if None is not api and starify ( api , True ) :
api = sickbeard . TRAKT_API
if None is not password and set ( ' * ' ) == set ( password ) :
password = sickbeard . TRAKT_PASSWORD
2015-02-23 02:19:25 +00:00
result = notifiers . trakt_notifier . test_notify ( api , username , password )
if result :
return ' Test notice sent successfully to Trakt '
else :
return ' Test notice failed to Trakt '
def loadShowNotifyLists ( self , * args , * * kwargs ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2014-06-21 22:46:59 +00:00
myDB = db . DBConnection ( )
2015-02-23 02:19:25 +00:00
rows = myDB . select ( ' SELECT show_id, show_name, notify_list FROM tv_shows ORDER BY show_name ASC ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
data = { }
size = 0
for r in rows :
data [ r [ ' show_id ' ] ] = { ' id ' : r [ ' show_id ' ] , ' name ' : r [ ' show_name ' ] , ' list ' : r [ ' notify_list ' ] }
size + = 1
data [ ' _size ' ] = size
return json . dumps ( data )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testEmail ( self , host = None , port = None , smtp_from = None , use_tls = None , user = None , pwd = None , to = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not pwd and set ( ' * ' ) == set ( pwd ) :
pwd = sickbeard . EMAIL_PASSWORD
2015-02-23 02:19:25 +00:00
host = config . clean_host ( host )
2015-02-24 13:37:27 +00:00
2015-02-23 02:19:25 +00:00
if notifiers . email_notifier . test_notify ( host , port , smtp_from , use_tls , user , pwd , to ) :
return ' Test email sent successfully! Check inbox. '
else :
return ' ERROR: %s ' % notifiers . email_notifier . last_err
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testNMA ( self , nma_api = None , nma_priority = 0 ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not nma_api and starify ( nma_api , True ) :
nma_api = sickbeard . NMA_API
2015-02-23 02:19:25 +00:00
result = notifiers . nma_notifier . test_notify ( nma_api , nma_priority )
if result :
return ' Test NMA notice sent successfully '
else :
return ' Test NMA notice failed '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def testPushalot ( self , authorizationToken = None ) :
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2015-02-24 13:37:27 +00:00
if None is not authorizationToken and starify ( authorizationToken , True ) :
authorizationToken = sickbeard . PUSHALOT_AUTHORIZATIONTOKEN
2015-02-23 02:19:25 +00:00
result = notifiers . pushalot_notifier . test_notify ( authorizationToken )
if result :
return ' Pushalot notification succeeded. Check your Pushalot clients to make sure it worked '
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
return ' Error sending Pushalot notification '
2014-03-10 05:18:05 +00:00
2015-02-24 07:32:45 +00:00
def testPushbullet ( self , accessToken = None , device_iden = None ) :
2015-02-23 02:19:25 +00:00
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not accessToken and starify ( accessToken , True ) :
accessToken = sickbeard . PUSHBULLET_ACCESS_TOKEN
2015-06-15 14:46:34 +00:00
return notifiers . pushbullet_notifier . test_notify ( accessToken , device_iden )
2014-03-10 05:18:05 +00:00
2015-02-24 07:32:45 +00:00
def getPushbulletDevices ( self , accessToken = None ) :
2015-02-23 02:19:25 +00:00
self . set_header ( ' Cache-Control ' , ' max-age=0,no-cache,no-store ' )
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if None is not accessToken and starify ( accessToken , True ) :
accessToken = sickbeard . PUSHBULLET_ACCESS_TOKEN
2015-06-15 14:46:34 +00:00
return notifiers . pushbullet_notifier . get_devices ( accessToken )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def shutdown ( self , pid = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if str ( pid ) != str ( sickbeard . PID ) :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . events . put ( sickbeard . events . SystemEvent . SHUTDOWN )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
title = ' Shutting down '
message = ' SickGear is shutting down... '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return self . _genericMessage ( title , message )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def restart ( self , pid = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if str ( pid ) != str ( sickbeard . PID ) :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' restart.tmpl ' )
t . submenu = self . HomeMenu ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# restart
sickbeard . events . put ( sickbeard . events . SystemEvent . RESTART )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def update ( self , pid = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if str ( pid ) != str ( sickbeard . PID ) :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
updated = sickbeard . versionCheckScheduler . action . update ( ) # @UndefinedVariable
if updated :
# do a hard restart
sickbeard . events . put ( sickbeard . events . SystemEvent . RESTART )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' restart_bare.tmpl ' )
return t . respond ( )
else :
return self . _genericMessage ( ' Update Failed ' ,
" Update wasn ' t successful, not restarting. Check your log for more information. " )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def branchCheckout ( self , branch ) :
sickbeard . BRANCH = branch
ui . notifications . message ( ' Checking out branch: ' , branch )
return self . update ( sickbeard . PID )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def pullRequestCheckout ( self , branch ) :
pull_request = branch
branch = branch . split ( ' : ' ) [ 1 ]
fetched = sickbeard . versionCheckScheduler . action . fetch ( pull_request )
if fetched :
sickbeard . BRANCH = branch
ui . notifications . message ( ' Checking out branch: ' , branch )
return self . update ( sickbeard . PID )
else :
self . redirect ( ' /home/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def displayShow ( self , show = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' Invalid show ID ' )
else :
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Show not in show list ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
seasonResults = myDB . select (
' SELECT DISTINCT season FROM tv_episodes WHERE showid = ? ORDER BY season desc ' ,
[ showObj . indexerid ]
)
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sqlResults = myDB . select (
' SELECT * FROM tv_episodes WHERE showid = ? ORDER BY season DESC, episode DESC ' ,
[ showObj . indexerid ]
)
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' displayShow.tmpl ' )
t . submenu = [ { ' title ' : ' Edit ' , ' path ' : ' home/editShow?show= %d ' % showObj . indexerid } ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
try :
t . showLoc = ( showObj . location , True )
except sickbeard . exceptions . ShowDirNotFoundException :
t . showLoc = ( showObj . _location , False )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
show_message = ' '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . showQueueScheduler . action . isBeingAdded ( showObj ) : # @UndefinedVariable
show_message = ' This show is in the process of being downloaded - the info below is incomplete. '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isBeingUpdated ( showObj ) : # @UndefinedVariable
show_message = ' The information on this page is in the process of being updated. '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isBeingRefreshed ( showObj ) : # @UndefinedVariable
show_message = ' The episodes below are currently being refreshed from disk '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isBeingSubtitled ( showObj ) : # @UndefinedVariable
show_message = ' Currently downloading subtitles for this show '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isInRefreshQueue ( showObj ) : # @UndefinedVariable
show_message = ' This show is queued to be refreshed. '
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isInUpdateQueue ( showObj ) : # @UndefinedVariable
show_message = ' This show is queued and awaiting an update. '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
elif sickbeard . showQueueScheduler . action . isInSubtitleQueue ( showObj ) : # @UndefinedVariable
show_message = ' This show is queued and awaiting subtitles download. '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not sickbeard . showQueueScheduler . action . isBeingAdded ( showObj ) : # @UndefinedVariable
if not sickbeard . showQueueScheduler . action . isBeingUpdated ( showObj ) : # @UndefinedVariable
t . submenu . append (
{ ' title ' : ' Remove ' , ' path ' : ' home/deleteShow?show= %d ' % showObj . indexerid , ' confirm ' : True } )
t . submenu . append ( { ' title ' : ' Re-scan files ' , ' path ' : ' home/refreshShow?show= %d ' % showObj . indexerid } )
t . submenu . append (
2015-03-17 00:53:12 +00:00
{ ' title ' : ' Force Full Update ' , ' path ' : ' home/updateShow?show= %d &force=1&web=1 ' % showObj . indexerid } )
2015-02-23 02:19:25 +00:00
t . submenu . append ( { ' title ' : ' Update show in XBMC ' ,
' path ' : ' home/updateXBMC?showName= %s ' % urllib . quote_plus (
2015-02-25 12:33:40 +00:00
showObj . name . encode ( ' utf-8 ' ) ) , ' requires ' : self . haveXBMC } )
t . submenu . append ( { ' title ' : ' Update show in Kodi ' ,
' path ' : ' home/updateKODI?showName= %s ' % urllib . quote_plus (
2015-02-25 19:31:58 +00:00
showObj . name . encode ( ' utf-8 ' ) ) , ' requires ' : self . haveKODI } )
2015-02-23 02:19:25 +00:00
t . submenu . append ( { ' title ' : ' Preview Rename ' , ' path ' : ' home/testRename?show= %d ' % showObj . indexerid } )
if sickbeard . USE_SUBTITLES and not sickbeard . showQueueScheduler . action . isBeingSubtitled (
showObj ) and showObj . subtitles :
t . submenu . append (
{ ' title ' : ' Download Subtitles ' , ' path ' : ' home/subtitleShow?show= %d ' % showObj . indexerid } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . show = showObj
t . sqlResults = sqlResults
t . seasonResults = seasonResults
t . show_message = show_message
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
epCounts = { }
epCats = { }
epCounts [ Overview . SKIPPED ] = 0
epCounts [ Overview . WANTED ] = 0
epCounts [ Overview . QUAL ] = 0
epCounts [ Overview . GOOD ] = 0
epCounts [ Overview . UNAIRED ] = 0
epCounts [ Overview . SNATCHED ] = 0
2015-03-23 02:16:33 +00:00
epCounts [ ' videos ' ] = { }
epCounts [ ' archived ' ] = { }
epCounts [ ' totals ' ] = { }
highest_season = 0
latest_season = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for curResult in sqlResults :
curEpCat = showObj . getOverview ( int ( curResult [ ' status ' ] ) )
if curEpCat :
epCats [ str ( curResult [ ' season ' ] ) + ' x ' + str ( curResult [ ' episode ' ] ) ] = curEpCat
epCounts [ curEpCat ] + = 1
2015-03-23 02:16:33 +00:00
if ' ' != curResult [ ' location ' ] :
if curResult [ ' season ' ] not in epCounts [ ' videos ' ] :
epCounts [ ' videos ' ] [ curResult [ ' season ' ] ] = 1
else :
epCounts [ ' videos ' ] [ curResult [ ' season ' ] ] + = 1
if curResult [ ' season ' ] not in epCounts [ ' totals ' ] :
epCounts [ ' totals ' ] [ curResult [ ' season ' ] ] = 1
else :
epCounts [ ' totals ' ] [ curResult [ ' season ' ] ] + = 1
if ARCHIVED == curResult [ ' status ' ] :
if curResult [ ' season ' ] not in epCounts [ ' archived ' ] :
epCounts [ ' archived ' ] [ curResult [ ' season ' ] ] = 1
else :
epCounts [ ' archived ' ] [ curResult [ ' season ' ] ] + = 1
if highest_season < curResult [ ' season ' ] and 1000 < curResult [ ' airdate ' ] and UNAIRED < curResult [ ' status ' ] :
highest_season = curResult [ ' season ' ]
2015-05-03 16:24:08 +00:00
if 0 < len ( epCounts [ ' totals ' ] ) :
latest_season = int ( sorted ( epCounts [ ' totals ' ] ) [ - 1 : : ] [ 0 ] )
2015-03-23 02:16:33 +00:00
display_seasons = [ ]
if 1 < highest_season :
display_seasons + = [ 1 ]
display_seasons + = [ highest_season ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def titler ( x ) :
return ( remove_article ( x ) , x ) [ not x or sickbeard . SORT_ARTICLE ]
2014-03-10 05:18:05 +00:00
2015-04-07 03:10:50 +00:00
if sickbeard . SHOWLIST_TAGVIEW == ' custom ' :
t . sortedShowLists = [ ]
for tag in sickbeard . SHOW_TAGS :
results = filter ( lambda x : x . tag == tag , sickbeard . showList )
if results :
t . sortedShowLists . append ( [ tag , sorted ( results , lambda x , y : cmp ( titler ( x . name ) , titler ( y . name ) ) ) ] )
elif sickbeard . SHOWLIST_TAGVIEW == ' anime ' :
2015-02-23 02:19:25 +00:00
shows = [ ]
anime = [ ]
for show in sickbeard . showList :
if show . is_anime :
anime . append ( show )
else :
shows . append ( show )
t . sortedShowLists = [ [ ' Shows ' , sorted ( shows , lambda x , y : cmp ( titler ( x . name ) , titler ( y . name ) ) ) ] ,
[ ' Anime ' , sorted ( anime , lambda x , y : cmp ( titler ( x . name ) , titler ( y . name ) ) ) ] ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
else :
t . sortedShowLists = [
2015-04-07 03:10:50 +00:00
[ ' Show List ' , sorted ( sickbeard . showList , lambda x , y : cmp ( titler ( x . name ) , titler ( y . name ) ) ) ] ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
tvshows = [ ]
2015-04-07 03:10:50 +00:00
tvshow_names = [ ]
2015-02-23 02:19:25 +00:00
for tvshow_types in t . sortedShowLists :
for tvshow in tvshow_types [ 1 ] :
tvshows . append ( tvshow . indexerid )
2015-04-07 03:10:50 +00:00
tvshow_names . append ( tvshow . name )
if showObj . indexerid == tvshow . indexerid :
cur_sel = len ( tvshow_names )
2015-02-23 02:19:25 +00:00
t . tvshow_id_csv = ' , ' . join ( str ( x ) for x in tvshows )
2014-06-11 08:34:28 +00:00
2015-04-07 03:10:50 +00:00
last_item = len ( tvshow_names )
t . prev_title = ' Prev show, %s ' % tvshow_names [ ( cur_sel - 2 , last_item - 1 ) [ 1 == cur_sel ] ]
t . next_title = ' Next show, %s ' % tvshow_names [ ( cur_sel , 0 ) [ last_item == cur_sel ] ]
2015-02-23 02:19:25 +00:00
t . bwl = None
if showObj . is_anime :
t . bwl = showObj . release_groups
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . epCounts = epCounts
t . epCats = epCats
2015-03-23 02:16:33 +00:00
t . display_seasons = display_seasons
t . latest_season = latest_season
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj . exceptions = scene_exceptions . get_scene_exceptions ( showObj . indexerid )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexerid = int ( showObj . indexerid )
indexer = int ( showObj . indexer )
t . all_scene_exceptions = showObj . exceptions
t . scene_numbering = get_scene_numbering_for_show ( indexerid , indexer )
t . xem_numbering = get_xem_numbering_for_show ( indexerid , indexer )
t . scene_absolute_numbering = get_scene_absolute_numbering_for_show ( indexerid , indexer )
t . xem_absolute_numbering = get_xem_absolute_numbering_for_show ( indexerid , indexer )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def plotDetails ( self , show , season , episode ) :
myDB = db . DBConnection ( )
result = myDB . select (
' SELECT description FROM tv_episodes WHERE showid = ? AND season = ? AND episode = ? ' ,
( int ( show ) , int ( season ) , int ( episode ) ) )
return result [ 0 ] [ ' description ' ] if result else ' Episode not found. '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def sceneExceptions ( self , show ) :
exceptionsList = sickbeard . scene_exceptions . get_all_scene_exceptions ( show )
if not exceptionsList :
return ' No scene exceptions '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
out = [ ]
2015-06-13 14:52:46 +00:00
for season , names in iter ( sorted ( iteritems ( exceptionsList ) ) ) :
2015-02-23 02:19:25 +00:00
if season == - 1 :
season = ' * '
out . append ( ' S ' + str ( season ) + ' : ' + ' , ' . join ( names ) )
return ' <br/> ' . join ( out )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def editShow ( self , show = None , location = None , anyQualities = [ ] , bestQualities = [ ] , exceptions_list = [ ] ,
flatten_folders = None , paused = None , directCall = False , air_by_date = None , sports = None , dvdorder = None ,
indexerLang = None , subtitles = None , archive_firstmatch = None , rls_ignore_words = None ,
rls_require_words = None , anime = None , blacklist = None , whitelist = None ,
2015-04-07 03:10:50 +00:00
scene = None , tag = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
errString = ' Invalid show ID: ' + str ( show )
if directCall :
return [ errString ]
else :
return self . _genericMessage ( ' Error ' , errString )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not showObj :
errString = ' Unable to find the specified show: ' + str ( show )
if directCall :
return [ errString ]
else :
return self . _genericMessage ( ' Error ' , errString )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj . exceptions = scene_exceptions . get_scene_exceptions ( showObj . indexerid )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not location and not anyQualities and not bestQualities and not flatten_folders :
t = PageTemplate ( headers = self . request . headers , file = ' editShow.tmpl ' )
t . submenu = self . HomeMenu ( )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if showObj . is_anime :
2015-07-16 23:09:27 +00:00
if not showObj . release_groups :
showObj . release_groups = BlackAndWhiteList ( showObj . indexerid )
2015-02-23 02:19:25 +00:00
t . whitelist = showObj . release_groups . whitelist
t . blacklist = showObj . release_groups . blacklist
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . groups = [ ]
if helpers . set_up_anidb_connection ( ) :
try :
anime = adba . Anime ( sickbeard . ADBA_CONNECTION , name = showObj . name )
t . groups = anime . get_groups ( )
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-02-23 02:19:25 +00:00
t . groups . append ( dict ( [ ( ' name ' , ' Fail:AniDB connect. Restart sg else check debug log ' ) , ( ' rating ' , ' ' ) , ( ' range ' , ' ' ) ] ) )
else :
t . groups . append ( dict ( [ ( ' name ' , ' Did not initialise AniDB. Check debug log if reqd. ' ) , ( ' rating ' , ' ' ) , ( ' range ' , ' ' ) ] ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
with showObj . lock :
t . show = showObj
t . scene_exceptions = get_scene_exceptions ( showObj . indexerid )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
flatten_folders = config . checkbox_to_value ( flatten_folders )
dvdorder = config . checkbox_to_value ( dvdorder )
archive_firstmatch = config . checkbox_to_value ( archive_firstmatch )
paused = config . checkbox_to_value ( paused )
air_by_date = config . checkbox_to_value ( air_by_date )
scene = config . checkbox_to_value ( scene )
sports = config . checkbox_to_value ( sports )
anime = config . checkbox_to_value ( anime )
subtitles = config . checkbox_to_value ( subtitles )
2014-10-16 20:45:07 +00:00
2015-02-23 02:19:25 +00:00
if indexerLang and indexerLang in sickbeard . indexerApi ( showObj . indexer ) . indexer ( ) . config [ ' valid_languages ' ] :
indexer_lang = indexerLang
else :
indexer_lang = showObj . lang
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if we changed the language then kick off an update
if indexer_lang == showObj . lang :
do_update = False
else :
do_update = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if scene == showObj . scene and anime == showObj . anime :
do_update_scene_numbering = False
else :
do_update_scene_numbering = True
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if type ( anyQualities ) != list :
anyQualities = [ anyQualities ]
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
if type ( bestQualities ) != list :
bestQualities = [ bestQualities ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if type ( exceptions_list ) != list :
exceptions_list = [ exceptions_list ]
2014-03-10 05:18:05 +00:00
2015-04-07 03:10:50 +00:00
# If directCall from mass_edit_update no scene exceptions handling or blackandwhite list handling or tags
2015-02-23 02:19:25 +00:00
if directCall :
do_update_exceptions = False
else :
if set ( exceptions_list ) == set ( showObj . exceptions ) :
do_update_exceptions = False
else :
do_update_exceptions = True
2014-05-30 06:22:01 +00:00
2015-02-23 02:19:25 +00:00
with showObj . lock :
if anime :
if not showObj . release_groups :
showObj . release_groups = BlackAndWhiteList ( showObj . indexerid )
if whitelist :
shortwhitelist = short_group_names ( whitelist )
showObj . release_groups . set_white_keywords ( shortwhitelist )
else :
showObj . release_groups . set_white_keywords ( [ ] )
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
if blacklist :
shortblacklist = short_group_names ( blacklist )
showObj . release_groups . set_black_keywords ( shortblacklist )
else :
showObj . release_groups . set_black_keywords ( [ ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
errors = [ ]
with showObj . lock :
newQuality = Quality . combineQualities ( map ( int , anyQualities ) , map ( int , bestQualities ) )
showObj . quality = newQuality
showObj . archive_firstmatch = archive_firstmatch
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# reversed for now
if bool ( showObj . flatten_folders ) != bool ( flatten_folders ) :
showObj . flatten_folders = flatten_folders
try :
sickbeard . showQueueScheduler . action . refreshShow ( showObj ) # @UndefinedVariable
2015-06-08 12:47:01 +00:00
except exceptions . CantRefreshException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to refresh this show: ' + ex ( e ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj . paused = paused
showObj . scene = scene
showObj . anime = anime
showObj . sports = sports
showObj . subtitles = subtitles
showObj . air_by_date = air_by_date
2015-04-07 03:10:50 +00:00
showObj . tag = tag
2014-10-16 20:45:07 +00:00
2015-02-23 02:19:25 +00:00
if not directCall :
showObj . lang = indexer_lang
showObj . dvdorder = dvdorder
showObj . rls_ignore_words = rls_ignore_words . strip ( )
showObj . rls_require_words = rls_require_words . strip ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if we change location clear the db of episodes, change it, write to db, and rescan
if os . path . normpath ( showObj . _location ) != os . path . normpath ( location ) :
logger . log ( os . path . normpath ( showObj . _location ) + ' != ' + os . path . normpath ( location ) , logger . DEBUG )
if not ek . ek ( os . path . isdir , location ) and not sickbeard . CREATE_MISSING_SHOW_DIRS :
errors . append ( ' New location <tt> %s </tt> does not exist ' % location )
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
# don't bother if we're going to update anyway
elif not do_update :
# change it
try :
showObj . location = location
try :
sickbeard . showQueueScheduler . action . refreshShow ( showObj ) # @UndefinedVariable
2015-06-08 12:47:01 +00:00
except exceptions . CantRefreshException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to refresh this show: ' + ex ( e ) )
# grab updated info from TVDB
# showObj.loadEpisodesFromIndexer()
# rescan the episodes in the new folder
except exceptions . NoNFOException :
errors . append (
" The folder at <tt> %s </tt> doesn ' t contain a tvshow.nfo - copy your files to that folder before you change the directory in SickGear. " % location )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# save it to the DB
showObj . saveToDB ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# force the update
if do_update :
try :
sickbeard . showQueueScheduler . action . updateShow ( showObj , True ) # @UndefinedVariable
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to force an update on the show. ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if do_update_exceptions :
try :
scene_exceptions . update_scene_exceptions ( showObj . indexerid , exceptions_list ) # @UndefinedVdexerid)
2015-03-16 16:38:38 +00:00
buildNameCache ( showObj )
2015-02-23 02:19:25 +00:00
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to force an update on scene exceptions of the show. ' )
2014-05-30 06:22:01 +00:00
2015-02-23 02:19:25 +00:00
if do_update_scene_numbering :
try :
sickbeard . scene_numbering . xem_refresh ( showObj . indexerid , showObj . indexer ) # @UndefinedVariable
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to force an update on scene numbering of the show. ' )
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
if directCall :
return errors
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
if len ( errors ) > 0 :
ui . notifications . error ( ' %d error %s while saving changes: ' % ( len ( errors ) , ' ' if len ( errors ) == 1 else ' s ' ) ,
' <ul> ' + ' \n ' . join ( [ ' <li> %s </li> ' % error for error in errors ] ) + ' </ul> ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + show )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def deleteShow ( self , show = None , full = 0 ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' Invalid show ID ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Unable to find the specified show ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . showQueueScheduler . action . isBeingAdded (
showObj ) or sickbeard . showQueueScheduler . action . isBeingUpdated ( showObj ) : # @UndefinedVariable
return self . _genericMessage ( " Error " , " Shows can ' t be deleted while they ' re being added or updated. " )
2014-10-16 20:45:07 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . USE_TRAKT and sickbeard . TRAKT_SYNC :
# remove show from trakt.tv library
sickbeard . traktCheckerScheduler . action . removeShowFromTraktLibrary ( showObj )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj . deleteShow ( bool ( full ) )
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' %s with %s ' % ( ( ' Deleting ' , ' Trashing ' ) [ sickbeard . TRASH_REMOVE_SHOW ] ,
( ' media left untouched ' , ' all related media ' ) [ bool ( full ) ] ) ,
' <b> %s </b> ' % showObj . name )
self . redirect ( ' /home/ ' )
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
def refreshShow ( self , show = None ) :
2014-07-31 02:32:12 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' Invalid show ID ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Unable to find the specified show ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# force the update from the DB
try :
sickbeard . showQueueScheduler . action . refreshShow ( showObj ) # @UndefinedVariable
2015-06-08 12:47:01 +00:00
except exceptions . CantRefreshException as e :
2015-02-23 02:19:25 +00:00
ui . notifications . error ( ' Unable to refresh this show. ' ,
ex ( e ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + str ( showObj . indexerid ) )
2014-03-10 05:18:05 +00:00
2015-03-17 00:53:12 +00:00
def updateShow ( self , show = None , force = 0 , web = 0 ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' Invalid show ID ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Unable to find the specified show ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# force the update
try :
2015-03-17 00:53:12 +00:00
sickbeard . showQueueScheduler . action . updateShow ( showObj , bool ( force ) , bool ( web ) )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2015-02-23 02:19:25 +00:00
ui . notifications . error ( ' Unable to update this show. ' ,
ex ( e ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# just give it some time
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + str ( showObj . indexerid ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def subtitleShow ( self , show = None , force = 0 ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' Invalid show ID ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-08-29 17:05:15 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Unable to find the specified show ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# search and download subtitles
sickbeard . showQueueScheduler . action . downloadSubtitles ( showObj , bool ( force ) ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
time . sleep ( cpu_presets [ sickbeard . CPU_PRESET ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + str ( showObj . indexerid ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def updateXBMC ( self , showName = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# only send update to first host in the list -- workaround for xbmc sql backend users
if sickbeard . XBMC_UPDATE_ONLYFIRST :
# only send update to first host in the list -- workaround for xbmc sql backend users
host = sickbeard . XBMC_HOST . split ( ' , ' ) [ 0 ] . strip ( )
else :
host = sickbeard . XBMC_HOST
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if notifiers . xbmc_notifier . update_library ( showName = showName ) :
ui . notifications . message ( ' Library update command sent to XBMC host(s): ' + host )
else :
ui . notifications . error ( ' Unable to contact one or more XBMC host(s): ' + host )
self . redirect ( ' /home/ ' )
2014-03-10 05:18:05 +00:00
2015-02-25 12:33:40 +00:00
def updateKODI ( self , showName = None ) :
# only send update to first host in the list -- workaround for kodi sql backend users
if sickbeard . KODI_UPDATE_ONLYFIRST :
# only send update to first host in the list -- workaround for kodi sql backend users
host = sickbeard . KODI_HOST . split ( ' , ' ) [ 0 ] . strip ( )
else :
host = sickbeard . KODI_HOST
if notifiers . kodi_notifier . update_library ( showName = showName ) :
ui . notifications . message ( ' Library update command sent to Kodi host(s): ' + host )
else :
ui . notifications . error ( ' Unable to contact one or more Kodi host(s): ' + host )
2015-02-25 19:31:58 +00:00
self . redirect ( ' /home/ ' )
2015-02-25 12:33:40 +00:00
2015-02-23 02:19:25 +00:00
def updatePLEX ( self , * args , * * kwargs ) :
result = notifiers . plex_notifier . update_library ( )
if None is result :
ui . notifications . message (
' Library update command sent to ' , ' Plex Media Server host(s): ' + sickbeard . PLEX_SERVER_HOST . replace ( ' , ' , ' , ' ) )
else :
ui . notifications . error ( ' Unable to contact ' , ' Plex Media Server host(s): ' + result . replace ( ' , ' , ' , ' ) )
self . redirect ( ' /home/ ' )
2014-08-29 17:05:15 +00:00
2015-02-23 02:19:25 +00:00
def setStatus ( self , show = None , eps = None , status = None , direct = False ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None or eps is None or status is None :
errMsg = ' You must specify a show and at least one episode '
if direct :
ui . notifications . error ( ' Error ' , errMsg )
return json . dumps ( { ' result ' : ' error ' } )
else :
return self . _genericMessage ( ' Error ' , errMsg )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not statusStrings . has_key ( int ( status ) ) :
errMsg = ' Invalid status '
if direct :
ui . notifications . error ( ' Error ' , errMsg )
return json . dumps ( { ' result ' : ' error ' } )
else :
return self . _genericMessage ( ' Error ' , errMsg )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
errMsg = ' Error ' , ' Show not in show list '
if direct :
ui . notifications . error ( ' Error ' , errMsg )
return json . dumps ( { ' result ' : ' error ' } )
else :
return self . _genericMessage ( ' Error ' , errMsg )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
segments = { }
if eps is not None :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sql_l = [ ]
for curEp in eps . split ( ' | ' ) :
2014-03-10 05:18:05 +00:00
2015-03-18 22:22:54 +00:00
logger . log ( u ' Attempting to set status on episode %s to %s ' % ( curEp , status ) , logger . DEBUG )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
epInfo = curEp . split ( ' x ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
epObj = showObj . getEpisode ( int ( epInfo [ 0 ] ) , int ( epInfo [ 1 ] ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if epObj is None :
return self . _genericMessage ( " Error " , " Episode couldn ' t be retrieved " )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if int ( status ) in [ WANTED , FAILED ] :
# figure out what episodes are wanted so we can backlog them
if epObj . season in segments :
segments [ epObj . season ] . append ( epObj )
else :
segments [ epObj . season ] = [ epObj ]
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
with epObj . lock :
# don't let them mess up UNAIRED episodes
if epObj . status == UNAIRED :
logger . log ( u ' Refusing to change status of ' + curEp + ' because it is UNAIRED ' , logger . ERROR )
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if int (
status ) in Quality . DOWNLOADED and epObj . status not in Quality . SNATCHED + Quality . SNATCHED_PROPER + Quality . DOWNLOADED + [
IGNORED ] and not ek . ek ( os . path . isfile , epObj . location ) :
logger . log (
u ' Refusing to change status of ' + curEp + " to DOWNLOADED because it ' s not SNATCHED/DOWNLOADED " ,
logger . ERROR )
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if int (
status ) == FAILED and epObj . status not in Quality . SNATCHED + Quality . SNATCHED_PROPER + Quality . DOWNLOADED :
logger . log (
u ' Refusing to change status of ' + curEp + " to FAILED because it ' s not SNATCHED/DOWNLOADED " ,
logger . ERROR )
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
epObj . status = int ( status )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# mass add to database
result = epObj . get_sql ( )
if None is not result :
sql_l . append ( result )
2014-03-10 05:18:05 +00:00
2015-03-29 11:20:29 +00:00
if 0 < len ( sql_l ) :
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
myDB . mass_action ( sql_l )
2014-06-11 08:34:28 +00:00
2015-03-10 00:26:46 +00:00
if WANTED == int ( status ) :
season_list = ' '
season_wanted = [ ]
2015-02-23 02:19:25 +00:00
for season , segment in segments . items ( ) :
2015-03-10 00:26:46 +00:00
if not showObj . paused :
cur_backlog_queue_item = search_queue . BacklogQueueItem ( showObj , segment )
sickbeard . searchQueueScheduler . action . add_item ( cur_backlog_queue_item ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-03-10 00:26:46 +00:00
if season not in season_wanted :
season_wanted + = [ season ]
season_list + = u ' <li>Season %s </li> ' % season
logger . log ( ( u ' Not adding wanted eps to backlog search for %s season %s because show is paused ' ,
u ' Starting backlog search for %s season %s because eps were set to wanted ' ) [
not showObj . paused ] % ( showObj . name , season ) )
2014-03-10 05:18:05 +00:00
2015-03-10 00:26:46 +00:00
( title , msg ) = ( ( ' Not starting backlog ' , u ' Paused show prevented backlog search ' ) ,
( ' Backlog started ' , u ' Backlog search started ' ) ) [ not showObj . paused ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if segments :
2015-03-10 00:26:46 +00:00
ui . notifications . message ( title ,
u ' %s for the following seasons of <b> %s </b>:<br /><ul> %s </ul> '
% ( msg , showObj . name , season_list ) )
2014-03-10 05:18:05 +00:00
2015-03-10 00:26:46 +00:00
elif FAILED == int ( status ) :
2015-02-23 02:19:25 +00:00
msg = ' Retrying Search was automatically started for the following season of <b> ' + showObj . name + ' </b>:<br /> '
msg + = ' <ul> '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for season , segment in segments . items ( ) :
2015-04-05 18:12:15 +00:00
cur_failed_queue_item = search_queue . FailedQueueItem ( showObj , segment )
2015-02-23 02:19:25 +00:00
sickbeard . searchQueueScheduler . action . add_item ( cur_failed_queue_item ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
msg + = ' <li>Season ' + str ( season ) + ' </li> '
logger . log ( u ' Retrying Search for ' + showObj . name + ' season ' + str (
season ) + ' because some eps were set to failed ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
msg + = ' </ul> '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if segments :
ui . notifications . message ( ' Retry Search started ' , msg )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if direct :
return json . dumps ( { ' result ' : ' success ' } )
2014-06-21 22:46:59 +00:00
else :
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + show )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
def testRename ( self , show = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None :
return self . _genericMessage ( ' Error ' , ' You must specify a show ' )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
return self . _genericMessage ( ' Error ' , ' Show not in show list ' )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
try :
show_loc = showObj . location # @UnusedVariable
except exceptions . ShowDirNotFoundException :
return self . _genericMessage ( ' Error ' , " Can ' t rename episodes when the show dir is missing. " )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
ep_obj_rename_list = [ ]
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
ep_obj_list = showObj . getAllEpisodes ( has_location = True )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for cur_ep_obj in ep_obj_list :
# Only want to rename if we have a location
if cur_ep_obj . location :
if cur_ep_obj . relatedEps :
# do we have one of multi-episodes in the rename list already
have_already = False
for cur_related_ep in cur_ep_obj . relatedEps + [ cur_ep_obj ] :
if cur_related_ep in ep_obj_rename_list :
have_already = True
break
if not have_already :
ep_obj_rename_list . append ( cur_ep_obj )
else :
ep_obj_rename_list . append ( cur_ep_obj )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if ep_obj_rename_list :
# present season DESC episode DESC on screen
ep_obj_rename_list . reverse ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' testRename.tmpl ' )
t . submenu = [ { ' title ' : ' Edit ' , ' path ' : ' home/editShow?show= %d ' % showObj . indexerid } ]
t . ep_obj_list = ep_obj_rename_list
t . show = showObj
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def doRename ( self , show = None , eps = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show is None or eps is None :
errMsg = ' You must specify a show and at least one episode '
return self . _genericMessage ( ' Error ' , errMsg )
2014-06-07 21:32:38 +00:00
2015-02-23 02:19:25 +00:00
show_obj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( show ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if show_obj is None :
errMsg = ' Error ' , ' Show not in show list '
return self . _genericMessage ( ' Error ' , errMsg )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
try :
show_loc = show_obj . location # @UnusedVariable
except exceptions . ShowDirNotFoundException :
return self . _genericMessage ( ' Error ' , " Can ' t rename episodes when the show dir is missing. " )
if eps is None :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/displayShow?show= ' + show )
2014-03-10 05:18:05 +00:00
2014-06-21 22:46:59 +00:00
myDB = db . DBConnection ( )
2015-02-23 02:19:25 +00:00
for curEp in eps . split ( ' | ' ) :
2014-06-07 21:32:38 +00:00
2015-02-23 02:19:25 +00:00
epInfo = curEp . split ( ' x ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# this is probably the worst possible way to deal with double eps but I've kinda painted myself into a corner here with this stupid database
ep_result = myDB . select (
' SELECT * FROM tv_episodes WHERE showid = ? AND season = ? AND episode = ? AND 5=5 ' ,
[ show , epInfo [ 0 ] , epInfo [ 1 ] ] )
if not ep_result :
logger . log ( u ' Unable to find an episode for ' + curEp + ' , skipping ' , logger . WARNING )
continue
related_eps_result = myDB . select ( ' SELECT * FROM tv_episodes WHERE location = ? AND episode != ? ' ,
[ ep_result [ 0 ] [ ' location ' ] , epInfo [ 1 ] ] )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
root_ep_obj = show_obj . getEpisode ( int ( epInfo [ 0 ] ) , int ( epInfo [ 1 ] ) )
root_ep_obj . relatedEps = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for cur_related_ep in related_eps_result :
related_ep_obj = show_obj . getEpisode ( int ( cur_related_ep [ ' season ' ] ) , int ( cur_related_ep [ ' episode ' ] ) )
if related_ep_obj not in root_ep_obj . relatedEps :
root_ep_obj . relatedEps . append ( related_ep_obj )
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
root_ep_obj . rename ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/displayShow?show= ' + show )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def searchEpisode ( self , show = None , season = None , episode = None ) :
2014-10-24 15:31:11 +00:00
2015-02-23 02:19:25 +00:00
# retrieve the episode object and fail if we can't get one
ep_obj = self . _getEpisode ( show , season , episode )
if isinstance ( ep_obj , str ) :
return json . dumps ( { ' result ' : ' failure ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# make a queue item for it and put it on the queue
ep_queue_item = search_queue . ManualSearchQueueItem ( ep_obj . show , ep_obj )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . searchQueueScheduler . action . add_item ( ep_queue_item ) # @UndefinedVariable
if ep_queue_item . success :
return returnManualSearchResult ( ep_queue_item )
if not ep_queue_item . started and ep_queue_item . success is None :
return json . dumps ( { ' result ' : ' success ' } ) #I Actually want to call it queued, because the search hasnt been started yet!
if ep_queue_item . started and ep_queue_item . success is None :
return json . dumps ( { ' result ' : ' success ' } )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' result ' : ' failure ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
### Returns the current ep_queue_item status for the current viewed show.
# Possible status: Downloaded, Snatched, etc...
# Returns {'show': 279530, 'episodes' : ['episode' : 6, 'season' : 1, 'searchstatus' : 'queued', 'status' : 'running', 'quality': '4013']
def getManualSearchStatus ( self , show = None , season = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
episodes = [ ]
currentManualSearchThreadsQueued = [ ]
currentManualSearchThreadActive = [ ]
finishedManualSearchThreadItems = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Queued Searches
currentManualSearchThreadsQueued = sickbeard . searchQueueScheduler . action . get_all_ep_from_queue ( show )
# Running Searches
if ( sickbeard . searchQueueScheduler . action . is_manualsearch_in_progress ( ) ) :
currentManualSearchThreadActive = sickbeard . searchQueueScheduler . action . currentItem
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Finished Searches
finishedManualSearchThreadItems = sickbeard . search_queue . MANUAL_SEARCH_HISTORY
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if currentManualSearchThreadsQueued :
for searchThread in currentManualSearchThreadsQueued :
searchstatus = ' queued '
if isinstance ( searchThread , sickbeard . search_queue . ManualSearchQueueItem ) :
episodes . append ( { ' episode ' : searchThread . segment . episode ,
' episodeindexid ' : searchThread . segment . indexerid ,
' season ' : searchThread . segment . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ searchThread . segment . status ] ,
' quality ' : self . getQualityClass ( searchThread . segment ) } )
2015-06-03 11:33:26 +00:00
elif hasattr ( searchThread , ' segment ' ) :
2015-02-23 02:19:25 +00:00
for epObj in searchThread . segment :
episodes . append ( { ' episode ' : epObj . episode ,
' episodeindexid ' : epObj . indexerid ,
' season ' : epObj . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ epObj . status ] ,
' quality ' : self . getQualityClass ( epObj ) } )
2014-05-13 09:58:27 +00:00
2015-02-23 02:19:25 +00:00
if currentManualSearchThreadActive :
searchThread = currentManualSearchThreadActive
searchstatus = ' searching '
if searchThread . success :
searchstatus = ' finished '
else :
searchstatus = ' searching '
2015-04-05 18:12:15 +00:00
if isinstance ( searchThread , sickbeard . search_queue . ManualSearchQueueItem ) :
episodes . append ( { ' episode ' : searchThread . segment . episode ,
' episodeindexid ' : searchThread . segment . indexerid ,
' season ' : searchThread . segment . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ searchThread . segment . status ] ,
' quality ' : self . getQualityClass ( searchThread . segment ) } )
2015-06-03 11:33:26 +00:00
elif hasattr ( searchThread , ' segment ' ) :
2015-04-05 18:12:15 +00:00
for epObj in searchThread . segment :
episodes . append ( { ' episode ' : epObj . episode ,
' episodeindexid ' : epObj . indexerid ,
' season ' : epObj . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ epObj . status ] ,
' quality ' : self . getQualityClass ( epObj ) } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if finishedManualSearchThreadItems :
for searchThread in finishedManualSearchThreadItems :
if isinstance ( searchThread , sickbeard . search_queue . ManualSearchQueueItem ) :
if str ( searchThread . show . indexerid ) == show and not [ x for x in episodes if x [ ' episodeindexid ' ] == searchThread . segment . indexerid ] :
searchstatus = ' finished '
episodes . append ( { ' episode ' : searchThread . segment . episode ,
' episodeindexid ' : searchThread . segment . indexerid ,
' season ' : searchThread . segment . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ searchThread . segment . status ] ,
' quality ' : self . getQualityClass ( searchThread . segment ) } )
2015-06-03 11:33:26 +00:00
### These are only Failed Downloads/Retry SearchThreadItems.. lets loop through the segement/episodes
elif hasattr ( searchThread , ' segment ' ) and str ( searchThread . show . indexerid ) == show :
for epObj in searchThread . segment :
if not [ x for x in episodes if x [ ' episodeindexid ' ] == epObj . indexerid ] :
searchstatus = ' finished '
episodes . append ( { ' episode ' : epObj . episode ,
' episodeindexid ' : epObj . indexerid ,
' season ' : epObj . season ,
' searchstatus ' : searchstatus ,
' status ' : statusStrings [ epObj . status ] ,
' quality ' : self . getQualityClass ( epObj ) } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' show ' : show , ' episodes ' : episodes } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
#return json.dumps()
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def getQualityClass ( self , ep_obj ) :
# return the correct json value
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Find the quality class for the episode
quality_class = Quality . qualityStrings [ Quality . UNKNOWN ]
ep_status , ep_quality = Quality . splitCompositeStatus ( ep_obj . status )
for x in ( SD , HD720p , HD1080p ) :
if ep_quality in Quality . splitQuality ( x ) [ 0 ] :
quality_class = qualityPresetStrings [ x ]
break
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return quality_class
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def searchEpisodeSubtitles ( self , show = None , season = None , episode = None ) :
# retrieve the episode object and fail if we can't get one
ep_obj = self . _getEpisode ( show , season , episode )
if isinstance ( ep_obj , str ) :
return json . dumps ( { ' result ' : ' failure ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# try do download subtitles for that episode
previous_subtitles = set ( subliminal . language . Language ( x ) for x in ep_obj . subtitles )
try :
ep_obj . subtitles = set ( x . language for x in ep_obj . downloadSubtitles ( ) . values ( ) [ 0 ] )
except :
return json . dumps ( { ' result ' : ' failure ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# return the correct json value
if previous_subtitles != ep_obj . subtitles :
status = ' New subtitles downloaded: %s ' % ' ' . join ( [
" <img src= ' " + sickbeard . WEB_ROOT + " /images/flags/ " + x . alpha2 +
" .png ' alt= ' " + x . name + " ' /> " for x in
sorted ( list ( ep_obj . subtitles . difference ( previous_subtitles ) ) ) ] )
else :
status = ' No subtitles downloaded '
ui . notifications . message ( ' Subtitles Search ' , status )
return json . dumps ( { ' result ' : status , ' subtitles ' : ' , ' . join ( sorted ( [ x . alpha2 for x in
ep_obj . subtitles . union ( previous_subtitles ) ] ) ) } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def setSceneNumbering ( self , show , indexer , forSeason = None , forEpisode = None , forAbsolute = None , sceneSeason = None ,
sceneEpisode = None , sceneAbsolute = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# sanitize:
2015-05-08 02:46:54 +00:00
show = None if show in [ None , ' null ' , ' ' ] else int ( show )
indexer = None if indexer in [ None , ' null ' , ' ' ] else int ( indexer )
show_obj = sickbeard . helpers . findCertainShow ( sickbeard . showList , show )
if not show_obj . is_anime :
for_season = None if forSeason in [ None , ' null ' , ' ' ] else int ( forSeason )
for_episode = None if forEpisode in [ None , ' null ' , ' ' ] else int ( forEpisode )
scene_season = None if sceneSeason in [ None , ' null ' , ' ' ] else int ( sceneSeason )
scene_episode = None if sceneEpisode in [ None , ' null ' , ' ' ] else int ( sceneEpisode )
action_log = u ' Set episode scene numbering to %s x %s for episode %s x %s of " %s " ' \
% ( scene_season , scene_episode , for_season , for_episode , show_obj . name )
ep_args = { ' show ' : show , ' season ' : for_season , ' episode ' : for_episode }
scene_args = { ' indexer_id ' : show , ' indexer ' : indexer , ' season ' : for_season , ' episode ' : for_episode ,
' sceneSeason ' : scene_season , ' sceneEpisode ' : scene_episode }
result = { ' forSeason ' : for_season , ' forEpisode ' : for_episode , ' sceneSeason ' : None , ' sceneEpisode ' : None }
2015-02-23 02:19:25 +00:00
else :
2015-05-08 02:46:54 +00:00
for_absolute = None if forAbsolute in [ None , ' null ' , ' ' ] else int ( forAbsolute )
scene_absolute = None if sceneAbsolute in [ None , ' null ' , ' ' ] else int ( sceneAbsolute )
action_log = u ' Set absolute scene numbering to %s for episode %s of " %s " ' \
% ( scene_absolute , for_absolute , show_obj . name )
ep_args = { ' show ' : show , ' absolute ' : for_absolute }
scene_args = { ' indexer_id ' : show , ' indexer ' : indexer , ' absolute_number ' : for_absolute ,
' sceneAbsolute ' : scene_absolute }
result = { ' forAbsolute ' : for_absolute , ' sceneAbsolute ' : None }
ep_obj = self . _getEpisode ( * * ep_args )
result [ ' success ' ] = not isinstance ( ep_obj , str )
if result [ ' success ' ] :
logger . log ( action_log , logger . DEBUG )
set_scene_numbering ( * * scene_args )
show_obj . flushEpisodes ( )
2015-02-23 02:19:25 +00:00
else :
result [ ' errorMessage ' ] = ep_obj
2014-03-10 05:18:05 +00:00
2015-05-08 02:46:54 +00:00
if not show_obj . is_anime :
scene_numbering = get_scene_numbering ( show , indexer , for_season , for_episode )
if scene_numbering :
( result [ ' sceneSeason ' ] , result [ ' sceneEpisode ' ] ) = scene_numbering
2015-02-23 02:19:25 +00:00
else :
2015-05-08 02:46:54 +00:00
scene_numbering = get_scene_absolute_numbering ( show , indexer , for_absolute )
if scene_numbering :
result [ ' sceneAbsolute ' ] = scene_numbering
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( result )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def retryEpisode ( self , show , season , episode ) :
2014-05-04 18:47:09 +00:00
2015-02-23 02:19:25 +00:00
# retrieve the episode object and fail if we can't get one
ep_obj = self . _getEpisode ( show , season , episode )
if isinstance ( ep_obj , str ) :
return json . dumps ( { ' result ' : ' failure ' } )
2014-10-08 06:00:48 +00:00
2015-02-23 02:19:25 +00:00
# make a queue item for it and put it on the queue
ep_queue_item = search_queue . FailedQueueItem ( ep_obj . show , [ ep_obj ] )
sickbeard . searchQueueScheduler . action . add_item ( ep_queue_item ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if ep_queue_item . success :
return returnManualSearchResult ( ep_queue_item )
if not ep_queue_item . started and ep_queue_item . success is None :
return json . dumps ( { ' result ' : ' success ' } ) #I Actually want to call it queued, because the search hasnt been started yet!
if ep_queue_item . started and ep_queue_item . success is None :
return json . dumps ( { ' result ' : ' success ' } )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' result ' : ' failure ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
@staticmethod
def fetch_releasegroups ( show_name ) :
2014-06-18 12:56:26 +00:00
2015-02-23 02:19:25 +00:00
if helpers . set_up_anidb_connection ( ) :
try :
anime = adba . Anime ( sickbeard . ADBA_CONNECTION , name = show_name )
groups = anime . get_groups ( )
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-02-23 02:19:25 +00:00
logger . log ( u ' exception msg: ' + str ( e ) , logger . DEBUG )
return json . dumps ( { ' result ' : ' fail ' , ' resp ' : ' connect ' } )
2014-07-27 14:39:33 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' result ' : ' success ' , ' groups ' : groups } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' result ' : ' fail ' , ' resp ' : ' init ' } )
2014-06-11 08:34:28 +00:00
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class HomePostProcess ( Home ) :
def index ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' home_postprocess.tmpl ' )
t . submenu = self . HomeMenu ( )
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def processEpisode ( self , dir = None , nzbName = None , jobName = None , quiet = None , process_method = None , force = None ,
2015-05-08 02:46:54 +00:00
force_replace = None , failed = ' 0 ' , type = ' auto ' , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not dir :
self . redirect ( ' /home/postprocess/ ' )
else :
2015-05-08 02:46:54 +00:00
result = processTV . processDir ( dir , nzbName , process_method = process_method , type = type ,
cleanup = ' cleanup ' in kwargs and kwargs [ ' cleanup ' ] in [ ' on ' , ' 1 ' ] ,
force = force in [ ' on ' , ' 1 ' ] ,
force_replace = force_replace in [ ' on ' , ' 1 ' ] ,
failed = not ' 0 ' == failed )
result = re . sub ( r ' (?i)<br(?:[ \ s/]+)> ' , ' \n ' , result )
if None is not quiet and 1 == int ( quiet ) :
return u ' %s ' % re . sub ( ' (?i)<a[^>]+>([^<]+)<[/]a> ' , r ' \ 1 ' , result )
2014-03-10 05:18:05 +00:00
2015-05-08 02:46:54 +00:00
return self . _genericMessage ( ' Postprocessing results ' , u ' <pre> %s </pre> ' % result )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class NewHomeAddShows ( Home ) :
def index ( self , * args , * * kwargs ) :
2014-05-12 10:34:18 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' home_addShows.tmpl ' )
t . submenu = self . HomeMenu ( )
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def getIndexerLanguages ( self , * args , * * kwargs ) :
result = sickbeard . indexerApi ( ) . config [ ' valid_languages ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Make sure list is sorted alphabetically but 'en' is in front
if ' en ' in result :
del result [ result . index ( ' en ' ) ]
result . sort ( )
result . insert ( 0 , ' en ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' results ' : result } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def sanitizeFileName ( self , name ) :
return helpers . sanitizeFileName ( name )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def searchIndexersForShowName ( self , search_term , lang = ' en ' , indexer = None ) :
if not lang or lang == ' null ' :
lang = ' en '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
search_term = search_term . encode ( ' utf-8 ' )
2014-07-27 14:39:33 +00:00
2015-02-23 02:19:25 +00:00
results = { }
final_results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Query Indexers for each search term and build the list of results
for indexer in sickbeard . indexerApi ( ) . indexers if not int ( indexer ) else [ int ( indexer ) ] :
lINDEXER_API_PARMS = sickbeard . indexerApi ( indexer ) . api_params . copy ( )
lINDEXER_API_PARMS [ ' language ' ] = lang
lINDEXER_API_PARMS [ ' custom_ui ' ] = classes . AllShowsListUI
t = sickbeard . indexerApi ( indexer ) . indexer ( * * lINDEXER_API_PARMS )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
logger . log ( ' Searching for Show with searchterm: %s on Indexer: %s ' % (
search_term , sickbeard . indexerApi ( indexer ) . name ) , logger . DEBUG )
try :
# add search results
results . setdefault ( indexer , [ ] ) . extend ( t [ search_term ] )
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-02-23 02:19:25 +00:00
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
map ( final_results . extend ,
( [ [ sickbeard . indexerApi ( id ) . name , id , sickbeard . indexerApi ( id ) . config [ ' show_url ' ] , int ( show [ ' id ' ] ) ,
show [ ' seriesname ' ] , show [ ' firstaired ' ] ] for show in shows ] for id , shows in results . items ( ) ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
lang_id = sickbeard . indexerApi ( ) . config [ ' langabbv_to_id ' ] [ lang ]
return json . dumps ( { ' results ' : final_results , ' langid ' : lang_id } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def massAddTable ( self , rootDir = None ) :
t = PageTemplate ( headers = self . request . headers , file = ' home_massAddTable.tmpl ' )
t . submenu = self . HomeMenu ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not rootDir :
return ' No folders selected. '
elif type ( rootDir ) != list :
root_dirs = [ rootDir ]
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
root_dirs = rootDir
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
root_dirs = [ urllib . unquote_plus ( x ) for x in root_dirs ]
if sickbeard . ROOT_DIRS :
default_index = int ( sickbeard . ROOT_DIRS . split ( ' | ' ) [ 0 ] )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
default_index = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( root_dirs ) > default_index :
tmp = root_dirs [ default_index ]
if tmp in root_dirs :
root_dirs . remove ( tmp )
root_dirs = [ tmp ] + root_dirs
2014-05-20 20:51:54 +00:00
2015-02-23 02:19:25 +00:00
dir_list = [ ]
2014-05-01 01:29:13 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
for root_dir in root_dirs :
try :
file_list = ek . ek ( os . listdir , root_dir )
except :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for cur_file in file_list :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
cur_path = ek . ek ( os . path . normpath , ek . ek ( os . path . join , root_dir , cur_file ) )
if not ek . ek ( os . path . isdir , cur_path ) :
continue
2014-09-28 09:20:42 +00:00
2015-02-23 02:19:25 +00:00
cur_dir = {
' dir ' : cur_path ,
' display_dir ' : ' <span class= " filepath " > ' + ek . ek ( os . path . dirname , cur_path ) + os . sep + ' </span> ' + ek . ek (
os . path . basename ,
cur_path ) ,
}
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# see if the folder is in XBMC already
dirResults = myDB . select ( ' SELECT * FROM tv_shows WHERE location = ? ' , [ cur_path ] )
2014-04-28 09:15:29 +00:00
2015-02-23 02:19:25 +00:00
if dirResults :
cur_dir [ ' added_already ' ] = True
else :
cur_dir [ ' added_already ' ] = False
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
dir_list . append ( cur_dir )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer_id = show_name = indexer = None
for cur_provider in sickbeard . metadata_provider_dict . values ( ) :
if indexer_id and show_name :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
( indexer_id , show_name , indexer ) = cur_provider . retrieveShowMetadata ( cur_path )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# default to TVDB if indexer was not detected
if show_name and not ( indexer or indexer_id ) :
( sn , idx , id ) = helpers . searchIndexerForShowID ( show_name , indexer , indexer_id )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# set indexer and indexer_id from found info
if not indexer and idx :
indexer = idx
2014-06-07 08:17:12 +00:00
2015-02-23 02:19:25 +00:00
if not indexer_id and id :
indexer_id = id
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
cur_dir [ ' existing_info ' ] = ( indexer_id , show_name , indexer )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if indexer_id and helpers . findCertainShow ( sickbeard . showList , indexer_id ) :
cur_dir [ ' added_already ' ] = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . dirList = dir_list
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-06-07 08:17:12 +00:00
2015-02-23 02:19:25 +00:00
def newShow ( self , show_to_add = None , other_shows = None , use_show_name = None ) :
"""
Display the new show page which collects a tvdb id , folder , and extra options and
posts them to addNewShow
"""
self . set_header ( ' Cache-Control ' , ' no-cache, no-store, must-revalidate ' )
self . set_header ( ' Pragma ' , ' no-cache ' )
self . set_header ( ' Expires ' , ' 0 ' )
2014-06-07 08:17:12 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' home_newShow.tmpl ' )
t . submenu = self . HomeMenu ( )
2015-03-10 22:20:02 +00:00
t . enable_anime_options = True
2015-03-10 00:13:58 +00:00
t . enable_default_wanted = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer , show_dir , indexer_id , show_name = self . split_extra_show ( show_to_add )
2014-04-28 09:15:29 +00:00
2015-02-23 02:19:25 +00:00
if indexer_id and indexer and show_name :
use_provided_info = True
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
use_provided_info = False
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# tell the template whether we're giving it show name & Indexer ID
t . use_provided_info = use_provided_info
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# use the given show_dir for the indexer search if available
if use_show_name :
t . default_show_name = show_name
elif not show_dir :
t . default_show_name = ' '
elif not show_name :
t . default_show_name = ek . ek ( os . path . basename , ek . ek ( os . path . normpath , show_dir ) ) . replace ( ' . ' , ' ' )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
t . default_show_name = show_name
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# carry a list of other dirs if given
if not other_shows :
other_shows = [ ]
elif type ( other_shows ) != list :
other_shows = [ other_shows ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if use_provided_info :
t . provided_indexer_id = int ( indexer_id or 0 )
t . provided_indexer_name = show_name
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . provided_show_dir = show_dir
t . other_shows = other_shows
t . provided_indexer = int ( indexer or sickbeard . INDEXER_DEFAULT )
t . indexers = sickbeard . indexerApi ( ) . indexers
t . whitelist = [ ]
t . blacklist = [ ]
t . groups = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def recommendedShows ( self , * args , * * kwargs ) :
"""
Display the new show page which collects a tvdb id , folder , and extra options and
posts them to addNewShow
"""
self . set_header ( ' Cache-Control ' , ' no-cache, no-store, must-revalidate ' )
self . set_header ( ' Pragma ' , ' no-cache ' )
self . set_header ( ' Expires ' , ' 0 ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' home_recommendedShows.tmpl ' )
t . submenu = self . HomeMenu ( )
2015-03-10 22:20:02 +00:00
t . enable_anime_options = False
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def getRecommendedShows ( self , * args , * * kwargs ) :
final_results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
logger . log ( u ' Getting recommended shows from Trakt.tv ' , logger . DEBUG )
recommendedlist = TraktCall ( ' recommendations/shows.json/ % API % ' , sickbeard . TRAKT_API , sickbeard . TRAKT_USERNAME ,
sickbeard . TRAKT_PASSWORD )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if recommendedlist == ' NULL ' :
logger . log ( u ' No shows found in your recommendedlist, aborting recommendedlist update ' , logger . DEBUG )
return
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if recommendedlist is None :
logger . log ( u ' Could not connect to trakt service, aborting recommended list update ' , logger . ERROR )
return
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
map ( final_results . append ,
( [ show [ ' url ' ] ,
show [ ' title ' ] ,
show [ ' overview ' ] ,
sbdatetime . sbdatetime . sbfdate ( datetime . date . fromtimestamp ( int ( show [ ' first_aired ' ] ) ) ) ,
sickbeard . indexerApi ( 1 ) . name ,
sickbeard . indexerApi ( 1 ) . config [ ' icon ' ] ,
int ( show [ ' tvdb_id ' ] or 0 ) ,
' %s %s ' % ( sickbeard . indexerApi ( 1 ) . config [ ' show_url ' ] , int ( show [ ' tvdb_id ' ] or 0 ) ) ,
sickbeard . indexerApi ( 2 ) . name ,
sickbeard . indexerApi ( 2 ) . config [ ' icon ' ] ,
int ( show [ ' tvrage_id ' ] or 0 ) ,
' %s %s ' % ( sickbeard . indexerApi ( 2 ) . config [ ' show_url ' ] , int ( show [ ' tvrage_id ' ] or 0 ) )
] for show in recommendedlist if not helpers . findCertainShow ( sickbeard . showList , indexerid = int ( show [ ' tvdb_id ' ] ) ) ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . set_header ( ' Content-Type ' , ' application/json ' )
return json . dumps ( { ' results ' : final_results } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def addRecommendedShow ( self , whichSeries = None , indexerLang = ' en ' , rootDir = None , defaultStatus = None ,
anyQualities = None , bestQualities = None , flatten_folders = None , subtitles = None ,
fullShowPath = None , other_shows = None , skipShow = None , providedIndexer = None , anime = None ,
scene = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer = 1
indexer_name = sickbeard . indexerApi ( int ( indexer ) ) . name
show_url = whichSeries . split ( ' | ' ) [ 1 ]
indexer_id = whichSeries . split ( ' | ' ) [ 0 ]
show_name = whichSeries . split ( ' | ' ) [ 2 ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return self . addNewShow ( ' | ' . join ( [ indexer_name , str ( indexer ) , show_url , indexer_id , show_name , ' ' ] ) ,
indexerLang , rootDir ,
defaultStatus ,
anyQualities , bestQualities , flatten_folders , subtitles , fullShowPath , other_shows ,
skipShow , providedIndexer , anime , scene )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def trendingShows ( self , * args , * * kwargs ) :
"""
Display the new show page which collects a tvdb id , folder , and extra options and
posts them to addNewShow
"""
t = PageTemplate ( headers = self . request . headers , file = ' home_trendingShows.tmpl ' )
t . submenu = self . HomeMenu ( )
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
t . trending_shows = TraktCall ( ' shows/trending.json/ % API % ' , sickbeard . TRAKT_API_KEY )
t . trending_inlibrary = 0
if None is not t . trending_shows :
for item in t . trending_shows :
tvdbs = [ ' tvdb_id ' , ' tvrage_id ' ]
for index , tvdb in enumerate ( tvdbs ) :
try :
item [ u ' show_id ' ] = str ( item [ tvdb ] )
tvshow = helpers . findCertainShow ( sickbeard . showList , int ( item [ tvdb ] ) )
except :
continue
# check tvshow indexer is not using the same id from another indexer
if tvshow and ( index + 1 ) == tvshow . indexer :
item [ u ' show_id ' ] = u ' %s : %s ' % ( tvshow . indexer , item [ tvdb ] )
t . trending_inlibrary + = 1
break
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
def existingShows ( self , * args , * * kwargs ) :
"""
Prints out the page to add existing shows from a root dir
"""
t = PageTemplate ( headers = self . request . headers , file = ' home_addExistingShow.tmpl ' )
t . submenu = self . HomeMenu ( )
2015-03-10 22:20:02 +00:00
t . enable_anime_options = False
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
def addTraktShow ( self , indexer_id , showName ) :
2015-03-10 22:26:32 +00:00
if helpers . findCertainShow ( sickbeard . showList , config . to_int ( indexer_id , ' ' ) ) :
2015-02-23 02:19:25 +00:00
return
return self . newShow ( ' | ' . join ( [ ' ' , ' ' , indexer_id , showName ] ) , use_show_name = True )
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
def addNewShow ( self , whichSeries = None , indexerLang = ' en ' , rootDir = None , defaultStatus = None ,
anyQualities = None , bestQualities = None , flatten_folders = None , subtitles = None ,
fullShowPath = None , other_shows = None , skipShow = None , providedIndexer = None , anime = None ,
2015-04-07 03:10:50 +00:00
scene = None , blacklist = None , whitelist = None , wanted_begin = None , wanted_latest = None , tag = None ) :
2015-02-23 02:19:25 +00:00
"""
Receive tvdb id , dir , and other options and create a show from them . If extra show dirs are
provided then it forwards back to newShow , if not it goes to / home .
"""
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
# grab our list of other dirs if given
if not other_shows :
other_shows = [ ]
elif type ( other_shows ) != list :
other_shows = [ other_shows ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def finishAddShow ( ) :
# if there are no extra shows then go home
if not other_shows :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# peel off the next one
next_show_dir = other_shows [ 0 ]
rest_of_show_dirs = other_shows [ 1 : ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# go to add the next show
return self . newShow ( next_show_dir , rest_of_show_dirs )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if we're skipping then behave accordingly
if skipShow :
return finishAddShow ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# sanity check on our inputs
if ( not rootDir and not fullShowPath ) or not whichSeries :
return ' Missing params, no Indexer ID or folder: ' + repr ( whichSeries ) + ' and ' + repr (
rootDir ) + ' / ' + repr ( fullShowPath )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# figure out what show we're adding and where
series_pieces = whichSeries . split ( ' | ' )
if ( whichSeries and rootDir ) or ( whichSeries and fullShowPath and len ( series_pieces ) > 1 ) :
if len ( series_pieces ) < 6 :
logger . log ( ' Unable to add show due to show selection. Not enough arguments: %s ' % ( repr ( series_pieces ) ) ,
logger . ERROR )
ui . notifications . error ( ' Unknown error. Unable to add show due to problem with show selection. ' )
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/addShows/existingShows/ ' )
2015-02-23 02:19:25 +00:00
indexer = int ( series_pieces [ 1 ] )
indexer_id = int ( series_pieces [ 3 ] )
show_name = series_pieces [ 4 ]
else :
# if no indexer was provided use the default indexer set in General settings
if not providedIndexer :
providedIndexer = sickbeard . INDEXER_DEFAULT
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer = int ( providedIndexer )
indexer_id = int ( whichSeries )
show_name = os . path . basename ( os . path . normpath ( fullShowPath ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# use the whole path if it's given, or else append the show name to the root dir to get the full show path
if fullShowPath :
show_dir = ek . ek ( os . path . normpath , fullShowPath )
else :
show_dir = ek . ek ( os . path . join , rootDir , helpers . sanitizeFileName ( show_name ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# blanket policy - if the dir exists you should have used 'add existing show' numbnuts
if ek . ek ( os . path . isdir , show_dir ) and not fullShowPath :
2015-03-10 00:13:58 +00:00
ui . notifications . error ( ' Unable to add show ' , u ' Found existing folder: ' + show_dir )
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/addShows/existingShows/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# don't create show dir if config says not to
if sickbeard . ADD_SHOWS_WO_DIR :
2015-03-10 00:13:58 +00:00
logger . log ( u ' Skipping initial creation due to config.ini setting (add_shows_wo_dir) ' )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
dir_exists = helpers . makeDir ( show_dir )
if not dir_exists :
2015-03-10 00:13:58 +00:00
logger . log ( u ' Unable to add show because can \' t create folder: ' + show_dir , logger . ERROR )
ui . notifications . error ( ' Unable to add show ' , u ' Can \' t create folder: ' + show_dir )
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
helpers . chmodAsParent ( show_dir )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# prepare the inputs for passing along
scene = config . checkbox_to_value ( scene )
anime = config . checkbox_to_value ( anime )
flatten_folders = config . checkbox_to_value ( flatten_folders )
subtitles = config . checkbox_to_value ( subtitles )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if whitelist :
whitelist = short_group_names ( whitelist )
if blacklist :
blacklist = short_group_names ( blacklist )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not anyQualities :
anyQualities = [ ]
if not bestQualities :
bestQualities = [ ]
if type ( anyQualities ) != list :
anyQualities = [ anyQualities ]
if type ( bestQualities ) != list :
bestQualities = [ bestQualities ]
newQuality = Quality . combineQualities ( map ( int , anyQualities ) , map ( int , bestQualities ) )
2014-03-10 05:18:05 +00:00
2015-03-14 19:51:14 +00:00
wanted_begin = config . minimax ( wanted_begin , 0 , - 1 , 10 )
wanted_latest = config . minimax ( wanted_latest , 0 , - 1 , 10 )
2015-03-10 00:13:58 +00:00
2015-02-23 02:19:25 +00:00
# add the show
sickbeard . showQueueScheduler . action . addShow ( indexer , indexer_id , show_dir , int ( defaultStatus ) , newQuality ,
flatten_folders , indexerLang , subtitles , anime ,
2015-03-10 00:13:58 +00:00
scene , None , blacklist , whitelist ,
2015-04-07 03:10:50 +00:00
wanted_begin , wanted_latest , tag ) # @UndefinedVariable
2015-03-10 00:13:58 +00:00
# ui.notifications.message('Show added', 'Adding the specified show into ' + show_dir)
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return finishAddShow ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def split_extra_show ( self , extra_show ) :
if not extra_show :
return ( None , None , None , None )
split_vals = extra_show . split ( ' | ' )
if len ( split_vals ) < 4 :
indexer = split_vals [ 0 ]
show_dir = split_vals [ 1 ]
return ( indexer , show_dir , None , None )
indexer = split_vals [ 0 ]
show_dir = split_vals [ 1 ]
indexer_id = split_vals [ 2 ]
show_name = ' | ' . join ( split_vals [ 3 : ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return ( indexer , show_dir , indexer_id , show_name )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def addExistingShows ( self , shows_to_add = None , promptForSettings = None ) :
"""
Receives a dir list and add them . Adds the ones with given TVDB IDs first , then forwards
along to the newShow page .
"""
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# grab a list of other shows to add, if provided
if not shows_to_add :
shows_to_add = [ ]
elif type ( shows_to_add ) != list :
shows_to_add = [ shows_to_add ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
shows_to_add = [ urllib . unquote_plus ( x ) for x in shows_to_add ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
promptForSettings = config . checkbox_to_value ( promptForSettings )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer_id_given = [ ]
dirs_only = [ ]
# separate all the ones with Indexer IDs
for cur_dir in shows_to_add :
if ' | ' in cur_dir :
split_vals = cur_dir . split ( ' | ' )
if len ( split_vals ) < 3 :
dirs_only . append ( cur_dir )
if not ' | ' in cur_dir :
dirs_only . append ( cur_dir )
else :
indexer , show_dir , indexer_id , show_name = self . split_extra_show ( cur_dir )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not show_dir or not indexer_id or not show_name :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
indexer_id_given . append ( ( int ( indexer ) , show_dir , int ( indexer_id ) , show_name ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if they want me to prompt for settings then I will just carry on to the newShow page
if promptForSettings and shows_to_add :
return self . newShow ( shows_to_add [ 0 ] , shows_to_add [ 1 : ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if they don't want me to prompt for settings then I can just add all the nfo shows now
num_added = 0
for cur_show in indexer_id_given :
indexer , show_dir , indexer_id , show_name = cur_show
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if indexer is not None and indexer_id is not None :
# add the show
sickbeard . showQueueScheduler . action . addShow ( indexer , indexer_id , show_dir ,
default_status = sickbeard . STATUS_DEFAULT ,
quality = sickbeard . QUALITY_DEFAULT ,
flatten_folders = sickbeard . FLATTEN_FOLDERS_DEFAULT ,
subtitles = sickbeard . SUBTITLES_DEFAULT ,
anime = sickbeard . ANIME_DEFAULT ,
scene = sickbeard . SCENE_DEFAULT )
num_added + = 1
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if num_added :
ui . notifications . message ( ' Shows Added ' ,
' Automatically added ' + str ( num_added ) + ' from their existing metadata files ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if we're done then go home
if not dirs_only :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /home/ ' )
2014-05-23 16:35:01 +00:00
2015-02-23 02:19:25 +00:00
# for the remaining shows we need to prompt for each one, so forward this on to the newShow page
return self . newShow ( dirs_only [ 0 ] , dirs_only [ 1 : ] )
2014-09-01 15:57:52 +00:00
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class Manage ( MainHandler ) :
def ManageMenu ( self ) :
manageMenu = [
{ ' title ' : ' Backlog Overview ' , ' path ' : ' manage/backlogOverview/ ' } ,
{ ' title ' : ' Manage Searches ' , ' path ' : ' manage/manageSearches/ ' } ,
2015-05-04 19:14:29 +00:00
{ ' title ' : ' Show Queue Overview ' , ' path ' : ' manage/showQueueOverview/ ' } ,
2015-02-23 02:19:25 +00:00
{ ' title ' : ' Episode Status Management ' , ' path ' : ' manage/episodeStatuses/ ' } , ]
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . USE_TORRENTS and sickbeard . TORRENT_METHOD != ' blackhole ' \
and ( sickbeard . ENABLE_HTTPS and sickbeard . TORRENT_HOST [ : 5 ] == ' https '
or not sickbeard . ENABLE_HTTPS and sickbeard . TORRENT_HOST [ : 5 ] == ' http: ' ) :
manageMenu . append ( { ' title ' : ' Manage Torrents ' , ' path ' : ' manage/manageTorrents/ ' } )
2014-08-29 10:29:56 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . USE_SUBTITLES :
manageMenu . append ( { ' title ' : ' Missed Subtitle Management ' , ' path ' : ' manage/subtitleMissed/ ' } )
2014-08-29 10:29:56 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . USE_FAILED_DOWNLOADS :
manageMenu . append ( { ' title ' : ' Failed Downloads ' , ' path ' : ' manage/failedDownloads/ ' } )
2014-08-29 10:29:56 +00:00
2015-02-23 02:19:25 +00:00
return manageMenu
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' manage.tmpl ' )
t . submenu = self . ManageMenu ( )
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def showEpisodeStatuses ( self , indexer_id , whichStatus ) :
status_list = [ int ( whichStatus ) ]
if status_list [ 0 ] == SNATCHED :
status_list = Quality . SNATCHED + Quality . SNATCHED_PROPER
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
cur_show_results = myDB . select (
2015-03-18 22:22:54 +00:00
' SELECT season, episode, name, airdate FROM tv_episodes WHERE showid = ? AND season != 0 AND status IN ( ' + ' , ' . join (
2015-02-23 02:19:25 +00:00
[ ' ? ' ] * len ( status_list ) ) + ' ) ' , [ int ( indexer_id ) ] + status_list )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = { }
for cur_result in cur_show_results :
cur_season = int ( cur_result [ ' season ' ] )
cur_episode = int ( cur_result [ ' episode ' ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if cur_season not in result :
result [ cur_season ] = { }
2014-03-10 05:18:05 +00:00
2015-03-18 22:22:54 +00:00
result [ cur_season ] [ cur_episode ] = { ' name ' : cur_result [ ' name ' ] , ' airdate_never ' : ( True , False ) [ 1000 < int ( cur_result [ ' airdate ' ] ) ] }
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( result )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def episodeStatuses ( self , whichStatus = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if whichStatus :
whichStatus = int ( whichStatus )
status_list = [ whichStatus ]
if status_list [ 0 ] == SNATCHED :
status_list = Quality . SNATCHED + Quality . SNATCHED_PROPER
else :
status_list = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_episodeStatuses.tmpl ' )
t . submenu = self . ManageMenu ( )
t . whichStatus = whichStatus
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if we have no status then this is as far as we need to go
if not status_list :
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
status_results = myDB . select (
2015-03-18 22:22:54 +00:00
' SELECT show_name, tv_shows.indexer_id as indexer_id, airdate FROM tv_episodes, tv_shows WHERE tv_episodes.status IN ( ' + ' , ' . join (
2015-02-23 02:19:25 +00:00
[ ' ? ' ] * len (
status_list ) ) + ' ) AND season != 0 AND tv_episodes.showid = tv_shows.indexer_id ORDER BY show_name ' ,
status_list )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
ep_counts = { }
2015-03-18 22:22:54 +00:00
ep_count = 0
never_counts = { }
2015-02-23 02:19:25 +00:00
show_names = { }
sorted_show_ids = [ ]
for cur_status_result in status_results :
cur_indexer_id = int ( cur_status_result [ ' indexer_id ' ] )
if cur_indexer_id not in ep_counts :
ep_counts [ cur_indexer_id ] = 1
else :
ep_counts [ cur_indexer_id ] + = 1
2015-03-18 22:22:54 +00:00
ep_count + = 1
if cur_indexer_id not in never_counts :
never_counts [ cur_indexer_id ] = 0
if 1000 > int ( cur_status_result [ ' airdate ' ] ) :
never_counts [ cur_indexer_id ] + = 1
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
show_names [ cur_indexer_id ] = cur_status_result [ ' show_name ' ]
if cur_indexer_id not in sorted_show_ids :
sorted_show_ids . append ( cur_indexer_id )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . show_names = show_names
t . ep_counts = ep_counts
2015-03-18 22:22:54 +00:00
t . ep_count = ep_count
t . never_counts = never_counts
2015-02-23 02:19:25 +00:00
t . sorted_show_ids = sorted_show_ids
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-03-18 22:22:54 +00:00
def changeEpisodeStatuses ( self , oldStatus , newStatus , wantedStatus = sickbeard . common . UNKNOWN , * args , * * kwargs ) :
2015-02-23 02:19:25 +00:00
status_list = [ int ( oldStatus ) ]
if status_list [ 0 ] == SNATCHED :
status_list = Quality . SNATCHED + Quality . SNATCHED_PROPER
2014-05-20 16:06:11 +00:00
2015-02-23 02:19:25 +00:00
to_change = { }
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# make a list of all shows and their associated args
for arg in kwargs :
# we don't care about unchecked checkboxes
if kwargs [ arg ] != ' on ' :
continue
2014-03-10 05:18:05 +00:00
2015-03-18 22:22:54 +00:00
indexer_id , what = arg . split ( ' - ' )
2015-02-23 02:19:25 +00:00
if indexer_id not in to_change :
to_change [ indexer_id ] = [ ]
2014-05-13 09:58:27 +00:00
2015-02-23 02:19:25 +00:00
to_change [ indexer_id ] . append ( what )
2014-03-10 05:18:05 +00:00
2015-03-18 22:22:54 +00:00
if sickbeard . common . WANTED == int ( wantedStatus ) :
newStatus = sickbeard . common . WANTED
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
for cur_indexer_id in to_change :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# get a list of all the eps we want to change if they just said 'all'
if ' all ' in to_change [ cur_indexer_id ] :
all_eps_results = myDB . select (
' SELECT season, episode FROM tv_episodes WHERE status IN ( ' + ' , ' . join (
[ ' ? ' ] * len ( status_list ) ) + ' ) AND season != 0 AND showid = ? ' ,
status_list + [ cur_indexer_id ] )
all_eps = [ str ( x [ ' season ' ] ) + ' x ' + str ( x [ ' episode ' ] ) for x in all_eps_results ]
to_change [ cur_indexer_id ] = all_eps
2014-05-19 20:06:32 +00:00
2015-02-23 02:19:25 +00:00
Home ( self . application , self . request ) . setStatus ( cur_indexer_id , ' | ' . join ( to_change [ cur_indexer_id ] ) ,
newStatus , direct = True )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/episodeStatuses/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def showSubtitleMissed ( self , indexer_id , whichSubs ) :
myDB = db . DBConnection ( )
cur_show_results = myDB . select (
" SELECT season, episode, name, subtitles FROM tv_episodes WHERE showid = ? AND season != 0 AND status LIKE ' % 4 ' " ,
[ int ( indexer_id ) ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = { }
for cur_result in cur_show_results :
if whichSubs == ' all ' :
if len ( set ( cur_result [ ' subtitles ' ] . split ( ' , ' ) ) . intersection ( set ( subtitles . wantedLanguages ( ) ) ) ) > = len (
subtitles . wantedLanguages ( ) ) :
continue
elif whichSubs in cur_result [ ' subtitles ' ] . split ( ' , ' ) :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
cur_season = int ( cur_result [ ' season ' ] )
cur_episode = int ( cur_result [ ' episode ' ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if cur_season not in result :
result [ cur_season ] = { }
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if cur_episode not in result [ cur_season ] :
result [ cur_season ] [ cur_episode ] = { }
2014-08-29 05:16:25 +00:00
2015-02-23 02:19:25 +00:00
result [ cur_season ] [ cur_episode ] [ ' name ' ] = cur_result [ ' name ' ]
2014-05-18 16:39:30 +00:00
2015-02-23 02:19:25 +00:00
result [ cur_season ] [ cur_episode ] [ ' subtitles ' ] = ' , ' . join (
subliminal . language . Language ( subtitle ) . alpha2 for subtitle in cur_result [ ' subtitles ' ] . split ( ' , ' ) ) if not \
cur_result [ ' subtitles ' ] == ' ' else ' '
2014-05-20 02:14:06 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( result )
2014-05-20 02:14:06 +00:00
2015-02-23 02:19:25 +00:00
def subtitleMissed ( self , whichSubs = None ) :
2014-04-26 13:35:37 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_subtitleMissed.tmpl ' )
t . submenu = self . ManageMenu ( )
t . whichSubs = whichSubs
2014-05-20 20:51:54 +00:00
2015-02-23 02:19:25 +00:00
if not whichSubs :
return t . respond ( )
2014-05-20 20:51:54 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
status_results = myDB . select (
" SELECT show_name, tv_shows.indexer_id as indexer_id, tv_episodes.subtitles subtitles FROM tv_episodes, tv_shows WHERE tv_shows.subtitles = 1 AND tv_episodes.status LIKE ' % 4 ' AND tv_episodes.season != 0 AND tv_episodes.showid = tv_shows.indexer_id ORDER BY show_name " )
2014-08-29 05:16:25 +00:00
2015-02-23 02:19:25 +00:00
ep_counts = { }
show_names = { }
sorted_show_ids = [ ]
for cur_status_result in status_results :
if whichSubs == ' all ' :
if len ( set ( cur_status_result [ ' subtitles ' ] . split ( ' , ' ) ) . intersection (
set ( subtitles . wantedLanguages ( ) ) ) ) > = len ( subtitles . wantedLanguages ( ) ) :
continue
elif whichSubs in cur_status_result [ ' subtitles ' ] . split ( ' , ' ) :
continue
2014-05-20 02:29:51 +00:00
2015-02-23 02:19:25 +00:00
cur_indexer_id = int ( cur_status_result [ ' indexer_id ' ] )
if cur_indexer_id not in ep_counts :
ep_counts [ cur_indexer_id ] = 1
else :
ep_counts [ cur_indexer_id ] + = 1
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
show_names [ cur_indexer_id ] = cur_status_result [ ' show_name ' ]
if cur_indexer_id not in sorted_show_ids :
sorted_show_ids . append ( cur_indexer_id )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . show_names = show_names
t . ep_counts = ep_counts
t . sorted_show_ids = sorted_show_ids
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def downloadSubtitleMissed ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
to_download = { }
2014-07-27 14:39:33 +00:00
2015-02-23 02:19:25 +00:00
# make a list of all shows and their associated args
for arg in kwargs :
indexer_id , what = arg . split ( ' - ' )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
# we don't care about unchecked checkboxes
if kwargs [ arg ] != ' on ' :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if indexer_id not in to_download :
to_download [ indexer_id ] = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
to_download [ indexer_id ] . append ( what )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for cur_indexer_id in to_download :
# get a list of all the eps we want to download subtitles if they just said 'all'
if ' all ' in to_download [ cur_indexer_id ] :
myDB = db . DBConnection ( )
all_eps_results = myDB . select (
" SELECT season, episode FROM tv_episodes WHERE status LIKE ' % 4 ' AND season != 0 AND showid = ? " ,
[ cur_indexer_id ] )
to_download [ cur_indexer_id ] = [ str ( x [ ' season ' ] ) + ' x ' + str ( x [ ' episode ' ] ) for x in all_eps_results ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for epResult in to_download [ cur_indexer_id ] :
season , episode = epResult . split ( ' x ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
show = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( cur_indexer_id ) )
subtitles = show . getEpisode ( int ( season ) , int ( episode ) ) . downloadSubtitles ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/subtitleMissed/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def backlogShow ( self , indexer_id ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
show_obj = helpers . findCertainShow ( sickbeard . showList , int ( indexer_id ) )
2014-05-07 14:23:06 +00:00
2015-02-23 02:19:25 +00:00
if show_obj :
sickbeard . backlogSearchScheduler . action . searchBacklog ( [ show_obj ] ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/backlogOverview/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def backlogOverview ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_backlogOverview.tmpl ' )
t . submenu = self . ManageMenu ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showCounts = { }
showCats = { }
showSQLResults = { }
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
for curShow in sickbeard . showList :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
epCounts = { }
epCats = { }
epCounts [ Overview . SKIPPED ] = 0
epCounts [ Overview . WANTED ] = 0
epCounts [ Overview . QUAL ] = 0
epCounts [ Overview . GOOD ] = 0
epCounts [ Overview . UNAIRED ] = 0
epCounts [ Overview . SNATCHED ] = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sqlResults = myDB . select (
' SELECT * FROM tv_episodes WHERE showid = ? ORDER BY season DESC, episode DESC ' ,
[ curShow . indexerid ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for curResult in sqlResults :
curEpCat = curShow . getOverview ( int ( curResult [ ' status ' ] ) )
if curEpCat :
epCats [ str ( curResult [ ' season ' ] ) + ' x ' + str ( curResult [ ' episode ' ] ) ] = curEpCat
epCounts [ curEpCat ] + = 1
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showCounts [ curShow . indexerid ] = epCounts
showCats [ curShow . indexerid ] = epCats
showSQLResults [ curShow . indexerid ] = sqlResults
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . showCounts = showCounts
t . showCats = showCats
t . showSQLResults = showSQLResults
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def massEdit ( self , toEdit = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_massEdit.tmpl ' )
t . submenu = self . ManageMenu ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not toEdit :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /manage/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showIDs = toEdit . split ( ' | ' )
showList = [ ]
for curID in showIDs :
curID = int ( curID )
showObj = helpers . findCertainShow ( sickbeard . showList , curID )
if showObj :
showList . append ( showObj )
2014-07-27 14:39:33 +00:00
2015-02-23 02:19:25 +00:00
archive_firstmatch_all_same = True
last_archive_firstmatch = None
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
flatten_folders_all_same = True
last_flatten_folders = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
paused_all_same = True
last_paused = None
2014-03-10 05:18:05 +00:00
2015-04-07 03:10:50 +00:00
tag_all_same = True
last_tag = None
2015-02-23 02:19:25 +00:00
anime_all_same = True
last_anime = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sports_all_same = True
last_sports = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
quality_all_same = True
last_quality = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
subtitles_all_same = True
last_subtitles = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
scene_all_same = True
last_scene = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
air_by_date_all_same = True
last_air_by_date = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
root_dir_list = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for curShow in showList :
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
cur_root_dir = ek . ek ( os . path . dirname , curShow . _location )
if cur_root_dir not in root_dir_list :
root_dir_list . append ( cur_root_dir )
2014-07-27 14:39:33 +00:00
2015-02-23 02:19:25 +00:00
if archive_firstmatch_all_same :
# if we had a value already and this value is different then they're not all the same
if last_archive_firstmatch not in ( None , curShow . archive_firstmatch ) :
archive_firstmatch_all_same = False
else :
last_archive_firstmatch = curShow . archive_firstmatch
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
# if we know they're not all the same then no point even bothering
if paused_all_same :
# if we had a value already and this value is different then they're not all the same
if last_paused not in ( None , curShow . paused ) :
paused_all_same = False
else :
last_paused = curShow . paused
2014-06-11 08:34:28 +00:00
2015-04-07 03:10:50 +00:00
if tag_all_same :
# if we had a value already and this value is different then they're not all the same
if last_tag not in ( None , curShow . tag ) :
tag_all_same = False
else :
last_tag = curShow . tag
2015-02-23 02:19:25 +00:00
if anime_all_same :
# if we had a value already and this value is different then they're not all the same
if last_anime not in ( None , curShow . is_anime ) :
anime_all_same = False
else :
last_anime = curShow . anime
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if flatten_folders_all_same :
if last_flatten_folders not in ( None , curShow . flatten_folders ) :
flatten_folders_all_same = False
else :
last_flatten_folders = curShow . flatten_folders
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if quality_all_same :
if last_quality not in ( None , curShow . quality ) :
quality_all_same = False
else :
last_quality = curShow . quality
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if subtitles_all_same :
if last_subtitles not in ( None , curShow . subtitles ) :
subtitles_all_same = False
else :
last_subtitles = curShow . subtitles
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if scene_all_same :
if last_scene not in ( None , curShow . scene ) :
scene_all_same = False
else :
last_scene = curShow . scene
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
if sports_all_same :
if last_sports not in ( None , curShow . sports ) :
sports_all_same = False
else :
last_sports = curShow . sports
2014-06-07 12:36:50 +00:00
2015-02-23 02:19:25 +00:00
if air_by_date_all_same :
if last_air_by_date not in ( None , curShow . air_by_date ) :
air_by_date_all_same = False
else :
last_air_by_date = curShow . air_by_date
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t . showList = toEdit
t . archive_firstmatch_value = last_archive_firstmatch if archive_firstmatch_all_same else None
t . paused_value = last_paused if paused_all_same else None
2015-04-07 03:10:50 +00:00
t . tag_value = last_tag if tag_all_same else None
2015-02-23 02:19:25 +00:00
t . anime_value = last_anime if anime_all_same else None
t . flatten_folders_value = last_flatten_folders if flatten_folders_all_same else None
t . quality_value = last_quality if quality_all_same else None
t . subtitles_value = last_subtitles if subtitles_all_same else None
t . scene_value = last_scene if scene_all_same else None
t . sports_value = last_sports if sports_all_same else None
t . air_by_date_value = last_air_by_date if air_by_date_all_same else None
t . root_dir_list = root_dir_list
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-06-07 12:36:50 +00:00
2015-04-07 03:10:50 +00:00
def massEditSubmit ( self , archive_firstmatch = None , paused = None , anime = None , sports = None , scene = None ,
flatten_folders = None , quality_preset = False , subtitles = None , air_by_date = None , anyQualities = [ ] ,
bestQualities = [ ] , toEdit = None , tag = None , * args , * * kwargs ) :
2014-06-16 05:45:52 +00:00
2015-02-23 02:19:25 +00:00
dir_map = { }
for cur_arg in kwargs :
if not cur_arg . startswith ( ' orig_root_dir_ ' ) :
continue
which_index = cur_arg . replace ( ' orig_root_dir_ ' , ' ' )
end_dir = kwargs [ ' new_root_dir_ ' + which_index ]
dir_map [ kwargs [ cur_arg ] ] = end_dir
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showIDs = toEdit . split ( ' | ' )
errors = [ ]
for curShow in showIDs :
curErrors = [ ]
showObj = helpers . findCertainShow ( sickbeard . showList , int ( curShow ) )
if not showObj :
continue
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
cur_root_dir = ek . ek ( os . path . dirname , showObj . _location )
cur_show_dir = ek . ek ( os . path . basename , showObj . _location )
if cur_root_dir in dir_map and cur_root_dir != dir_map [ cur_root_dir ] :
new_show_dir = ek . ek ( os . path . join , dir_map [ cur_root_dir ] , cur_show_dir )
logger . log (
u ' For show ' + showObj . name + ' changing dir from ' + showObj . _location + ' to ' + new_show_dir )
else :
new_show_dir = showObj . _location
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if archive_firstmatch == ' keep ' :
new_archive_firstmatch = showObj . archive_firstmatch
else :
new_archive_firstmatch = True if archive_firstmatch == ' enable ' else False
new_archive_firstmatch = ' on ' if new_archive_firstmatch else ' off '
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if paused == ' keep ' :
new_paused = showObj . paused
else :
new_paused = True if paused == ' enable ' else False
new_paused = ' on ' if new_paused else ' off '
2014-05-08 15:23:26 +00:00
2015-04-07 03:10:50 +00:00
if tag == ' keep ' :
new_tag = showObj . tag
else :
new_tag = tag
2015-02-23 02:19:25 +00:00
if anime == ' keep ' :
new_anime = showObj . anime
else :
new_anime = True if anime == ' enable ' else False
new_anime = ' on ' if new_anime else ' off '
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if sports == ' keep ' :
new_sports = showObj . sports
else :
new_sports = True if sports == ' enable ' else False
new_sports = ' on ' if new_sports else ' off '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if scene == ' keep ' :
new_scene = showObj . is_scene
else :
new_scene = True if scene == ' enable ' else False
new_scene = ' on ' if new_scene else ' off '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if air_by_date == ' keep ' :
new_air_by_date = showObj . air_by_date
else :
new_air_by_date = True if air_by_date == ' enable ' else False
new_air_by_date = ' on ' if new_air_by_date else ' off '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if flatten_folders == ' keep ' :
new_flatten_folders = showObj . flatten_folders
else :
new_flatten_folders = True if flatten_folders == ' enable ' else False
new_flatten_folders = ' on ' if new_flatten_folders else ' off '
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if subtitles == ' keep ' :
new_subtitles = showObj . subtitles
else :
new_subtitles = True if subtitles == ' enable ' else False
2014-03-16 22:57:38 +00:00
2015-02-23 02:19:25 +00:00
new_subtitles = ' on ' if new_subtitles else ' off '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if quality_preset == ' keep ' :
anyQualities , bestQualities = Quality . splitQuality ( showObj . quality )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
exceptions_list = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
curErrors + = Home ( self . application , self . request ) . editShow ( curShow , new_show_dir , anyQualities ,
bestQualities , exceptions_list ,
archive_firstmatch = new_archive_firstmatch ,
flatten_folders = new_flatten_folders ,
paused = new_paused , sports = new_sports ,
subtitles = new_subtitles , anime = new_anime ,
scene = new_scene , air_by_date = new_air_by_date ,
2015-04-07 03:10:50 +00:00
tag = new_tag , directCall = True )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curErrors :
logger . log ( u ' Errors: ' + str ( curErrors ) , logger . ERROR )
errors . append ( ' <b> %s :</b> \n <ul> ' % showObj . name + ' ' . join (
[ ' <li> %s </li> ' % error for error in curErrors ] ) + ' </ul> ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( errors ) > 0 :
ui . notifications . error ( ' %d error %s while saving changes: ' % ( len ( errors ) , ' ' if len ( errors ) == 1 else ' s ' ) ,
' ' . join ( errors ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def massUpdate ( self , toUpdate = None , toRefresh = None , toRename = None , toDelete = None , toRemove = None , toMetadata = None , toSubtitle = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toUpdate is not None :
toUpdate = toUpdate . split ( ' | ' )
else :
toUpdate = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toRefresh is not None :
toRefresh = toRefresh . split ( ' | ' )
else :
toRefresh = [ ]
2014-06-07 12:36:50 +00:00
2015-02-23 02:19:25 +00:00
if toRename is not None :
toRename = toRename . split ( ' | ' )
else :
toRename = [ ]
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if toSubtitle is not None :
toSubtitle = toSubtitle . split ( ' | ' )
else :
toSubtitle = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toDelete is not None :
toDelete = toDelete . split ( ' | ' )
else :
toDelete = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toRemove is not None :
toRemove = toRemove . split ( ' | ' )
else :
toRemove = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toMetadata is not None :
toMetadata = toMetadata . split ( ' | ' )
else :
toMetadata = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
errors = [ ]
refreshes = [ ]
updates = [ ]
renames = [ ]
subtitles = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for curShowID in set ( toUpdate + toRefresh + toRename + toSubtitle + toDelete + toRemove + toMetadata ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID == ' ' :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
showObj = sickbeard . helpers . findCertainShow ( sickbeard . showList , int ( curShowID ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if showObj is None :
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID in toDelete :
showObj . deleteShow ( True )
# don't do anything else if it's being deleted
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID in toRemove :
showObj . deleteShow ( )
# don't do anything else if it's being remove
2014-06-21 22:46:59 +00:00
continue
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID in toUpdate :
try :
2015-03-27 19:25:34 +00:00
sickbeard . showQueueScheduler . action . updateShow ( showObj , True , True ) # @UndefinedVariable
2015-02-23 02:19:25 +00:00
updates . append ( showObj . name )
2015-06-08 12:47:01 +00:00
except exceptions . CantUpdateException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to update show ' + showObj . name + ' : ' + ex ( e ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# don't bother refreshing shows that were updated anyway
if curShowID in toRefresh and curShowID not in toUpdate :
try :
sickbeard . showQueueScheduler . action . refreshShow ( showObj ) # @UndefinedVariable
refreshes . append ( showObj . name )
2015-06-08 12:47:01 +00:00
except exceptions . CantRefreshException as e :
2015-02-23 02:19:25 +00:00
errors . append ( ' Unable to refresh show ' + showObj . name + ' : ' + ex ( e ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID in toRename :
sickbeard . showQueueScheduler . action . renameShowEpisodes ( showObj ) # @UndefinedVariable
renames . append ( showObj . name )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if curShowID in toSubtitle :
sickbeard . showQueueScheduler . action . downloadSubtitles ( showObj ) # @UndefinedVariable
subtitles . append ( showObj . name )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( errors ) > 0 :
ui . notifications . error ( ' Errors encountered ' ,
' <br > \n ' . join ( errors ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
messageDetail = ' '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( updates ) > 0 :
messageDetail + = ' <br /><b>Updates</b><br /><ul><li> '
messageDetail + = ' </li><li> ' . join ( updates )
messageDetail + = ' </li></ul> '
2014-11-20 22:25:33 +00:00
2015-02-23 02:19:25 +00:00
if len ( refreshes ) > 0 :
messageDetail + = ' <br /><b>Refreshes</b><br /><ul><li> '
messageDetail + = ' </li><li> ' . join ( refreshes )
messageDetail + = ' </li></ul> '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( renames ) > 0 :
messageDetail + = ' <br /><b>Renames</b><br /><ul><li> '
messageDetail + = ' </li><li> ' . join ( renames )
messageDetail + = ' </li></ul> '
2014-05-01 10:00:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( subtitles ) > 0 :
messageDetail + = ' <br /><b>Subtitles</b><br /><ul><li> '
messageDetail + = ' </li><li> ' . join ( subtitles )
messageDetail + = ' </li></ul> '
2014-05-01 10:00:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( updates + refreshes + renames + subtitles ) > 0 :
ui . notifications . message ( ' The following actions were queued: ' ,
messageDetail )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def manageTorrents ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_torrents.tmpl ' )
t . info_download_station = ' '
t . submenu = self . ManageMenu ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if re . search ( ' localhost ' , sickbeard . TORRENT_HOST ) :
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . LOCALHOST_IP == ' ' :
t . webui_url = re . sub ( ' localhost ' , helpers . get_lan_ip ( ) , sickbeard . TORRENT_HOST )
else :
t . webui_url = re . sub ( ' localhost ' , sickbeard . LOCALHOST_IP , sickbeard . TORRENT_HOST )
else :
t . webui_url = sickbeard . TORRENT_HOST
2014-11-20 22:25:33 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . TORRENT_METHOD == ' utorrent ' :
t . webui_url = ' / ' . join ( s . strip ( ' / ' ) for s in ( t . webui_url , ' gui/ ' ) )
if sickbeard . TORRENT_METHOD == ' download_station ' :
if helpers . check_url ( t . webui_url + ' download/ ' ) :
t . webui_url = t . webui_url + ' download/ '
else :
t . info_download_station = ' <p>To have a better experience please set the Download Station alias as <code>download</code>, you can check this setting in the Synology DSM <b>Control Panel</b> > <b>Application Portal</b>. Make sure you allow DSM to be embedded with iFrames too in <b>Control Panel</b> > <b>DSM Settings</b> > <b>Security</b>.</p><br/><p>There is more information about this available <a href= " https://github.com/midgetspy/Sick-Beard/pull/338 " >here</a>.</p><br/> '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def failedDownloads ( self , limit = 100 , toRemove = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( ' failed.db ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if limit == ' 0 ' :
sqlResults = myDB . select ( ' SELECT * FROM failed ' )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
sqlResults = myDB . select ( ' SELECT * FROM failed LIMIT ? ' , [ limit ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
toRemove = toRemove . split ( ' | ' ) if toRemove is not None else [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for release in toRemove :
2015-05-07 00:36:40 +00:00
item = re . sub ( ' _ { 3,} ' , ' % ' , release )
myDB . action ( ' DELETE FROM failed WHERE release like ? ' , [ item ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if toRemove :
2015-03-10 00:57:43 +00:00
return self . redirect ( ' /manage/failedDownloads/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' manage_failedDownloads.tmpl ' )
t . failedResults = sqlResults
t . limit = limit
t . submenu = self . ManageMenu ( )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-11-20 22:25:33 +00:00
2014-06-27 11:00:16 +00:00
2015-02-23 02:19:25 +00:00
class ManageSearches ( Manage ) :
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' manage_manageSearches.tmpl ' )
# t.backlogPI = sickbeard.backlogSearchScheduler.action.getProgressIndicator()
2015-03-11 12:31:57 +00:00
t . backlogPaused = sickbeard . searchQueueScheduler . action . is_backlog_paused ( )
t . backlogRunning = sickbeard . searchQueueScheduler . action . is_backlog_in_progress ( )
2015-05-11 19:35:41 +00:00
t . standardBacklogRunning = sickbeard . searchQueueScheduler . action . is_standard_backlog_in_progress ( )
2015-05-04 19:14:29 +00:00
t . backlogRunningType = sickbeard . searchQueueScheduler . action . type_of_backlog_in_progress ( )
2015-03-11 12:31:57 +00:00
t . recentSearchStatus = sickbeard . searchQueueScheduler . action . is_recentsearch_in_progress ( )
2015-05-04 19:14:29 +00:00
t . findPropersStatus = sickbeard . searchQueueScheduler . action . is_propersearch_in_progress ( )
2015-02-23 02:19:25 +00:00
t . queueLength = sickbeard . searchQueueScheduler . action . queue_length ( )
2014-06-27 11:00:16 +00:00
2015-02-23 02:19:25 +00:00
t . submenu = self . ManageMenu ( )
2014-06-27 11:04:54 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-08-29 10:29:56 +00:00
2015-02-23 02:19:25 +00:00
def forceVersionCheck ( self , * args , * * kwargs ) :
# force a check to see if there is a new version
if sickbeard . versionCheckScheduler . action . check_for_new_version ( force = True ) :
logger . log ( u ' Forcing version check ' )
2014-08-08 23:07:17 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /home/ ' )
2014-06-29 03:28:00 +00:00
2015-05-04 19:14:29 +00:00
def forceLimitedBacklog ( self , * args , * * kwargs ) :
2015-02-23 02:19:25 +00:00
# force it to run the next time it looks
2015-05-04 19:14:29 +00:00
if not sickbeard . searchQueueScheduler . action . is_standard_backlog_in_progress ( ) :
sickbeard . backlogSearchScheduler . forceSearch ( force_type = LIMITED_BACKLOG )
logger . log ( u ' Limited Backlog search forced ' )
ui . notifications . message ( ' Limited Backlog search started ' )
time . sleep ( 5 )
self . redirect ( ' /manage/manageSearches/ ' )
def forceFullBacklog ( self , * args , * * kwargs ) :
# force it to run the next time it looks
if not sickbeard . searchQueueScheduler . action . is_standard_backlog_in_progress ( ) :
sickbeard . backlogSearchScheduler . forceSearch ( force_type = FULL_BACKLOG )
logger . log ( u ' Full Backlog search forced ' )
ui . notifications . message ( ' Full Backlog search started ' )
2014-11-20 22:25:33 +00:00
2015-05-04 19:14:29 +00:00
time . sleep ( 5 )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/manageSearches/ ' )
2014-06-27 11:00:16 +00:00
2015-02-23 02:19:25 +00:00
def forceSearch ( self , * args , * * kwargs ) :
2014-06-27 11:00:16 +00:00
2015-02-23 02:19:25 +00:00
# force it to run the next time it looks
2015-05-04 19:14:29 +00:00
if not sickbeard . searchQueueScheduler . action . is_recentsearch_in_progress ( ) :
result = sickbeard . recentSearchScheduler . forceRun ( )
if result :
logger . log ( u ' Recent search forced ' )
ui . notifications . message ( ' Recent search started ' )
2014-06-27 11:00:16 +00:00
2015-05-04 19:14:29 +00:00
time . sleep ( 5 )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/manageSearches/ ' )
2014-07-01 08:49:12 +00:00
2015-02-23 02:19:25 +00:00
def forceFindPropers ( self , * args , * * kwargs ) :
2014-07-01 08:49:12 +00:00
2015-02-23 02:19:25 +00:00
# force it to run the next time it looks
result = sickbeard . properFinderScheduler . forceRun ( )
if result :
logger . log ( u ' Find propers search forced ' )
ui . notifications . message ( ' Find propers search started ' )
2014-11-06 04:55:43 +00:00
2015-05-04 19:14:29 +00:00
time . sleep ( 5 )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/manageSearches/ ' )
2014-06-27 11:00:16 +00:00
2015-02-23 02:19:25 +00:00
def pauseBacklog ( self , paused = None ) :
if paused == ' 1 ' :
sickbeard . searchQueueScheduler . action . pause_backlog ( ) # @UndefinedVariable
else :
sickbeard . searchQueueScheduler . action . unpause_backlog ( ) # @UndefinedVariable
2014-03-10 05:18:05 +00:00
2015-05-04 19:14:29 +00:00
time . sleep ( 5 )
2015-02-23 02:19:25 +00:00
self . redirect ( ' /manage/manageSearches/ ' )
2014-06-11 08:34:28 +00:00
2015-05-04 19:14:29 +00:00
class showQueueOverview ( Manage ) :
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' manage_showQueueOverview.tmpl ' )
t . queueLength = sickbeard . showQueueScheduler . action . queue_length ( )
t . showList = sickbeard . showList
t . ShowUpdateRunning = sickbeard . showQueueScheduler . action . isShowUpdateRunning ( )
t . submenu = self . ManageMenu ( )
return t . respond ( )
def forceShowUpdate ( self , * args , * * kwargs ) :
result = sickbeard . showUpdateScheduler . forceRun ( )
if result :
logger . log ( u ' Show Update forced ' )
ui . notifications . message ( ' Forced Show Update started ' )
time . sleep ( 5 )
self . redirect ( ' /manage/showQueueOverview/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class History ( MainHandler ) :
def index ( self , limit = 100 ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# sqlResults = myDB.select('SELECT h.*, show_name, name FROM history h, tv_shows s, tv_episodes e WHERE h.showid=s.indexer_id AND h.showid=e.showid AND h.season=e.season AND h.episode=e.episode ORDER BY date DESC LIMIT '+str(numPerPage*(p-1))+', '+str(numPerPage))
myDB = db . DBConnection ( )
if limit == ' 0 ' :
sqlResults = myDB . select (
' SELECT h.*, show_name FROM history h, tv_shows s WHERE h.showid=s.indexer_id ORDER BY date DESC ' )
else :
sqlResults = myDB . select (
' SELECT h.*, show_name FROM history h, tv_shows s WHERE h.showid=s.indexer_id ORDER BY date DESC LIMIT ? ' ,
[ limit ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
history = { ' show_id ' : 0 , ' season ' : 0 , ' episode ' : 0 , ' quality ' : 0 ,
' actions ' : [ { ' time ' : ' ' , ' action ' : ' ' , ' provider ' : ' ' } ] }
compact = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for sql_result in sqlResults :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not any ( ( history [ ' show_id ' ] == sql_result [ ' showid ' ]
and history [ ' season ' ] == sql_result [ ' season ' ]
and history [ ' episode ' ] == sql_result [ ' episode ' ]
and history [ ' quality ' ] == sql_result [ ' quality ' ] )
for history in compact ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
history = { }
history [ ' show_id ' ] = sql_result [ ' showid ' ]
history [ ' season ' ] = sql_result [ ' season ' ]
history [ ' episode ' ] = sql_result [ ' episode ' ]
history [ ' quality ' ] = sql_result [ ' quality ' ]
history [ ' show_name ' ] = sql_result [ ' show_name ' ]
history [ ' resource ' ] = sql_result [ ' resource ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
action = { }
history [ ' actions ' ] = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
action [ ' time ' ] = sql_result [ ' date ' ]
action [ ' action ' ] = sql_result [ ' action ' ]
action [ ' provider ' ] = sql_result [ ' provider ' ]
action [ ' resource ' ] = sql_result [ ' resource ' ]
history [ ' actions ' ] . append ( action )
history [ ' actions ' ] . sort ( key = lambda x : x [ ' time ' ] )
compact . append ( history )
else :
index = [ i for i , dict in enumerate ( compact ) \
if dict [ ' show_id ' ] == sql_result [ ' showid ' ] \
and dict [ ' season ' ] == sql_result [ ' season ' ] \
and dict [ ' episode ' ] == sql_result [ ' episode ' ]
and dict [ ' quality ' ] == sql_result [ ' quality ' ] ] [ 0 ]
2014-06-02 03:43:37 +00:00
2015-02-23 02:19:25 +00:00
action = { }
history = compact [ index ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
action [ ' time ' ] = sql_result [ ' date ' ]
action [ ' action ' ] = sql_result [ ' action ' ]
action [ ' provider ' ] = sql_result [ ' provider ' ]
action [ ' resource ' ] = sql_result [ ' resource ' ]
history [ ' actions ' ] . append ( action )
history [ ' actions ' ] . sort ( key = lambda x : x [ ' time ' ] , reverse = True )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' history.tmpl ' )
t . historyResults = sqlResults
t . compactResults = compact
t . limit = limit
t . submenu = [
{ ' title ' : ' Clear History ' , ' path ' : ' history/clearHistory ' } ,
{ ' title ' : ' Trim History ' , ' path ' : ' history/trimHistory ' } ,
]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def clearHistory ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
myDB . action ( ' DELETE FROM history WHERE 1=1 ' )
2015-01-25 11:58:49 +00:00
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' History cleared ' )
self . redirect ( ' /history/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def trimHistory ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
myDB = db . DBConnection ( )
myDB . action ( ' DELETE FROM history WHERE date < ' + str (
( datetime . datetime . today ( ) - datetime . timedelta ( days = 30 ) ) . strftime ( history . dateFormat ) ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Removed history entries greater than 30 days old ' )
self . redirect ( ' /history/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class Config ( MainHandler ) :
@staticmethod
def ConfigMenu ( ) :
return [
{ ' title ' : ' General ' , ' path ' : ' config/general/ ' } ,
{ ' title ' : ' Search Settings ' , ' path ' : ' config/search/ ' } ,
{ ' title ' : ' Search Providers ' , ' path ' : ' config/providers/ ' } ,
{ ' title ' : ' Subtitles Settings ' , ' path ' : ' config/subtitles/ ' } ,
{ ' title ' : ' Post Processing ' , ' path ' : ' config/postProcessing/ ' } ,
{ ' title ' : ' Notifications ' , ' path ' : ' config/notifications/ ' } ,
{ ' title ' : ' Anime ' , ' path ' : ' config/anime/ ' } ,
]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' config.tmpl ' )
t . submenu = self . ConfigMenu
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class ConfigGeneral ( Config ) :
def index ( self , * args , * * kwargs ) :
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' config_general.tmpl ' )
t . submenu = self . ConfigMenu
2015-05-03 00:42:14 +00:00
t . show_tags = ' , ' . join ( sickbeard . SHOW_TAGS )
2015-02-23 02:19:25 +00:00
return t . respond ( )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
def saveRootDirs ( self , rootDirString = None ) :
sickbeard . ROOT_DIRS = rootDirString
2014-03-10 05:18:05 +00:00
2015-03-10 00:13:58 +00:00
def saveAddShowDefaults ( self , default_status , any_qualities = ' ' , best_qualities = ' ' , default_wanted_begin = None ,
default_wanted_latest = None , default_flatten_folders = False , default_scene = False ,
default_subtitles = False , default_anime = False ) :
any_qualities = ( [ ] , any_qualities . split ( ' , ' ) ) [ any ( any_qualities ) ]
best_qualities = ( [ ] , best_qualities . split ( ' , ' ) ) [ any ( best_qualities ) ]
sickbeard . STATUS_DEFAULT = int ( default_status )
sickbeard . QUALITY_DEFAULT = int ( Quality . combineQualities ( map ( int , any_qualities ) , map ( int , best_qualities ) ) )
sickbeard . WANTED_BEGIN_DEFAULT = config . minimax ( default_wanted_begin , 0 , - 1 , 10 )
sickbeard . WANTED_LATEST_DEFAULT = config . minimax ( default_wanted_latest , 0 , - 1 , 10 )
sickbeard . FLATTEN_FOLDERS_DEFAULT = config . checkbox_to_value ( default_flatten_folders )
sickbeard . SCENE_DEFAULT = config . checkbox_to_value ( default_scene )
sickbeard . SUBTITLES_DEFAULT = config . checkbox_to_value ( default_subtitles )
sickbeard . ANIME_DEFAULT = config . checkbox_to_value ( default_anime )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
def generateKey ( self , * args , * * kwargs ) :
""" Return a new randomized API_KEY
"""
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
try :
from hashlib import md5
except ImportError :
from md5 import md5
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Create some values to seed md5
t = str ( time . time ( ) )
r = str ( random . random ( ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Create the md5 instance and give it the current time
m = md5 ( t )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Update the md5 instance with the random variable
m . update ( r )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Return a hex digest of the md5, eg 49f68a5c8493ec2c0bf489821c21fc3b
logger . log ( u ' New API generated ' )
return m . hexdigest ( )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def saveGeneral ( self , log_dir = None , web_port = None , web_log = None , encryption_version = None , web_ipv6 = None ,
update_shows_on_start = None , show_update_hour = None , trash_remove_show = None , trash_rotate_logs = None , update_frequency = None , launch_browser = None , web_username = None ,
2015-03-24 20:52:23 +00:00
use_api = None , api_key = None , indexer_default = None , timezone_display = None , cpu_preset = None , file_logging_preset = None ,
2015-02-23 02:19:25 +00:00
web_password = None , version_notify = None , enable_https = None , https_cert = None , https_key = None ,
handle_reverse_proxy = None , home_search_focus = None , sort_article = None , auto_update = None , notify_on_update = None ,
proxy_setting = None , proxy_indexers = None , anon_redirect = None , git_path = None , git_remote = None , calendar_unprotected = None ,
fuzzy_dating = None , trim_zero = None , date_preset = None , date_preset_na = None , time_preset = None ,
2015-03-23 01:05:13 +00:00
indexer_timeout = None , rootDir = None , theme_name = None , default_home = None , use_imdb_info = None ,
2015-04-07 03:10:50 +00:00
display_background = None , display_background_transparent = None , display_all_seasons = None ,
show_tags = None , showlist_tagview = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# Misc
sickbeard . LAUNCH_BROWSER = config . checkbox_to_value ( launch_browser )
config . change_VERSION_NOTIFY ( config . checkbox_to_value ( version_notify ) )
sickbeard . AUTO_UPDATE = config . checkbox_to_value ( auto_update )
sickbeard . NOTIFY_ON_UPDATE = config . checkbox_to_value ( notify_on_update )
# sickbeard.LOG_DIR is set in config.change_LOG_DIR()
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . UPDATE_SHOWS_ON_START = config . checkbox_to_value ( update_shows_on_start )
sickbeard . SHOW_UPDATE_HOUR = config . minimax ( show_update_hour , 3 , 0 , 23 )
sickbeard . TRASH_REMOVE_SHOW = config . checkbox_to_value ( trash_remove_show )
sickbeard . TRASH_ROTATE_LOGS = config . checkbox_to_value ( trash_rotate_logs )
config . change_UPDATE_FREQUENCY ( update_frequency )
sickbeard . LAUNCH_BROWSER = config . checkbox_to_value ( launch_browser )
sickbeard . HOME_SEARCH_FOCUS = config . checkbox_to_value ( home_search_focus )
2015-03-14 02:48:38 +00:00
sickbeard . USE_IMDB_INFO = config . checkbox_to_value ( use_imdb_info )
2015-03-23 01:05:13 +00:00
sickbeard . DISPLAY_BACKGROUND = config . checkbox_to_value ( display_background )
sickbeard . DISPLAY_BACKGROUND_TRANSPARENT = display_background_transparent
sickbeard . DISPLAY_ALL_SEASONS = config . checkbox_to_value ( display_all_seasons )
2015-02-23 02:19:25 +00:00
sickbeard . SORT_ARTICLE = config . checkbox_to_value ( sort_article )
sickbeard . CPU_PRESET = cpu_preset
2015-03-24 20:52:23 +00:00
sickbeard . FILE_LOGGING_PRESET = file_logging_preset
2015-04-07 03:10:50 +00:00
sickbeard . SHOWLIST_TAGVIEW = showlist_tagview
2015-05-03 00:42:14 +00:00
# 'Show List' is the must have default fallback. Tags in use that are removed from config ui are restored, not deleted.
# Deduped list order preservation is key to feature function.
myDB = db . DBConnection ( )
2015-04-07 03:10:50 +00:00
sql_results = myDB . select ( ' SELECT DISTINCT tag FROM tv_shows ' )
2015-05-03 00:42:14 +00:00
new_names = [ u ' ' + v . strip ( ) for v in ( show_tags . split ( u ' , ' ) , [ ] ) [ None is show_tags ] if v . strip ( ) ]
orphans = [ item for item in [ v [ ' tag ' ] for v in sql_results or [ ] ] if item not in new_names ]
cleanser = [ ]
if 0 < len ( orphans ) :
cleanser = [ item for item in sickbeard . SHOW_TAGS if item in orphans or item in new_names ]
results + = [ u ' An attempt was prevented to remove a show list group name still in use ' ]
dedupe = { }
sickbeard . SHOW_TAGS = [ dedupe . setdefault ( item , item ) for item in ( cleanser + new_names + [ u ' Show List ' ] )
if item not in dedupe ]
2015-03-24 20:52:23 +00:00
logger . log_set_level ( )
2015-02-23 02:19:25 +00:00
sickbeard . ANON_REDIRECT = anon_redirect
sickbeard . PROXY_SETTING = proxy_setting
sickbeard . PROXY_INDEXERS = config . checkbox_to_value ( proxy_indexers )
sickbeard . GIT_PATH = git_path
sickbeard . GIT_REMOTE = git_remote
sickbeard . CALENDAR_UNPROTECTED = config . checkbox_to_value ( calendar_unprotected )
# sickbeard.LOG_DIR is set in config.change_LOG_DIR()
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . WEB_PORT = config . to_int ( web_port )
sickbeard . WEB_IPV6 = config . checkbox_to_value ( web_ipv6 )
# sickbeard.WEB_LOG is set in config.change_LOG_DIR()
sickbeard . ENCRYPTION_VERSION = config . checkbox_to_value ( encryption_version )
2014-03-10 05:18:05 +00:00
2015-02-28 04:23:46 +00:00
reload_page = False
2015-02-14 02:28:24 +00:00
2015-02-28 04:23:46 +00:00
if sickbeard . WEB_USERNAME != web_username :
sickbeard . WEB_USERNAME = web_username
reload_page = True
2014-03-10 05:18:05 +00:00
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( web_password ) :
sickbeard . WEB_PASSWORD = web_password
2015-02-28 04:23:46 +00:00
reload_page = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . FUZZY_DATING = config . checkbox_to_value ( fuzzy_dating )
sickbeard . TRIM_ZERO = config . checkbox_to_value ( trim_zero )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if date_preset :
sickbeard . DATE_PRESET = date_preset
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if indexer_default :
sickbeard . INDEXER_DEFAULT = config . to_int ( indexer_default )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if indexer_timeout :
sickbeard . INDEXER_TIMEOUT = config . to_int ( indexer_timeout )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if time_preset :
sickbeard . TIME_PRESET_W_SECONDS = time_preset
sickbeard . TIME_PRESET = sickbeard . TIME_PRESET_W_SECONDS . replace ( u ' : % S ' , u ' ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . TIMEZONE_DISPLAY = timezone_display
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_LOG_DIR ( log_dir , web_log ) :
results + = [ ' Unable to create directory ' + os . path . normpath ( log_dir ) + ' , log directory not changed. ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_API = config . checkbox_to_value ( use_api )
sickbeard . API_KEY = api_key
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . ENABLE_HTTPS = config . checkbox_to_value ( enable_https )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_HTTPS_CERT ( https_cert ) :
results + = [
' Unable to create directory ' + os . path . normpath ( https_cert ) + ' , https cert directory not changed. ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_HTTPS_KEY ( https_key ) :
results + = [
' Unable to create directory ' + os . path . normpath ( https_key ) + ' , https key directory not changed. ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . HANDLE_REVERSE_PROXY = config . checkbox_to_value ( handle_reverse_proxy )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . THEME_NAME = theme_name
2015-02-25 00:34:26 +00:00
sickbeard . DEFAULT_HOME = default_home
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
2015-05-03 00:42:14 +00:00
for v in results :
logger . log ( v , logger . ERROR )
2015-02-23 02:19:25 +00:00
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
2014-05-26 06:29:22 +00:00
else :
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2014-06-11 08:34:28 +00:00
2015-02-28 04:23:46 +00:00
if reload_page :
self . clear_cookie ( ' sickgear-session ' )
self . write ( ' reload ' )
2014-07-06 14:45:01 +00:00
2015-03-02 11:31:11 +00:00
@staticmethod
def fetch_pullrequests ( ) :
if sickbeard . BRANCH == ' master ' :
return json . dumps ( { ' result ' : ' success ' , ' pulls ' : [ ] } )
else :
try :
pulls = sickbeard . versionCheckScheduler . action . list_remote_pulls ( )
return json . dumps ( { ' result ' : ' success ' , ' pulls ' : pulls } )
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-03-02 11:31:11 +00:00
logger . log ( u ' exception msg: ' + str ( e ) , logger . DEBUG )
return json . dumps ( { ' result ' : ' fail ' } )
@staticmethod
def fetch_branches ( ) :
try :
branches = sickbeard . versionCheckScheduler . action . list_remote_branches ( )
return json . dumps ( { ' result ' : ' success ' , ' branches ' : branches } )
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-03-02 11:31:11 +00:00
logger . log ( u ' exception msg: ' + str ( e ) , logger . DEBUG )
return json . dumps ( { ' result ' : ' fail ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class ConfigSearch ( Config ) :
def index ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' config_search.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveSearch ( self , use_nzbs = None , use_torrents = None , nzb_dir = None , sab_username = None , sab_password = None ,
sab_apikey = None , sab_category = None , sab_host = None , nzbget_username = None , nzbget_password = None ,
nzbget_category = None , nzbget_priority = None , nzbget_host = None , nzbget_use_https = None ,
2015-04-05 18:12:15 +00:00
backlog_days = None , backlog_frequency = None , search_unaired = None , recentsearch_frequency = None ,
2015-02-23 02:19:25 +00:00
nzb_method = None , torrent_method = None , usenet_retention = None ,
download_propers = None , check_propers_interval = None , allow_high_priority = None ,
torrent_dir = None , torrent_username = None , torrent_password = None , torrent_host = None ,
torrent_label = None , torrent_path = None , torrent_verify_cert = None ,
torrent_seed_time = None , torrent_paused = None , torrent_high_bandwidth = None , ignore_words = None , require_words = None ) :
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_NZB_DIR ( nzb_dir ) :
results + = [ ' Unable to create directory ' + os . path . normpath ( nzb_dir ) + ' , dir not changed. ' ]
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_TORRENT_DIR ( torrent_dir ) :
results + = [ ' Unable to create directory ' + os . path . normpath ( torrent_dir ) + ' , dir not changed. ' ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
config . change_RECENTSEARCH_FREQUENCY ( recentsearch_frequency )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
config . change_BACKLOG_FREQUENCY ( backlog_frequency )
sickbeard . BACKLOG_DAYS = config . to_int ( backlog_days , default = 7 )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_NZBS = config . checkbox_to_value ( use_nzbs )
sickbeard . USE_TORRENTS = config . checkbox_to_value ( use_torrents )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . NZB_METHOD = nzb_method
sickbeard . TORRENT_METHOD = torrent_method
sickbeard . USENET_RETENTION = config . to_int ( usenet_retention , default = 500 )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . IGNORE_WORDS = ignore_words if ignore_words else ' '
sickbeard . REQUIRE_WORDS = require_words if require_words else ' '
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . DOWNLOAD_PROPERS = config . checkbox_to_value ( download_propers )
sickbeard . CHECK_PROPERS_INTERVAL = check_propers_interval
2014-03-10 05:18:05 +00:00
2015-04-05 18:12:15 +00:00
sickbeard . SEARCH_UNAIRED = config . checkbox_to_value ( search_unaired )
2015-02-23 02:19:25 +00:00
sickbeard . ALLOW_HIGH_PRIORITY = config . checkbox_to_value ( allow_high_priority )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . SAB_USERNAME = sab_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( sab_password ) :
sickbeard . SAB_PASSWORD = sab_password
key = sab_apikey . strip ( )
if not starify ( key , True ) :
sickbeard . SAB_APIKEY = key
2015-02-23 02:19:25 +00:00
sickbeard . SAB_CATEGORY = sab_category
sickbeard . SAB_HOST = config . clean_url ( sab_host )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . NZBGET_USERNAME = nzbget_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( nzbget_password ) :
sickbeard . NZBGET_PASSWORD = nzbget_password
2015-02-23 02:19:25 +00:00
sickbeard . NZBGET_CATEGORY = nzbget_category
sickbeard . NZBGET_HOST = config . clean_host ( nzbget_host )
sickbeard . NZBGET_USE_HTTPS = config . checkbox_to_value ( nzbget_use_https )
sickbeard . NZBGET_PRIORITY = config . to_int ( nzbget_priority , default = 100 )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . TORRENT_USERNAME = torrent_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( torrent_password ) :
sickbeard . TORRENT_PASSWORD = torrent_password
2015-02-23 02:19:25 +00:00
sickbeard . TORRENT_LABEL = torrent_label
sickbeard . TORRENT_VERIFY_CERT = config . checkbox_to_value ( torrent_verify_cert )
sickbeard . TORRENT_PATH = torrent_path
sickbeard . TORRENT_SEED_TIME = torrent_seed_time
sickbeard . TORRENT_PAUSED = config . checkbox_to_value ( torrent_paused )
sickbeard . TORRENT_HIGH_BANDWIDTH = config . checkbox_to_value ( torrent_high_bandwidth )
sickbeard . TORRENT_HOST = config . clean_url ( torrent_host )
2014-05-07 14:23:06 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
2014-05-07 14:23:06 +00:00
else :
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2014-05-07 14:23:06 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /config/search/ ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class ConfigPostProcessing ( Config ) :
def index ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' config_postProcessing.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def savePostProcessing ( self , naming_pattern = None , naming_multi_ep = None ,
xbmc_data = None , xbmc_12plus_data = None , mediabrowser_data = None , sony_ps3_data = None ,
2015-02-25 12:33:40 +00:00
wdtv_data = None , tivo_data = None , mede8er_data = None , kodi_data = None ,
2015-02-23 02:19:25 +00:00
keep_processed_dir = None , process_method = None , process_automatically = None ,
rename_episodes = None , airdate_episodes = None , unpack = None ,
move_associated_files = None , postpone_if_sync_files = None , nfo_rename = None , tv_download_dir = None , naming_custom_abd = None ,
naming_anime = None ,
naming_abd_pattern = None , naming_strip_year = None , use_failed_downloads = None ,
delete_failed = None , extra_scripts = None , skip_removed_files = None ,
naming_custom_sports = None , naming_sports_pattern = None ,
naming_custom_anime = None , naming_anime_pattern = None , naming_anime_multi_ep = None ,
autopostprocesser_frequency = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not config . change_TV_DOWNLOAD_DIR ( tv_download_dir ) :
results + = [ ' Unable to create directory ' + os . path . normpath ( tv_download_dir ) + ' , dir not changed. ' ]
2014-03-10 05:18:05 +00:00
2015-03-12 23:15:44 +00:00
new_val = config . checkbox_to_value ( process_automatically )
if new_val != sickbeard . PROCESS_AUTOMATICALLY :
if not sickbeard . PROCESS_AUTOMATICALLY and not sickbeard . autoPostProcesserScheduler . ident :
try :
sickbeard . autoPostProcesserScheduler . start ( )
except :
pass
sickbeard . PROCESS_AUTOMATICALLY = new_val
2015-02-23 02:19:25 +00:00
config . change_AUTOPOSTPROCESSER_FREQUENCY ( autopostprocesser_frequency )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . PROCESS_AUTOMATICALLY :
sickbeard . autoPostProcesserScheduler . silent = False
else :
sickbeard . autoPostProcesserScheduler . silent = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if unpack :
if self . isRarSupported ( ) != ' not supported ' :
sickbeard . UNPACK = config . checkbox_to_value ( unpack )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
sickbeard . UNPACK = 0
results . append ( ' Unpacking Not Supported, disabling unpack setting ' )
else :
sickbeard . UNPACK = config . checkbox_to_value ( unpack )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . KEEP_PROCESSED_DIR = config . checkbox_to_value ( keep_processed_dir )
sickbeard . PROCESS_METHOD = process_method
sickbeard . EXTRA_SCRIPTS = [ x . strip ( ) for x in extra_scripts . split ( ' | ' ) if x . strip ( ) ]
sickbeard . RENAME_EPISODES = config . checkbox_to_value ( rename_episodes )
sickbeard . AIRDATE_EPISODES = config . checkbox_to_value ( airdate_episodes )
sickbeard . MOVE_ASSOCIATED_FILES = config . checkbox_to_value ( move_associated_files )
sickbeard . POSTPONE_IF_SYNC_FILES = config . checkbox_to_value ( postpone_if_sync_files )
sickbeard . NAMING_CUSTOM_ABD = config . checkbox_to_value ( naming_custom_abd )
sickbeard . NAMING_CUSTOM_SPORTS = config . checkbox_to_value ( naming_custom_sports )
sickbeard . NAMING_CUSTOM_ANIME = config . checkbox_to_value ( naming_custom_anime )
sickbeard . NAMING_STRIP_YEAR = config . checkbox_to_value ( naming_strip_year )
sickbeard . USE_FAILED_DOWNLOADS = config . checkbox_to_value ( use_failed_downloads )
sickbeard . DELETE_FAILED = config . checkbox_to_value ( delete_failed )
2015-03-17 13:59:03 +00:00
sickbeard . SKIP_REMOVED_FILES = config . minimax ( skip_removed_files , IGNORED , 1 , IGNORED )
2015-02-23 02:19:25 +00:00
sickbeard . NFO_RENAME = config . checkbox_to_value ( nfo_rename )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . METADATA_XBMC = xbmc_data
sickbeard . METADATA_XBMC_12PLUS = xbmc_12plus_data
sickbeard . METADATA_MEDIABROWSER = mediabrowser_data
sickbeard . METADATA_PS3 = sony_ps3_data
sickbeard . METADATA_WDTV = wdtv_data
sickbeard . METADATA_TIVO = tivo_data
sickbeard . METADATA_MEDE8ER = mede8er_data
2015-02-25 12:33:40 +00:00
sickbeard . METADATA_KODI = kodi_data
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . metadata_provider_dict [ ' XBMC ' ] . set_config ( sickbeard . METADATA_XBMC )
sickbeard . metadata_provider_dict [ ' XBMC 12+ ' ] . set_config ( sickbeard . METADATA_XBMC_12PLUS )
sickbeard . metadata_provider_dict [ ' MediaBrowser ' ] . set_config ( sickbeard . METADATA_MEDIABROWSER )
sickbeard . metadata_provider_dict [ ' Sony PS3 ' ] . set_config ( sickbeard . METADATA_PS3 )
sickbeard . metadata_provider_dict [ ' WDTV ' ] . set_config ( sickbeard . METADATA_WDTV )
sickbeard . metadata_provider_dict [ ' TIVO ' ] . set_config ( sickbeard . METADATA_TIVO )
sickbeard . metadata_provider_dict [ ' Mede8er ' ] . set_config ( sickbeard . METADATA_MEDE8ER )
2015-02-25 12:33:40 +00:00
sickbeard . metadata_provider_dict [ ' Kodi ' ] . set_config ( sickbeard . METADATA_KODI )
2015-02-23 02:19:25 +00:00
if self . isNamingValid ( naming_pattern , naming_multi_ep , anime_type = naming_anime ) != ' invalid ' :
sickbeard . NAMING_PATTERN = naming_pattern
sickbeard . NAMING_MULTI_EP = int ( naming_multi_ep )
sickbeard . NAMING_ANIME = int ( naming_anime )
sickbeard . NAMING_FORCE_FOLDERS = naming . check_force_season_folders ( )
else :
if int ( naming_anime ) in [ 1 , 2 ] :
results . append ( ' You tried saving an invalid anime naming config, not saving your naming settings ' )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
results . append ( ' You tried saving an invalid naming config, not saving your naming settings ' )
2015-02-14 02:28:24 +00:00
2015-02-23 02:19:25 +00:00
if self . isNamingValid ( naming_anime_pattern , naming_anime_multi_ep , anime_type = naming_anime ) != ' invalid ' :
sickbeard . NAMING_ANIME_PATTERN = naming_anime_pattern
sickbeard . NAMING_ANIME_MULTI_EP = int ( naming_anime_multi_ep )
sickbeard . NAMING_ANIME = int ( naming_anime )
sickbeard . NAMING_FORCE_FOLDERS = naming . check_force_season_folders ( )
else :
if int ( naming_anime ) in [ 1 , 2 ] :
results . append ( ' You tried saving an invalid anime naming config, not saving your naming settings ' )
else :
results . append ( ' You tried saving an invalid naming config, not saving your naming settings ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if self . isNamingValid ( naming_abd_pattern , None , abd = True ) != ' invalid ' :
sickbeard . NAMING_ABD_PATTERN = naming_abd_pattern
else :
results . append (
' You tried saving an invalid air-by-date naming config, not saving your air-by-date settings ' )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if self . isNamingValid ( naming_sports_pattern , None , sports = True ) != ' invalid ' :
sickbeard . NAMING_SPORTS_PATTERN = naming_sports_pattern
else :
results . append (
' You tried saving an invalid sports naming config, not saving your sports settings ' )
2015-02-14 02:28:24 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2015-02-14 02:28:24 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
2015-02-14 02:28:24 +00:00
else :
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2015-02-14 02:28:24 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /config/postProcessing/ ' )
2015-02-14 02:28:24 +00:00
2015-02-23 02:19:25 +00:00
def testNaming ( self , pattern = None , multi = None , abd = False , sports = False , anime_type = None ) :
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if multi is not None :
multi = int ( multi )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if anime_type is not None :
anime_type = int ( anime_type )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = naming . test_name ( pattern , multi , abd , sports , anime_type )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
result = ek . ek ( os . path . join , result [ ' dir ' ] , result [ ' name ' ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return result
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
def isNamingValid ( self , pattern = None , multi = None , abd = False , sports = False , anime_type = None ) :
if pattern is None :
return ' invalid '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if multi is not None :
multi = int ( multi )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if anime_type is not None :
anime_type = int ( anime_type )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# air by date shows just need one check, we don't need to worry about season folders
if abd :
is_valid = naming . check_valid_abd_naming ( pattern )
require_season_folders = False
# sport shows just need one check, we don't need to worry about season folders
elif sports :
is_valid = naming . check_valid_sports_naming ( pattern )
require_season_folders = False
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
# check validity of single and multi ep cases for the whole path
is_valid = naming . check_valid_naming ( pattern , multi , anime_type )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# check validity of single and multi ep cases for only the file name
require_season_folders = naming . check_force_season_folders ( pattern , multi , anime_type )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if is_valid and not require_season_folders :
return ' valid '
elif is_valid and require_season_folders :
return ' seasonfolders '
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
return ' invalid '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def isRarSupported ( self , * args , * * kwargs ) :
"""
Test Packing Support :
- Simulating in memory rar extraction on test . rar file
"""
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
try :
rar_path = os . path . join ( sickbeard . PROG_DIR , ' lib ' , ' unrar2 ' , ' test.rar ' )
testing = RarFile ( rar_path ) . read_files ( ' *test.txt ' )
if testing [ 0 ] [ 1 ] == ' This is only a test. ' :
return ' supported '
logger . log ( u ' Rar Not Supported: Can not read the content of test file ' , logger . ERROR )
return ' not supported '
2015-06-08 12:47:01 +00:00
except Exception as e :
2015-02-23 02:19:25 +00:00
logger . log ( u ' Rar Not Supported: ' + ex ( e ) , logger . ERROR )
return ' not supported '
2014-06-07 21:32:38 +00:00
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class ConfigProviders ( Config ) :
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' config_providers.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def canAddNewznabProvider ( self , name ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not name :
return json . dumps ( { ' error ' : ' No Provider Name specified ' } )
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
providerDict = dict ( zip ( [ x . get_id ( ) for x in sickbeard . newznabProviderList ] , sickbeard . newznabProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
tempProvider = newznab . NewznabProvider ( name , ' ' )
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
if tempProvider . get_id ( ) in providerDict :
return json . dumps ( { ' error ' : ' Provider Name already exists as ' + providerDict [ tempProvider . get_id ( ) ] . name } )
2014-03-10 05:18:05 +00:00
else :
2015-07-13 09:39:20 +00:00
return json . dumps ( { ' success ' : tempProvider . get_id ( ) } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveNewznabProvider ( self , name , url , key = ' ' ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not name or not url :
return ' 0 '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
providerDict = dict ( zip ( [ x . name for x in sickbeard . newznabProviderList ] , sickbeard . newznabProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if name in providerDict :
if not providerDict [ name ] . default :
providerDict [ name ] . name = name
providerDict [ name ] . url = config . clean_url ( url )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
providerDict [ name ] . key = key
# a 0 in the key spot indicates that no key is needed
if key == ' 0 ' :
providerDict [ name ] . needs_auth = False
else :
providerDict [ name ] . needs_auth = True
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
return providerDict [ name ] . get_id ( ) + ' | ' + providerDict [ name ] . config_str ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
else :
newProvider = newznab . NewznabProvider ( name , url , key = key )
sickbeard . newznabProviderList . append ( newProvider )
2015-07-13 09:39:20 +00:00
return newProvider . get_id ( ) + ' | ' + newProvider . config_str ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def getNewznabCategories ( self , name , url , key ) :
'''
Retrieves a list of possible categories with category id ' s
Using the default url / api ? cat
http : / / yournewznaburl . com / api ? t = caps & apikey = yourapikey
'''
error = ' '
success = False
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not name :
error + = ' \n No Provider Name specified '
if not url :
error + = ' \n No Provider Url specified '
if not key :
error + = ' \n No Provider Api key specified '
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if error < > ' ' :
return json . dumps ( { ' success ' : False , ' error ' : error } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
#Get list with Newznabproviders
2015-07-13 09:39:20 +00:00
#providerDict = dict(zip([x.get_id() for x in sickbeard.newznabProviderList], sickbeard.newznabProviderList))
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
#Get newznabprovider obj with provided name
tempProvider = newznab . NewznabProvider ( name , url , key )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
success , tv_categories , error = tempProvider . get_newznab_categories ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return json . dumps ( { ' success ' : success , ' tv_categories ' : tv_categories , ' error ' : error } )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
def deleteNewznabProvider ( self , nnid ) :
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
providerDict = dict ( zip ( [ x . get_id ( ) for x in sickbeard . newznabProviderList ] , sickbeard . newznabProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if nnid not in providerDict or providerDict [ nnid ] . default :
return ' 0 '
2014-06-18 09:13:22 +00:00
2015-02-23 02:19:25 +00:00
# delete it from the list
sickbeard . newznabProviderList . remove ( providerDict [ nnid ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if nnid in sickbeard . PROVIDER_ORDER :
sickbeard . PROVIDER_ORDER . remove ( nnid )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
return ' 1 '
2014-12-29 04:22:34 +00:00
2015-02-23 02:19:25 +00:00
def canAddTorrentRssProvider ( self , name , url , cookies ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not name :
return json . dumps ( { ' error ' : ' Invalid name specified ' } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
providerDict = dict (
2015-07-13 09:39:20 +00:00
zip ( [ x . get_id ( ) for x in sickbeard . torrentRssProviderList ] , sickbeard . torrentRssProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
tempProvider = rsstorrent . TorrentRssProvider ( name , url , cookies )
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
if tempProvider . get_id ( ) in providerDict :
return json . dumps ( { ' error ' : ' Exists as ' + providerDict [ tempProvider . get_id ( ) ] . name } )
2015-02-23 02:19:25 +00:00
else :
2015-06-15 19:09:01 +00:00
( succ , errMsg ) = tempProvider . validate_feed ( )
2015-02-23 02:19:25 +00:00
if succ :
2015-07-13 09:39:20 +00:00
return json . dumps ( { ' success ' : tempProvider . get_id ( ) } )
2015-02-23 02:19:25 +00:00
else :
return json . dumps ( { ' error ' : errMsg } )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveTorrentRssProvider ( self , name , url , cookies ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if not name or not url :
return ' 0 '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
providerDict = dict ( zip ( [ x . name for x in sickbeard . torrentRssProviderList ] , sickbeard . torrentRssProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if name in providerDict :
providerDict [ name ] . name = name
providerDict [ name ] . url = config . clean_url ( url )
providerDict [ name ] . cookies = cookies
2014-03-10 05:18:05 +00:00
2015-07-13 09:39:20 +00:00
return providerDict [ name ] . get_id ( ) + ' | ' + providerDict [ name ] . config_str ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
else :
newProvider = rsstorrent . TorrentRssProvider ( name , url , cookies )
sickbeard . torrentRssProviderList . append ( newProvider )
2015-07-13 09:39:20 +00:00
return newProvider . get_id ( ) + ' | ' + newProvider . config_str ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def deleteTorrentRssProvider ( self , id ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
providerDict = dict (
2015-07-13 09:39:20 +00:00
zip ( [ x . get_id ( ) for x in sickbeard . torrentRssProviderList ] , sickbeard . torrentRssProviderList ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if id not in providerDict :
return ' 0 '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# delete it from the list
sickbeard . torrentRssProviderList . remove ( providerDict [ id ] )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if id in sickbeard . PROVIDER_ORDER :
sickbeard . PROVIDER_ORDER . remove ( id )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
return ' 1 '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveProviders ( self , newznab_string = ' ' , torrentrss_string = ' ' , provider_order = None , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
provider_str_list = provider_order . split ( )
provider_list = [ ]
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
newznabProviderDict = dict (
2015-07-13 09:39:20 +00:00
zip ( [ x . get_id ( ) for x in sickbeard . newznabProviderList ] , sickbeard . newznabProviderList ) )
2014-12-12 02:36:23 +00:00
2015-02-23 02:19:25 +00:00
finishedNames = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# add all the newznab info we got into our list
if newznab_string :
for curNewznabProviderStr in newznab_string . split ( ' !!! ' ) :
2014-12-13 04:06:05 +00:00
2015-02-23 02:19:25 +00:00
if not curNewznabProviderStr :
continue
2014-05-27 07:44:23 +00:00
2015-02-23 02:19:25 +00:00
cur_name , cur_url , cur_key , cur_cat = curNewznabProviderStr . split ( ' | ' )
cur_url = config . clean_url ( cur_url )
2014-03-25 05:57:24 +00:00
2015-02-24 13:37:27 +00:00
if starify ( cur_key , True ) :
cur_key = ' '
2015-02-23 02:19:25 +00:00
newProvider = newznab . NewznabProvider ( cur_name , cur_url , key = cur_key )
2014-06-20 23:49:57 +00:00
2015-07-13 09:39:20 +00:00
cur_id = newProvider . get_id ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# if it already exists then update it
if cur_id in newznabProviderDict :
newznabProviderDict [ cur_id ] . name = cur_name
newznabProviderDict [ cur_id ] . url = cur_url
2015-02-24 13:37:27 +00:00
if cur_key :
newznabProviderDict [ cur_id ] . key = cur_key
2015-06-19 16:47:52 +00:00
newznabProviderDict [ cur_id ] . cat_ids = cur_cat
2015-02-23 02:19:25 +00:00
# a 0 in the key spot indicates that no key is needed
if cur_key == ' 0 ' :
newznabProviderDict [ cur_id ] . needs_auth = False
else :
newznabProviderDict [ cur_id ] . needs_auth = True
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
try :
newznabProviderDict [ cur_id ] . search_mode = str ( kwargs [ cur_id + ' _search_mode ' ] ) . strip ( )
except :
pass
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
try :
newznabProviderDict [ cur_id ] . search_fallback = config . checkbox_to_value (
kwargs [ cur_id + ' _search_fallback ' ] )
except :
newznabProviderDict [ cur_id ] . search_fallback = 0
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
try :
newznabProviderDict [ cur_id ] . enable_recentsearch = config . checkbox_to_value (
kwargs [ cur_id + ' _enable_recentsearch ' ] )
except :
newznabProviderDict [ cur_id ] . enable_recentsearch = 0
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
try :
newznabProviderDict [ cur_id ] . enable_backlog = config . checkbox_to_value (
kwargs [ cur_id + ' _enable_backlog ' ] )
except :
newznabProviderDict [ cur_id ] . enable_backlog = 0
else :
sickbeard . newznabProviderList . append ( newProvider )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
finishedNames . append ( cur_id )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# delete anything that is missing
for curProvider in sickbeard . newznabProviderList :
2015-07-13 09:39:20 +00:00
if curProvider . get_id ( ) not in finishedNames :
2015-02-23 02:19:25 +00:00
sickbeard . newznabProviderList . remove ( curProvider )
2014-07-28 01:46:15 +00:00
2015-02-23 02:19:25 +00:00
torrentRssProviderDict = dict (
2015-07-13 09:39:20 +00:00
zip ( [ x . get_id ( ) for x in sickbeard . torrentRssProviderList ] , sickbeard . torrentRssProviderList ) )
2015-02-23 02:19:25 +00:00
finishedNames = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if torrentrss_string :
for curTorrentRssProviderStr in torrentrss_string . split ( ' !!! ' ) :
2014-07-27 14:14:41 +00:00
2015-02-23 02:19:25 +00:00
if not curTorrentRssProviderStr :
continue
2014-05-26 06:29:22 +00:00
2015-02-23 02:19:25 +00:00
curName , curURL , curCookies = curTorrentRssProviderStr . split ( ' | ' )
curURL = config . clean_url ( curURL , False )
2014-07-18 00:43:36 +00:00
2015-02-24 13:37:27 +00:00
if starify ( curCookies , True ) :
curCookies = ' '
2015-02-23 02:19:25 +00:00
newProvider = rsstorrent . TorrentRssProvider ( curName , curURL , curCookies )
2015-07-13 09:39:20 +00:00
curID = newProvider . get_id ( )
2015-02-23 02:19:25 +00:00
# if it already exists then update it
if curID in torrentRssProviderDict :
torrentRssProviderDict [ curID ] . name = curName
torrentRssProviderDict [ curID ] . url = curURL
2015-02-24 13:37:27 +00:00
if curCookies :
torrentRssProviderDict [ curID ] . cookies = curCookies
2015-02-16 03:17:56 +00:00
else :
2015-02-23 02:19:25 +00:00
sickbeard . torrentRssProviderList . append ( newProvider )
2014-07-18 00:43:36 +00:00
2015-02-23 02:19:25 +00:00
finishedNames . append ( curID )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# delete anything that is missing
for curProvider in sickbeard . torrentRssProviderList :
2015-07-13 09:39:20 +00:00
if curProvider . get_id ( ) not in finishedNames :
2015-02-23 02:19:25 +00:00
sickbeard . torrentRssProviderList . remove ( curProvider )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# do the enable/disable
for curProviderStr in provider_str_list :
curProvider , curEnabled = curProviderStr . split ( ' : ' )
curEnabled = config . to_int ( curEnabled )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
curProvObj = [ x for x in sickbeard . providers . sortedProviderList ( ) if
2015-07-13 09:39:20 +00:00
x . get_id ( ) == curProvider and hasattr ( x , ' enabled ' ) ]
2015-02-23 02:19:25 +00:00
if curProvObj :
curProvObj [ 0 ] . enabled = bool ( curEnabled )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
provider_list . append ( curProvider )
if curProvider in newznabProviderDict :
newznabProviderDict [ curProvider ] . enabled = bool ( curEnabled )
elif curProvider in torrentRssProviderDict :
torrentRssProviderDict [ curProvider ] . enabled = bool ( curEnabled )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
# dynamically load provider settings
for curTorrentProvider in [ curProvider for curProvider in sickbeard . providers . sortedProviderList ( ) if
curProvider . providerType == sickbeard . GenericProvider . TORRENT ] :
2014-05-30 06:22:01 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' minseed ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . minseed = int ( str ( kwargs [ curTorrentProvider . get_id ( ) + ' _minseed ' ] ) . strip ( ) )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . minseed = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' minleech ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . minleech = int ( str ( kwargs [ curTorrentProvider . get_id ( ) + ' _minleech ' ] ) . strip ( ) )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . minleech = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' ratio ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . ratio = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _ratio ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . ratio = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' digest ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . digest = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _digest ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . digest = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' hash ' ) :
try :
2015-07-13 09:39:20 +00:00
key = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _hash ' ] ) . strip ( )
2015-02-24 13:37:27 +00:00
if not starify ( key , True ) :
curTorrentProvider . hash = key
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . hash = None
2015-01-19 14:27:48 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' api_key ' ) :
try :
2015-07-13 09:39:20 +00:00
key = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _api_key ' ] ) . strip ( )
2015-02-24 13:37:27 +00:00
if not starify ( key , True ) :
curTorrentProvider . api_key = key
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . api_key = None
2014-05-27 07:44:23 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' username ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . username = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _username ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . username = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' password ' ) :
2014-03-10 05:18:05 +00:00
try :
2015-07-13 09:39:20 +00:00
key = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _password ' ] ) . strip ( )
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( key ) :
curTorrentProvider . password = key
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . password = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' passkey ' ) :
try :
2015-07-13 09:39:20 +00:00
key = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _passkey ' ] ) . strip ( )
2015-02-24 13:37:27 +00:00
if not starify ( key , True ) :
curTorrentProvider . passkey = key
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . passkey = None
2014-06-04 07:21:21 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' confirmed ' ) :
try :
curTorrentProvider . confirmed = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _confirmed ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . confirmed = 0
2014-04-24 05:18:16 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' proxy ' ) :
try :
curTorrentProvider . proxy . enabled = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _proxy ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . proxy . enabled = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider . proxy , ' url ' ) :
2014-03-10 05:18:05 +00:00
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . proxy . url = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _proxy_url ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . proxy . url = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' freeleech ' ) :
try :
curTorrentProvider . freeleech = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _freeleech ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . freeleech = 0
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' search_mode ' ) :
try :
2015-07-13 09:39:20 +00:00
curTorrentProvider . search_mode = str ( kwargs [ curTorrentProvider . get_id ( ) + ' _search_mode ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . search_mode = ' eponly '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' search_fallback ' ) :
try :
curTorrentProvider . search_fallback = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _search_fallback ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . search_fallback = 0 # these exceptions are catching unselected checkboxes
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' enable_recentsearch ' ) :
try :
curTorrentProvider . enable_recentsearch = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _enable_recentsearch ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . enable_recentsearch = 0 # these exceptions are actually catching unselected checkboxes
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curTorrentProvider , ' enable_backlog ' ) :
try :
curTorrentProvider . enable_backlog = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curTorrentProvider . get_id ( ) + ' _enable_backlog ' ] )
2015-02-23 02:19:25 +00:00
except :
curTorrentProvider . enable_backlog = 0 # these exceptions are actually catching unselected checkboxes
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
for curNzbProvider in [ curProvider for curProvider in sickbeard . providers . sortedProviderList ( ) if
curProvider . providerType == sickbeard . GenericProvider . NZB ] :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' api_key ' ) :
try :
2015-07-13 09:39:20 +00:00
key = str ( kwargs [ curNzbProvider . get_id ( ) + ' _api_key ' ] ) . strip ( )
2015-02-24 13:37:27 +00:00
if not starify ( key , True ) :
curNzbProvider . api_key = key
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . api_key = None
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' username ' ) :
try :
2015-07-13 09:39:20 +00:00
curNzbProvider . username = str ( kwargs [ curNzbProvider . get_id ( ) + ' _username ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . username = None
2014-06-29 03:28:00 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' search_mode ' ) :
try :
2015-07-13 09:39:20 +00:00
curNzbProvider . search_mode = str ( kwargs [ curNzbProvider . get_id ( ) + ' _search_mode ' ] ) . strip ( )
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . search_mode = ' eponly '
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' search_fallback ' ) :
try :
curNzbProvider . search_fallback = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curNzbProvider . get_id ( ) + ' _search_fallback ' ] )
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . search_fallback = 0 # these exceptions are actually catching unselected checkboxes
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' enable_recentsearch ' ) :
try :
curNzbProvider . enable_recentsearch = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curNzbProvider . get_id ( ) + ' _enable_recentsearch ' ] )
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . enable_recentsearch = 0 # these exceptions are actually catching unselected checkboxes
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if hasattr ( curNzbProvider , ' enable_backlog ' ) :
try :
curNzbProvider . enable_backlog = config . checkbox_to_value (
2015-07-13 09:39:20 +00:00
kwargs [ curNzbProvider . get_id ( ) + ' _enable_backlog ' ] )
2015-02-23 02:19:25 +00:00
except :
curNzbProvider . enable_backlog = 0 # these exceptions are actually catching unselected checkboxes
2014-03-10 05:18:05 +00:00
2015-06-19 16:47:52 +00:00
sickbeard . NEWZNAB_DATA = ' !!! ' . join ( [ x . config_str ( ) for x in sickbeard . newznabProviderList ] )
2015-02-23 02:19:25 +00:00
sickbeard . PROVIDER_ORDER = provider_list
2014-03-10 05:18:05 +00:00
2015-04-23 10:02:21 +00:00
helpers . clear_unused_providers ( )
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
else :
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /config/providers/ ' )
2014-03-10 05:18:05 +00:00
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
class ConfigNotifications ( Config ) :
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' config_notifications.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveNotifications ( self , use_xbmc = None , xbmc_always_on = None , xbmc_notify_onsnatch = None ,
xbmc_notify_ondownload = None ,
xbmc_notify_onsubtitledownload = None , xbmc_update_onlyfirst = None ,
xbmc_update_library = None , xbmc_update_full = None , xbmc_host = None , xbmc_username = None ,
xbmc_password = None ,
2015-02-25 12:33:40 +00:00
use_kodi = None , kodi_always_on = None , kodi_notify_onsnatch = None , kodi_notify_ondownload = None ,
kodi_notify_onsubtitledownload = None , kodi_update_onlyfirst = None , kodi_update_library = None ,
kodi_update_full = None , kodi_host = None , kodi_username = None , kodi_password = None ,
2015-02-23 02:19:25 +00:00
use_plex = None , plex_notify_onsnatch = None , plex_notify_ondownload = None ,
plex_notify_onsubtitledownload = None , plex_update_library = None ,
plex_server_host = None , plex_host = None , plex_username = None , plex_password = None ,
use_growl = None , growl_notify_onsnatch = None , growl_notify_ondownload = None ,
growl_notify_onsubtitledownload = None , growl_host = None , growl_password = None ,
use_prowl = None , prowl_notify_onsnatch = None , prowl_notify_ondownload = None ,
prowl_notify_onsubtitledownload = None , prowl_api = None , prowl_priority = 0 ,
use_twitter = None , twitter_notify_onsnatch = None , twitter_notify_ondownload = None ,
twitter_notify_onsubtitledownload = None ,
use_boxcar2 = None , boxcar2_notify_onsnatch = None , boxcar2_notify_ondownload = None ,
boxcar2_notify_onsubtitledownload = None , boxcar2_accesstoken = None , boxcar2_sound = None ,
use_pushover = None , pushover_notify_onsnatch = None , pushover_notify_ondownload = None ,
pushover_notify_onsubtitledownload = None , pushover_userkey = None , pushover_apikey = None ,
2015-02-25 11:30:52 +00:00
pushover_priority = None , pushover_device = None , pushover_sound = None , pushover_device_list = None ,
2015-02-23 02:19:25 +00:00
use_libnotify = None , libnotify_notify_onsnatch = None , libnotify_notify_ondownload = None ,
libnotify_notify_onsubtitledownload = None ,
use_nmj = None , nmj_host = None , nmj_database = None , nmj_mount = None , use_synoindex = None ,
use_nmjv2 = None , nmjv2_host = None , nmjv2_dbloc = None , nmjv2_database = None ,
use_trakt = None , trakt_username = None , trakt_password = None , trakt_api = None ,
trakt_remove_watchlist = None , trakt_use_watchlist = None , trakt_method_add = None ,
trakt_start_paused = None , trakt_use_recommended = None , trakt_sync = None ,
trakt_default_indexer = None , trakt_remove_serieslist = None ,
use_synologynotifier = None , synologynotifier_notify_onsnatch = None ,
synologynotifier_notify_ondownload = None , synologynotifier_notify_onsubtitledownload = None ,
use_pytivo = None , pytivo_notify_onsnatch = None , pytivo_notify_ondownload = None ,
pytivo_notify_onsubtitledownload = None , pytivo_update_library = None ,
pytivo_host = None , pytivo_share_name = None , pytivo_tivo_name = None ,
use_nma = None , nma_notify_onsnatch = None , nma_notify_ondownload = None ,
nma_notify_onsubtitledownload = None , nma_api = None , nma_priority = 0 ,
use_pushalot = None , pushalot_notify_onsnatch = None , pushalot_notify_ondownload = None ,
pushalot_notify_onsubtitledownload = None , pushalot_authorizationtoken = None ,
use_pushbullet = None , pushbullet_notify_onsnatch = None , pushbullet_notify_ondownload = None ,
2015-02-24 07:32:45 +00:00
pushbullet_notify_onsubtitledownload = None , pushbullet_access_token = None ,
pushbullet_device_iden = None , pushbullet_device_list = None ,
2015-02-23 02:19:25 +00:00
use_email = None , email_notify_onsnatch = None , email_notify_ondownload = None ,
email_notify_onsubtitledownload = None , email_host = None , email_port = 25 , email_from = None ,
email_tls = None , email_user = None , email_password = None , email_list = None , email_show_list = None ,
email_show = None ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_XBMC = config . checkbox_to_value ( use_xbmc )
sickbeard . XBMC_ALWAYS_ON = config . checkbox_to_value ( xbmc_always_on )
sickbeard . XBMC_NOTIFY_ONSNATCH = config . checkbox_to_value ( xbmc_notify_onsnatch )
sickbeard . XBMC_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( xbmc_notify_ondownload )
sickbeard . XBMC_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( xbmc_notify_onsubtitledownload )
sickbeard . XBMC_UPDATE_LIBRARY = config . checkbox_to_value ( xbmc_update_library )
sickbeard . XBMC_UPDATE_FULL = config . checkbox_to_value ( xbmc_update_full )
sickbeard . XBMC_UPDATE_ONLYFIRST = config . checkbox_to_value ( xbmc_update_onlyfirst )
sickbeard . XBMC_HOST = config . clean_hosts ( xbmc_host )
sickbeard . XBMC_USERNAME = xbmc_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( xbmc_password ) :
sickbeard . XBMC_PASSWORD = xbmc_password
2014-03-10 05:18:05 +00:00
2015-02-25 12:33:40 +00:00
sickbeard . USE_KODI = config . checkbox_to_value ( use_kodi )
sickbeard . KODI_ALWAYS_ON = config . checkbox_to_value ( kodi_always_on )
sickbeard . KODI_NOTIFY_ONSNATCH = config . checkbox_to_value ( kodi_notify_onsnatch )
sickbeard . KODI_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( kodi_notify_ondownload )
sickbeard . KODI_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( kodi_notify_onsubtitledownload )
sickbeard . KODI_UPDATE_LIBRARY = config . checkbox_to_value ( kodi_update_library )
sickbeard . KODI_UPDATE_FULL = config . checkbox_to_value ( kodi_update_full )
sickbeard . KODI_UPDATE_ONLYFIRST = config . checkbox_to_value ( kodi_update_onlyfirst )
sickbeard . KODI_HOST = config . clean_hosts ( kodi_host )
sickbeard . KODI_USERNAME = kodi_username
if set ( ' * ' ) != set ( kodi_password ) :
sickbeard . KODI_PASSWORD = kodi_password
2015-02-23 02:19:25 +00:00
sickbeard . USE_PLEX = config . checkbox_to_value ( use_plex )
sickbeard . PLEX_NOTIFY_ONSNATCH = config . checkbox_to_value ( plex_notify_onsnatch )
sickbeard . PLEX_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( plex_notify_ondownload )
sickbeard . PLEX_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( plex_notify_onsubtitledownload )
sickbeard . PLEX_UPDATE_LIBRARY = config . checkbox_to_value ( plex_update_library )
sickbeard . PLEX_HOST = config . clean_hosts ( plex_host )
sickbeard . PLEX_SERVER_HOST = config . clean_hosts ( plex_server_host )
sickbeard . PLEX_USERNAME = plex_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( plex_password ) :
sickbeard . PLEX_PASSWORD = plex_password
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_GROWL = config . checkbox_to_value ( use_growl )
sickbeard . GROWL_NOTIFY_ONSNATCH = config . checkbox_to_value ( growl_notify_onsnatch )
sickbeard . GROWL_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( growl_notify_ondownload )
sickbeard . GROWL_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( growl_notify_onsubtitledownload )
sickbeard . GROWL_HOST = config . clean_host ( growl_host , default_port = 23053 )
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( growl_password ) :
sickbeard . GROWL_PASSWORD = growl_password
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_PROWL = config . checkbox_to_value ( use_prowl )
sickbeard . PROWL_NOTIFY_ONSNATCH = config . checkbox_to_value ( prowl_notify_onsnatch )
sickbeard . PROWL_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( prowl_notify_ondownload )
sickbeard . PROWL_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( prowl_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = prowl_api . strip ( )
if not starify ( key , True ) :
sickbeard . PROWL_API = key
2015-02-23 02:19:25 +00:00
sickbeard . PROWL_PRIORITY = prowl_priority
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_TWITTER = config . checkbox_to_value ( use_twitter )
sickbeard . TWITTER_NOTIFY_ONSNATCH = config . checkbox_to_value ( twitter_notify_onsnatch )
sickbeard . TWITTER_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( twitter_notify_ondownload )
sickbeard . TWITTER_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( twitter_notify_onsubtitledownload )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_BOXCAR2 = config . checkbox_to_value ( use_boxcar2 )
sickbeard . BOXCAR2_NOTIFY_ONSNATCH = config . checkbox_to_value ( boxcar2_notify_onsnatch )
sickbeard . BOXCAR2_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( boxcar2_notify_ondownload )
sickbeard . BOXCAR2_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( boxcar2_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = boxcar2_accesstoken . strip ( )
if not starify ( key , True ) :
sickbeard . BOXCAR2_ACCESSTOKEN = key
2015-02-23 02:19:25 +00:00
sickbeard . BOXCAR2_SOUND = boxcar2_sound
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_PUSHOVER = config . checkbox_to_value ( use_pushover )
sickbeard . PUSHOVER_NOTIFY_ONSNATCH = config . checkbox_to_value ( pushover_notify_onsnatch )
sickbeard . PUSHOVER_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( pushover_notify_ondownload )
sickbeard . PUSHOVER_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( pushover_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = pushover_userkey . strip ( )
if not starify ( key , True ) :
sickbeard . PUSHOVER_USERKEY = key
key = pushover_apikey . strip ( )
if not starify ( key , True ) :
sickbeard . PUSHOVER_APIKEY = key
2015-02-25 11:30:52 +00:00
sickbeard . PUSHOVER_PRIORITY = pushover_priority
sickbeard . PUSHOVER_DEVICE = pushover_device
sickbeard . PUSHOVER_SOUND = pushover_sound
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_LIBNOTIFY = config . checkbox_to_value ( use_libnotify )
sickbeard . LIBNOTIFY_NOTIFY_ONSNATCH = config . checkbox_to_value ( libnotify_notify_onsnatch )
sickbeard . LIBNOTIFY_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( libnotify_notify_ondownload )
sickbeard . LIBNOTIFY_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( libnotify_notify_onsubtitledownload )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_NMJ = config . checkbox_to_value ( use_nmj )
sickbeard . NMJ_HOST = config . clean_host ( nmj_host )
sickbeard . NMJ_DATABASE = nmj_database
sickbeard . NMJ_MOUNT = nmj_mount
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_NMJv2 = config . checkbox_to_value ( use_nmjv2 )
sickbeard . NMJv2_HOST = config . clean_host ( nmjv2_host )
sickbeard . NMJv2_DATABASE = nmjv2_database
sickbeard . NMJv2_DBLOC = nmjv2_dbloc
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_SYNOINDEX = config . checkbox_to_value ( use_synoindex )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_SYNOLOGYNOTIFIER = config . checkbox_to_value ( use_synologynotifier )
sickbeard . SYNOLOGYNOTIFIER_NOTIFY_ONSNATCH = config . checkbox_to_value ( synologynotifier_notify_onsnatch )
sickbeard . SYNOLOGYNOTIFIER_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( synologynotifier_notify_ondownload )
sickbeard . SYNOLOGYNOTIFIER_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value (
synologynotifier_notify_onsubtitledownload )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_TRAKT = config . checkbox_to_value ( use_trakt )
sickbeard . TRAKT_USERNAME = trakt_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( trakt_password ) :
sickbeard . TRAKT_PASSWORD = trakt_password
key = trakt_api . strip ( )
if not starify ( key , True ) :
sickbeard . TRAKT_API = key
2015-02-23 02:19:25 +00:00
sickbeard . TRAKT_REMOVE_WATCHLIST = config . checkbox_to_value ( trakt_remove_watchlist )
sickbeard . TRAKT_REMOVE_SERIESLIST = config . checkbox_to_value ( trakt_remove_serieslist )
sickbeard . TRAKT_USE_WATCHLIST = config . checkbox_to_value ( trakt_use_watchlist )
sickbeard . TRAKT_METHOD_ADD = int ( trakt_method_add )
sickbeard . TRAKT_START_PAUSED = config . checkbox_to_value ( trakt_start_paused )
sickbeard . TRAKT_USE_RECOMMENDED = config . checkbox_to_value ( trakt_use_recommended )
sickbeard . TRAKT_SYNC = config . checkbox_to_value ( trakt_sync )
sickbeard . TRAKT_DEFAULT_INDEXER = int ( trakt_default_indexer )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if sickbeard . USE_TRAKT :
sickbeard . traktCheckerScheduler . silent = False
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
sickbeard . traktCheckerScheduler . silent = True
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_EMAIL = config . checkbox_to_value ( use_email )
sickbeard . EMAIL_NOTIFY_ONSNATCH = config . checkbox_to_value ( email_notify_onsnatch )
sickbeard . EMAIL_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( email_notify_ondownload )
sickbeard . EMAIL_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( email_notify_onsubtitledownload )
sickbeard . EMAIL_HOST = config . clean_host ( email_host )
sickbeard . EMAIL_PORT = config . to_int ( email_port , default = 25 )
sickbeard . EMAIL_FROM = email_from
sickbeard . EMAIL_TLS = config . checkbox_to_value ( email_tls )
sickbeard . EMAIL_USER = email_user
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( email_password ) :
sickbeard . EMAIL_PASSWORD = email_password
2015-02-23 02:19:25 +00:00
sickbeard . EMAIL_LIST = email_list
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_PYTIVO = config . checkbox_to_value ( use_pytivo )
sickbeard . PYTIVO_NOTIFY_ONSNATCH = config . checkbox_to_value ( pytivo_notify_onsnatch )
sickbeard . PYTIVO_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( pytivo_notify_ondownload )
sickbeard . PYTIVO_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( pytivo_notify_onsubtitledownload )
sickbeard . PYTIVO_UPDATE_LIBRARY = config . checkbox_to_value ( pytivo_update_library )
sickbeard . PYTIVO_HOST = config . clean_host ( pytivo_host )
sickbeard . PYTIVO_SHARE_NAME = pytivo_share_name
sickbeard . PYTIVO_TIVO_NAME = pytivo_tivo_name
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_NMA = config . checkbox_to_value ( use_nma )
sickbeard . NMA_NOTIFY_ONSNATCH = config . checkbox_to_value ( nma_notify_onsnatch )
sickbeard . NMA_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( nma_notify_ondownload )
sickbeard . NMA_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( nma_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = nma_api . strip ( )
if not starify ( key , True ) :
sickbeard . NMA_API = key
2015-02-23 02:19:25 +00:00
sickbeard . NMA_PRIORITY = nma_priority
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_PUSHALOT = config . checkbox_to_value ( use_pushalot )
sickbeard . PUSHALOT_NOTIFY_ONSNATCH = config . checkbox_to_value ( pushalot_notify_onsnatch )
sickbeard . PUSHALOT_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( pushalot_notify_ondownload )
sickbeard . PUSHALOT_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( pushalot_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = pushalot_authorizationtoken . strip ( )
if not starify ( key , True ) :
sickbeard . PUSHALOT_AUTHORIZATIONTOKEN = key
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_PUSHBULLET = config . checkbox_to_value ( use_pushbullet )
sickbeard . PUSHBULLET_NOTIFY_ONSNATCH = config . checkbox_to_value ( pushbullet_notify_onsnatch )
sickbeard . PUSHBULLET_NOTIFY_ONDOWNLOAD = config . checkbox_to_value ( pushbullet_notify_ondownload )
sickbeard . PUSHBULLET_NOTIFY_ONSUBTITLEDOWNLOAD = config . checkbox_to_value ( pushbullet_notify_onsubtitledownload )
2015-02-24 13:37:27 +00:00
key = pushbullet_access_token . strip ( )
if not starify ( key , True ) :
sickbeard . PUSHBULLET_ACCESS_TOKEN = key
2015-02-24 07:32:45 +00:00
sickbeard . PUSHBULLET_DEVICE_IDEN = pushbullet_device_iden
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2014-06-11 08:34:28 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /config/notifications/ ' )
2014-03-10 05:18:05 +00:00
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
class ConfigSubtitles ( Config ) :
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' config_subtitles.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-05-30 05:48:02 +00:00
2015-02-23 02:19:25 +00:00
def saveSubtitles ( self , use_subtitles = None , subtitles_plugins = None , subtitles_languages = None , subtitles_dir = None ,
service_order = None , subtitles_history = None , subtitles_finder_frequency = None ) :
results = [ ]
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
if subtitles_finder_frequency == ' ' or subtitles_finder_frequency is None :
subtitles_finder_frequency = 1
2014-05-30 05:48:02 +00:00
2015-02-23 02:19:25 +00:00
if use_subtitles == ' on ' and not sickbeard . subtitlesFinderScheduler . isAlive ( ) :
sickbeard . subtitlesFinderScheduler . silent = False
sickbeard . subtitlesFinderScheduler . start ( )
else :
sickbeard . subtitlesFinderScheduler . stop . set ( )
sickbeard . subtitlesFinderScheduler . silent = True
logger . log ( u ' Waiting for the SUBTITLESFINDER thread to exit ' )
try :
sickbeard . subtitlesFinderScheduler . join ( 5 )
except :
pass
2014-05-30 05:48:02 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_SUBTITLES = config . checkbox_to_value ( use_subtitles )
sickbeard . SUBTITLES_LANGUAGES = [ lang . alpha2 for lang in subtitles . isValidLanguage (
subtitles_languages . replace ( ' ' , ' ' ) . split ( ' , ' ) ) ] if subtitles_languages != ' ' else ' '
sickbeard . SUBTITLES_DIR = subtitles_dir
sickbeard . SUBTITLES_HISTORY = config . checkbox_to_value ( subtitles_history )
sickbeard . SUBTITLES_FINDER_FREQUENCY = config . to_int ( subtitles_finder_frequency , default = 1 )
2014-05-30 05:48:02 +00:00
2015-02-23 02:19:25 +00:00
# Subtitles services
services_str_list = service_order . split ( )
subtitles_services_list = [ ]
subtitles_services_enabled = [ ]
for curServiceStr in services_str_list :
curService , curEnabled = curServiceStr . split ( ' : ' )
subtitles_services_list . append ( curService )
subtitles_services_enabled . append ( int ( curEnabled ) )
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . SUBTITLES_SERVICES_LIST = subtitles_services_list
sickbeard . SUBTITLES_SERVICES_ENABLED = subtitles_services_enabled
2014-03-25 05:57:24 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2014-05-03 10:09:46 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
2014-03-10 05:18:05 +00:00
else :
2015-02-23 02:19:25 +00:00
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
self . redirect ( ' /config/subtitles/ ' )
2014-03-25 05:57:24 +00:00
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
class ConfigAnime ( Config ) :
def index ( self , * args , * * kwargs ) :
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' config_anime.tmpl ' )
t . submenu = self . ConfigMenu
return t . respond ( )
2014-03-10 05:18:05 +00:00
2015-02-23 02:19:25 +00:00
def saveAnime ( self , use_anidb = None , anidb_username = None , anidb_password = None , anidb_use_mylist = None ,
split_home = None , anime_treat_as_hdtv = None ) :
2014-11-06 04:55:43 +00:00
2015-02-23 02:19:25 +00:00
results = [ ]
2015-01-25 11:58:49 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . USE_ANIDB = config . checkbox_to_value ( use_anidb )
sickbeard . ANIDB_USERNAME = anidb_username
2015-02-24 13:37:27 +00:00
if set ( ' * ' ) != set ( anidb_password ) :
sickbeard . ANIDB_PASSWORD = anidb_password
2015-02-23 02:19:25 +00:00
sickbeard . ANIDB_USE_MYLIST = config . checkbox_to_value ( anidb_use_mylist )
sickbeard . ANIME_SPLIT_HOME = config . checkbox_to_value ( split_home )
sickbeard . ANIME_TREAT_AS_HDTV = config . checkbox_to_value ( anime_treat_as_hdtv )
2015-02-16 03:17:56 +00:00
2015-02-23 02:19:25 +00:00
sickbeard . save_config ( )
2015-01-25 11:58:49 +00:00
2015-02-23 02:19:25 +00:00
if len ( results ) > 0 :
for x in results :
logger . log ( x , logger . ERROR )
ui . notifications . error ( ' Error(s) Saving Configuration ' ,
' <br /> \n ' . join ( results ) )
else :
ui . notifications . message ( ' Configuration Saved ' , ek . ek ( os . path . join , sickbeard . CONFIG_FILE ) )
self . redirect ( ' /config/anime/ ' )
2015-01-25 11:58:49 +00:00
2014-07-01 08:49:12 +00:00
2014-06-19 08:37:44 +00:00
class UI ( MainHandler ) :
2014-06-29 10:05:33 +00:00
def add_message ( self ) :
2014-03-10 05:18:05 +00:00
ui . notifications . message ( ' Test 1 ' , ' This is test number 1 ' )
ui . notifications . error ( ' Test 2 ' , ' This is test number 2 ' )
2015-02-23 02:19:25 +00:00
return ' ok '
2014-06-11 08:34:28 +00:00
2014-06-29 10:05:33 +00:00
def get_messages ( self ) :
2014-03-10 05:18:05 +00:00
messages = { }
cur_notification_num = 1
2014-06-30 03:58:34 +00:00
for cur_notification in ui . notifications . get_notifications ( self . request . remote_ip ) :
2014-03-10 05:18:05 +00:00
messages [ ' notification- ' + str ( cur_notification_num ) ] = { ' title ' : cur_notification . title ,
2014-07-01 08:49:12 +00:00
' message ' : cur_notification . message ,
' type ' : cur_notification . type }
2014-03-10 05:18:05 +00:00
cur_notification_num + = 1
2014-08-08 07:18:03 +00:00
return json . dumps ( messages )
2015-02-23 02:19:25 +00:00
class ErrorLogs ( MainHandler ) :
@staticmethod
def ErrorLogsMenu ( ) :
return [ { ' title ' : ' Clear Errors ' , ' path ' : ' errorlogs/clearerrors/ ' } , ]
def index ( self , * args , * * kwargs ) :
t = PageTemplate ( headers = self . request . headers , file = ' errorlogs.tmpl ' )
t . submenu = self . ErrorLogsMenu
return t . respond ( )
def clearerrors ( self , * args , * * kwargs ) :
classes . ErrorViewer . clear ( )
self . redirect ( ' /errorlogs/ ' )
def viewlog ( self , minLevel = logger . MESSAGE , maxLines = 500 ) :
t = PageTemplate ( headers = self . request . headers , file = ' viewlogs.tmpl ' )
t . submenu = self . ErrorLogsMenu
minLevel = int ( minLevel )
data = [ ]
if os . path . isfile ( logger . sb_log_instance . log_file_path ) :
with ek . ek ( open , logger . sb_log_instance . log_file_path ) as f :
data = f . readlines ( )
regex = ' ^( \ d \ d \ d \ d) \ -( \ d \ d) \ -( \ d \ d) \ s*( \ d \ d) \ :( \ d \ d):( \ d \ d) \ s*([A-Z]+) \ s*(.+?) \ s* \ : \ : \ s*(.*)$ '
finalData = [ ]
numLines = 0
lastLine = False
numToShow = min ( maxLines , len ( data ) )
for x in reversed ( data ) :
x = x . decode ( ' utf-8 ' , ' replace ' )
match = re . match ( regex , x )
if match :
level = match . group ( 7 )
if level not in logger . reverseNames :
lastLine = False
continue
if logger . reverseNames [ level ] > = minLevel :
lastLine = True
finalData . append ( x )
else :
lastLine = False
continue
elif lastLine :
finalData . append ( ' AA ' + x )
numLines + = 1
if numLines > = numToShow :
break
result = ' ' . join ( finalData )
t . logLines = result
t . minLevel = minLevel
return t . respond ( )
class WebFileBrowser ( MainHandler ) :
def index ( self , path = ' ' , includeFiles = False , * args , * * kwargs ) :
self . set_header ( ' Content-Type ' , ' application/json ' )
return json . dumps ( foldersAtPath ( path , True , bool ( int ( includeFiles ) ) ) )
def complete ( self , term , includeFiles = 0 ) :
self . set_header ( ' Content-Type ' , ' application/json ' )
paths = [ entry [ ' path ' ] for entry in foldersAtPath ( os . path . dirname ( term ) , includeFiles = bool ( int ( includeFiles ) ) ) if ' path ' in entry ]
return json . dumps ( paths )
class ApiBuilder ( MainHandler ) :
def index ( self ) :
""" expose the api-builder template """
t = PageTemplate ( headers = self . request . headers , file = ' apiBuilder.tmpl ' )
def titler ( x ) :
return ( remove_article ( x ) , x ) [ not x or sickbeard . SORT_ARTICLE ]
t . sortedShowList = sorted ( sickbeard . showList , lambda x , y : cmp ( titler ( x . name ) , titler ( y . name ) ) )
seasonSQLResults = { }
episodeSQLResults = { }
myDB = db . DBConnection ( row_type = ' dict ' )
for curShow in t . sortedShowList :
seasonSQLResults [ curShow . indexerid ] = myDB . select (
' SELECT DISTINCT season FROM tv_episodes WHERE showid = ? ORDER BY season DESC ' , [ curShow . indexerid ] )
for curShow in t . sortedShowList :
episodeSQLResults [ curShow . indexerid ] = myDB . select (
' SELECT DISTINCT season,episode FROM tv_episodes WHERE showid = ? ORDER BY season DESC, episode DESC ' ,
[ curShow . indexerid ] )
t . seasonSQLResults = seasonSQLResults
t . episodeSQLResults = episodeSQLResults
if len ( sickbeard . API_KEY ) == 32 :
t . apikey = sickbeard . API_KEY
else :
t . apikey = ' api key not generated '
2015-01-13 14:18:52 +00:00
return t . respond ( )
class Cache ( MainHandler ) :
def index ( self ) :
myDB = db . DBConnection ( ' cache.db ' )
2015-03-29 15:46:46 +00:00
sql_results = myDB . select ( ' SELECT * FROM provider_cache ' )
if not sql_results :
sql_results = [ ]
2015-01-13 14:18:52 +00:00
t = PageTemplate ( headers = self . request . headers , file = ' cache.tmpl ' )
2015-03-29 15:46:46 +00:00
t . cacheResults = sql_results
2015-01-13 14:18:52 +00:00
2015-03-01 04:09:49 +00:00
return t . respond ( )