Evan Reichard
1930dfc099
All checks were successful
continuous-integration/drone/push Build is passing
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
from flask import Blueprint
|
|
from flask import make_response, render_template, send_from_directory
|
|
from html_sanitizer import Sanitizer
|
|
from markdown import markdown
|
|
from vreader.config import Config
|
|
import os
|
|
|
|
|
|
bp = Blueprint("common", __name__)
|
|
sanitizer = Sanitizer()
|
|
|
|
|
|
@bp.route("/static/<path:path>")
|
|
def static(path):
|
|
return send_from_directory("static", path)
|
|
|
|
|
|
@bp.route("/manifest.json")
|
|
def manifest():
|
|
return send_from_directory("static", "manifest.json")
|
|
|
|
|
|
@bp.route("/", methods=["GET"])
|
|
def main_entry():
|
|
# Get Files
|
|
directory = str(Config.DATA_PATH)
|
|
all_files = os.listdir(directory)
|
|
|
|
# Sort Files
|
|
markdown_files = [file for file in all_files if file.endswith(".md")]
|
|
markdown_files = sorted(markdown_files, reverse=True)
|
|
|
|
# Get Article Metadata
|
|
articles = [get_article_metadata(filename, directory) for filename in markdown_files]
|
|
|
|
return make_response(render_template("index.html", articles=articles))
|
|
|
|
|
|
@bp.route("/articles/<id>", methods=["GET"])
|
|
def article_item(id):
|
|
|
|
if len(id) != 11:
|
|
return make_response(
|
|
render_template("error.html", status=404, message="Invalid Article")
|
|
), 404
|
|
|
|
metadata = find_article(id)
|
|
if not metadata:
|
|
return make_response(
|
|
render_template("error.html", status=404, message="Invalid Article")
|
|
), 404
|
|
|
|
try:
|
|
with open(metadata["filepath"], 'r', encoding='utf-8') as file:
|
|
article_contents = file.read()
|
|
|
|
markdown_html = sanitizer.sanitize(markdown(article_contents))
|
|
|
|
return make_response(
|
|
render_template("article.html", metadata=metadata, markdown_html=markdown_html)
|
|
)
|
|
except Exception as e:
|
|
return make_response(
|
|
render_template("error.html", status=404, message=e)
|
|
), 404
|
|
|
|
|
|
def find_article(id):
|
|
directory = str(Config.DATA_PATH)
|
|
files = os.listdir(directory)
|
|
|
|
# Find Filename
|
|
filename = next((x for x in files if x[15:26] == id and x.endswith(".md")), None)
|
|
if filename is None:
|
|
return None
|
|
|
|
# Normalize File Info
|
|
return get_article_metadata(filename, directory)
|
|
|
|
|
|
def get_article_metadata(filename, directory):
|
|
return {
|
|
"date": filename[:14],
|
|
"video_id": filename[15:26],
|
|
"title": filename[27:][:-3],
|
|
"filepath": os.path.join(directory, filename)
|
|
}
|