def saveAddShowDefaults(defaultStatus, anyQualities, bestQualities, defaultSeasonFolders, subtitles=False,
anime=False, scene=False, defaultStatusAfter=WANTED):
if anyQualities:
anyQualities = anyQualities.split(',')
else:
anyQualities = []
if bestQualities:
bestQualities = bestQualities.split(',')
else:
bestQualities = []
newQuality = Quality.combineQualities([int(quality) for quality in anyQualities], [int(quality) for quality in bestQualities])
sickbeard.STATUS_DEFAULT = int(defaultStatus)
sickbeard.STATUS_DEFAULT_AFTER = int(defaultStatusAfter)
sickbeard.QUALITY_DEFAULT = int(newQuality)
sickbeard.SEASON_FOLDERS_DEFAULT = config.checkbox_to_value(defaultSeasonFolders)
sickbeard.SUBTITLES_DEFAULT = config.checkbox_to_value(subtitles)
sickbeard.ANIME_DEFAULT = config.checkbox_to_value(anime)
sickbeard.SCENE_DEFAULT = config.checkbox_to_value(scene)
sickbeard.save_config()
ui.notifications.message(_('Saved Defaults'), _('Your "add show" defaults have been set to your current selections.'))
def saveAddShowDefaults(defaultStatus, anyQualities, bestQualities, defaultFlattenFolders, subtitles=False,
anime=False, scene=False, defaultStatusAfter=WANTED):
allowed_qualities = anyQualities.split(',') if anyQualities else []
preferred_qualities = bestQualities.split(',') if bestQualities else []
new_quality = Quality.combineQualities([int(quality) for quality in allowed_qualities], [int(quality) for quality in preferred_qualities])
sickbeard.STATUS_DEFAULT = int(defaultStatus)
sickbeard.STATUS_DEFAULT_AFTER = int(defaultStatusAfter)
sickbeard.QUALITY_DEFAULT = int(new_quality)
sickbeard.FLATTEN_FOLDERS_DEFAULT = config.checkbox_to_value(defaultFlattenFolders)
sickbeard.SUBTITLES_DEFAULT = config.checkbox_to_value(subtitles)
sickbeard.ANIME_DEFAULT = config.checkbox_to_value(anime)
sickbeard.SCENE_DEFAULT = config.checkbox_to_value(scene)
sickbeard.save_config()
def saveAddShowDefaults(self, defaultSeasonFolders, defaultStatus, anyQualities, bestQualities):
if anyQualities:
anyQualities = anyQualities.split(',')
else:
anyQualities = []
if bestQualities:
bestQualities = bestQualities.split(',')
else:
bestQualities = []
newQuality = Quality.combineQualities(map(int, anyQualities), map(int, bestQualities))
sickbeard.STATUS_DEFAULT = int(defaultStatus)
sickbeard.QUALITY_DEFAULT = int(newQuality)
if defaultSeasonFolders == "true":
defaultSeasonFolders = 1
else:
defaultSeasonFolders = 0
sickbeard.SEASON_FOLDERS_DEFAULT = int(defaultSeasonFolders)
def render_body(context, **pageargs):
__M_caller = context.caller_stack._push_frame()
try:
__M_locals = __M_dict_builtin(pageargs=pageargs)
show = context.get('show', UNDEFINED)
int = context.get('int', UNDEFINED)
len = context.get('len', UNDEFINED)
filter = context.get('filter', UNDEFINED)
sorted = context.get('sorted', UNDEFINED)
_ = context.get('_', UNDEFINED)
__M_writer = context.writer()
__M_writer(u'\n\n')
if show is not UNDEFINED:
__quality = int(show.quality)
else:
__quality = int(sickbeard.QUALITY_DEFAULT)
anyQualities, bestQualities = Quality.splitQuality(__quality)
overall_quality = Quality.combineQualities(anyQualities, bestQualities)
selected = None
__M_locals_builtin_stored = __M_locals_builtin()
__M_locals.update(
__M_dict_builtin([
(__M_key, __M_locals_builtin_stored[__M_key]) for __M_key in [
'__quality', 'anyQualities', 'overall_quality', 'selected',
'bestQualities'
] if __M_key in __M_locals_builtin_stored
]))
__M_writer(
u'\n\n<div class="row">\n <div class="col-md-12">\n <select id="qualityPreset" name="quality_preset" class="form-control input-sm input100" title="qualityPreset">\n <option value="0">Custom</option>\n'
)
for curPreset in qualityPresets:
__M_writer(u' <option value="')
__M_writer(unicode(curPreset))
__M_writer(u'" ')
__M_writer(
unicode(
('', 'selected="selected"')[curPreset == overall_quality]))
__M_writer(u' ')
__M_writer(
unicode(('', 'style="padding-left: 15px;"'
)[qualityPresetStrings[curPreset].endswith("0p")]))
__M_writer(u'>')
__M_writer(unicode(qualityPresetStrings[curPreset]))
__M_writer(u'</option>\n')
__M_writer(
u' </select>\n </div>\n</div>\n<div class="row">\n <div class="col-md-12">\n <div id="customQualityWrapper">\n <div id="customQuality" style="padding-left: 0;">\n '
)
__M_writer(
unicode(
_('<p><b><u>Preferred</u></b> qualities will replace those in <b><u>allowed</u></b>, even if they are lower.</p>'
)))
__M_writer(
u'\n\n <div style="padding-right: 40px; text-align: left; float: left;">\n <h5>'
)
__M_writer(unicode(_('Allowed')))
__M_writer(u'</h5>\n ')
anyQualityList = filter(lambda x: x > Quality.NONE,
Quality.qualityStrings)
__M_locals_builtin_stored = __M_locals_builtin()
__M_locals.update(
__M_dict_builtin([(__M_key, __M_locals_builtin_stored[__M_key])
for __M_key in ['anyQualityList']
if __M_key in __M_locals_builtin_stored]))
__M_writer(
u'\n <select id="anyQualities" name="anyQualities" multiple="multiple" size="'
)
__M_writer(unicode(len(anyQualityList)))
__M_writer(
u'" class="form-control form-control-inline input-sm" title="anyQualities">\n'
)
for curQuality in sorted(anyQualityList):
__M_writer(u' <option value="')
__M_writer(unicode(curQuality))
__M_writer(u'" ')
__M_writer(
unicode(
('', 'selected="selected"')[curQuality in anyQualities]))
__M_writer(u'>')
__M_writer(unicode(Quality.qualityStrings[curQuality]))
__M_writer(u'</option>\n')
__M_writer(
u' </select>\n </div>\n\n <div style="text-align: left; float: left;">\n <h5>'
)
__M_writer(unicode(_('Preferred')))
__M_writer(u'</h5>\n ')
bestQualityList = filter(lambda x: Quality.SDTV <= x < Quality.UNKNOWN,
Quality.qualityStrings)
__M_locals_builtin_stored = __M_locals_builtin()
__M_locals.update(
__M_dict_builtin([(__M_key, __M_locals_builtin_stored[__M_key])
for __M_key in ['bestQualityList']
if __M_key in __M_locals_builtin_stored]))
__M_writer(
u'\n <select id="bestQualities" name="bestQualities" multiple="multiple" size="'
)
__M_writer(unicode(len(bestQualityList)))
__M_writer(
u'" class="form-control form-control-inline input-sm" title="bestQualities">\n'
)
for curQuality in sorted(bestQualityList):
__M_writer(u' <option value="')
__M_writer(unicode(curQuality))
__M_writer(u'" ')
__M_writer(
unicode(
('', 'selected="selected"')[curQuality in bestQualities]))
__M_writer(u'>')
__M_writer(unicode(Quality.qualityStrings[curQuality]))
__M_writer(u'</option>\n')
__M_writer(
u' </select>\n </div>\n </div>\n </div>\n </div>\n</div>\n'
)
return ''
finally:
context.caller_stack._pop_frame()
def addNewShow(self,
whichSeries=None,
indexerLang=None,
rootDir=None,
defaultStatus=None,
quality_preset=None,
anyQualities=None,
bestQualities=None,
season_folders=None,
subtitles=None,
subtitles_sr_metadata=None,
fullShowPath=None,
other_shows=None,
skipShow=None,
providedIndexer=None,
anime=None,
scene=None,
blacklist=None,
whitelist=None,
defaultStatusAfter=None):
"""
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.
"""
if not indexerLang:
indexerLang = sickbeard.INDEXER_DEFAULT_LANGUAGE
# grab our list of other dirs if given
if not other_shows:
other_shows = []
elif not isinstance(other_shows, list):
other_shows = [other_shows]
def finishAddShow():
# if there are no extra shows then go home
if not other_shows:
return self.redirect('/home/')
# peel off the next one
next_show_dir = other_shows[0]
rest_of_show_dirs = other_shows[1:]
# go to add the next show
return self.newShow(next_show_dir, rest_of_show_dirs)
# if we're skipping then behave accordingly
if skipShow:
return finishAddShow()
# sanity check on our inputs
if (not rootDir and not fullShowPath) or not whichSeries:
return _(
"Missing params, no Indexer ID or folder: {show_to_add} and {root_dir}/{show_path}"
).format(show_to_add=whichSeries,
root_dir=rootDir,
show_path=fullShowPath)
# 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: {0}"
.format((repr(series_pieces))), logger.ERROR)
ui.notifications.error(
_("Unknown error. Unable to add show due to problem with show selection."
))
return self.redirect('/addShows/existingShows/')
indexer = int(series_pieces[1])
indexer_id = int(series_pieces[3])
# Show name was sent in UTF-8 in the form
show_name = xhtml_unescape(series_pieces[4]).decode('utf-8')
else:
# if no indexer was provided use the default indexer set in General settings
if not providedIndexer:
providedIndexer = sickbeard.INDEXER_DEFAULT
indexer = int(providedIndexer)
indexer_id = int(whichSeries)
show_name = ek(os.path.basename,
ek(os.path.normpath, xhtml_unescape(fullShowPath)))
# 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(os.path.normpath, xhtml_unescape(fullShowPath))
extra_check_dir = show_dir
else:
folder_name = show_name
s = sickchill.indexer.series_by_id(indexerid=indexer_id,
indexer=indexer,
language=indexerLang)
if sickbeard.ADD_SHOWS_WITH_YEAR and s.firstAired:
try:
year = '({0})'.format(
dateutil.parser.parse(s.firstAired).year)
if year not in folder_name:
folder_name = '{0} {1}'.format(s.seriesName, year)
except (TypeError, ValueError):
logger.log(
_('Could not append the show year folder for the show: {0}'
).format(folder_name))
show_dir = ek(os.path.join, rootDir,
sanitize_filename(xhtml_unescape(folder_name)))
extra_check_dir = ek(os.path.join, rootDir,
sanitize_filename(xhtml_unescape(show_name)))
# blanket policy - if the dir exists you should have used "add existing show" numbnuts
if ek(os.path.isdir, show_dir) or ek(
os.path.isdir, extra_check_dir) and not fullShowPath:
ui.notifications.error(
_("Unable to add show"),
_("Folder {show_dir} exists already").format(
show_dir=show_dir))
return self.redirect('/addShows/existingShows/')
# don't create show dir if config says not to
if sickbeard.ADD_SHOWS_WO_DIR:
logger.log("Skipping initial creation of " + show_dir +
" due to config.ini setting")
else:
dir_exists = helpers.makeDir(show_dir)
if not dir_exists:
logger.log(
"Unable to create the folder " + show_dir +
", can't add the show", logger.ERROR)
ui.notifications.error(
_("Unable to add show"),
_("Unable to create the folder {show_dir}, can't add the show"
).format(show_dir=show_dir))
# Don't redirect to default page because user wants to see the new show
return self.redirect("/home/")
else:
helpers.chmodAsParent(show_dir)
# prepare the inputs for passing along
scene = config.checkbox_to_value(scene)
anime = config.checkbox_to_value(anime)
season_folders = config.checkbox_to_value(season_folders)
subtitles = config.checkbox_to_value(subtitles)
subtitles_sr_metadata = config.checkbox_to_value(subtitles_sr_metadata)
if whitelist:
whitelist = short_group_names(whitelist)
if blacklist:
blacklist = short_group_names(blacklist)
if not anyQualities:
anyQualities = []
if not bestQualities or try_int(quality_preset, None):
bestQualities = []
if not isinstance(anyQualities, list):
anyQualities = [anyQualities]
if not isinstance(bestQualities, list):
bestQualities = [bestQualities]
newQuality = Quality.combineQualities([int(q) for q in anyQualities],
[int(q) for q in bestQualities])
# add the show
sickbeard.showQueueScheduler.action.add_show(
indexer,
indexer_id,
showDir=show_dir,
default_status=int(defaultStatus),
quality=newQuality,
season_folders=season_folders,
lang=indexerLang,
subtitles=subtitles,
subtitles_sr_metadata=subtitles_sr_metadata,
anime=anime,
scene=scene,
paused=None,
blacklist=blacklist,
whitelist=whitelist,
default_status_after=int(defaultStatusAfter),
root_dir=rootDir)
ui.notifications.message(
_('Show added'),
_('Adding the specified show into {show_dir}').format(
show_dir=show_dir))
return finishAddShow()
def addShowByID(self,
indexer_id,
show_name,
indexer="TVDB",
which_series=None,
indexer_lang=None,
root_dir=None,
default_status=None,
quality_preset=None,
any_qualities=None,
best_qualities=None,
season_folders=None,
subtitles=None,
full_show_path=None,
other_shows=None,
skip_show=None,
provided_indexer=None,
anime=None,
scene=None,
blacklist=None,
whitelist=None,
default_status_after=None,
default_season_folders=None,
configure_show_options=None):
if indexer != "TVDB":
indexer_id = helpers.tvdbid_from_remote_id(indexer_id,
indexer.upper())
if not indexer_id:
logger.log(
"Unable to to find tvdb ID to add {0}".format(show_name))
ui.notifications.error(
"Unable to add {0}".format(show_name),
"Could not add {0}. We were unable to locate the tvdb id at this time."
.format(show_name))
return
indexer_id = try_int(indexer_id)
if indexer_id <= 0 or Show.find(sickbeard.showList, indexer_id):
return
# Sanitize the parameter anyQualities and bestQualities. As these would normally be passed as lists
any_qualities = any_qualities.split(',') if any_qualities else []
best_qualities = best_qualities.split(',') if best_qualities else []
# If configure_show_options is enabled let's use the provided settings
if config.checkbox_to_value(configure_show_options):
# prepare the inputs for passing along
scene = config.checkbox_to_value(scene)
anime = config.checkbox_to_value(anime)
season_folders = config.checkbox_to_value(season_folders)
subtitles = config.checkbox_to_value(subtitles)
if whitelist:
whitelist = short_group_names(whitelist)
if blacklist:
blacklist = short_group_names(blacklist)
if not any_qualities:
any_qualities = []
if not best_qualities or try_int(quality_preset, None):
best_qualities = []
if not isinstance(any_qualities, list):
any_qualities = [any_qualities]
if not isinstance(best_qualities, list):
best_qualities = [best_qualities]
quality = Quality.combineQualities(
[int(q) for q in any_qualities],
[int(q) for q in best_qualities])
location = root_dir
else:
default_status = sickbeard.STATUS_DEFAULT
quality = sickbeard.QUALITY_DEFAULT
season_folders = sickbeard.SEASON_FOLDERS_DEFAULT
subtitles = sickbeard.SUBTITLES_DEFAULT
anime = sickbeard.ANIME_DEFAULT
scene = sickbeard.SCENE_DEFAULT
default_status_after = sickbeard.STATUS_DEFAULT_AFTER
if sickbeard.ROOT_DIRS:
root_dirs = sickbeard.ROOT_DIRS.split('|')
location = root_dirs[int(root_dirs[0]) + 1]
else:
location = None
if not location:
logger.log(
"There was an error creating the show, no root directory setting found"
)
return _("No root directories setup, please go back and add one.")
show_name = sickchill.indexer[1].get_series_by_id(
indexer_id, indexer_lang).seriesName
show_dir = None
if not show_name:
ui.notifications.error(_('Unable to add show'))
return self.redirect('/home/')
# add the show
sickbeard.showQueueScheduler.action.add_show(
indexer=1,
indexer_id=indexer_id,
showDir=show_dir,
default_status=default_status,
quality=quality,
season_folders=season_folders,
lang=indexer_lang,
subtitles=subtitles,
subtitles_sr_metadata=None,
anime=anime,
scene=scene,
paused=None,
blacklist=blacklist,
whitelist=whitelist,
default_status_after=default_status_after,
root_dir=location)
ui.notifications.message(
_('Show added'),
_('Adding the specified show {show_name}').format(
show_name=show_name))
# done adding show
return self.redirect('/home/')
import datetime
import unittest
import sickbeard
import test_lib as test
from sickbeard import db
from sickbeard.common import Quality, UNAIRED, SKIPPED, WANTED, WantedQualities, statusStrings
from sickbeard.show_queue import QueueItemAdd
from sickbeard.tv import TVEpisode, TVShow
# noinspection DuplicatedCode
wanted_tests = [
dict(name='Start and End',
show=dict(indexer=1,
indexerid=1,
quality=Quality.combineQualities([Quality.SDTV], [])),
episodes=[
dict(season=1,
episode=1,
status=SKIPPED,
quality=Quality.NONE,
airdate=datetime.date(2019, 1, 1)),
dict(season=1,
episode=2,
status=SKIPPED,
quality=Quality.NONE,
airdate=datetime.date(2019, 1, 1)),
dict(season=1,
episode=3,
status=UNAIRED,
quality=Quality.NONE,
def addNewShow(self, whichSeries=None, indexerLang=None, rootDir=None, defaultStatus=None,
quality_preset=None, anyQualities=None, bestQualities=None, flatten_folders=None, subtitles=None,
fullShowPath=None, other_shows=None, skipShow=None, providedIndexer=None, anime=None,
scene=None, blacklist=None, whitelist=None, defaultStatusAfter=None):
"""
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.
"""
provided_indexer = providedIndexer
allowed_qualities = anyQualities
preferred_qualities = bestQualities
indexer_lang = sickbeard.INDEXER_DEFAULT_LANGUAGE if not indexerLang else indexerLang
# grab our list of other dirs if given
if not other_shows:
other_shows = []
elif not isinstance(other_shows, list):
other_shows = [other_shows]
def finishAddShow():
# if there are no extra shows then go home
if not other_shows:
return self.redirect('/home/')
# peel off the next one
next_show_dir = other_shows[0]
rest_of_show_dirs = other_shows[1:]
# go to add the next show
return self.newShow(next_show_dir, rest_of_show_dirs)
# if we're skipping then behave accordingly
if skipShow:
return finishAddShow()
# sanity check on our inputs
if (not rootDir and not fullShowPath) or not whichSeries:
return 'Missing params, no Indexer ID or folder:{series!r} and {root!r}/{path!r}'.format(
series=whichSeries, root=rootDir, path=fullShowPath)
# 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(u'Unable to add show due to show selection. Not anough arguments: %s' % (repr(series_pieces)),
logger.ERROR)
ui.notifications.error('Unknown error. Unable to add show due to problem with show selection.')
return self.redirect('/addShows/existingShows/')
indexer = int(series_pieces[1])
indexer_id = int(series_pieces[3])
# Show name was sent in UTF-8 in the form
show_name = series_pieces[4].decode('utf-8')
else:
# if no indexer was provided use the default indexer set in General settings
if not provided_indexer:
provided_indexer = sickbeard.INDEXER_DEFAULT
indexer = int(provided_indexer)
indexer_id = int(whichSeries)
show_name = ek(os.path.basename, ek(os.path.normpath, fullShowPath))
# 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(os.path.normpath, fullShowPath)
else:
show_dir = ek(os.path.join, rootDir, sanitize_filename(show_name))
# blanket policy - if the dir exists you should have used 'add existing show' numbnuts
if ek(os.path.isdir, show_dir) and not fullShowPath:
ui.notifications.error('Unable to add show', 'Folder {path} exists already'.format(path=show_dir))
return self.redirect('/addShows/existingShows/')
# don't create show dir if config says not to
if sickbeard.ADD_SHOWS_WO_DIR:
logger.log(u'Skipping initial creation of {path} due to config.ini setting'.format
(path=show_dir))
else:
dir_exists = helpers.makeDir(show_dir)
if not dir_exists:
logger.log(u'Unable to create the folder {path}, can\'t add the show'.format
(path=show_dir), logger.ERROR)
ui.notifications.error('Unable to add show',
'Unable to create the folder {path}, can\'t add the show'.format(path=show_dir))
# Don't redirect to default page because user wants to see the new show
return self.redirect('/home/')
else:
helpers.chmodAsParent(show_dir)
# 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)
if whitelist:
whitelist = short_group_names(whitelist)
if blacklist:
blacklist = short_group_names(blacklist)
if not allowed_qualities:
allowed_qualities = []
if not preferred_qualities or try_int(quality_preset, None):
preferred_qualities = []
if not isinstance(allowed_qualities, list):
allowed_qualities = [allowed_qualities]
if not isinstance(preferred_qualities, list):
preferred_qualities = [preferred_qualities]
new_quality = Quality.combineQualities([int(q) for q in allowed_qualities], [int(q) for q in preferred_qualities])
# add the show
sickbeard.showQueueScheduler.action.addShow(indexer, indexer_id, show_dir, int(defaultStatus), new_quality,
flatten_folders, indexer_lang, subtitles, anime,
scene, None, blacklist, whitelist, int(defaultStatusAfter))
ui.notifications.message('Show added', 'Adding the specified show into {path}'.format(path=show_dir))
return finishAddShow()
def addShowByID(self, indexer_id, show_name, indexer='TVDB', which_series=None,
indexer_lang=None, root_dir=None, default_status=None,
quality_preset=None, any_qualities=None, best_qualities=None,
flatten_folders=None, subtitles=None, full_show_path=None,
other_shows=None, skip_show=None, provided_indexer=None,
anime=None, scene=None, blacklist=None, whitelist=None,
default_status_after=None, default_flatten_folders=None,
configure_show_options=None):
if indexer != 'TVDB':
tvdb_id = helpers.getTVDBFromID(indexer_id, indexer.upper())
if not tvdb_id:
logger.log(u'Unable to to find tvdb ID to add %s' % show_name)
ui.notifications.error(
'Unable to add %s' % show_name,
'Could not add %s. We were unable to locate the tvdb id at this time.' % show_name
)
return
indexer_id = try_int(tvdb_id, None)
if Show.find(sickbeard.showList, int(indexer_id)):
return
# Sanitize the paramater anyQualities and bestQualities. As these would normally be passed as lists
if any_qualities:
any_qualities = any_qualities.split(',')
else:
any_qualities = []
if best_qualities:
best_qualities = best_qualities.split(',')
else:
best_qualities = []
# If configure_show_options is enabled let's use the provided settings
configure_show_options = config.checkbox_to_value(configure_show_options)
if configure_show_options:
# 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)
if whitelist:
whitelist = short_group_names(whitelist)
if blacklist:
blacklist = short_group_names(blacklist)
if not any_qualities:
any_qualities = []
if not best_qualities or try_int(quality_preset, None):
best_qualities = []
if not isinstance(any_qualities, list):
any_qualities = [any_qualities]
if not isinstance(best_qualities, list):
best_qualities = [best_qualities]
quality = Quality.combineQualities([int(q) for q in any_qualities], [int(q) for q in best_qualities])
location = root_dir
else:
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
default_status_after = sickbeard.STATUS_DEFAULT_AFTER
if sickbeard.ROOT_DIRS:
root_dirs = sickbeard.ROOT_DIRS.split('|')
location = root_dirs[int(root_dirs[0]) + 1]
else:
location = None
if not location:
logger.log(u'There was an error creating the show, '
u'no root directory setting found', logger.WARNING)
return 'No root directories setup, please go back and add one.'
show_name = get_showname_from_indexer(1, indexer_id)
show_dir = None
# add the show
sickbeard.showQueueScheduler.action.addShow(1, int(indexer_id), show_dir, int(default_status), quality, flatten_folders,
indexer_lang, subtitles, anime, scene, None, blacklist, whitelist,
int(default_status_after), root_dir=location)
ui.notifications.message('Show added', 'Adding the specified show {0}'.format(show_name))
# done adding show
return self.redirect('/home/')
def editShow(self, show=None, location=None, anyQualities=[], bestQualities=[], seasonfolders=None, paused=None, directCall=False, air_by_date=None, tvdbLang=None):
if show == None:
errString = "Invalid show ID: "+str(show)
if directCall:
return [errString]
else:
return _genericMessage("Error", errString)
showObj = sickbeard.helpers.findCertainShow(sickbeard.showList, int(show))
if showObj == None:
errString = "Unable to find the specified show: "+str(show)
if directCall:
return [errString]
else:
return _genericMessage("Error", errString)
if not location and not anyQualities and not bestQualities and not seasonfolders:
t = PageTemplate(file="editShow.tmpl")
t.submenu = HomeMenu()
with showObj.lock:
t.show = showObj
return _munge(t)
if seasonfolders == "on":
seasonfolders = 1
else:
seasonfolders = 0
if paused == "on":
paused = 1
else:
paused = 0
if air_by_date == "on":
air_by_date = 1
else:
air_by_date = 0
if tvdbLang and tvdbLang in tvdb_api.Tvdb().config['valid_languages']:
tvdb_lang = tvdbLang
else:
tvdb_lang = showObj.lang
# if we changed the language then kick off an update
if tvdb_lang == showObj.lang:
do_update = False
else:
do_update = True
if type(anyQualities) != list:
anyQualities = [anyQualities]
if type(bestQualities) != list:
bestQualities = [bestQualities]
errors = []
with showObj.lock:
newQuality = Quality.combineQualities(map(int, anyQualities), map(int, bestQualities))
showObj.quality = newQuality
if bool(showObj.seasonfolders) != bool(seasonfolders):
showObj.seasonfolders = seasonfolders
try:
sickbeard.showQueueScheduler.action.refreshShow(showObj) #@UndefinedVariable
except exceptions.CantRefreshException, e:
errors.append("Unable to refresh this show: "+ex(e))
showObj.paused = paused
showObj.air_by_date = air_by_date
showObj.lang = tvdb_lang
# 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))
if not ek.ek(os.path.isdir, location):
errors.append("New location <tt>%s</tt> does not exist" % location)
# 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
except exceptions.CantRefreshException, e:
errors.append("Unable to refresh this show:"+ex(e))
# grab updated info from TVDB
#showObj.loadEpisodesFromTVDB()
# 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 Sick Beard." % location)
# save it to the DB
showObj.saveToDB()
def addNewShow(self, whichSeries=None, tvdbLang="en", rootDir=None, defaultStatus=None,
anyQualities=None, bestQualities=None, seasonFolders=None, fullShowPath=None,
other_shows=None, skipShow=None):
"""
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.
"""
# grab our list of other dirs if given
if not other_shows:
other_shows = []
elif type(other_shows) != list:
other_shows = [other_shows]
def finishAddShow():
# if there are no extra shows then go home
if not other_shows:
redirect('/home')
# peel off the next one
next_show_dir = other_shows[0]
rest_of_show_dirs = other_shows[1:]
# go to add the next show
return self.newShow(next_show_dir, rest_of_show_dirs)
# if we're skipping then behave accordingly
if skipShow:
return finishAddShow()
# sanity check on our inputs
if (not rootDir and not fullShowPath) or not whichSeries:
return "Missing params, no tvdb id or folder:"+repr(whichSeries)+" and "+repr(rootDir)+"/"+repr(fullShowPath)
# figure out what show we're adding and where
series_pieces = whichSeries.partition('|')
if len(series_pieces) < 3:
return "Error with show selection."
tvdb_id = int(series_pieces[0])
show_name = series_pieces[2]
# 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))
# 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:
ui.notifications.error("Unable to add show", "Folder "+show_dir+" exists already")
redirect('/home')
# create the dir and make sure it worked
dir_exists = helpers.makeDir(show_dir)
if not dir_exists:
logger.log(u"Unable to create the folder "+show_dir+", can't add the show", logger.ERROR)
ui.notifications.error("Unable to add show", "Unable to create the folder "+show_dir+", can't add the show")
redirect("/home")
else:
helpers.chmodAsParent(show_dir)
# prepare the inputs for passing along
if seasonFolders == "on":
seasonFolders = 1
else:
seasonFolders = 0
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))
# add the show
sickbeard.showQueueScheduler.action.addShow(tvdb_id, show_dir, int(defaultStatus), newQuality, seasonFolders, tvdbLang) #@UndefinedVariable
ui.notifications.message('Show added', 'Adding the specified show into '+show_dir)
return finishAddShow()
