98 lines
4.0 KiB
Python
98 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from pathlib import Path
|
|
import frontmatter
|
|
import markdown
|
|
from jinja2 import Environment, FileSystemLoader
|
|
import shutil
|
|
from config import Config
|
|
from logger import logger
|
|
|
|
content_dir = Path('content')
|
|
template_dir = Path('templates')
|
|
output_dir = Path(Config.OUTPUT_DIR)
|
|
img_dir = Path('public/images')
|
|
assets_css_dir = Path(output_dir / 'static/css')
|
|
assets_js_dir = Path(output_dir / 'static/js')
|
|
static_dir = Path('static')
|
|
header_image = Config.HEADER_IMAGE or '/static/header.jpg'
|
|
|
|
class ContentItemPrototype:
|
|
def render_content(self):
|
|
if self.image_src_file.exists():
|
|
shutil.copyfile(self.image_src_file, output_dir / self.image)
|
|
if self.custom_css_src_file.exists():
|
|
shutil.copyfile(self.custom_css_src_file,output_dir / self.custom_css)
|
|
if self.custom_js_src_file.exists():
|
|
shutil.copyfile(self.custom_js_src_file, output_dir / self.custom_js)
|
|
with self.content_output_path.open("w", encoding="utf-8") as f:
|
|
f.write(content_item_template.render(content_item = self, page_title = self.title))
|
|
|
|
def __init__(self, md_file):
|
|
self.slug = md_file.stem
|
|
self.data = frontmatter.load(md_file)
|
|
self.html = markdown.markdown(self.data.content)
|
|
self.preview = self.html[:300] + "<a href=" + f"{self.slug}.html" + ">... read more</a>"
|
|
self.title = self.data.get("title", self.slug)
|
|
self.omit_second_title = self.data.get("omit_second_title", False)
|
|
self.date = self.data.get("data", "2000-01-01T00:00:00+03:00")
|
|
self.content_output_path = output_dir / f"{self.slug}.html"
|
|
self.image_src_file = Path(f"{content_dir}/{md_file.stem}.jpg")
|
|
self.image = f"images/{md_file.stem}.jpg" if self.image_src_file.exists() else None
|
|
self.custom_css_src_file = Path(f"{content_dir}/{md_file.stem}.css")
|
|
self.custom_css = f"static/css/{md_file.stem}.css" if self.custom_css_src_file.exists() else None
|
|
logger.debug(f"Custom CSS: {self.slug}: {self.custom_css}")
|
|
self.custom_js_src_file = Path(f"{content_dir}/{md_file.stem}.js")
|
|
self.custom_js = f"static/js/{md_file.stem}.js" if self.custom_js_src_file.exists() else None
|
|
logger.debug(f"Custom JS: {self.slug}: {self.custom_js}")
|
|
# Special handling for the 'about' item
|
|
if self.slug == 'about':
|
|
self.image = 'static/about.jpg'
|
|
|
|
logger.info(f"Building the '{Config.MAIN_PAGE_TITLE}' site.")
|
|
|
|
# Recreate the output dir if needed
|
|
if output_dir.exists():
|
|
shutil.rmtree(output_dir)
|
|
output_dir.mkdir()
|
|
|
|
# Create public subdirs
|
|
subdirs = [img_dir, assets_css_dir, assets_js_dir]
|
|
for subdir in subdirs:
|
|
subdir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Copy static files if exist
|
|
if static_dir.exists():
|
|
shutil.copytree(static_dir, output_dir / "static", dirs_exist_ok=True)
|
|
|
|
# Prepare template rendering engine
|
|
env = Environment(loader=FileSystemLoader(str(template_dir)))
|
|
env.globals['header_image'] = header_image
|
|
index_template = env.get_template("index.html")
|
|
content_item_template = env.get_template("content_item.html")
|
|
|
|
# Parse the content files
|
|
content_items = []
|
|
for md_file in content_dir.glob("*.md"):
|
|
current_content_item = ContentItemPrototype(md_file)
|
|
current_content_item.render_content()
|
|
content_items.append({
|
|
"slug": current_content_item.slug,
|
|
"title": current_content_item.title,
|
|
"date": current_content_item.date,
|
|
"preview": markdown.markdown(current_content_item.preview),
|
|
"image": current_content_item.image,
|
|
})
|
|
|
|
# Render the index file
|
|
with (output_dir / "index.html").open("w", encoding="utf-8") as f:
|
|
f.write(index_template.render(page_title = Config.MAIN_PAGE_TITLE, content_items=content_items))
|
|
|
|
# Render the about file
|
|
about_content = ContentItemPrototype(Path('static/about.md'))
|
|
about_content.render_content()
|
|
|
|
# Move 'robots.txt' into output_dir
|
|
shutil.copyfile(static_dir / 'robots.txt', output_dir / 'robots.txt')
|
|
|
|
logger.info(f"Created {len(content_items)} content items.") |