Add files via upload

tasks.py

@@ -0,0 +1,59 @@
+import asyncio
+
+from loguru import logger
+
+from lnbits.core.models import Payment
+from lnbits.core.services import create_invoice, pay_invoice, websocketUpdater
+from lnbits.helpers import get_current_extension_name
+from lnbits.tasks import register_invoice_listener
+
+from .crud import get_temp
+
+
+#######################################
+########## RUN YOU TASKS HERE #########
+#######################################
+
+
+# the usual task is to listen to invoices related to this extension
+
+async def wait_for_paid_invoices():
+    invoice_queue = asyncio.Queue()
+    register_invoice_listener(invoice_queue, get_current_extension_name())
+
+    while True:
+        payment = await invoice_queue.get()
+        await on_invoice_paid(payment)
+
+
+# do somethhing when an invoice related top this extension is paid
+
+async def on_invoice_paid(payment: Payment) -> None:
+    if payment.extra.get("tag") != "temp":
+        return
+
+    temp_id = payment.extra.get("tempId")
+    assert temp_id
+
+    temp = await get_temp(temp_id)
+    assert temp
+
+    # update something
+    
+    data_to_update = {
+        "total" temp.total + payment.amount
+    }
+
+    await update_temp(temp_id=temp_id, **data_to_update.dict())
+
+
+    # here we could send some data to a websocket on wss://<your-lnbits>/api/v1/ws/<temp_id>
+
+    some_payment_data = {
+        "name": temp.name,
+        "amount": payment.amount,
+        "fee": payment.fee,
+        "checking_id": payment.checking_id
+    }
+
+    await websocketUpdater(temp_id, str(some_payment_data))