def _report_results(self) -> None:
if self.success_count == 0:
message_type = MessageType.ERROR
elif self.success_count == self.total_count:
message_type = MessageType.SUCCESS
else:
message_type = MessageType.ALERT
tty.msg(
f"\n{self.success_count} out of {self.total_count} files processed successfully",
message_type,
)
def _announce_file(self, target: Target):
media_type = target.metadata.media.value.title()
description = (f"{media_type} Subtitle" if is_subtitle(
target.metadata.container) else media_type)
filename_label = target.source.name
filesize_label = get_filesize(target.source)
tty.msg(
f'\nProcessing {description} "{filename_label}" ({filesize_label})',
MessageType.HEADING,
)
tty.msg(target.source, debug=True)
def _list_details(self, target: Target):
tty.msg(f"using {target.provider_type.value}",
MessageType.ALERT,
debug=True)
tty.msg("\nsearch parameters", debug=True)
tty.msg(target.metadata.as_dict(), debug=True)
tty.msg("", debug=True)
def _handle_directives(self) -> None:
if self.settings.version:
tty.msg(f"mnamer version {VERSION}")
raise SystemExit(0)
if self.settings.config_dump:
print(self.settings.as_json())
raise SystemExit(0)
if self.settings.clear_cache:
clear_cache()
tty.msg("cache cleared", MessageType.ALERT)
raise SystemExit(0)
if self.settings.test:
tty.msg("testing mode", MessageType.ALERT)
if self.settings.config_path:
tty.msg(f"loaded config from '{self.settings.config_path}'",
MessageType.ALERT)
def _rename_and_move_file(self, target: Target):
tty.msg(
f"moving to {target.destination.absolute()}",
MessageType.SUCCESS,
)
if self.settings.test:
self.success_count += 1
return
try:
target.relocate()
except MnamerException:
tty.msg("FAILED!", MessageType.ERROR)
else:
tty.msg("OK!", MessageType.SUCCESS)
self.success_count += 1
def _remove_empty_source_directory(self, target: Target):
if self.settings.remove_empty_source_directory:
try:
directory_removed = remove_empty_directory(
target.source.parent, self.settings.test)
if directory_removed:
tty.msg(
"removed empty source directory " +
str(target.source.parent.absolute()),
MessageType.SUCCESS,
)
else:
tty.msg(
"skipped removal of non empty source directory " +
str(target.source.parent.absolute()),
MessageType.SUCCESS,
)
except:
tty.msg(
"failed to remove empty source directory " +
str(target.source.parent.absolute()),
MessageType.ERROR,
)
def run(load_configuration=True, load_arguments=True):
"""The main program loop."""
# setup arguments and load runtime configuration
try:
settings = Settings(
load_configuration=load_configuration, load_arguments=load_arguments
)
except MnamerException as e:
tty.msg(str(e), MessageType.ERROR)
raise SystemExit(1)
targets = Target.populate_paths(settings)
tty.configure(settings)
# handle directives and configuration
if settings.version:
tty.msg(f"mnamer version {VERSION}")
raise SystemExit(0)
if settings.config_dump:
print(settings.as_json)
raise SystemExit(0)
tty.msg("Starting mnamer", MessageType.HEADING)
if settings.no_cache:
clear_cache()
tty.msg("cache cleared", MessageType.ALERT)
if settings.test:
tty.msg("testing mode", MessageType.ALERT)
if settings.configuration_path:
tty.msg(
f"loaded config from '{settings.configuration_path}'",
MessageType.ALERT,
)
# print configuration details
tty.msg("\nsystem", debug=True)
tty.msg(SYSTEM, debug=True)
tty.msg("\nsettings", debug=True)
tty.msg(settings.as_dict, debug=True)
tty.msg("\ntargets", debug=True)
tty.msg(targets or [None], debug=True)
# exit early if no media files are found
total_count = len(targets)
if total_count == 0:
tty.msg("", debug=True)
tty.msg("no media files found", MessageType.ALERT)
raise SystemExit(0)
# main program loop
success_count = 0
for target in targets:
# announce file
media_label = target.metadata.media.value.title()
filename_label = target.source.name
filesize_label = get_filesize(target.source)
tty.msg(
f'\nProcessing {media_label} "{filename_label}" ({filesize_label})',
MessageType.HEADING,
)
tty.msg(target.source, debug=True)
# list details
tty.msg(
f"using {target.provider_type.value}", MessageType.ALERT, debug=True
)
tty.msg("\nsearch parameters", debug=True)
tty.msg(target.metadata.as_dict, debug=True)
tty.msg("", debug=True)
# find match for target
matches = []
try:
matches = target.query()
except MnamerNotFoundException:
tty.msg("no matches found", MessageType.ALERT)
except MnamerNetworkException:
tty.msg("network error", MessageType.ALERT)
if not matches and settings.no_guess:
tty.msg("skipping (--no-guess)", MessageType.ALERT)
continue
try:
if settings.batch:
match = matches[0] if matches else target.metadata
elif not matches:
match = tty.confirm_guess(target.metadata)
else:
tty.msg("results")
match = tty.prompt(matches)
except MnamerSkipException:
tty.msg("skipping (user request)", MessageType.ALERT)
continue
except MnamerAbortException:
tty.msg("aborting (user request)", MessageType.ERROR)
break
target.metadata.update(match)
# sanity check move
if target.destination == target.source:
tty.msg(
f"skipping (source and destination paths are the same)",
MessageType.ALERT,
)
continue
if settings.no_overwrite and target.destination.exists():
tty.msg(
f"skipping (--no-overwrite)", MessageType.ALERT,
)
continue
tty.msg(
f"moving to {target.destination.absolute()}", MessageType.SUCCESS,
)
# rename and move file
if settings.test:
success_count += 1
continue
try:
target.relocate()
except MnamerException:
tty.msg("FAILED!", MessageType.ERROR)
else:
tty.msg("OK!", MessageType.SUCCESS)
success_count += 1
# report results
if success_count == 0:
message_type = MessageType.ERROR
elif success_count == total_count:
message_type = MessageType.SUCCESS
else:
message_type = MessageType.ALERT
tty.msg(
f"\n{success_count} out of {total_count} files processed successfully",
message_type,
)
def _process_targets(self) -> None:
for target in self.targets:
self._announce_file(target)
self._list_details(target)
# find match for target
matches = []
try:
matches = target.query()
except MnamerNotFoundException:
tty.msg("no matches found", MessageType.ALERT)
except MnamerNetworkException:
tty.msg("network error", MessageType.ALERT)
if not matches and self.settings.no_guess:
tty.msg("skipping (--no-guess)", MessageType.ALERT)
continue
try:
if self.settings.batch:
match = matches[0] if matches else target.metadata
elif not matches:
match = tty.metadata_guess(target.metadata)
else:
match = tty.metadata_prompt(matches)
except MnamerSkipException:
tty.msg("skipping (user request)", MessageType.ALERT)
continue
except MnamerAbortException:
tty.msg("aborting (user request)", MessageType.ERROR)
break
target.metadata.update(match)
if (is_subtitle(target.metadata.container)
and not target.metadata.language_sub):
if self.settings.batch:
tty.msg(
"skipping (subtitle language can't be detected)",
MessageType.ALERT,
)
continue
try:
target.metadata.language_sub = tty.subtitle_prompt()
except MnamerSkipException:
tty.msg("skipping (user request)", MessageType.ALERT)
continue
except MnamerAbortException:
tty.msg("aborting (user request)", MessageType.ERROR)
break
# sanity check move
if target.destination == target.source:
tty.msg(
"skipping (source and destination paths are the same)",
MessageType.ALERT,
)
continue
if self.settings.no_overwrite and target.destination.exists():
tty.msg(
"skipping (--no-overwrite)",
MessageType.ALERT,
)
continue
self._rename_and_move_file(target)
def _ensure_targets(self) -> None:
if not self.targets:
tty.msg("", debug=True)
tty.msg("no media files found", MessageType.ALERT)
raise SystemExit(0)
def launch(self) -> None:
tty.msg("Starting mnamer", MessageType.HEADING)
self._ensure_targets()
self._process_targets()
self._report_results()
def _print_configuration(self) -> None:
tty.msg("\nsystem", debug=True)
tty.msg(SYSTEM, debug=True)
tty.msg("\nsettings", debug=True)
tty.msg(self.settings.as_dict(), debug=True)
tty.msg("\ntargets", debug=True)
tty.msg(self.targets or [None], debug=True)
