def process(
self, names, filenames=None, destination_filename=None,
in_extention=None, get_content=False, metadata=None, keep=False):
is_merged = False
out_ext = in_extention
cmds_config = metatask.config.get("cmds", {})
cmds = []
for cmd in names:
if isinstance(cmd, str):
c = cmds_config.get(cmd)
if c is None:
raise Exception("Missing command '{0!s}' in `cmds`".format(cmd))
c["name"] = cmd
cmds.append(c)
else:
cmds.append(cmd)
filename = filenames[0] if isinstance(filenames, list) else filenames
if destination_filename is None:
destination_filename = filename
if filename is not None:
dst, _, types, _ = self.destination_filename(names, filename, metadata=metadata)
if types == set():
return None, None
if types == set(["rename"]):
if filename != dst:
directory = os.path.dirname(dst)
with self.lock:
if directory != '' and not os.path.exists(directory):
os.makedirs(directory)
if not os.path.exists(dst):
shutil.move(filename, dst)
return None, None
original_filename = filename
if cmds[0].get("inplace") is True or cmds[0].get("type" == "metadata"):
if in_extention is None:
out_name = NamedTemporaryFile(mode='w+b').name
else:
out_name = NamedTemporaryFile(
mode='w+b',
suffix='.' + in_extention
).name
shutil.copyfile(filename, out_name)
filename = out_name
for no, cmd in enumerate(cmds):
if isinstance(cmd, str):
cmd = dict(cmd=cmd)
if cmd.get('type') == 'rename':
destination_filename = self._rename(cmd, destination_filename, metadata)
if cmd.get("type") == "metadata":
value = cmd.get('value_format')
value = self._format(
filename,
cmd.get("value_get"),
cmd.get("value_format"),
)
subprocess.check_output(['exiftool', '-{0!s}={1!s}'.format(cmd.get('tag'), value), filename])
else:
if 'out_ext' in cmd:
out_ext = cmd['out_ext']
inplace = cmd.get('inplace', False)
cmd_cmd = cmd.get('cmd')
if inplace:
out_name = filename
else:
if out_ext is None:
out_name = NamedTemporaryFile(mode='w+b').name
else:
out_name = NamedTemporaryFile(
mode='w+b',
suffix='.' + out_ext
).name
params = {}
if metadata is not None:
params.update(metadata)
# it's a merge
if not is_merged and isinstance(filenames, list):
params["in"] = " ".join([
"'{0!s}'".format(f.replace("'", "'\"'\"'")) for f in filenames
])
# do the merge only one time
is_merged = True
elif filename is not None:
params["in"] = "'{0!s}'".format(filename.replace("'", "'\"'\"'"))
if not inplace:
params["out"] = "'{0!s}'".format(out_name.replace("'", "'\"'\"'"))
try:
cmd_cmd = cmd_cmd.format(**params)
except Exception:
print("Error in {name}: {cmd}, with {params}".format(
name=cmd["name"], cmd=cmd_cmd, params=params))
raise
self.progress.emit(no, cmd["name"], cmd_cmd, cmd)
if self.cancel is True:
return None, None
print("{name}: {cmd}".format(
name=bashcolor.colorize(cmd["name"], bashcolor.BLUE), cmd=cmd_cmd
))
subprocess.check_call(cmd_cmd, shell=True)
if filename != original_filename and not inplace:
os.unlink(filename)
filename = out_name
if get_content:
content = None
if os.path.exists(filename):
with open(filename, encoding='utf-8') as f:
content = f.read()
if original_filename is None or original_filename != filename:
os.unlink(filename)
return content, out_ext
else:
if out_ext is not None:
destination_filename = "{0!s}.{1!s}".format(re.sub(
r"\.[a-z0-9A-Z]{2,5}$", "",
destination_filename
), out_ext)
if filename != destination_filename:
directory = os.path.dirname(destination_filename)
with self.lock:
if directory != "" and not os.path.exists(directory):
os.makedirs(directory)
if filename != destination_filename and (
# apply on new file
filenames is None or
# apply a transformation on the file
len(filenames) == 1 and filenames[0] == destination_filename or
not os.path.exists(destination_filename)
):
print("{name}: {file}".format(
name=bashcolor.colorize('copy', bashcolor.BLUE),
file=bashcolor.colorize(destination_filename, bashcolor.YELLOW),
))
shutil.move(filename, destination_filename)
if not keep and original_filename != destination_filename:
if isinstance(filenames, list):
for f in filenames:
if f != filename:
os.unlink(f)
elif original_filename is not None and original_filename != filename:
os.unlink(original_filename)
return destination_filename, out_ext
