mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import asyncio
|
|
|
|
from lnbits.core.models import Payment
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import (
|
|
create_invoice_payment,
|
|
get_invoice,
|
|
get_invoice_items,
|
|
get_invoice_payments,
|
|
get_invoice_total,
|
|
get_payments_total,
|
|
update_invoice_internal,
|
|
)
|
|
from .models import InvoiceStatusEnum
|
|
|
|
|
|
async def wait_for_paid_invoices():
|
|
invoice_queue = asyncio.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 not payment.extra:
|
|
return
|
|
|
|
if payment.extra.get("tag") != "invoices":
|
|
return
|
|
|
|
invoice_id = payment.extra.get("invoice_id")
|
|
assert invoice_id
|
|
|
|
amount = payment.extra.get("famount")
|
|
assert amount
|
|
|
|
await create_invoice_payment(invoice_id=invoice_id, amount=amount)
|
|
|
|
invoice = await get_invoice(invoice_id)
|
|
assert invoice
|
|
|
|
invoice_items = await get_invoice_items(invoice_id)
|
|
invoice_total = await get_invoice_total(invoice_items)
|
|
|
|
invoice_payments = await get_invoice_payments(invoice_id)
|
|
payments_total = await get_payments_total(invoice_payments)
|
|
|
|
if payments_total >= invoice_total:
|
|
invoice.status = InvoiceStatusEnum.paid
|
|
await update_invoice_internal(invoice.wallet, invoice)
|
|
|
|
return
|