comic basic layout

1 files changed, 8 insertions, 0 deletions

diff --git a/__main__.py b/__main__.py
index 16c2243..fd26157 100644
--- a/__main__.py
+++ b/__main__.py
@@ -4,6 +4,7 @@ import jinja2
 
 from .config import Config
 from .modules import blog_generate
+from .modules import comic_generate
 
 
 def init_jinja_env() -> jinja2.Environment:
@@ -14,9 +15,16 @@ def init_jinja_env() -> jinja2.Environment:
 def main(output_root_path: str, local: bool):
     if local:
         Config.BLOG_ROOT_URL = "http://localhost"
+        Config.COMIC_ROOT_URL = "http://localhost"
 
     jinja_env = init_jinja_env()
+    jinja_env.globals.update(site={
+        "assets_path": Config.ASSETS_IMPORT_PATH,
+        "assets_path_static": Config.ASSETS_IMPORT_PATH_STATIC,
+    })
+
     blog_generate.generate(jinja_env, os.path.join(output_root_path, "blog"), local)
+    comic_generate.generate(jinja_env, os.path.join(output_root_path, "comic"), local)
 
 
 if __name__ == '__main__':