summaryrefslogtreecommitdiff
path: root/__main__.py
blob: 16c224395b41c0fb80f25bafbb5d8754e145301f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import os.path
import sys
import jinja2

from .config import Config
from .modules import blog_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"

    jinja_env = init_jinja_env()
    blog_generate.generate(jinja_env, os.path.join(output_root_path, "blog"), local)


if __name__ == '__main__':
    main(sys.argv[1], bool(int(sys.argv[2])) if len(sys.argv) > 2 else False)