doordesk-js/dennis/src/dennis.py
2023-06-27 18:01:14 -04:00

71 lines
1.8 KiB
Python

import os
import markdown
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from datetime import date, datetime
STATIC_PATH = "./static"
api = FastAPI()
api.add_middleware(
CORSMiddleware,
allow_origins=['*'],
allow_credentials=True,
allow_methods=["GET"],
allow_headers=[],
)
class Article(BaseModel):
content_type: str
title: str
date: date
content: str
def get_articles_from_path(path: str) -> list[Article]:
md = markdown.Markdown(extensions=["meta"]);
articles: list[Article] = []
for root, _, filenames in os.walk(path):
for filename in filenames:
if filename[-2:] == "md":
with open(f"{root}/{filename}") as file:
html: str = md.convert(file.read());
meta: dict = md.Meta;
articles.append(
Article(
content_type=meta.get("content_type")[0],
title=meta.get("title")[0],
date=datetime.strptime(meta.get("date")[0], "%Y %m %d"),
content=html,
));
return articles;
DB = sorted(
get_articles_from_path(STATIC_PATH),
key=lambda article: article.date,
reverse=True
);
@api.get('/home')
async def serve_home():
return DB
@api.get('/blog')
async def serve_blog():
return [entry for entry in DB if entry.content_type == 'blog']
@api.get('/projects')
async def serve_projects():
return [entry for entry in DB if entry.content_type == 'project' or entry.content_type == 'game']
@api.get('/bots')
async def serve_bots():
return [entry for entry in DB if entry.content_type == 'chatbot']