import os import shutil import jinja2 import markdown import pathlib from datetime import datetime from dataclasses import dataclass __here__ = pathlib.Path(__file__).resolve().parent date = datetime.now().strftime('%Y-%m-%d %H:%M:%S') md = markdown.Markdown(extensions=['meta', "toc", "extra"]) env = jinja2.Environment(loader = jinja2.FileSystemLoader(str(__here__ / "templates"))) # grab posts -------------------------------------------------------------------------------------- @dataclass class Post: title: str date: str content: str path: str tags: list posts = [] tags = [] for post in os.listdir(__here__ / "posts"): if len(post) < 3: continue if not post.endswith(".md"): continue with open(__here__ / "posts" / post, "r") as f: content = md.convert(f.read()) kwargs = dict() kwargs["title"] = md.Meta["title"][0] kwargs["date"] = md.Meta["date"][0] kwargs["content"] = content kwargs["path"] = post[:-3] kwargs["tags"] = md.Meta.get("tags", [""])[0].split() tags += kwargs["tags"] posts.append(Post(**kwargs)) posts.sort(key=lambda p: p.date, reverse=True) tags = list(set(tags)) tags.sort() # index ------------------------------------------------------------------------------------------- if not os.path.isdir(__here__ / "public"): os.mkdir(__here__ / "public") shutil.copytree("posts", "public", dirs_exist_ok=True) template = env.get_template("index.html") with open(__here__ / "public" / "index.html", "w") as f: f.write(template.render(posts=posts, title="blog", date=date)) # posts ------------------------------------------------------------------------------------------- template = env.get_template("post.html") for post in posts: print(post.path) if not os.path.isdir(__here__ / "public" / post.path): os.mkdir(__here__ / "public" / post.path) with open(__here__ / "public" / post.path / "index.html", "w") as f: f.write(template.render(post=post, title=post.title, date=date)) # tags -------------------------------------------------------------------------------------------- template = env.get_template("tags.html") if not os.path.isdir(__here__ / "public" / "tags"): os.mkdir(__here__ / "public" / "tags") with open(__here__ / "public" / "tags" / "index.html", "w") as f: f.write(template.render(posts=posts, tags=tags, title="tags", date=date)) # css --------------------------------------------------------------------------------------------- template = env.get_template('style.css') for d, _, _ in os.walk(__here__ / "public", topdown=False): with open(os.path.join(d, "style.css"), 'w') as f: f.write(template.render())