1 files changed, 5 insertions, 4 deletions
diff --git a/modules/main_generate.py b/modules/main_generate.py
index 7137537..a166a93 100644
--- a/modules/main_generate.py
+++ b/modules/main_generate.py
@@ -11,16 +11,17 @@ import jinja2
 from ..config import Config
 
 
+def init(jinja_env: jinja2.Environment, local: bool):
+    jinja_env.globals.update(main={
+        "main_url": Config.MAIN_ROOT_URL,
+    })
+
 def generate(jinja_env: jinja2.Environment, output_root_path: str, local: bool):
     if os.path.exists(output_root_path):
         shutil.rmtree(output_root_path)
 
     os.mkdir(output_root_path)
 
-    jinja_env.globals.update(main={
-        "main_url": Config.MAIN_ROOT_URL,
-    })
-
     root_template = jinja_env.get_template("main.html.j2")
     with open(os.path.join(output_root_path, "index.html"), "w") as f:
         f.write(root_template.render({"url" : Config.MAIN_ROOT_URL, "title": Config.MAIN_NAME}))