summaryrefslogtreecommitdiff
path: root/web.py
blob: e2d7fdfa5a3948c26b97554df8b1c8f42c56f648 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import re

from fastapi import FastAPI, Response
from fastapi.responses import FileResponse

import htmlcache
import texgen

app = FastAPI()

utaten_pattern = re.compile(r'[a-z0-9]+')
tex_generator = texgen.TexGenerator('pdf_cache', 'temp', 20)
html_cache = htmlcache.HtmlCache('html_cache')


@app.get("/utaten/{item_id}.pdf")
async def get_utaten_lyric_pdf(item_id: str):
    try:
        print('_get_utaten_tex_source')
        tex = await html_cache.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 html_cache.get_utaten_tex_source(item_id)
        return Response(content=tex, media_type='application/x-tex')
    except htmlcache.TexSourceGenerationError as e:
        return Response(content=str(e), status_code=503)