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

use proper typing

Browse source
This commit is contained in:
callebtc 2023-01-03 11:35:10 +01:00
parent f3e0bd4907
commit 6d29cc85b0
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
lnbits/extensions/gerty/crud.py
View file

@ -50,15 +50,13 @@ async def create_gerty(wallet_id: str, data: Gerty) -> Gerty:
return gerty return gerty
async def update_gerty(gerty_id: str, **kwargs) -> Gerty: async def update_gerty(gerty_id: str, **kwargs) -> Optional[Gerty]:
q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()]) q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()])
await db.execute( await db.execute(
f"UPDATE gerty.gertys SET {q} WHERE id = ?", (*kwargs.values(), gerty_id) f"UPDATE gerty.gertys SET {q} WHERE id = ?", (*kwargs.values(), gerty_id)
) )
gerty = await get_gerty(gerty_id) return await get_gerty(gerty_id)
assert gerty
return gerty
async def get_gerty(gerty_id: str) -> Optional[Gerty]: async def get_gerty(gerty_id: str) -> Optional[Gerty]:

3
lnbits/extensions/gerty/views_api.py
View file

@ -55,6 +55,9 @@ async def api_link_create_or_update(
data.wallet = wallet.wallet.id data.wallet = wallet.wallet.id
gerty = await update_gerty(gerty_id, **data.dict()) gerty = await update_gerty(gerty_id, **data.dict())
assert gerty, HTTPException(
status_code=HTTPStatus.NOT_FOUND, detail="Gerty does not exist"
)
else: else:
gerty = await create_gerty(wallet_id=wallet.wallet.id, data=data) gerty = await create_gerty(wallet_id=wallet.wallet.id, data=data)