mirror of
https://github.com/rjNemo/vf-site
synced 2026-06-06 01:16:38 +00:00
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
import tomllib
|
|
from distutils.dir_util import copy_tree
|
|
from time import perf_counter
|
|
|
|
from rich import print
|
|
|
|
from lib.config import Config, load
|
|
from lib.engine import FileSystemRenderer
|
|
|
|
|
|
def main():
|
|
start = perf_counter()
|
|
config = parse_config()
|
|
|
|
fs = FileSystemRenderer(config)
|
|
|
|
print(f"🏁 Start building [italic red]{config.name}[/italic red]")
|
|
clean_dist(config.out_dir)
|
|
|
|
for page in os.scandir(config.templates_dir):
|
|
if page.is_file():
|
|
with open(os.path.join(config.out_dir, page.name), "w") as fd:
|
|
data = parse_data(page, config.data_dir)
|
|
print(f"📃 Render [italic cyan]{page.name}[/italic cyan]")
|
|
fd.write(fs.render(page.name, data))
|
|
|
|
print("⏩ Copy static assets to build")
|
|
copy_tree(config.static_dir, os.path.join(config.out_dir))
|
|
|
|
end = perf_counter()
|
|
print(f"🎉 Done… in [italic red]{(end - start) * 1000:.2f} ms[/italic red]")
|
|
|
|
|
|
def parse_config() -> Config:
|
|
try:
|
|
config = load()
|
|
except FileNotFoundError:
|
|
print(
|
|
"🛑 configuration file [italic red]'config.toml'[/italic red] was not found. "
|
|
"Please verify it exists at the root level"
|
|
)
|
|
sys.exit(1)
|
|
else:
|
|
return config
|
|
|
|
|
|
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()
|