aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: embeddable img link in withdraw extension

Browse source
This commit is contained in:
Stefan Stammberger 2021-10-10 18:49:41 +02:00
parent c1a82dad9d
commit 9a18720a96
No known key found for this signature in database
GPG key ID: 645FA807E935D9D5
1 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

19
lnbits/extensions/withdraw/views.py
View file

@ -10,7 +10,7 @@ from fastapi.params import Depends
from fastapi.templating import Jinja2Templates from fastapi.templating import Jinja2Templates
from starlette.exceptions import HTTPException from starlette.exceptions import HTTPException
from starlette.responses import HTMLResponse from starlette.responses import HTMLResponse, StreamingResponse
from lnbits.core.models import User from lnbits.core.models import User
templates = Jinja2Templates(directory="templates") templates = Jinja2Templates(directory="templates")
@ -36,7 +36,7 @@ async def display(request: Request, link_id):
return withdraw_renderer().TemplateResponse("withdraw/display.html", {"request":request,"link":{**link.dict(), "lnurl": link.lnurl(request)}, "unique":True}) return withdraw_renderer().TemplateResponse("withdraw/display.html", {"request":request,"link":{**link.dict(), "lnurl": link.lnurl(request)}, "unique":True})
@withdraw_ext.get("/img/{link_id}", response_class=HTMLResponse) @withdraw_ext.get("/img/{link_id}", response_class=StreamingResponse)
async def img(request: Request, link_id): async def img(request: Request, link_id):
link = await get_withdraw_link(link_id, 0) link = await get_withdraw_link(link_id, 0)
if not link: if not link:
@ -50,16 +50,19 @@ async def img(request: Request, link_id):
print(qr) print(qr)
stream = BytesIO() stream = BytesIO()
qr.svg(stream, scale=3) qr.svg(stream, scale=3)
return ( stream.seek(0)
stream.getvalue(),
200, async def _generator(stream: BytesIO):
{ yield stream.getvalue()
return StreamingResponse(
_generator(stream),
headers={
"Content-Type": "image/svg+xml", "Content-Type": "image/svg+xml",
"Cache-Control": "no-cache, no-store, must-revalidate", "Cache-Control": "no-cache, no-store, must-revalidate",
"Pragma": "no-cache", "Pragma": "no-cache",
"Expires": "0", "Expires": "0",
}, })
)
@withdraw_ext.get("/print/{link_id}", response_class=HTMLResponse) @withdraw_ext.get("/print/{link_id}", response_class=HTMLResponse)