Skip to content

Commit

Permalink
fix: properly start and stop tasks (#31)
Browse files Browse the repository at this point in the history
  • Loading branch information
dni authored Apr 15, 2024
1 parent 631624c commit 825e603
Showing 1 changed file with 14 additions and 4 deletions.
18 changes: 14 additions & 4 deletions __init__.py
Original file line number Diff line number Diff line change
@@ -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")

Expand All @@ -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
Expand Down

0 comments on commit 825e603

Please sign in to comment.