mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
191ad1e224
this makes handling in tasks.py so much easier
40 lines
988 B
Python
40 lines
988 B
Python
import asyncio
|
|
|
|
from loguru import logger
|
|
|
|
from lnbits.core.models import Payment
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import (
|
|
get_market_order_details,
|
|
get_market_order_invoiceid,
|
|
set_market_order_paid,
|
|
update_market_product_stock,
|
|
)
|
|
|
|
|
|
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 payment.extra.get("tag") != "market":
|
|
return
|
|
|
|
order = await get_market_order_invoiceid(payment.payment_hash)
|
|
if not order:
|
|
logger.error("this should never happen", payment)
|
|
return
|
|
|
|
# set order as paid
|
|
await set_market_order_paid(payment.payment_hash)
|
|
|
|
# deduct items sold from stock
|
|
details = await get_market_order_details(order.id)
|
|
await update_market_product_stock(details)
|