chore: mypy fixes

This commit is contained in:
Vlad Stan 2023-02-15 18:18:46 +02:00
parent d0f38346e3
commit bc1af610db
7 changed files with 27 additions and 11 deletions

View file

@ -26,12 +26,19 @@ async def on_invoice_paid(payment: Payment):
relay_id = payment.extra.get("relay_id")
pubkey = payment.extra.get("pubkey")
if not relay_id or not pubkey:
logger.warning(f"Invoice extra data missing for 'relay_id' and 'pubkey'. Payment hash: {payment.payment_hash}")
return
if payment.extra.get("action") == "join":
await invoice_paid_to_join(relay_id, pubkey, payment.amount)
return
if payment.extra.get("action") == "storage":
storage_to_buy = payment.extra.get("storage_to_buy")
if not storage_to_buy:
logger.warning(f"Invoice extra data missing for 'storage_to_buy'. Payment hash: {payment.payment_hash}")
return
await invoice_paid_for_storage(relay_id, pubkey, storage_to_buy, payment.amount)
return
@ -56,7 +63,7 @@ async def invoice_paid_to_join(relay_id: str, pubkey: str, amount: int):
logger.warning(ex)
async def invoice_paid_for_storage(relay_id: str, pubkey: str, storage_to_buy: int, amount: str):
async def invoice_paid_for_storage(relay_id: str, pubkey: str, storage_to_buy: int, amount: int):
try:
account = await get_account(relay_id, pubkey)
if not account: