mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from quart import g, abort, render_template
|
|
from http import HTTPStatus
|
|
import pyqrcode
|
|
from io import BytesIO
|
|
from lnbits.decorators import check_user_exists, validate_uuids
|
|
|
|
from . import withdraw_ext
|
|
from .crud import get_withdraw_link, chunks
|
|
|
|
|
|
@withdraw_ext.route("/")
|
|
@validate_uuids(["usr"], required=True)
|
|
@check_user_exists()
|
|
async def index():
|
|
return await render_template("withdraw/index.html", user=g.user)
|
|
|
|
|
|
@withdraw_ext.route("/<link_id>")
|
|
async def display(link_id):
|
|
link = await get_withdraw_link(link_id, 0) or abort(
|
|
HTTPStatus.NOT_FOUND, "Withdraw link does not exist."
|
|
)
|
|
return await render_template("withdraw/display.html", link=link, unique=True)
|
|
|
|
|
|
@withdraw_ext.route("/img/<link_id>")
|
|
async def img(link_id):
|
|
link = await get_withdraw_link(link_id, 0) or abort(
|
|
HTTPStatus.NOT_FOUND, "Withdraw link does not exist."
|
|
)
|
|
qr = pyqrcode.create(link.lnurl)
|
|
stream = BytesIO()
|
|
qr.svg(stream, scale=3)
|
|
return (
|
|
stream.getvalue(),
|
|
200,
|
|
{
|
|
"Content-Type": "image/svg+xml",
|
|
"Cache-Control": "no-cache, no-store, must-revalidate",
|
|
"Pragma": "no-cache",
|
|
"Expires": "0",
|
|
},
|
|
)
|
|
|
|
|
|
@withdraw_ext.route("/print/<link_id>")
|
|
async def print_qr(link_id):
|
|
link = await get_withdraw_link(link_id) or abort(
|
|
HTTPStatus.NOT_FOUND, "Withdraw link does not exist."
|
|
)
|
|
if link.uses == 0:
|
|
return await render_template("withdraw/print_qr.html", link=link, unique=False)
|
|
links = []
|
|
count = 0
|
|
for x in link.usescsv.split(","):
|
|
linkk = await get_withdraw_link(link_id, count) or abort(
|
|
HTTPStatus.NOT_FOUND, "Withdraw link does not exist."
|
|
)
|
|
links.append(str(linkk.lnurl))
|
|
count = count + 1
|
|
page_link = list(chunks(links, 2))
|
|
linked = list(chunks(page_link, 5))
|
|
return await render_template("withdraw/print_qr.html", link=linked, unique=True)
|