lnbits-exts/satmachineclient
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

added confetti

Browse source
This commit is contained in:
benarc 2024-01-19 17:40:43 +00:00
parent 1735cc0d66
commit d90babebc8
15 changed files with 177 additions and 135 deletions
Download patch file Download diff file Expand all files Collapse all files

26
crud.py
View file

@ -8,7 +8,7 @@ from loguru import logger
from fastapi import Request
from lnurl import encode as lnurl_encode
async def create_myextension(wallet_id: str, data: CreateMyExtensionData) -> MyExtension:
async def create_myextension(wallet_id: str, data: CreateMyExtensionData, req: Request) -> MyExtension:
myextension_id = urlsafe_short_hash()
await db.execute(
"""
@ -23,21 +23,23 @@ async def create_myextension(wallet_id: str, data: CreateMyExtensionData) -> MyE
data.lnurlwithdrawamount
),
)
myextension = await get_myextension(myextension_id)
myextension = await get_myextension(myextension_id, req)
assert myextension, "Newly created table couldn't be retrieved"
return myextension
async def get_myextension(myextension_id: str, req: Request) -> Optional[MyExtension]:
async def get_myextension(myextension_id: str, req: Optional[Request] = None) -> Optional[MyExtension]:
logger.debug(myextension_id)
row = await db.fetchone("SELECT * FROM myextension.maintable WHERE id = ?", (myextension_id,))
if not row:
return None
rowAmended = MyExtension(**row)
rowAmended.lnurlpay = lnurl_encode(req.url_for("myextension.api_lnurl_pay", myextension_id=row.id)._url)
rowAmended.lnurlwithdraw = lnurl_encode(req.url_for("myextension.api_lnurl_withdraw", myextension_id=row.id)._url)
if req:
rowAmended.lnurlpay = lnurl_encode(req.url_for("myextension.api_lnurl_pay", myextension_id=row.id)._url)
rowAmended.lnurlwithdraw = lnurl_encode(req.url_for("myextension.api_lnurl_withdraw", myextension_id=row.id)._url)
return rowAmended
async def get_myextensions(wallet_ids: Union[str, List[str]], req: Request) -> List[MyExtension]:
async def get_myextensions(wallet_ids: Union[str, List[str]], req: Optional[Request] = None) -> List[MyExtension]:
if isinstance(wallet_ids, str):
wallet_ids = [wallet_ids]
@ -46,17 +48,19 @@ async def get_myextensions(wallet_ids: Union[str, List[str]], req: Request) -> L
f"SELECT * FROM myextension.maintable WHERE wallet IN ({q})", (*wallet_ids,)
)
tempRows = [MyExtension(**row) for row in rows]
for row in tempRows:
row.lnurlpay = lnurl_encode(req.url_for("myextension.api_lnurl_pay", myextension_id=row.id)._url)
row.lnurlwithdraw = lnurl_encode(req.url_for("myextension.api_lnurl_withdraw", myextension_id=row.id)._url)
if req:
for row in tempRows:
row.lnurlpay = lnurl_encode(req.url_for("myextension.api_lnurl_pay", myextension_id=row.id)._url)
row.lnurlwithdraw = lnurl_encode(req.url_for("myextension.api_lnurl_withdraw", myextension_id=row.id)._url)
return tempRows
async def update_myextension(myextension_id: str, **kwargs) -> MyExtension:
async def update_myextension(myextension_id: str, req: Optional[Request] = None, **kwargs) -> MyExtension:
q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()])
logger.debug( kwargs.items())
await db.execute(
f"UPDATE myextension.maintable SET {q} WHERE id = ?", (*kwargs.values(), myextension_id)
)
myextension = await get_myextension(myextension_id)
myextension = await get_myextension(myextension_id, req)
assert myextension, "Newly updated myextension couldn't be retrieved"
return myextension