modules/main_generate.py


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

import os.path
import shutil
import jinja2

from ..config import Config
import os.path
import shutil

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)

    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}))

    shutil.copytree(Config.CODENAMES_ASSETS_SOURCE_DIR, output_root_path + Config.ASSETS_IMPORT_PATH)