summaryrefslogtreecommitdiff
path: root/__main__.py
blob: fd26157e4ddda5d8410640d7cb41098985ccb08f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import os.path
import sys
import jinja2

from .config import Config
from .modules import blog_generate
from .modules import comic_generate


def init_jinja_env() -> jinja2.Environment:
    jinja_env = jinja2.Environment(loader=jinja2.FileSystemLoader(Config.TEMPLATES_SOURCE_DIR))
    return jinja_env


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__':
    main(sys.argv[1], bool(int(sys.argv[2])) if len(sys.argv) > 2 else False)