197 lines
7.7 KiB
Python
197 lines
7.7 KiB
Python
import csv
|
||
import os
|
||
import shutil
|
||
from pathlib import Path
|
||
|
||
from watchdog.observers import Observer
|
||
from watchdog.events import FileSystemEventHandler
|
||
from jinja2 import Environment, FileSystemLoader, select_autoescape
|
||
import yaml
|
||
import frontmatter
|
||
import markdown2
|
||
import fire
|
||
import minify_html
|
||
|
||
from common.functions import *
|
||
|
||
|
||
def load_translations(file_path, _dict):
|
||
with open(file_path) as f:
|
||
reader = csv.DictReader(f, delimiter='|') #
|
||
for row in reader: # проходим по строкам csv, каждая из которых является словарём
|
||
_dict[row['id']] = row # добавляем ключ - значение ключа id, значение - словарь row
|
||
del row[
|
||
'id'] # удаляем ключ по названием "id" из словаря row, так как его значение уже является ключом в словаре _dict
|
||
|
||
|
||
# Функции, доступные в теме
|
||
def translate(id, language):
|
||
global translations
|
||
return translations[id][language]
|
||
|
||
|
||
translations = {} # здесь будут переводы от темы и от сайта
|
||
config = yaml.safe_load(read_file('config.yaml')) # Читаем конфиг сайта
|
||
running = False # нужно для проверки
|
||
|
||
|
||
# класс для watchdog
|
||
# Во время разработки, при обнаружении изменений в папках content и themes/{config['theme']},
|
||
# перегенерировать папку public
|
||
class Develop(FileSystemEventHandler):
|
||
def on_modified(self, event):
|
||
print(f'event type: {event.event_type} path : {event.src_path}')
|
||
franca()
|
||
|
||
def on_created(self, event):
|
||
print(f'event type: {event.event_type} path : {event.src_path}')
|
||
franca()
|
||
|
||
def on_deleted(self, event):
|
||
print(f'event type: {event.event_type} path : {event.src_path}')
|
||
franca()
|
||
|
||
|
||
# Функция для запуска разработки
|
||
def develop(prod):
|
||
global running
|
||
if not prod and not running:
|
||
event_handler = Develop()
|
||
observer = Observer()
|
||
observer.schedule(event_handler, path='content', recursive=True)
|
||
observer.schedule(event_handler, path='assets', recursive=True)
|
||
observer.schedule(event_handler, path='static', recursive=True)
|
||
observer.schedule(event_handler, path='config.yaml')
|
||
observer.schedule(event_handler, path=f"themes/{config['theme']}", recursive=True)
|
||
observer.start()
|
||
running = True
|
||
try:
|
||
start_httpd(directory='public')
|
||
except KeyboardInterrupt:
|
||
pass
|
||
|
||
|
||
# Функция для генерации сайта
|
||
def franca(prod=False):
|
||
if not prod:
|
||
config['base_url'] = "http://127.0.0.1:8000"
|
||
# 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")
|
||
index = templates.get_template("index.j2")
|
||
|
||
# Load translations
|
||
global 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']:
|
||
content_dir = Path(f"content/{language}")
|
||
posts[language] = {}
|
||
|
||
# Create posts dict
|
||
for item in list(content_dir.rglob("*.md")):
|
||
post_data = frontmatter.loads(read_file(item))
|
||
|
||
post_path = str(item).replace(
|
||
"content", "public").rstrip(".md")
|
||
os.makedirs(post_path, exist_ok=True)
|
||
|
||
content = markdown2.markdown(post_data.content, extras=['fenced-code-blocks'])
|
||
description = content.partition('<!--more-->')[0]
|
||
content = "{% import 'shortcodes.j2' as shortcodes %}" + content
|
||
|
||
url = post_path.replace(f"public/{language}", "")
|
||
section = "/" if len(url.split('/')) == 2 else url.split('/')[1]
|
||
date = post_data['date']
|
||
posts[language].setdefault(date, {})
|
||
posts[language][date].setdefault(section, {})
|
||
|
||
posts[language][date][section][url] = {
|
||
'title': post_data['title'],
|
||
'description': description,
|
||
'date': date,
|
||
'content': templates.from_string(content).render()
|
||
}
|
||
|
||
image = post_data.get('image', None)
|
||
if image:
|
||
os.makedirs(os.path.dirname(f'public{image}'), exist_ok=True)
|
||
filename = image.split('/')[-1].split('.')[0]
|
||
|
||
create_thumbnail(f'assets{image}', f'public/images/{filename}_600.jpg', 600)
|
||
posts[language][date][section][url]['image'] = f'/images/{filename}_600.jpg'
|
||
|
||
create_thumbnail(f'assets{image}', f'public/images/{filename}_400.jpg', 400)
|
||
posts[language][date][section][url]['thumbnail'] = f'/images/{filename}_400.jpg'
|
||
|
||
posts[language] = dict(sorted(posts[language].items(), reverse=True))
|
||
|
||
for date, sections in posts[language].items():
|
||
for section, urls in sections.items():
|
||
if section != "/":
|
||
html = base.render(config=config, section=section,
|
||
language=language, posts=posts)
|
||
|
||
write_file(f"public/{language}/{section}/index.html", minify_html.minify(html, minify_js=True))
|
||
|
||
for url, post in urls.items():
|
||
html = base.render(config=config, post=post,
|
||
language=language, url=url, posts=posts)
|
||
|
||
write_file(f"public/{language}{url}/index.html", minify_html.minify(html, minify_js=True))
|
||
|
||
html = base.render(config=config, posts=posts,
|
||
language=language, home=True)
|
||
|
||
write_file(f"public/{language}/index.html", minify_html.minify(html, minify_js=True))
|
||
|
||
# copy images/css/js from theme
|
||
shutil.copytree(f"themes/{config['theme']}/static/images", 'public/images', dirs_exist_ok=True)
|
||
css = read_file(f"themes/{config['theme']}/static/css/style.css")
|
||
minify_css('public/css/style.css', css)
|
||
if 'css_includes' in config:
|
||
for include in config['css_includes']:
|
||
css = read_file(f"themes/{config['theme']}/static/css/{include}")
|
||
minify_css(f'public/css/{include}', css)
|
||
if 'js_includes' in config:
|
||
for include in config['js_includes']:
|
||
copy_file(f"themes/{config['theme']}/static/js/{include}", 'public/js/')
|
||
|
||
# copy css/images from site static folder
|
||
if 'custom_css' in config:
|
||
shutil.copytree('static/css', 'public/css', dirs_exist_ok=True)
|
||
|
||
copy_file('static/logo.svg', 'public/')
|
||
copy_file('static/favicon.ico', 'public/')
|
||
|
||
# Write main index.html
|
||
html = index.render(config=config)
|
||
write_file('public/index.html', minify_html.minify(html, minify_js=True))
|
||
# Write robots.txt
|
||
robots_content = "User-agent: *"
|
||
if not config.get('search_engines', None) == "allow":
|
||
robots_content += "\nDisallow: /"
|
||
write_file('public/robots.txt', robots_content)
|
||
|
||
if prod:
|
||
if 'pagefind' in config:
|
||
os.system("npx pagefind --source public") # build search index
|
||
|
||
develop(prod)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
fire.Fire(franca)
|