mirror of
https://github.com/rjNemo/vf-site
synced 2026-06-06 01:16:38 +00:00
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
import tomllib
|
|
from distutils.dir_util import copy_tree
|
|
from time import perf_counter
|
|
|
|
from loguru import logger
|
|
|
|
from lib.config import load
|
|
from lib.engine import FileSystemRenderer
|
|
|
|
|
|
def main():
|
|
start = perf_counter()
|
|
try:
|
|
config = load()
|
|
except FileNotFoundError:
|
|
logger.error("The configuration file 'config.toml' was not found. Please verify it exists at the root level")
|
|
sys.exit()
|
|
|
|
destination_path = config.out_dir
|
|
fs = FileSystemRenderer(config.templates_dir)
|
|
|
|
logger.info(f"🏁 Start building {config.name}")
|
|
|
|
clean_dist(destination_path)
|
|
|
|
for page in os.scandir(config.templates_dir):
|
|
if page.is_file():
|
|
with open(os.path.join(destination_path, page.name), "w") as fd:
|
|
data = parse_data(page, config.data_dir)
|
|
logger.info(f"📃Render '{page.name}'")
|
|
fd.write(fs.render(page.name, data))
|
|
|
|
logger.info("⏩ Start copying staticfiles to build")
|
|
copy_tree(config.static_dir, os.path.join(config.out_dir))
|
|
|
|
end = perf_counter()
|
|
logger.info(f"🎉 Done… in {(end - start) * 1000:.2f} ms")
|
|
|
|
|
|
def parse_data(page: os.DirEntry, data_dir: str) -> dict:
|
|
data_file_path = os.path.join(data_dir, f"{page.name.split('.')[0]}.toml")
|
|
data = {}
|
|
if os.path.isfile(data_file_path):
|
|
with open(data_file_path, "rb") as f:
|
|
data = tomllib.load(f)
|
|
return data
|
|
|
|
|
|
def clean_dist(destination_path: str):
|
|
if os.path.exists(destination_path) and os.path.isdir(destination_path):
|
|
shutil.rmtree(destination_path)
|
|
os.mkdir(destination_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|