nostrmarket/tasks.py
Vlad Stan c073bf55c9 Fix high load (#84)
* chore: testing

* fix: resubscribe when merchant added/removed

* chore: code clean-up

* fix: temp subscribe when new merchant is added

* fix: new customer profile
2023-09-20 09:17:01 +03:00

41 lines
No EOL
1 KiB
Python

from asyncio import Queue
from lnbits.core.models import Payment
from lnbits.tasks import register_invoice_listener
from .nostr.nostr_client import NostrClient
from .services import (
handle_order_paid,
process_nostr_message,
subscribe_to_all_merchants,
)
async def wait_for_paid_invoices():
invoice_queue = Queue()
register_invoice_listener(invoice_queue)
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") != "nostrmarket":
return
order_id = payment.extra.get("order_id")
merchant_pubkey = payment.extra.get("merchant_pubkey")
if not order_id or not merchant_pubkey:
return None
await handle_order_paid(order_id, merchant_pubkey)
async def wait_for_nostr_events(nostr_client: NostrClient):
await subscribe_to_all_merchants()
while True:
message = await nostr_client.get_event()
await process_nostr_message(message)