Evan Reichard
146c2e25d0
All checks were successful
continuous-integration/drone/push Build is passing
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
from flask import Blueprint
|
|
from flask import make_response, render_template
|
|
from html_sanitizer import Sanitizer
|
|
from markdown import markdown
|
|
from vreader.config import Config
|
|
import os
|
|
|
|
bp = Blueprint("common", __name__)
|
|
sanitizer = Sanitizer()
|
|
|
|
@bp.route("/", methods=["GET"])
|
|
def main_entry():
|
|
# Get Files
|
|
directory = str(Config.DATA_PATH)
|
|
all_files = os.listdir(directory)
|
|
markdown_files = [file for file in all_files if file.endswith(".md")]
|
|
|
|
# Get Create Time
|
|
file_info_list = []
|
|
for filename in markdown_files:
|
|
file_path = os.path.join(directory, filename)
|
|
creation_time = os.path.getctime(file_path)
|
|
file_info_list.append((filename, creation_time))
|
|
|
|
# Sort Create Time (Recent First)
|
|
file_info_list.sort(key=lambda x: x[1], reverse=True)
|
|
|
|
# Get Articles
|
|
articles = [parse_filename(item[0]) for item in file_info_list]
|
|
|
|
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("404.html")), 404
|
|
|
|
metadata = get_article_metadata(id)
|
|
if not metadata:
|
|
return make_response(render_template("404.html")), 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 _:
|
|
return make_response(render_template("404.html")), 404
|
|
|
|
|
|
def get_article_metadata(id):
|
|
directory = str(Config.DATA_PATH)
|
|
files = os.listdir(directory)
|
|
for file_name in files:
|
|
if file_name.startswith(id) and file_name.endswith(".md"):
|
|
file_path = os.path.join(directory, file_name)
|
|
metadata = parse_filename(file_name)
|
|
metadata["filepath"] = file_path
|
|
return metadata
|
|
return None
|
|
|
|
|
|
def parse_filename(filename):
|
|
video_id = filename[:11]
|
|
title = filename[12:][:-3]
|
|
|
|
return {
|
|
"video_id": video_id,
|
|
"title": title
|
|
}
|