def processDir(dirName, nzbName=None, process_method=None, force=False, is_priority=None, delete_on=False, failed=False,
proc_type="auto", **kwargs):
"""
Scans through the files in dirName and processes whatever media files it finds
:param dirName: The folder name to look in
:param nzbName: The NZB name which resulted in this folder being downloaded
:param force: True to postprocess already postprocessed files
:param failed: Boolean for whether or not the download failed
:param proc_type: Type of postprocessing auto or manual
"""
result = ProcessResult()
result.output += logHelper("Processing folder %s" % dirName, sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("TV_DOWNLOAD_DIR: %s" % sickrage.srCore.srConfig.TV_DOWNLOAD_DIR,
sickrage.srCore.srLogger.DEBUG)
postpone = False
# if they passed us a real dir then assume it's the one we want
if os.path.isdir(dirName):
dirName = os.path.realpath(dirName)
# if the client and SickRage are not on the same machine translate the Dir in a network dir
elif sickrage.srCore.srConfig.TV_DOWNLOAD_DIR and os.path.isdir(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR) \
and os.path.normpath(dirName) != os.path.normpath(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR):
dirName = os.path.join(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR,
os.path.abspath(dirName).split(os.path.sep)[-1])
result.output += logHelper("Trying to use folder %s" % dirName, sickrage.srCore.srLogger.DEBUG)
# if we didn't find a real dir then quit
if not os.path.isdir(dirName):
result.output += logHelper(
"Unable to figure out what folder to process. If your downloader and SiCKRAGE aren't on the same PC make sure you fill out your TV download dir in the config.",
sickrage.srCore.srLogger.DEBUG)
return result.output
path, dirs, files = get_path_dir_files(dirName, nzbName, proc_type)
files = [x for x in files if notTorNZBFile(x)]
SyncFiles = [x for x in files if isSyncFile(x)]
# Don't post process if files are still being synced and option is activated
if SyncFiles and sickrage.srCore.srConfig.POSTPONE_IF_SYNC_FILES:
postpone = True
nzbNameOriginal = nzbName
if not postpone:
result.output += logHelper("PostProcessing Path: %s" % path, sickrage.srCore.srLogger.INFO)
result.output += logHelper("PostProcessing Dirs: [%s]" % ", ".join(dirs), sickrage.srCore.srLogger.DEBUG)
rarFiles = [x for x in files if isRarFile(x)]
rarContent = unRAR(path, rarFiles, force, result)
files += rarContent
videoFiles = [x for x in files if isMediaFile(x)]
videoInRar = [x for x in rarContent if isMediaFile(x)]
result.output += logHelper("PostProcessing Files: [%s]" % ", ".join(files), sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing VideoFiles: [%s]" % ", ".join(videoFiles),
sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing RarContent: [%s]" % ", ".join(rarContent),
sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing VideoInRar: [%s]" % ", ".join(videoInRar),
sickrage.srCore.srLogger.DEBUG)
# If nzbName is set and there's more than one videofile in the folder, files will be lost (overwritten).
if len(videoFiles) >= 2:
nzbName = None
if not process_method:
process_method = sickrage.srCore.srConfig.PROCESS_METHOD
result.result = True
# Don't Link media when the media is extracted from a rar in the same path
if process_method in ('hardlink', 'symlink') and videoInRar:
process_media(path, videoInRar, nzbName, 'move', force, is_priority, result)
delete_files(path, rarContent, result)
for video in set(videoFiles) - set(videoInRar):
process_media(path, [video], nzbName, process_method, force, is_priority, result)
elif sickrage.srCore.srConfig.DELRARCONTENTS and videoInRar:
process_media(path, videoInRar, nzbName, process_method, force, is_priority, result)
delete_files(path, rarContent, result, True)
for video in set(videoFiles) - set(videoInRar):
process_media(path, [video], nzbName, process_method, force, is_priority, result)
else:
for video in videoFiles:
process_media(path, [video], nzbName, process_method, force, is_priority, result)
else:
result.output += logHelper("Found temporary sync files, skipping post processing for: %s" % path)
result.output += logHelper("Sync Files: [%s] in path %s" % (", ".join(SyncFiles), path))
result.missedfiles.append("%s : Syncfiles found" % path)
# Process Video File in all TV Subdir
for curDir in [x for x in dirs if validateDir(path, x, nzbNameOriginal, failed, result)]:
result.result = True
for processPath, _, fileList in os.walk(os.path.join(path, curDir), topdown=False):
if not validateDir(path, processPath, nzbNameOriginal, failed, result):
continue
postpone = False
SyncFiles = [x for x in fileList if isSyncFile(x)]
# Don't post process if files are still being synced and option is activated
if SyncFiles and sickrage.srCore.srConfig.POSTPONE_IF_SYNC_FILES:
postpone = True
if not postpone:
rarFiles = [x for x in fileList if isRarFile(x)]
rarContent = unRAR(processPath, rarFiles, force, result)
fileList = set(fileList + rarContent)
videoFiles = [x for x in fileList if isMediaFile(x)]
videoInRar = [x for x in rarContent if isMediaFile(x)]
notwantedFiles = [x for x in fileList if x not in videoFiles]
if notwantedFiles:
result.output += logHelper("Found unwanted files: [%s]" % ", ".join(notwantedFiles),
sickrage.srCore.srLogger.DEBUG)
# Don't Link media when the media is extracted from a rar in the same path
if process_method in ('hardlink', 'symlink') and videoInRar:
process_media(processPath, videoInRar, nzbName, 'move', force, is_priority, result)
process_media(processPath, set(videoFiles) - set(videoInRar), nzbName, process_method, force,
is_priority, result)
delete_files(processPath, rarContent, result)
elif sickrage.srCore.srConfig.DELRARCONTENTS and videoInRar:
process_media(processPath, videoInRar, nzbName, process_method, force, is_priority, result)
process_media(processPath, set(videoFiles) - set(videoInRar), nzbName, process_method, force,
is_priority, result)
delete_files(processPath, rarContent, result, True)
else:
process_media(processPath, videoFiles, nzbName, process_method, force, is_priority, result)
# Delete all file not needed
if process_method != "move" or not result.result \
or (
proc_type == "manual" and not delete_on): # Avoid to delete files if is Manual PostProcessing
continue
delete_files(processPath, notwantedFiles, result)
if (
not sickrage.srCore.srConfig.NO_DELETE or proc_type == "manual") and process_method == "move" and \
os.path.normpath(processPath) != os.path.normpath(
sickrage.srCore.srConfig.TV_DOWNLOAD_DIR):
if delete_folder(processPath, check_empty=True):
result.output += logHelper("Deleted folder: %s" % processPath,
sickrage.srCore.srLogger.DEBUG)
else:
result.output += logHelper("Found temporary sync files, skipping post processing for: %s" % processPath)
result.output += logHelper("Sync Files: [%s] in path %s" % (", ".join(SyncFiles), processPath))
result.missedfiles.append("%s : Syncfiles found" % processPath)
if result.aggresult:
result.output += logHelper("Processing completed")
if result.missedfiles:
result.output += logHelper("I did encounter some unprocessable items: [%s]" % ", ".join(result.missedfiles))
else:
result.output += logHelper(
"Problem(s) during processing, failed the following files/folders: [%s]" % ", ".join(
result.missedfiles),
sickrage.srCore.srLogger.WARNING)
return result.output
def processDir(dirName, nzbName=None, process_method=None, force=False, is_priority=None, delete_on=False, failed=False,
proc_type="auto", **kwargs):
"""
Scans through the files in dirName and processes whatever media files it finds
:param dirName: The folder name to look in
:param nzbName: The NZB name which resulted in this folder being downloaded
:param force: True to postprocess already postprocessed files
:param failed: Boolean for whether or not the download failed
:param proc_type: Type of postprocessing auto or manual
"""
result = ProcessResult()
result.output += logHelper("Processing folder %s" % dirName, sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("TV_DOWNLOAD_DIR: %s" % sickrage.srCore.srConfig.TV_DOWNLOAD_DIR,
sickrage.srCore.srLogger.DEBUG)
postpone = False
# if they passed us a real dir then assume it's the one we want
if os.path.isdir(dirName):
dirName = os.path.realpath(dirName)
# if the client and SickRage are not on the same machine translate the Dir in a network dir
elif sickrage.srCore.srConfig.TV_DOWNLOAD_DIR and os.path.isdir(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR) \
and os.path.normpath(dirName) != os.path.normpath(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR):
dirName = os.path.join(sickrage.srCore.srConfig.TV_DOWNLOAD_DIR,
os.path.abspath(dirName).split(os.path.sep)[-1])
result.output += logHelper("Trying to use folder %s" % dirName, sickrage.srCore.srLogger.DEBUG)
# if we didn't find a real dir then quit
if not os.path.isdir(dirName):
result.output += logHelper(
"Unable to figure out what folder to process. If your downloader and SiCKRAGE aren't on the same PC make sure you fill out your TV download dir in the config.",
sickrage.srCore.srLogger.DEBUG)
return result.output
path, dirs, files = get_path_dir_files(dirName, nzbName, proc_type)
files = [x for x in files if notTorNZBFile(x)]
SyncFiles = [x for x in files if isSyncFile(x)]
# Don't post process if files are still being synced and option is activated
if SyncFiles and sickrage.srCore.srConfig.POSTPONE_IF_SYNC_FILES:
postpone = True
nzbNameOriginal = nzbName
if not postpone:
result.output += logHelper("PostProcessing Path: %s" % path, sickrage.srCore.srLogger.INFO)
result.output += logHelper("PostProcessing Dirs: [%s]" % ", ".join(dirs), sickrage.srCore.srLogger.DEBUG)
rarFiles = [x for x in files if isRarFile(x)]
rarContent = unRAR(path, rarFiles, force, result)
files += rarContent
videoFiles = [x for x in files if isMediaFile(x)]
videoInRar = [x for x in rarContent if isMediaFile(x)]
result.output += logHelper("PostProcessing Files: [%s]" % ", ".join(files), sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing VideoFiles: [%s]" % ", ".join(videoFiles),
sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing RarContent: [%s]" % ", ".join(rarContent),
sickrage.srCore.srLogger.DEBUG)
result.output += logHelper("PostProcessing VideoInRar: [%s]" % ", ".join(videoInRar),
sickrage.srCore.srLogger.DEBUG)
# If nzbName is set and there's more than one videofile in the folder, files will be lost (overwritten).
if len(videoFiles) >= 2:
nzbName = None
if not process_method:
process_method = sickrage.srCore.srConfig.PROCESS_METHOD
result.result = True
# Don't Link media when the media is extracted from a rar in the same path
if process_method in ('hardlink', 'symlink') and videoInRar:
process_media(path, videoInRar, nzbName, 'move', force, is_priority, result)
delete_files(path, rarContent, result)
for video in set(videoFiles) - set(videoInRar):
process_media(path, [video], nzbName, process_method, force, is_priority, result)
elif sickrage.srCore.srConfig.DELRARCONTENTS and videoInRar:
process_media(path, videoInRar, nzbName, process_method, force, is_priority, result)
delete_files(path, rarContent, result, True)
for video in set(videoFiles) - set(videoInRar):
process_media(path, [video], nzbName, process_method, force, is_priority, result)
else:
for video in videoFiles:
process_media(path, [video], nzbName, process_method, force, is_priority, result)
else:
result.output += logHelper("Found temporary sync files, skipping post processing for: %s" % path)
result.output += logHelper("Sync Files: [%s] in path %s" % (", ".join(SyncFiles), path))
result.missedfiles.append("%s : Syncfiles found" % path)
# Process Video File in all TV Subdir
for curDir in [x for x in dirs if validateDir(path, x, nzbNameOriginal, failed, result)]:
result.result = True
for processPath, _, fileList in os.walk(os.path.join(path, curDir), topdown=False):
if not validateDir(path, processPath, nzbNameOriginal, failed, result):
continue
postpone = False
SyncFiles = [x for x in fileList if isSyncFile(x)]
# Don't post process if files are still being synced and option is activated
if SyncFiles and sickrage.srCore.srConfig.POSTPONE_IF_SYNC_FILES:
postpone = True
if not postpone:
rarFiles = [x for x in fileList if isRarFile(x)]
rarContent = unRAR(processPath, rarFiles, force, result)
fileList = set(fileList + rarContent)
videoFiles = [x for x in fileList if isMediaFile(x)]
videoInRar = [x for x in rarContent if isMediaFile(x)]
notwantedFiles = [x for x in fileList if x not in videoFiles]
if notwantedFiles:
result.output += logHelper("Found unwanted files: [%s]" % ", ".join(notwantedFiles),
sickrage.srCore.srLogger.DEBUG)
# Don't Link media when the media is extracted from a rar in the same path
if process_method in ('hardlink', 'symlink') and videoInRar:
process_media(processPath, videoInRar, nzbName, 'move', force, is_priority, result)
process_media(processPath, set(videoFiles) - set(videoInRar), nzbName, process_method, force,
is_priority, result)
delete_files(processPath, rarContent, result)
elif sickrage.srCore.srConfig.DELRARCONTENTS and videoInRar:
process_media(processPath, videoInRar, nzbName, process_method, force, is_priority, result)
process_media(processPath, set(videoFiles) - set(videoInRar), nzbName, process_method, force,
is_priority, result)
delete_files(processPath, rarContent, result, True)
else:
process_media(processPath, videoFiles, nzbName, process_method, force, is_priority, result)
# Delete all file not needed
if process_method != "move" or not result.result \
or (
proc_type == "manual" and not delete_on): # Avoid to delete files if is Manual PostProcessing
continue
delete_files(processPath, notwantedFiles, result)
if (
not sickrage.srCore.srConfig.NO_DELETE or proc_type == "manual") and process_method == "move" and \
os.path.normpath(processPath) != os.path.normpath(
sickrage.srCore.srConfig.TV_DOWNLOAD_DIR):
if delete_folder(processPath, check_empty=True):
result.output += logHelper("Deleted folder: %s" % processPath,
sickrage.srCore.srLogger.DEBUG)
else:
result.output += logHelper("Found temporary sync files, skipping post processing for: %s" % processPath)
result.output += logHelper("Sync Files: [%s] in path %s" % (", ".join(SyncFiles), processPath))
result.missedfiles.append("%s : Syncfiles found" % processPath)
if result.aggresult:
result.output += logHelper("Processing completed")
if result.missedfiles:
result.output += logHelper("I did encounter some unprocessable items: [%s]" % ", ".join(result.missedfiles))
else:
result.output += logHelper(
"Problem(s) during processing, failed the following files/folders: [%s]" % ", ".join(
result.missedfiles),
sickrage.srCore.srLogger.WARNING)
return result.output