def get_subtitles_path(self, video_path):
if Path(sickrage.app.config.subtitles_dir).is_absolute() and Path(sickrage.app.config.subtitles_dir).exists():
new_subtitles_path = sickrage.app.config.subtitles_dir
elif sickrage.app.config.subtitles_dir:
new_subtitles_path = Path(os.path.dirname(video_path)).joinpath(sickrage.app.config.subtitles_dir.strip('/'))
dir_exists = make_dir(new_subtitles_path)
if not dir_exists:
sickrage.app.log.warning('Unable to create subtitles folder {}'.format(new_subtitles_path))
else:
chmod_as_parent(new_subtitles_path)
else:
new_subtitles_path = Path(video_path).parent
return new_subtitles_path
def start(self):
# remove all handlers
self.handlers.clear()
# console log handler
if self.consoleLogging:
console_handler = logging.StreamHandler()
formatter = logging.Formatter('%(asctime)s %(levelname)s::%(threadName)s::%(message)s', '%H:%M:%S')
console_handler.setFormatter(formatter)
console_handler.setLevel(self.logLevels['INFO'] if not self.debugLogging else self.logLevels['DEBUG'])
self.addHandler(console_handler)
# file log handlers
if self.logFile:
# make logs folder if it doesn't exist
if not os.path.exists(os.path.dirname(self.logFile)):
if not make_dir(os.path.dirname(self.logFile)):
return
if sickrage.app.developer:
rfh = FileHandler(
filename=self.logFile,
)
else:
rfh = RotatingFileHandler(
filename=self.logFile,
maxBytes=self.logSize,
backupCount=self.logNr
)
rfh_errors = RotatingFileHandler(
filename=self.logFile.replace('.log', '.error.log'),
maxBytes=self.logSize,
backupCount=self.logNr
)
formatter = logging.Formatter('%(asctime)s %(levelname)s::%(threadName)s::%(message)s', '%Y-%m-%d %H:%M:%S')
rfh.setFormatter(formatter)
rfh.setLevel(self.logLevels['INFO'] if not self.debugLogging else self.logLevels['DEBUG'])
self.addHandler(rfh)
rfh_errors.setFormatter(formatter)
rfh_errors.setLevel(self.logLevels['ERROR'])
self.addHandler(rfh_errors)
def get_subtitles_path(self, video_path):
if os.path.isabs(sickrage.app.config.subtitles_dir):
new_subtitles_path = sickrage.app.config.subtitles_dir
elif sickrage.app.config.subtitles_dir:
new_subtitles_path = os.path.join(
os.path.dirname(video_path), sickrage.app.config.subtitles_dir)
dir_exists = make_dir(new_subtitles_path)
if not dir_exists:
sickrage.app.log.warning(
'Unable to create subtitles folder {}'.format(
new_subtitles_path))
else:
chmod_as_parent(new_subtitles_path)
else:
new_subtitles_path = os.path.dirname(video_path)
return new_subtitles_path
def add_default_show(indexer, indexer_id, name, status):
"""
Adds a new show with the default settings
"""
if not find_show(int(indexer_id)):
sickrage.app.log.info("Adding show " + str(indexer_id))
root_dirs = sickrage.app.config.root_dirs.split('|')
try:
location = root_dirs[int(root_dirs[0]) + 1]
except Exception:
location = None
if location:
showPath = os.path.join(location, sanitize_file_name(name))
dir_exists = make_dir(showPath)
if not dir_exists:
sickrage.app.log.warning(
"Unable to create the folder %s , can't add the show" %
showPath)
return
else:
chmod_as_parent(showPath)
sickrage.app.show_queue.add_show(
int(indexer),
int(indexer_id),
showPath,
default_status=status,
quality=int(sickrage.app.config.quality_default),
flatten_folders=int(
sickrage.app.config.flatten_folders_default),
paused=sickrage.app.config.trakt_start_paused,
default_status_after=status,
scene=sickrage.app.config.scene_default,
skip_downloaded=sickrage.app.config.skip_downloaded_default
)
else:
sickrage.app.log.warning(
"There was an error creating the show, no root directory setting found"
)
return
def change_nzb_dir(nzb_dir):
"""
Change NZB blackhole directory
:param nzb_dir: New NZB Folder location
:return: True on success, False on failure
"""
if nzb_dir == '':
sickrage.app.config.blackhole.nzb_dir = ''
return True
if os.path.normpath(sickrage.app.config.blackhole.nzb_dir
) != os.path.normpath(nzb_dir):
if make_dir(nzb_dir):
sickrage.app.config.blackhole.nzb_dir = os.path.normpath(nzb_dir)
sickrage.app.log.info("Changed NZB folder to " + nzb_dir)
else:
return False
return True
def change_https_key(https_key):
"""
Replace HTTPS Key file path
:param https_key: path to the new key file
:return: True on success, False on failure
"""
if https_key == '':
sickrage.app.config.general.https_key = ''
return True
if os.path.normpath(sickrage.app.config.general.https_key
) != os.path.normpath(https_key):
if make_dir(os.path.dirname(os.path.abspath(https_key))):
sickrage.app.config.general.https_key = os.path.normpath(https_key)
sickrage.app.log.info("Changed https key path to " + https_key)
else:
return False
return True
def change_torrent_dir(torrent_dir):
"""
Change Torrent blackhole directory
:param torrent_dir: New torrent directory
:return: True on success, False on failure
"""
if torrent_dir == '':
sickrage.app.config.blackhole.torrent_dir = ''
return True
if os.path.normpath(sickrage.app.config.blackhole.torrent_dir
) != os.path.normpath(torrent_dir):
if make_dir(torrent_dir):
sickrage.app.config.blackhole.torrent_dir = os.path.normpath(
torrent_dir)
sickrage.app.log.info("Changed torrent folder to " + torrent_dir)
else:
return False
return True
def change_tv_download_dir(tv_download_dir):
"""
Change TV_DOWNLOAD directory (used by postprocessor)
:param tv_download_dir: New tv download directory
:return: True on success, False on failure
"""
if tv_download_dir == '':
sickrage.app.config.general.tv_download_dir = ''
return True
if os.path.normpath(sickrage.app.config.general.tv_download_dir
) != os.path.normpath(tv_download_dir):
if make_dir(tv_download_dir):
sickrage.app.config.general.tv_download_dir = os.path.normpath(
tv_download_dir)
sickrage.app.log.info("Changed TV download folder to " +
tv_download_dir)
else:
return False
return True
def change_https_cert(https_cert):
"""
Replace HTTPS Certificate file path
:param https_cert: path to the new certificate file
:return: True on success, False on failure
"""
if https_cert == '':
sickrage.app.config.general.https_cert = ''
return True
if os.path.normpath(sickrage.app.config.general.https_cert
) != os.path.normpath(https_cert):
if make_dir(os.path.dirname(os.path.abspath(https_cert))):
sickrage.app.config.general.https_cert = os.path.normpath(
https_cert)
sickrage.app.log.info("Changed https cert path to " + https_cert)
else:
return False
return True
async def post(self, *args, **kwargs):
"""
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.
"""
whichSeries = self.get_argument('whichSeries', None)
rootDir = self.get_argument('rootDir', None)
fullShowPath = self.get_argument('fullShowPath', None)
providedName = self.get_argument('providedName', None)
indexerLang = self.get_argument('indexerLang', None)
defaultStatus = self.get_argument('defaultStatus', None)
quality_preset = self.get_argument('quality_preset', None)
anyQualities = self.get_arguments('anyQualities')
bestQualities = self.get_arguments('bestQualities')
flatten_folders = self.get_argument('flatten_folders', None)
subtitles = self.get_argument('subtitles', None)
sub_use_sr_metadata = self.get_argument('sub_use_sr_metadata', None)
other_shows = self.get_arguments('other_shows')
skipShow = self.get_argument('skipShow', None)
providedIndexer = self.get_argument('providedIndexer', None)
anime = self.get_argument('anime', None)
scene = self.get_argument('scene', None)
blacklist = self.get_argument('blacklist', None)
whitelist = self.get_argument('whitelist', None)
defaultStatusAfter = self.get_argument('defaultStatusAfter', None)
skip_downloaded = self.get_argument('skip_downloaded', None)
add_show_year = self.get_argument('add_show_year', None)
indexerLang = indexerLang or sickrage.app.config.indexer_default_language
# if we're skipping then behave accordingly
if skipShow:
return self.write(await self.finish_add_show(other_shows))
if not whichSeries:
return self.redirect("/home/")
# 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:
sickrage.app.log.error(
'Unable to add show due to show selection. Not anough arguments: %s'
% (repr(series_pieces)))
sickrage.app.alerts.error(
_('Unknown error. Unable to add show due to problem with show selection.'
))
return self.redirect('/home/addShows/existingShows/')
indexer = int(series_pieces[1])
indexer_id = int(series_pieces[3])
show_name = series_pieces[4]
else:
indexer = int(providedIndexer
or sickrage.app.config.indexer_default)
indexer_id = int(whichSeries)
if fullShowPath:
show_name = os.path.basename(os.path.normpath(fullShowPath))
else:
show_name = providedName
# 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 = os.path.normpath(fullShowPath)
else:
show_dir = os.path.join(rootDir, sanitize_file_name(show_name))
if add_show_year and not re.match(r'.*\(\d+\)$',
show_dir) and re.search(
r'\d{4}', series_pieces[5]):
show_dir = "{} ({})".format(
show_dir,
re.search(r'\d{4}', series_pieces[5]).group(0))
# blanket policy - if the dir exists you should have used "add existing show" numbnuts
if os.path.isdir(show_dir) and not fullShowPath:
sickrage.app.alerts.error(
_("Unable to add show"),
_("Folder ") + show_dir + _(" exists already"))
return self.redirect('/home/addShows/existingShows/')
# don't create show dir if config says not to
if sickrage.app.config.add_shows_wo_dir:
sickrage.app.log.info("Skipping initial creation of " + show_dir +
" due to SiCKRAGE configuation setting")
else:
dir_exists = make_dir(show_dir)
if not dir_exists:
sickrage.app.log.warning("Unable to create the folder " +
show_dir + ", can't add the show")
sickrage.app.alerts.error(
_("Unable to add show"),
_("Unable to create the folder " + show_dir +
", can't add the show"))
# Don't redirect to default page because user wants to see the new show
return self.redirect("/home/")
else:
chmod_as_parent(show_dir)
# prepare the inputs for passing along
scene = checkbox_to_value(scene)
anime = checkbox_to_value(anime)
flatten_folders = checkbox_to_value(flatten_folders)
subtitles = checkbox_to_value(subtitles)
sub_use_sr_metadata = checkbox_to_value(sub_use_sr_metadata)
skip_downloaded = checkbox_to_value(skip_downloaded)
if whitelist:
whitelist = short_group_names(whitelist)
if blacklist:
blacklist = short_group_names(blacklist)
newQuality = try_int(quality_preset, None)
if not newQuality:
newQuality = Quality.combine_qualities(map(int, anyQualities),
map(int, bestQualities))
# add the show
sickrage.app.show_queue.add_show(
indexer=indexer,
indexer_id=indexer_id,
showDir=show_dir,
default_status=int(defaultStatus),
quality=newQuality,
flatten_folders=flatten_folders,
lang=indexerLang,
subtitles=subtitles,
sub_use_sr_metadata=sub_use_sr_metadata,
anime=anime,
scene=scene,
paused=None,
blacklist=blacklist,
whitelist=whitelist,
default_status_after=int(defaultStatusAfter),
skip_downloaded=skip_downloaded)
sickrage.app.alerts.message(
_('Adding Show'),
_('Adding the specified show into ') + show_dir)
return self.write(await self.finish_add_show(other_shows))
def post(self):
data = json_decode(self.request.body)
is_existing = data.get('isExisting', 'false')
root_directory = data.get('rootDirectory', None)
series_id = data.get('seriesId', None)
series_name = data.get('seriesName', None)
series_directory = data.get('seriesDirectory', None)
first_aired = data.get('firstAired', None)
series_provider_slug = data.get('seriesProviderSlug', None)
series_provider_language = data.get('seriesProviderLanguage', None)
default_status = data.get('defaultStatus', None)
default_status_after = data.get('defaultStatusAfter', None)
quality_preset = data.get('qualityPreset', None)
allowed_qualities = data.get('allowedQualities', [])
preferred_qualities = data.get('preferredQualities', [])
subtitles = self._parse_boolean(
data.get('subtitles', sickrage.app.config.subtitles.default))
sub_use_sr_metadata = self._parse_boolean(
data.get('subUseSrMetadata', 'false'))
flatten_folders = self._parse_boolean(
data.get('flattenFolders',
sickrage.app.config.general.flatten_folders_default))
is_anime = self._parse_boolean(
data.get('isAnime', sickrage.app.config.general.anime_default))
is_scene = self._parse_boolean(
data.get('isScene', sickrage.app.config.general.scene_default))
search_format = data.get(
'searchFormat',
sickrage.app.config.general.search_format_default.name)
dvd_order = self._parse_boolean(data.get('dvdOrder', 'false'))
skip_downloaded = self._parse_boolean(
data.get('skipDownloaded',
sickrage.app.config.general.skip_downloaded_default))
add_show_year = self._parse_boolean(data.get('addShowYear', 'false'))
if not series_id:
return self.send_error(
400, error=f"Missing seriesId parameter: {series_id}")
series_provider_id = SeriesProviderID.by_slug(series_provider_slug)
if not series_provider_id:
return self.send_error(
404,
error="Unable to identify a series provider using provided slug"
)
series = find_show(int(series_id), series_provider_id)
if series:
return self.send_error(400,
error=f"Already exists series: {series_id}")
if is_existing and not series_directory:
return self.send_error(400,
error="Missing seriesDirectory parameter")
if not is_existing:
series_directory = os.path.join(root_directory,
sanitize_file_name(series_name))
if first_aired:
series_year = re.search(r'\d{4}', first_aired)
if add_show_year and not re.match(
r'.*\(\d+\)$', series_directory) and series_year:
series_directory = f"{series_directory} ({series_year.group()})"
if os.path.isdir(series_directory):
sickrage.app.alerts.error(
_("Unable to add show"),
_("Folder ") + series_directory + _(" exists already"))
return self.send_error(
400,
error=f"Show directory {series_directory} already exists!")
if not make_dir(series_directory):
sickrage.app.log.warning(
f"Unable to create the folder {series_directory}, can't add the show"
)
sickrage.app.alerts.error(
_("Unable to add show"),
_(f"Unable to create the folder {series_directory}, can't add the show"
))
return self.send_error(
400,
error=
f"Unable to create the show folder {series_directory}, can't add the show"
)
chmod_as_parent(series_directory)
try:
new_quality = Qualities[quality_preset.upper()]
except KeyError:
new_quality = Quality.combine_qualities(
[Qualities[x.upper()] for x in allowed_qualities],
[Qualities[x.upper()] for x in preferred_qualities])
sickrage.app.show_queue.add_show(
series_provider_id=series_provider_id,
series_id=int(series_id),
showDir=series_directory,
default_status=EpisodeStatus[default_status.upper()],
default_status_after=EpisodeStatus[default_status_after.upper()],
quality=new_quality,
flatten_folders=flatten_folders,
lang=series_provider_language,
subtitles=subtitles,
sub_use_sr_metadata=sub_use_sr_metadata,
anime=is_anime,
dvd_order=dvd_order,
search_format=SearchFormat[search_format.upper()],
paused=False,
# blacklist=blacklist,
# whitelist=whitelist,
scene=is_scene,
skip_downloaded=skip_downloaded)
sickrage.app.alerts.message(
_('Adding Show'),
_(f'Adding the specified show into {series_directory}'))
return self.write_json({'message': True})
