Added key option to tmv

1 files changed, 7 insertions, 4 deletions

diff --git a/patternutils/commands/tmv.py b/patternutils/commands/tmv.py
index 7d0be21..8147c11 100644
--- a/patternutils/commands/tmv.py
+++ b/patternutils/commands/tmv.py
@@ -24,6 +24,8 @@ def move_file(old_filename: str, new_filename: str) -> None:
               help='force rename')
 @click.option('-t', '--template-engine', default='python', type=click.Choice(['python', 'jinja2']),
               help='select the template engine to use')
+@click.option('-k', '--key', default='_abspath',
+              help='select the key to use as the source file')
 @click.option('-p', '--use-predefined-template', is_flag=True, default=False,
               help='use predefined template')
 @click.option('--read-from', default='-', type=click.File('r'))
@@ -43,6 +45,7 @@ def tmv(
     abort_on_path_exist: bool,
     use_predefined_template: bool,
     read_from: click.File,
+    key: str,
     dry_run: bool,
     verbose: bool,
 ) -> None:
@@ -50,9 +53,9 @@ def tmv(
     """Rename files based on the OUTPUT_PATTERN.
 
        It read a JSON list containing objects. The object needs to have a
-       key named _abspath, which points to the file to move. The rest of the
-       keys will be supplied to the templating engine to generate the new
-       filename.
+       key named as the option KEY (default: _abspath), which points to
+       the file to move. The rest of the keys will be supplied to the
+       templating engine to generate the new filename.
     """
 
     try:
@@ -77,7 +80,7 @@ def tmv(
         click.secho('jinja2 not installed!', fg='red', err=True)
         raise SystemExit
 
-    new_filenames = dict((entry['_abspath'], render(**entry)) for entry in data)
+    new_filenames = dict((entry[key], render(**entry)) for entry in data)
 
     has_existing = False
     non_existing = {}