Feat: Enable configfile loading

bopytex/bopytex.py

@@ -2,14 +2,3 @@ import bopytex.default_config as DEFAULT
 from bopytex.service import orcherstrator
 
 
-def bopytex(**options):
-
-    config = options.copy()
-    config["jinja2"] = {}
-    config["jinja2"]["environment"] = DEFAULT.jinja2_env
-
-    orcherstre = orcherstrator(
-        config, planner=DEFAULT.planner, dispatcher=DEFAULT.dispatcher
-    )
-    for message in orcherstre:
-        yield message

bopytex/default_config.py

@@ -1,4 +1,4 @@
-import jinja2
+import jinja2 as j2
 from bopytex.planner.generate_compile_join_planner import planner
 from bopytex.worker import Dispatcher
 from bopytex.worker.clean import clean
@@ -6,7 +6,9 @@ from bopytex.worker.compile import latexmk
 from bopytex.worker.generate import generate
 from bopytex.worker.join_pdf import pdfjam
 
-jinja2_env = jinja2.Environment(loader=jinja2.FileSystemLoader("./"))
+jinja2 = {
+    "environment": j2.Environment(loader=j2.FileSystemLoader("./")),
+}
 
 dispatcher = Dispatcher(
     {"GENERATE": generate, "COMPILE": latexmk, "JOIN": pdfjam, "CLEAN": clean}

bopytex/script.py

@@ -71,14 +71,21 @@ logger.addHandler(steam_handler)
     help="Activate correction and compile only from existing subjects",
 )
 @click.option(
-    "-c",
+    "-C",
     "--corr",
     is_flag=True,
     default=False,
     help="Create and compile correction while making subjects",
 )
+@click.option(
+    "-c",
+    "--configfile",
+    type=str,
+    default="bopyptex_config.py",
+    help="Config file path",
+)
 def new(**options):
-    for message in main(**options):
+    for message in main(options):
         try:
             assert message.status == 0
         except AssertionError:

bopytex/service.py

@@ -5,8 +5,10 @@
 Producing then compiling templates
 """
 
+import importlib.util
+from pathlib import Path
 from bopytex.scheduler import Scheduler
-import bopytex.default_config as DEFAULT
+from bopytex import default_config
 
 
 def orcherstrator(
@@ -22,18 +24,63 @@ def orcherstrator(
     for message in scheduler.backlog():
         yield message
 
+
+def load_module(modulefile: str):
+    spec = importlib.util.spec_from_file_location("module.name", modulefile)
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    return module
+
+
+def clean_vars_keys(
+    vars: dict,
+    keys: list[str] = [
+        "__name__",
+        "__doc__",
+        "__package__",
+        "__loader__",
+        "__spec__",
+        "__file__",
+        "__cached__",
+        "__builtins__",
+    ],
+) -> dict:
+    new_dict = vars.copy()
+    for k in keys:
+        del new_dict[k]
+    return new_dict
+
+
+def get_config(options: dict) -> dict:
+    """ Look for options["configfile"] to load it with default_config and options """
+    try:
+        options["configfile"]
+    except KeyError:
+        local_config = {}
+    else:
+        if Path(options["configfile"]).exists():
+            local_config = vars(load_module(options["configfile"]))
+            local_config = clean_vars_keys(local_config)
+        else:
+            local_config = {}
+
+    config = clean_vars_keys(vars(default_config))
+    config.update(local_config)
+    config.update(options)
+    return config
+
+
 def main(**options):
 
-    config = options.copy()
-    config["jinja2"] = {}
-    config["jinja2"]["environment"] = DEFAULT.jinja2_env
+    config = get_config(options)
 
     orcherstre = orcherstrator(
-        config, planner=DEFAULT.planner, dispatcher=DEFAULT.dispatcher
+        config, planner=default_config.planner, dispatcher=default_config.dispatcher
     )
     for message in orcherstre:
         yield message
 
+
 # -----------------------------
 # Reglages pour 'vim'
 # vim:set autoindent expandtab tabstop=4 shiftwidth=4: