gopher-wp-bridge/server.py
EamonnMR 73889a9a82 Nice little FastAPI inspired syntax for adding path handlers.
Shame the junk wordpress is returning breaks bs4. Will need to play with the data a bit more.
2022-12-31 02:04:57 -05:00

56 lines
No EOL
1.3 KiB
Python

from os import getenv
from urllib.parse import urljoin
from html import unescape
import pituophis
from pituophis import Item, Request
from requests import get
from parse import parse
from bs4 import BeautifulSoup
wordpress_url = getenv("URL")
handlers = {}
def register_handler(path: str):
def decorator_handler(func):
handlers[path] = func
return func
return decorator_handler
@register_handler("/post/{id}")
def post(request: Request, id: int):
post = get(urljoin(wordpress_url, f"wp-json/wp/v2/posts/{id}")).json()
return BeautifulSoup(post["content"]).get_text()
def handle(request):
for path, handler in handlers.items():
parse_result = parse(path, request.path)
if parse_result is not None:
return handler(request, **parse_result.named)
posts = get(urljoin(wordpress_url, "wp-json/wp/v2/posts")).json()
menu = [
Item(itype=0, path=f"/post/{post['id']}", text=unescape(post['title']['rendered']), host=request.host, port=request.port)
for i, post in enumerate(posts)
]
return menu
if __name__ == '__main__':
pituophis.serve("127.0.0.1", int(getenv("PORT")), handler=handle)
# Itypes:
# 0: FILE
# 1: dir
# 2: CSO
# 3: UNKN
# 4: HQX
# 5: BIN
# 6: UUE
# 7: ?
# 8: TEL
# 9: BIN