lnbits-legend/lnbits/extensions/lnticket/tasks.py
Pavol Rusnak 191ad1e224
make payment.extra non-optional
this makes handling in tasks.py so much easier
2023-01-05 09:22:34 +01:00

32 lines
863 B
Python

import asyncio
from loguru import logger
from lnbits.core.models import Payment
from lnbits.helpers import get_current_extension_name
from lnbits.tasks import register_invoice_listener
from .crud import get_ticket, set_ticket_paid
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)
async def on_invoice_paid(payment: Payment) -> None:
if payment.extra.get("tag") != "lnticket":
# not a lnticket invoice
return
ticket = await get_ticket(payment.checking_id)
if not ticket:
logger.error("this should never happen", payment)
return
await payment.set_pending(False)
await set_ticket_paid(payment.payment_hash)