website/start_site.py

#!/usr/bin/env python

import os
import toml
from datetime import datetime
import pytz
from flask import Flask, url_for, render_template, send_from_directory
from flask import make_response
from feedgen.feed import FeedGenerator
from flaskext.markdown import Markdown
import markdown
import markdown.extensions.fenced_code
import markdown.extensions.codehilite
import markdown.extensions.toc
from markdown.extensions.toc import TocExtension
from pygments.formatters import HtmlFormatter

app = Flask(__name__)
Markdown(app)

meta_data = {
    root[len("./blog/"):]: datetime.strptime(
        open(os.path.join(root, "index.md"), encoding='UTF-8')
        .readline().rstrip(),
        "%Y-%m-%d"
        )
    for root, dirs, files in os.walk("./blog")
    if "index.md" in files
        }

app.config["blog"] = toml.load("settings.toml")
content_path = app.config["blog"]["content"]["path"]
highlight_style = app.config["blog"]["content"]["style"]
STYLESHEET = "stylesheet.css"
STYLESHEET_AUTO_COMPLETE = "auto-complete.css"
project_name = app.config["blog"]["project"]["name"]
project_title = app.config["blog"]["project"]["title"]

app.config["blog"]["style"] = toml.load("style.toml")
colors = app.config["blog"]["style"][highlight_style]


@app.route('/')
def index(_paths=meta_data):
    ''' Main Site.
    '''
    sorted_meta_data = dict(
        sorted(
            meta_data.items(),
            reverse=True,
            key=lambda item: item[1]
            )
        )
    return render_template("index.html", colors=colors, _paths=sorted_meta_data)


@app.route('/blog/<blog_item>/index.html')
def blog(blog_item, _date=meta_data):
    ''' Blog Pages.
    '''
    md_file_path = os.path.join(f"blog/{blog_item}/index.md")
    with open(md_file_path, "r") as _f:
        md_file = _f.read()
    md_extensions = [
        'toc',
        TocExtension(toc_class="", title=""),
        "fenced_code",
        "codehilite",
        "tables",
        "mdx_math"
    ]

    md_configs = {"mdx_math": {"enable_dollar_delimiter": True}}
    md = markdown.Markdown(
        extensions=md_extensions,
        extension_configs=md_configs
        )

    html = md.convert(md_file)

    formatter = HtmlFormatter(
        style=highlight_style,
        full=True,
        cssclass="codehilite"
        )
    css_string = formatter.get_style_defs()
    md_css_string = "<style>" + css_string + "</style>"
    md_template = md_css_string + html
    res = render_template(
        "blog.html",
        #toc=md.toc,
        md_doc=md_template,
        colors=colors,
        stylesheet=STYLESHEET,
        #stylesheet_auto_complete=STYLESHEET_AUTO_COMPLETE,
        project_name=project_name,
        project_title=project_title,
        # tree=cut_path_tree(
        #     #rem_readme(content_path, make_tree(content_path)),
        #     make_tree(content_path),
        #     content_path,
        #     ".md" #     )
        _date=meta_data[blog_item]
        )
    response = make_response(res)
    response.headers["Content-Type"] = "text/html; charset=utf-8"
    return response


    # return render_template(
    #     f"blog/{blog_item}/index.html",
    #     _date=meta_data[blog_item]
    #     )


@app.route("/about.html")
def about():
    ''' About Page.
    '''
    return render_template("about.html", colors=colors)


@app.route("/contact.html")
def contact():
    ''' Contact Page.
    '''
    return render_template("contact.html", colors=colors)


@app.route("/rss.xml")
def rss(_items=meta_data):
    ''' RSS Feed.
    Generates RSS feed as XML
    '''
    # rss_feed = []
    _tz = pytz.timezone('Europe/Berlin')
    _fg = FeedGenerator()
    _fg.title("Website of Stefan Friese")
    _fg.description("test")
    _fg.language("en-us")
    # _fg.author({'name': "Stefan Friese", 'email': 'stefan@stefan.works'})
    _fg.link(href="https://stefan.works", rel="self")
    for key in meta_data.keys():
        _fe = _fg.add_entry()
        _fe.id(f"https://stefan.works/blog/{key}/index.html")
        _fe.title(key)
        # _fe.description("test")
        # _fe.author({'name': "Stefan Friese", 'email': 'stefan@stefan.works'})
        _fe.link(href=f"https://stefan.works/blog/{key}/index.html")
        _fe.pubDate(pubDate=_tz.localize(meta_data[key]))
    _fg.rss_str(pretty=True)
    _fg.rss_file('./static/rss.xml')
    return send_from_directory(
        os.path.join(
            app.root_path,
            'static'
            ),
        'rss.xml'
        )


@app.route('/favicon.ico')
def favicon():
    ''' Provides favicon.
    '''
    return send_from_directory(
        os.path.join(
            app.root_path,
            'static'
            ),
        'favicon.ico'
        )


@app.errorhandler(404)
def page_not_found(_error):
    ''' Error Handling.
    Error 404
    '''
    return render_template("/status_code/404.html", colors=colors), 404


@app.errorhandler(400)
def bad_request(_error):
    ''' Error Handling.
    Error 400
    '''
    return render_template("/status_code/400.html", colors=colors), 400


@app.errorhandler(500)
def internal_server_error(_error):
    ''' Error Handling.
    Error 500
    '''
    return render_template("/status_code/500.html", colors=colors), 500


with app.test_request_context():
    print(url_for("index"))
    # print(url_for("about"))
    # print(url_for("contact"))
    # print(url_for("rss"))
    # print(url_for("static", filename="stylesheet.css"))
    # print(meta_data)
    # print(url_for("blog", blog_item="first_blog"))