Merge pull request #1274 from prusnak/mypy-extension-lnticket

fix mypy for lnbits/extensions/lnticket

lnbits/extensions/lnticket/tasks.py

@@ -19,7 +19,7 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if payment.extra.get("tag") != "lnticket":
+    if not payment.extra or payment.extra.get("tag") != "lnticket":
         # not a lnticket invoice
         return
 

lnbits/extensions/lnticket/views.py

@@ -33,6 +33,7 @@ async def display(request: Request, form_id):
         )
 
     wallet = await get_wallet(form.wallet)
+    assert wallet
 
     return lnticket_renderer().TemplateResponse(
         "lnticket/display.html",

lnbits/extensions/lnticket/views_api.py

@@ -1,8 +1,7 @@
 import re
 from http import HTTPStatus
 
-from fastapi import Query
-from fastapi.params import Depends
+from fastapi import Depends, Query
 from starlette.exceptions import HTTPException
 
 from lnbits.core.crud import get_user
@@ -35,7 +34,8 @@ async def api_forms_get(
     wallet_ids = [wallet.wallet.id]
 
     if all_wallets:
-        wallet_ids = (await get_user(wallet.wallet.user)).wallet_ids
+        user = await get_user(wallet.wallet.user)
+        wallet_ids = user.wallet_ids if user else []
 
     return [form.dict() for form in await get_forms(wallet_ids)]
 
@@ -91,7 +91,8 @@ async def api_tickets(
     wallet_ids = [wallet.wallet.id]
 
     if all_wallets:
-        wallet_ids = (await get_user(wallet.wallet.user)).wallet_ids
+        user = await get_user(wallet.wallet.user)
+        wallet_ids = user.wallet_ids if user else []
 
     return [form.dict() for form in await get_tickets(wallet_ids)]
 

pyproject.toml

@@ -98,7 +98,6 @@ exclude = """(?x)(
     | ^lnbits/extensions/livestream.
     | ^lnbits/extensions/lnaddress.
     | ^lnbits/extensions/lndhub.
-    | ^lnbits/extensions/lnticket.
     | ^lnbits/extensions/lnurldevice.
     | ^lnbits/extensions/lnurlp.
     | ^lnbits/extensions/offlineshop.