Merge pull request #1280 from prusnak/unify-payment-extra

make payment.extra non-optional

lnbits/core/models.py

@@ -88,7 +88,7 @@ class Payment(BaseModel):
     preimage: str
     payment_hash: str
     expiry: Optional[float]
-    extra: Optional[Dict] = {}
+    extra: Dict = {}
     wallet_id: str
     webhook: Optional[str]
     webhook_status: Optional[int]

lnbits/extensions/cashu/tasks.py

@@ -28,6 +28,7 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if payment.extra and not payment.extra.get("tag") == "cashu":
+    if payment.extra.get("tag") != "cashu":
         return
+
     return

lnbits/extensions/copilot/tasks.py

@@ -24,12 +24,12 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    webhook = None
-    data = None
-    if not payment.extra or payment.extra.get("tag") != "copilot":
+    if payment.extra.get("tag") != "copilot":
         # not an copilot invoice
         return
 
+    webhook = None
+    data = None
     copilot = await get_copilot(payment.extra.get("copilotid", -1))
 
     if not copilot:

lnbits/extensions/invoices/tasks.py

@@ -25,9 +25,6 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if not payment.extra:
-        return
-
     if payment.extra.get("tag") != "invoices":
         return
 

lnbits/extensions/jukebox/tasks.py

@@ -17,8 +17,8 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if payment.extra:
-        if payment.extra.get("tag") != "jukebox":
-            # not a jukebox invoice
-            return
-        await update_jukebox_payment(payment.payment_hash, paid=True)
+    if payment.extra.get("tag") != "jukebox":
+        # not a jukebox invoice
+        return
+
+    await update_jukebox_payment(payment.payment_hash, paid=True)

lnbits/extensions/lnticket/tasks.py

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

lnbits/extensions/lnurlp/tasks.py

@@ -22,7 +22,7 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment):
-    if not payment.extra or payment.extra.get("tag") != "lnurlp":
+    if payment.extra.get("tag") != "lnurlp":
         return
 
     if payment.extra.get("wh_status"):

lnbits/extensions/market/tasks.py

@@ -23,9 +23,6 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if not payment.extra:
-        return
-
     if payment.extra.get("tag") != "market":
         return
 

lnbits/extensions/nostrnip5/tasks.py

@@ -18,8 +18,6 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if not payment.extra:
-        return
     if payment.extra.get("tag") != "nostrnip5":
         return
 

lnbits/extensions/scrub/tasks.py

@@ -27,7 +27,7 @@ async def wait_for_paid_invoices():
 
 async def on_invoice_paid(payment: Payment):
     # (avoid loops)
-    if payment.extra and payment.extra.get("tag") == "scrubed":
+    if payment.extra.get("tag") == "scrubed":
         # already scrubbed
         return
 

lnbits/extensions/splitpayments/tasks.py

@@ -20,7 +20,7 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if not payment.extra or payment.extra.get("tag") == "splitpayments":
+    if payment.extra.get("tag") == "splitpayments":
         # already a splitted payment, ignore
         return
 

lnbits/extensions/subdomains/tasks.py

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

lnbits/extensions/tpos/tasks.py

@@ -20,8 +20,6 @@ async def wait_for_paid_invoices():
 
 
 async def on_invoice_paid(payment: Payment) -> None:
-    if not payment.extra:
-        return
     if payment.extra.get("tag") != "tpos":
         return