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

import asyncio

from loguru import logger

from lnbits.core.models import Payment
from lnbits.tasks import register_invoice_listener

from .crud import get_email_by_payment_hash, get_emailaddress, set_email_paid
from .smtp import send_mail


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") != "smtp":
        return

    email = await get_email_by_payment_hash(payment.checking_id)
    if not email:
        logger.error("SMTP: email can not by fetched")
        return

    emailaddress = await get_emailaddress(email.emailaddress_id)
    if not emailaddress:
        logger.error("SMTP: emailaddress can not by fetched")
        return

    await payment.set_pending(False)
    await send_mail(emailaddress, email)
    await set_email_paid(payment_hash=payment.payment_hash)