fix: properly start and stop tasks (#31)

This commit is contained in:
dni ⚡ 2024-04-15 10:32:47 +02:00 committed by GitHub
commit 825e603a11
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,10 +1,10 @@
import asyncio
from loguru import logger
from fastapi import APIRouter
from lnbits.db import Database
from lnbits.helpers import template_renderer
from lnbits.tasks import catch_everything_and_restart
from lnbits.tasks import create_permanent_unique_task
db = Database("ext_boltcards")
@ -25,10 +25,20 @@ def boltcards_renderer():
from .lnurl import * # noqa: F401,F403
from .tasks import * # noqa: F401,F403
scheduled_tasks: list[asyncio.Task] = []
def boltcards_stop():
for task in scheduled_tasks:
try:
task.cancel()
except Exception as ex:
logger.warning(ex)
def boltcards_start():
loop = asyncio.get_event_loop()
loop.create_task(catch_everything_and_restart(wait_for_paid_invoices)) # noqa: F405
task = create_permanent_unique_task(wait_for_paid_invoices)
scheduled_tasks.append(task)
from .views import * # noqa: F401,F403