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 fn(*args):
Target.reset_providers()
for arg in args:
sys.argv.append(arg)
try:
run()
except SystemExit as e:
code = e.code
else:
code = 0
return E2EResult(
code,
strip_format(capsys.readouterr().out.strip()) +
strip_format(capsys.readouterr().err.strip()),
)
def fn(*args):
Target.reset_providers()
for arg in args:
sys.argv.append(arg)
try:
run(load_configuration=False)
except SystemExit as e:
code = e.code
else:
code = 0
out = strip_format(capsys.readouterr().out.strip())
out += strip_format(capsys.readouterr().err.strip())
with open(E2E_LOG, "a+") as fp:
fp.write("=" * 10 + "\n")
fp.write(request.node.name + "\n")
fp.write("-" * 10 + "\n")
fp.write(out + "\n\n")
return E2EResult(code, out)
def fn(*args):
Target.reset_providers()
out = ""
code = 0
for arg in args:
sys.argv.append(arg)
try:
settings = SettingStore()
settings.load()
Cli(settings).launch()
except MnamerException as e:
out += str(e)
code = 2
except SystemExit as e:
code = e.code
out += strip_format(capsys.readouterr().out.strip())
out += strip_format(capsys.readouterr().err.strip())
with open(E2E_LOG, "a+") as fp:
fp.write("=" * 10 + "\n")
fp.write(request.node.name + "\n")
fp.write("-" * 10 + "\n")
fp.write(out + "\n\n")
return E2EResult(code, out)
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 __init__(self, settings: SettingStore):
self.settings = settings
self.targets = Target.populate_paths(self.settings)
tty.configure(self.settings)
self._handle_directives()
self._print_configuration()