mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-25 23:21:21 +01:00
7fbea79e62
* allow for custom print of vouchers * lnbits defaul voucher * reduce png size * add template for custom design * rename lnbits voucher * send voucher amount to UI * add some tweaks for default voucher * added bonus in readme * blacked * make default string as constant
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
from sqlite3 import Row
|
|
|
|
import shortuuid # type: ignore
|
|
from fastapi.param_functions import Query
|
|
from lnurl import Lnurl, LnurlWithdrawResponse
|
|
from lnurl import encode as lnurl_encode # type: ignore
|
|
from pydantic import BaseModel
|
|
from starlette.requests import Request
|
|
|
|
|
|
class CreateWithdrawData(BaseModel):
|
|
title: str = Query(...)
|
|
min_withdrawable: int = Query(..., ge=1)
|
|
max_withdrawable: int = Query(..., ge=1)
|
|
uses: int = Query(..., ge=1)
|
|
wait_time: int = Query(..., ge=1)
|
|
is_unique: bool
|
|
webhook_url: str = Query(None)
|
|
custom_url: str = Query(None)
|
|
|
|
|
|
class WithdrawLink(BaseModel):
|
|
id: str
|
|
wallet: str = Query(None)
|
|
title: str = Query(None)
|
|
min_withdrawable: int = Query(0)
|
|
max_withdrawable: int = Query(0)
|
|
uses: int = Query(0)
|
|
wait_time: int = Query(0)
|
|
is_unique: bool = Query(False)
|
|
unique_hash: str = Query(0)
|
|
k1: str = Query(None)
|
|
open_time: int = Query(0)
|
|
used: int = Query(0)
|
|
usescsv: str = Query(None)
|
|
number: int = Query(0)
|
|
webhook_url: str = Query(None)
|
|
custom_url: str = Query(None)
|
|
|
|
@property
|
|
def is_spent(self) -> bool:
|
|
return self.used >= self.uses
|
|
|
|
def lnurl(self, req: Request) -> Lnurl:
|
|
if self.is_unique:
|
|
usescssv = self.usescsv.split(",")
|
|
tohash = self.id + self.unique_hash + usescssv[self.number]
|
|
multihash = shortuuid.uuid(name=tohash)
|
|
url = req.url_for(
|
|
"withdraw.api_lnurl_multi_response",
|
|
unique_hash=self.unique_hash,
|
|
id_unique_hash=multihash,
|
|
)
|
|
else:
|
|
url = req.url_for(
|
|
"withdraw.api_lnurl_response", unique_hash=self.unique_hash
|
|
)
|
|
|
|
return lnurl_encode(url)
|
|
|
|
def lnurl_response(self, req: Request) -> LnurlWithdrawResponse:
|
|
url = req.url_for(
|
|
name="withdraw.api_lnurl_callback", unique_hash=self.unique_hash
|
|
)
|
|
return LnurlWithdrawResponse(
|
|
callback=url,
|
|
k1=self.k1,
|
|
min_withdrawable=self.min_withdrawable * 1000,
|
|
max_withdrawable=self.max_withdrawable * 1000,
|
|
default_description=self.title,
|
|
)
|
|
|
|
|
|
class HashCheck(BaseModel):
|
|
id: str
|
|
lnurl_id: str
|
|
|
|
@classmethod
|
|
def from_row(cls, row: Row) -> "Hash":
|
|
return cls(**dict(row))
|