trimill.xyz/src/main.py

from flask import Flask, render_template, send_from_directory, abort, request, make_response
import markdown2
import datetime
import os
import ast

data = {
    "feet": [
        "🐝",
        "Best viewed using Internet Explorer 6 or earlier",
        "The HORSE is a noble animal",
        "🦀",
        "<code>segmentation fault (core dumped)</code>",
        "Bees land on thyme",
        "☃",
        "<code># cat /dev/urandom > /dev/sda</code>",
        "<code>:(){ :|: & };:</code>",
        "Formal complaints will recieve responses within 5-7 business days",
        "<code>++++[-&gt;++++&lt;]&gt;+[-&gt;++++++&gt;+++++++&gt;++&lt;&lt;&lt;]&gt;.&gt;--------..+++++.&lt;-.&gt;--.&gt;--.&lt;++.&lt;.&gt;++++.----.</code>",
        "Copywrong © 3034. All rights unreserved.",
        "[<span style='text-decoration: underline;'>citation needed</span>]",
        "Best viewed with eyes",
        "Your browser does not support 7D graphics. Please update for the best user experience.",
        "Press SPACE to jump",
        "🐀",
        "If problems persist, please return to the nearest Blockbuster Video® establishment",
        "Oversalt to taste",
        "curl -s -L http://bit.ly/10hA8iC | bash",
        "Submit footer text via carrier pigeon to <code>[REDACTED]</code>",
        "GEORGE is inevitable."
    ],
    "next_theme": {
        "system": "dark",
        "dark": "light",
        "light": "contrast",
        "contrast": "system"
    }
}

markdown_extras = ["fenced-code-blocks", "footnotes", "strike", "tables", "metadata"]

app = Flask(__name__)


projects = []
proj_dir = os.path.join(app.root_path, app.template_folder, "projects")
for root, dirs, files in os.walk(proj_dir):
    for file in files:
        try:
            with open(os.path.join(root, file)) as f:
                metaline = f.read().splitlines()[0]
            if metaline.startswith("{% set meta="):
                meta = metaline[12:-2].strip()
                meta = ast.literal_eval(meta)
                meta["path"] = os.path.relpath(root, start=proj_dir)
                meta["file"] = file
                projects.append(meta)
        except Exception as e:
            print(e)
data["projects"] = sorted(projects, key=lambda p: [p.get("star") != True, p.get("title")])

blogposts = []
blog_dir = os.path.join(app.root_path, "blog")
for file in os.listdir(blog_dir):
    with open(os.path.join(blog_dir, file)) as f:
        contents = f.read()
    html = markdown2.markdown(contents, extras=markdown_extras)
    meta = html.metadata
    meta["file"] = file
    date_parts = file.removesuffix(".md").split("-")
    date = datetime.date(int(date_parts[0]), int(date_parts[1]), int(date_parts[2]))
    meta["timestamp"] = datetime.datetime.fromisoformat(meta["timestamp"])
    meta["date"] = date
    meta["n"] = int(date_parts[3])
    meta["url"] = f"/blog/{date.isoformat().replace('-','/')}/{meta['n']}"
    blogposts.append(meta)
data["blogposts"] = sorted(blogposts, key=lambda post:[post["date"], post["n"]], reverse=True)


@app.route("/favicon.ico")
def favicon():
    path = os.path.join(app.root_path, "static")
    return send_from_directory(path, "favicon.ico", mimetype="image/vnd.microsoft.icon")

@app.errorhandler(404)
def four_oh_four(e):
    theme = request.cookies.get("theme") or "dark"
    return render_template("404.html", data=data, theme=theme)

def load_page(url):
    if url.endswith(".html"):
        path = os.path.join(app.root_path, app.template_folder, url)
        if os.path.exists(path):
            theme = request.cookies.get("theme") or "dark"
            return render_template(url, data=data, theme=theme)
        else:
            return abort(404)
    else:
        return send_from_directory("templates", url)

@app.route("/")
@app.route("/index.html")
def home():
    return load_page("index.html")

@app.route("/projects/")
@app.route("/projects/index.html")
def projects_index():
    return load_page("projects.html")

@app.route("/projects/<page>/")
@app.route("/projects/<page>/<file>")
def project(page, file="index.html"):
    return load_page(f"projects/{page}/{file}")

@app.route("/blog/")
def blog_list():
    theme = request.cookies.get("theme") or "dark"
    return render_template("blog.html", data=data, theme=theme)


@app.route("/blog/<int:y>/<int:m>/<int:d>/")
@app.route("/blog/<int:y>/<int:m>/<int:d>/<int:n>")
def blog_page(y, m, d, n=0):
    date = datetime.date(y, m, d)
    print(date.isoformat())
    path = os.path.join(app.root_path, "blog", f"{date.isoformat()}-{n}.md")
    if os.path.exists(path):
        with open(path) as f:
            contents = f.read()
        content = markdown2.markdown(contents, extras=markdown_extras)
        meta = content.metadata
        theme = request.cookies.get("theme") or "dark"
        return render_template("_blog.html", data=data, theme=theme, content=content, date=date, meta=meta)
    else:
        return abort(404)

@app.route("/blog/rss.xml")
def blog_rss():
    xml = render_template("rss.xml", data=data)
    response = make_response(xml)
    response.headers["Content-Type"] = "application/rss+xml; charset=utf-8"
    return response

@app.route("/blog/atom.xml")
def blog_atom():
    xml = render_template("atom.xml", data=data)
    response = make_response(xml)
    response.headers["Content-Type"] = "application/atom+xml; charset=utf-8"
    return response