✨ support cli option -e, user defined extension. resolves #257

Author: chfw

moban/constants.py

@@ -37,7 +37,9 @@
 LABEL_VERSION = "version"
 LABEL_GROUP = "group"
 LABEL_DEFINE = "define"
+LABEL_EXTENSION = "extension"
 CLI_DICT = "user_dict"
+EXTENSION_DICT = "user_extensions"
 
 DEFAULT_CONFIGURATION_DIRNAME = ".%s.cd" % PROGRAM_NAME
 DEFAULT_TEMPLATE_DIRNAME = ".%s.td" % PROGRAM_NAME

moban/core/moban_factory.py

@@ -22,7 +22,13 @@ def __init__(self):
         self.options_registry = {}
 
     def register_extensions(self, extensions):
-        self.extensions.update(extensions)
+        for template_type in extensions.keys():
+            if template_type in self.extensions:
+                self.extensions[template_type] = self.extensions[
+                    template_type
+                ].union(extensions[template_type])
+            else:
+                self.extensions[template_type] = extensions[template_type]
 
     def register_options(self, template_types):
         # need the value of 'template_types'

moban/main.py

@@ -12,6 +12,7 @@
 import logging
 import argparse
 import logging.config
+from collections import defaultdict
 
 from moban import (
     core,
@@ -41,6 +42,7 @@ def main():
     options[constants.CLI_DICT] = handle_custom_variables(
         options.pop(constants.LABEL_DEFINE)
     )
+    handle_custom_variables(options.pop(constants.LABEL_EXTENSION))
     OPTIONS.update(options)
     handle_verbose(options[constants.LABEL_VERBOSE])
 
@@ -155,6 +157,12 @@ def create_parser():
         nargs="+",
         help="to take a list of VAR=VALUEs",
     )
+    parser.add_argument(
+        "-e",
+        "--%s" % constants.LABEL_EXTENSION,
+        nargs="+",
+        help="to add an extension to TEMPLATE_TYPE=EXTENSION_NAME",
+    )
     return parser
 
 
@@ -265,6 +273,16 @@ def handle_custom_variables(list_of_definitions):
     return custom_data
 
 
+def handle_custom_extensions(list_of_definitions):
+    user_extensions = defaultdict(set)
+    if list_of_definitions:
+        for definition in list_of_definitions:
+            key, value = definition.split("=")
+            user_extensions[key].append(value)
+
+    core.ENGINES.register_options(user_extensions)
+
+
 def handle_verbose(verbose_level):
     if verbose_level > len(LOG_LEVEL):
         verbose_level = 3