mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import asyncio
|
|
import json
|
|
|
|
from loguru import logger
|
|
|
|
from lnbits.core.models import Payment
|
|
from lnbits.extensions.satspay.crud import check_address_balance, get_charge
|
|
from lnbits.helpers import get_current_extension_name
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import update_charge
|
|
from .helpers import call_webhook
|
|
|
|
|
|
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") != "charge":
|
|
# not a charge invoice
|
|
return
|
|
|
|
charge = await get_charge(payment.memo)
|
|
if not charge:
|
|
logger.error("this should never happen", payment)
|
|
return
|
|
|
|
await payment.set_pending(False)
|
|
charge = await check_address_balance(charge_id=charge.id)
|
|
|
|
if charge.must_call_webhook():
|
|
resp = await call_webhook(charge)
|
|
extra = {**charge.config.dict(), **resp}
|
|
await update_charge(charge_id=charge.id, extra=json.dumps(extra))
|