Add the feature of predefined template and regexes

3 files changed, 59 insertions, 1 deletions

diff --git a/patternutils/commands/pmatch.py b/patternutils/commands/pmatch.py
index da32f9c..5633c29 100644
--- a/patternutils/commands/pmatch.py
+++ b/patternutils/commands/pmatch.py
@@ -10,6 +10,7 @@ from typing import Generator, List, Iterator, Pattern, Any, Tuple, Dict, Match
 import click
 
 from .. import match as mat
+from .. import config
 
 
 def walk(dir: str, recursive: bool=False, match_directories: bool=False) -> Generator[os.DirEntry, None, None]:
@@ -58,6 +59,8 @@ def apply_regex(
               help='exit if a name does not match the INPUT_PATTERN')
 @click.option('--match-full-path', is_flag=True, default=False,
               help='match on full path instead of only filename')
+@click.option('-p', '--use-predefined-pattern', is_flag=True, default=False,
+              help='use predefinged reges pattern')
 @click.option('-v', '--verbose', is_flag=True, default=False,
               help='explain what is being done')
 @click.version_option()
@@ -69,9 +72,13 @@ def pmatch(
     recursive: bool,
     abort_on_no_match: bool,
     match_full_path: bool,
+    use_predefined_pattern: bool,
     verbose: bool,
 ) -> None:
 
+    if use_predefined_pattern:
+        regex_pattern, _ = config.load_template(regex_pattern, 'pmatch')
+
     try:
         regex_pattern_c = re.compile(regex_pattern)
     except re.error as e:
diff --git a/patternutils/commands/targs.py b/patternutils/commands/targs.py
index 59a8ab3..c1db5e4 100644
--- a/patternutils/commands/targs.py
+++ b/patternutils/commands/targs.py
@@ -15,6 +15,7 @@ from typing import (
 import click
 
 from .. import template
+from .. import config
 
 
 def run_command(
@@ -41,6 +42,8 @@ def run_command(
 @click.option('-v', '--verbose', is_flag=True, default=False,
               help='explain what is being done')
 @click.option('-P', '--max-procs', default=1, type=click.IntRange(min=0, clamp=True))
+@click.option('-p', '--use-predefined-template', is_flag=True, default=False,
+              help='use predefined template')
 @click.option('--redirect-stdout', is_flag=True, default=False,
               help='Redirect stdout to /dev/null')
 @click.option('--redirect-stderr', is_flag=True, default=False,
@@ -55,6 +58,7 @@ def targs(
     max_procs: int,
     redirect_stdout: bool,
     redirect_stderr: bool,
+    use_predefined_template: bool,
     verbose: bool,
 ) -> None:
 
@@ -64,7 +68,9 @@ def targs(
         click.secho('JSON not valid!', fg='red', err=True)
         raise SystemExit
 
-    use_jinja = True if template_engine == 'jinja2' else False
+    if use_predefined_template:
+        template_engine, comand = config.load_template(command, 'targs')
+
     try:
         render = template.get_render_function(command, engine=template_engine)
     except RuntimeError:
diff --git a/patternutils/config.py b/patternutils/config.py
new file mode 100644
index 0000000..a8feb36
--- /dev/null
+++ b/patternutils/config.py
@@ -0,0 +1,45 @@
+import os.path
+import glob
+
+import click
+
+from typing import Tuple, List, Union
+
+CONFIG_PATH = click.get_app_dir('patternutils')
+
+
+def load_template(template_name: str, command_name: str) -> Tuple[str, str]:
+    try:
+        files = find_templates(template_name, command_name)
+        if len(files) == 1:
+            with open(files[0], encoding='utf-8') as f:
+                template_lines = f.read()
+        elif len(files) == 0:
+            raise FileNotFoundError
+        elif len(files) > 1:
+            click.secho('More then one file found!', fg='red', err=True)
+            raise SystemExit
+    except FileNotFoundError:
+        raise
+    else:
+        output_pattern = template_lines.replace('\n', '')
+        template_file = files[0]
+        if template_file.endswith('regex'):
+            template_engine = 'regex'
+        elif template_file.endswith('jinja2'):
+            template_engine = 'jinja'
+        else:
+            template_engine = 'python'
+
+    return output_pattern, template_engine
+
+
+def find_templates(template_name: str, command_name: Union[str, None]) -> List[str]:
+    template_glob = '.'.join((template_name, '*'))
+
+    if command_name is None:
+        command_glob = '*'
+    else:
+        command_glob = os.path.join(CONFIG_PATH, command_name)
+
+    return glob.glob(os.path.join(command_glob, template_glob))