From 77c5ef83402aa28fc6d3a9d2982ccb03fea70d2f Mon Sep 17 00:00:00 2001 From: Keuin Date: Thu, 1 Jun 2023 20:14:00 +0800 Subject: texgen (WIP) --- web.py | 38 +++++++++++++++++++++++++++++--------- 1 file changed, 29 insertions(+), 9 deletions(-) (limited to 'web.py') diff --git a/web.py b/web.py index 7aadac4..007e93d 100644 --- a/web.py +++ b/web.py @@ -2,26 +2,46 @@ import re import aiohttp from fastapi import FastAPI, Response +from fastapi.responses import FileResponse import main +import texgen app = FastAPI() utaten_pattern = re.compile(r'[a-z0-9]+') +tex_generator = texgen.TexGenerator('pdf_cache', 'temp', 20) -@app.get("/utaten/{item_id}/pdf") -async def get_utaten_lyric_pdf(item_id: str): - raise NotImplementedError +class TexSourceGenerationError(Exception): + pass -@app.get("/utaten/{item_id}.tex") -async def get_utaten_lyric_pdf(item_id: str, resp: Response): +async def _get_utaten_tex_source(item_id: str) -> str: async with aiohttp.ClientSession() as ses: async with ses.get(f'https://utaten.com/lyric/{item_id}/') as r: if not r.ok: - resp.status_code = 503 - return + raise TexSourceGenerationError('HTTP request failed when reading page source') html = await r.text() - tex = main.html_to_tex(html) - return Response(content=tex, media_type='application/x-tex') + return main.html_to_tex(html) + + +@app.get("/utaten/{item_id}.pdf") +async def get_utaten_lyric_pdf(item_id: str): + try: + print('_get_utaten_tex_source') + tex = await _get_utaten_tex_source(item_id) + print('xelatex') + pdf_path = await tex_generator.xelatex(tex) + return FileResponse(pdf_path, media_type='application/pdf') + except texgen.TexGenerationError as e: + return Response(content=f'Failed to generate tex file: {e}', status_code=502) + + +@app.get("/utaten/{item_id}.tex") +async def get_utaten_lyric_tex(item_id: str): + try: + tex = await _get_utaten_tex_source(item_id) + return Response(content=tex, media_type='application/x-tex') + except TexSourceGenerationError as e: + return Response(content=str(e), status_code=503) -- cgit v1.2.3