140 lines
4.2 KiB
Python
140 lines
4.2 KiB
Python
import csv
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
import http.server
|
|
import socketserver
|
|
|
|
from jinja2 import Environment, FileSystemLoader, select_autoescape
|
|
import yaml
|
|
import frontmatter
|
|
import markdown
|
|
import fire
|
|
|
|
|
|
translations = {}
|
|
|
|
def copy_file(src, dst):
|
|
os.makedirs(os.path.dirname(dst), exist_ok=True)
|
|
shutil.copy2(src, dst)
|
|
|
|
|
|
def load_translations(file_path, _dict):
|
|
with open(file_path) as f:
|
|
reader = csv.DictReader(f, delimiter='|')
|
|
for row in reader:
|
|
_dict[row['id']] = row
|
|
del row['id']
|
|
|
|
|
|
def translate(id, language):
|
|
global translations
|
|
return translations[id][language]
|
|
|
|
|
|
def develop():
|
|
PORT = 8000
|
|
os.chdir('public')
|
|
handler = http.server.SimpleHTTPRequestHandler
|
|
server = socketserver.TCPServer(("", PORT), handler)
|
|
print("Server started at port 8000. Press CTRL+C to close the server.")
|
|
try:
|
|
server.serve_forever()
|
|
except KeyboardInterrupt:
|
|
server.server_close()
|
|
print("Server Closed")
|
|
|
|
|
|
def site_creator(prod=False):
|
|
global translations
|
|
# Read config of site
|
|
with open('config.yaml', 'r') as file:
|
|
config = yaml.safe_load(file)
|
|
|
|
# Load theme's jinja templates
|
|
templates = Environment(loader=FileSystemLoader(
|
|
f"themes/{config['theme']}/templates/"), autoescape=select_autoescape())
|
|
|
|
# Load base jinja template
|
|
base = templates.get_template("base.j2")
|
|
|
|
# Load translations
|
|
load_translations(f"themes/{config['theme']}/i18n.csv", translations)
|
|
load_translations("i18n.csv", translations)
|
|
|
|
# Add functions to base template
|
|
base.globals.update({"translate": translate})
|
|
|
|
# Remove public folder if exists before generating new content
|
|
shutil.rmtree('public', ignore_errors=True)
|
|
|
|
posts = {}
|
|
# Create new public folder
|
|
for language in config['languages']:
|
|
path = f"content/{language}"
|
|
content_dir = pathlib.Path(path)
|
|
content_files = content_dir.rglob("*.md")
|
|
posts[language] = {}
|
|
|
|
# Create posts dict
|
|
for item in list(content_files):
|
|
with open(item, 'r') as f:
|
|
data = frontmatter.loads(f.read())
|
|
|
|
post_path = str(item).replace(
|
|
"content", "public").rstrip(".md")
|
|
os.makedirs(post_path, exist_ok=True)
|
|
|
|
content = "{% import 'shortcodes.j2' as shortcodes %}" + \
|
|
markdown.markdown(data.content)
|
|
|
|
content = templates.from_string(content).render()
|
|
description = content.partition('<!--more-->')[0]
|
|
url = post_path.replace(f"public/{language}", "")
|
|
section = url.split('/')[1]
|
|
|
|
if not os.path.dirname(url) == "/":
|
|
posts[language].setdefault(section, {})
|
|
|
|
posts[language][section][url] = {
|
|
'title': data['title'],
|
|
'description': description,
|
|
'date': data['date'],
|
|
'content': content
|
|
}
|
|
|
|
image = data.get('image', None)
|
|
if image:
|
|
posts[language][section][url]['image'] = image
|
|
copy_file(f'assets{image}', f'public{image}')
|
|
|
|
for section, urls in posts[language].items():
|
|
html = base.render(config=config, section=section,
|
|
language=language, posts=posts)
|
|
|
|
with open(f"public/{language}/{section}/index.html", 'w') as f:
|
|
f.write(html)
|
|
|
|
for url, post in urls.items():
|
|
html = base.render(config=config, post=post,
|
|
language=language, url=url, posts=posts)
|
|
|
|
with open(f"public/{language}{url}/index.html", 'w') as f:
|
|
f.write(html)
|
|
|
|
html = base.render(config=config, posts=posts,
|
|
language=language, home=True)
|
|
|
|
with open(f"public/{language}/index.html", 'w') as f:
|
|
f.write(html)
|
|
|
|
shutil.copytree('static', 'public', dirs_exist_ok=True)
|
|
shutil.copytree(
|
|
f"themes/{config['theme']}/static", 'public', dirs_exist_ok=True)
|
|
if not prod:
|
|
develop()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
fire.Fire(site_creator)
|