56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import asyncio
|
|
|
|
from fastapi import APIRouter
|
|
from lnbits.tasks import create_permanent_unique_task
|
|
from loguru import logger
|
|
|
|
from .crud import db
|
|
from .tasks import wait_for_paid_invoices, hourly_transaction_polling
|
|
from .views import satmachineadmin_generic_router
|
|
from .views_api import satmachineadmin_api_router
|
|
|
|
logger.debug(
|
|
"This logged message is from satmachineadmin/__init__.py, you can debug in your "
|
|
"extension using 'import logger from loguru' and 'logger.debug(<thing-to-log>)'."
|
|
)
|
|
|
|
|
|
satmachineadmin_ext: APIRouter = APIRouter(prefix="/satmachineadmin", tags=["DCA Admin"])
|
|
satmachineadmin_ext.include_router(satmachineadmin_generic_router)
|
|
satmachineadmin_ext.include_router(satmachineadmin_api_router)
|
|
|
|
satmachineadmin_static_files = [
|
|
{
|
|
"path": "/satmachineadmin/static",
|
|
"name": "satmachineadmin_static",
|
|
}
|
|
]
|
|
|
|
scheduled_tasks: list[asyncio.Task] = []
|
|
|
|
|
|
def satmachineadmin_stop():
|
|
for task in scheduled_tasks:
|
|
try:
|
|
task.cancel()
|
|
except Exception as ex:
|
|
logger.warning(ex)
|
|
|
|
|
|
def satmachineadmin_start():
|
|
# Start invoice listener task
|
|
invoice_task = create_permanent_unique_task("ext_satmachineadmin", wait_for_paid_invoices)
|
|
scheduled_tasks.append(invoice_task)
|
|
|
|
# Start hourly transaction polling task
|
|
polling_task = create_permanent_unique_task("ext_satmachineadmin_polling", hourly_transaction_polling)
|
|
scheduled_tasks.append(polling_task)
|
|
|
|
|
|
__all__ = [
|
|
"db",
|
|
"satmachineadmin_ext",
|
|
"satmachineadmin_static_files",
|
|
"satmachineadmin_start",
|
|
"satmachineadmin_stop",
|
|
]
|