fix: properly start and stop tasks (#31)
This commit is contained in:
parent
631624c54f
commit
825e603a11
1 changed files with 14 additions and 4 deletions
18
__init__.py
18
__init__.py
|
|
@ -1,10 +1,10 @@
|
||||||
import asyncio
|
import asyncio
|
||||||
|
from loguru import logger
|
||||||
|
|
||||||
from fastapi import APIRouter
|
from fastapi import APIRouter
|
||||||
|
|
||||||
from lnbits.db import Database
|
from lnbits.db import Database
|
||||||
from lnbits.helpers import template_renderer
|
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")
|
db = Database("ext_boltcards")
|
||||||
|
|
||||||
|
|
@ -25,10 +25,20 @@ def boltcards_renderer():
|
||||||
from .lnurl import * # noqa: F401,F403
|
from .lnurl import * # noqa: F401,F403
|
||||||
from .tasks 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():
|
def boltcards_start():
|
||||||
loop = asyncio.get_event_loop()
|
task = create_permanent_unique_task(wait_for_paid_invoices)
|
||||||
loop.create_task(catch_everything_and_restart(wait_for_paid_invoices)) # noqa: F405
|
scheduled_tasks.append(task)
|
||||||
|
|
||||||
|
|
||||||
from .views import * # noqa: F401,F403
|
from .views import * # noqa: F401,F403
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue