feat: make events dynamic (#43)
--------- Co-authored-by: dni <office@dnilabs.com>
This commit is contained in:
parent
f06bd9a668
commit
9e477ac959
21 changed files with 1164 additions and 1143 deletions
25
tasks.py
25
tasks.py
|
|
@ -5,8 +5,24 @@ from lnbits.tasks import register_invoice_listener
|
|||
from loguru import logger
|
||||
|
||||
from .crud import get_ticket
|
||||
from .models import Ticket
|
||||
from .services import set_ticket_paid
|
||||
|
||||
payment_listeners: dict[str, list[asyncio.Queue[Ticket]]] = {}
|
||||
|
||||
|
||||
def register_payment_listener(payment_hash, queue: asyncio.Queue[Ticket]) -> None:
|
||||
if payment_hash not in payment_listeners:
|
||||
payment_listeners[payment_hash] = []
|
||||
payment_listeners[payment_hash].append(queue)
|
||||
|
||||
|
||||
def deregister_payment_listener(payment_hash, queue: asyncio.Queue[Ticket]) -> None:
|
||||
if payment_hash in payment_listeners:
|
||||
payment_listeners[payment_hash].remove(queue)
|
||||
if not payment_listeners[payment_hash]:
|
||||
del payment_listeners[payment_hash]
|
||||
|
||||
|
||||
async def wait_for_paid_invoices():
|
||||
invoice_queue = asyncio.Queue()
|
||||
|
|
@ -21,13 +37,12 @@ async def on_invoice_paid(payment: Payment) -> None:
|
|||
if not payment.extra or "events" != payment.extra.get("tag"):
|
||||
return
|
||||
|
||||
if not payment.extra.get("name") or not payment.extra.get("email"):
|
||||
logger.warning(f"Ticket {payment.payment_hash} missing name or email.")
|
||||
return
|
||||
|
||||
ticket = await get_ticket(payment.payment_hash)
|
||||
if not ticket:
|
||||
logger.warning(f"Ticket for payment {payment.payment_hash} not found.")
|
||||
return
|
||||
|
||||
await set_ticket_paid(ticket)
|
||||
ticket = await set_ticket_paid(ticket)
|
||||
if payment_listeners.get(payment.payment_hash):
|
||||
for paid_ticket_queue in payment_listeners[payment.payment_hash]:
|
||||
paid_ticket_queue.put_nowait(ticket)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue