def processFile(tvdb_instance, episode):
"""Gets episode name, prompts user for input
"""
p("#" * 20)
p("# Processing file: %s" % episode.fullfilename)
if len(Config['input_filename_replacements']) > 0:
replaced = applyCustomInputReplacements(episode.fullfilename)
p("# With custom replacements: %s" % (replaced))
# Use force_name option. Done after input_filename_replacements so
# it can be used to skip the replacements easily
if Config['force_name'] is not None:
episode.seriesname = Config['force_name']
p("# Detected series: %s (%s)" %
(episode.seriesname, episode.number_string()))
try:
episode.populateFromTvdb(tvdb_instance,
force_name=Config['force_name'],
series_id=Config['series_id'])
except (DataRetrievalError, ShowNotFound) as errormsg:
if Config['always_rename'] and Config['skip_file_on_error'] is True:
if Config['skip_behaviour'] == 'exit':
warn("Exiting due to error: %s" % errormsg)
raise SkipBehaviourAbort()
warn("Skipping file due to error: %s" % errormsg)
return
else:
warn(errormsg)
except (SeasonNotFound, EpisodeNotFound, EpisodeNameNotFound) as errormsg:
# Show was found, so use corrected series name
if Config['always_rename'] and Config['skip_file_on_error']:
if Config['skip_behaviour'] == 'exit':
warn("Exiting due to error: %s" % errormsg)
raise SkipBehaviourAbort()
warn("Skipping file due to error: %s" % errormsg)
return
warn(errormsg)
cnamer = Renamer(episode.fullpath)
shouldRename = False
if Config["move_files_only"]:
newName = episode.fullfilename
shouldRename = True
else:
newName = episode.generateFilename()
if newName == episode.fullfilename:
p("#" * 20)
p("Existing filename is correct: %s" % episode.fullfilename)
p("#" * 20)
shouldRename = True
else:
p("#" * 20)
p("Old filename: %s" % episode.fullfilename)
if len(Config['output_filename_replacements']) > 0:
# Show filename without replacements
p("Before custom output replacements: %s" %
(episode.generateFilename(preview_orig_filename=False)))
p("New filename: %s" % newName)
if Config['always_rename']:
doRenameFile(cnamer, newName)
if Config['move_files_enable']:
if Config['move_files_destination_is_filepath']:
doMoveFile(cnamer=cnamer,
destFilepath=getMoveDestination(episode))
else:
doMoveFile(cnamer=cnamer,
destDir=getMoveDestination(episode))
return
elif Config['dry_run']:
p("%s will be renamed to %s" % (episode.fullfilename, newName))
if Config['move_files_enable']:
p("%s will be moved to %s" %
(newName, getMoveDestination(episode)))
return
ans = confirm("Rename?", options=['y', 'n', 'a', 'q'], default='y')
if ans == "a":
p("Always renaming")
Config['always_rename'] = True
shouldRename = True
elif ans == "q":
p("Quitting")
raise UserAbort("User exited with q")
elif ans == "y":
p("Renaming")
shouldRename = True
elif ans == "n":
p("Skipping")
else:
p("Invalid input, skipping")
if shouldRename:
doRenameFile(cnamer, newName)
if shouldRename and Config['move_files_enable']:
newPath = getMoveDestination(episode)
if Config['dry_run']:
p("%s will be moved to %s" %
(newName, getMoveDestination(episode)))
return
if Config['move_files_destination_is_filepath']:
doMoveFile(cnamer=cnamer,
destFilepath=newPath,
getPathPreview=True)
else:
doMoveFile(cnamer=cnamer, destDir=newPath, getPathPreview=True)
if not Config['batch'] and Config['move_files_confirmation']:
ans = confirm("Move file?", options=['y', 'n', 'q'], default='y')
else:
ans = 'y'
if ans == 'y':
p("Moving file")
doMoveFile(cnamer, newPath)
elif ans == 'q':
p("Quitting")
raise UserAbort("user exited with q")
def processFile(tvdb_instance, episode):
"""Gets episode name, prompts user for input
"""
p("#" * 20)
p("# Processing file: %s" % episode.fullfilename)
if len(Config['input_filename_replacements']) > 0:
replaced = applyCustomInputReplacements(episode.fullfilename)
p("# With custom replacements: %s" % (replaced))
# Use force_name option. Done after input_filename_replacements so
# it can be used to skip the replacements easily
if Config['force_name'] is not None:
episode.seriesname = Config['force_name']
p("# Detected series: %s (%s)" % (episode.seriesname, episode.number_string()))
try:
episode.populateFromTvdb(tvdb_instance, force_name=Config['force_name'], series_id=Config['series_id'])
except (DataRetrievalError, ShowNotFound) as errormsg:
if Config['always_rename'] and Config['skip_file_on_error'] is True:
if Config['skip_behaviour'] == 'exit':
warn("Exiting due to error: %s" % errormsg)
raise SkipBehaviourAbort()
warn("Skipping file due to error: %s" % errormsg)
return
else:
warn(errormsg)
except (SeasonNotFound, EpisodeNotFound, EpisodeNameNotFound) as errormsg:
# Show was found, so use corrected series name
if Config['always_rename'] and Config['skip_file_on_error']:
if Config['skip_behaviour'] == 'exit':
warn("Exiting due to error: %s" % errormsg)
raise SkipBehaviourAbort()
warn("Skipping file due to error: %s" % errormsg)
return
warn(errormsg)
cnamer = Renamer(episode.fullpath)
shouldRename = False
if Config["move_files_only"]:
newName = episode.fullfilename
shouldRename = True
else:
newName = episode.generateFilename()
if newName == episode.fullfilename:
p("#" * 20)
p("Existing filename is correct: %s" % episode.fullfilename)
p("#" * 20)
shouldRename = True
else:
p("#" * 20)
p("Old filename: %s" % episode.fullfilename)
if len(Config['output_filename_replacements']) > 0:
# Show filename without replacements
p("Before custom output replacements: %s" % (episode.generateFilename(preview_orig_filename = False)))
p("New filename: %s" % newName)
if Config['always_rename']:
doRenameFile(cnamer, newName)
if Config['move_files_enable']:
if Config['move_files_destination_is_filepath']:
doMoveFile(cnamer = cnamer, destFilepath = getMoveDestination(episode))
else:
doMoveFile(cnamer = cnamer, destDir = getMoveDestination(episode))
return
elif Config['dry_run']:
p("%s will be renamed to %s" % (episode.fullfilename, newName))
if Config['move_files_enable']:
p("%s will be moved to %s" % (newName, getMoveDestination(episode)))
return
ans = confirm("Rename?", options = ['y', 'n', 'a', 'q'], default = 'y')
if ans == "a":
p("Always renaming")
Config['always_rename'] = True
shouldRename = True
elif ans == "q":
p("Quitting")
raise UserAbort("User exited with q")
elif ans == "y":
p("Renaming")
shouldRename = True
elif ans == "n":
p("Skipping")
else:
p("Invalid input, skipping")
if shouldRename:
doRenameFile(cnamer, newName)
if shouldRename and Config['move_files_enable']:
newPath = getMoveDestination(episode)
if Config['dry_run']:
p("%s will be moved to %s" % (newName, getMoveDestination(episode)))
return
if Config['move_files_destination_is_filepath']:
doMoveFile(cnamer = cnamer, destFilepath = newPath, getPathPreview = True)
else:
doMoveFile(cnamer = cnamer, destDir = newPath, getPathPreview = True)
if not Config['batch'] and Config['move_files_confirmation']:
ans = confirm("Move file?", options = ['y', 'n', 'q'], default = 'y')
else:
ans = 'y'
if ans == 'y':
p("Moving file")
doMoveFile(cnamer, newPath)
elif ans == 'q':
p("Quitting")
raise UserAbort("user exited with q")