lnbits-legend/lnbits/extensions/example/tasks.py

30 lines
750 B
Python
Raw Normal View History

2023-01-06 14:40:42 +00:00
# tasks.py is for asynchronous when invoices get paid
2023-01-06 14:37:49 +00:00
# add your dependencies here
import asyncio
2023-01-06 15:28:32 +00:00
2023-01-06 14:37:49 +00:00
from loguru import logger
2023-01-06 15:28:32 +00:00
2023-01-06 14:37:49 +00:00
from lnbits.core.models import Payment
from lnbits.helpers import get_current_extension_name
from lnbits.tasks import register_invoice_listener
2023-01-06 15:28:32 +00:00
2023-01-06 14:37:49 +00:00
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)
2023-01-06 15:28:32 +00:00
2023-01-06 14:37:49 +00:00
async def on_invoice_paid(payment: Payment) -> None:
2023-01-06 15:28:32 +00:00
if (
payment.extra.get("tag") != "example"
): # Will grab any payment with the tag "example"
2023-01-06 14:37:49 +00:00
logger.debug(payment)
# Do something
2023-01-06 15:28:32 +00:00
return