#!/usr/bin/env python3
import fnmatch
import locale
import os
from dataclasses import dataclass
from datetime import datetime, timedelta
from html import escape
from pathlib import Path
from textwrap import dedent
from time import perf_counter
import mistune
from jinja2 import Environment as Env
from jinja2 import FileSystemLoader
from minicli import cli, run, wrap
from mistune.directives import DirectiveInclude
from PIL import Image
from slugify import slugify
# Useful for dates rendering within Jinja2.
locale.setlocale(locale.LC_ALL, "fr_FR.UTF-8")
HERE = Path(".")
DAVID = HERE / "david"
STATIC = HERE / ".." / "larlet-fr-static"
DOMAIN = "https://larlet.fr"
LOCAL_DOMAIN = "http://larlet.test:3579"
# Hardcoding publication at 12 in Paris timezone.
NORMALIZED_STRFTIME = "%Y-%m-%dT12:00:00+01:00"
TODAY = datetime.today() + timedelta(hours=6)
class MarkParser(mistune.InlineParser):
"""Parses `==foo==` as `foo`."""
MARK = (
r"(\={2})(?=[^\s*])("
r"(?:\\[\\*]|[^*])*"
r"(?:" + mistune.InlineParser.ESCAPE + r"|[^\s*]))\1"
)
RULE_NAMES = mistune.InlineParser.RULE_NAMES + ("mark",)
def parse_mark(self, m, state):
marker = m.group(1)
text = m.group(2)
return "mark", self.render(text, state)
class MarkRenderer(mistune.HTMLRenderer):
"""To use in conjunction with `MarkParser`."""
def mark(self, text):
return "" + text + ""
class ImgsWithSizesRenderer(MarkRenderer):
"""Renders images as
", 1)[0] + ""
def __lt__(self, other: "Page"):
if not isinstance(other, Page):
return NotImplemented
return self.date < other.date
@staticmethod
def all(source: Path, only_published=True, with_h2_anchors=True):
"""Retrieve all pages sorted by desc."""
page_list = []
md = markdown_with_h2_anchors if with_h2_anchors else markdown_with_img_sizes
for file_path in each_markdown_from(source):
result = md.read(file_path)
# Extract (and remove) the title from the generated page.
title, content = result.split("", 1)
h1_opening_size = len("
")
title = title[h1_opening_size:]
page = Page(title, content, file_path)
if only_published and page.is_draft:
continue
page_list.append(page)
return sorted(page_list, reverse=True)
@property
def is_draft(self):
return (
datetime(year=self.date.year, month=self.date.month, day=self.date.day)
> TODAY
)
@cli
def orphans():
"""Print out fragments not linked to any page."""
linked_fragments = []
for file_path in each_markdown_from(DAVID / "2020"):
for line in open(file_path).readlines():
if line.startswith(".. include:: fragments/"):
linked_fragments.append(line[len(".. include:: fragments/") : -1])
all_fragments = []
for file_path in each_markdown_from(DAVID / "2020" / "fragments"):
all_fragments.append(file_path[len("david/2020/fragments/") :])
for fragment_filename in set(all_fragments) - set(linked_fragments):
# Prepending path for easy command+click from fish.
print(f"Orphan: {DAVID / '2020' / 'fragments' / fragment_filename}")
@cli
def pages():
"""Build the agregations from fragments."""
root_path = DAVID / "2020"
for previous, page, next_ in neighborhood(
reversed(Page.all(source=root_path, only_published=False)),
first={
"url": "/david/stream/",
"title": "Streams 2009-2019",
"is_draft": False,
},
):
template = environment.get_template("article_2020.html")
content = template.render(page=page, prev=previous, next=next_,)
target_path = Path(page.url[1:])
target_path.mkdir(parents=True, exist_ok=True)
open(target_path / "index.html", "w").write(content)
if page.is_draft:
print(f"Draft: {LOCAL_DOMAIN}{page.url} ({page.title})")
template = environment.get_template("archives_2020.html")
content = template.render(page_list=Page.all(source=root_path))
open(root_path / "index.html", "w").write(content)
@cli
def home():
"""Build the home page with last published items."""
template = environment.get_template("profil.html")
content = template.render(page_list=Page.all(source=DAVID / "2020"),)
open(DAVID / "index.html", "w").write(content)
@cli
def feed():
"""Generate a feed from last published items."""
template = environment.get_template("feed.xml")
content = template.render(
page_list=Page.all(source=DAVID / "2020", with_h2_anchors=False),
current_dt=TODAY.strftime(NORMALIZED_STRFTIME),
BASE_URL=f"{DOMAIN}/david/",
)
open(DAVID / "log" / "index.xml", "w").write(content)
@wrap
def perf_wrapper():
start = perf_counter()
yield
elapsed = perf_counter() - start
print(f"Done in {elapsed:.5f} seconds.")
if __name__ == "__main__":
run()