mirrors/lnbits-legend
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits-legend.git synced 2025-01-19 05:33:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: order of migrations

Browse Source
This commit is contained in:
Vlad Stan 2023-01-18 18:35:02 +02:00
parent d3c05ef300
commit ce1639792a
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lnbits/app.py
View File

@ -21,10 +21,11 @@ from fastapi.staticfiles import StaticFiles
from loguru import logger
from lnbits.core.crud import get_installed_extensions
from lnbits.core.helpers import migrate_extension_database
from lnbits.core.tasks import register_task_listeners
from lnbits.settings import get_wallet_class, set_wallet_class, settings
from .commands import migrate_databases
from .commands import db_versions, migrate_databases
from .core import core_app, core_app_extra
from .core.services import check_admin_settings
from .core.views.generic import core_html_routes
@ -132,9 +133,11 @@ async def check_installed_extensions(app: FastAPI):
for ext in installed_extensions:
try:
is_installed = check_installed_extension(ext)
if not is_installed:
installed = check_installed_extension(ext)
if not installed:
register_ext_routes(app, Extension(ext.id, True, False))
current_version = (await db_versions()).get(ext.id, 0)
await migrate_extension_database(ext, current_version)
except:
logger.warning(f"Failed to re-install extension: {ext.id}")
@ -204,9 +207,6 @@ def register_startup(app: FastAPI):
async def lnbits_startup():
try:
# check extensions after restart
await check_installed_extensions(app)
# wait till migration is done
await migrate_databases()
@ -221,6 +221,9 @@ def register_startup(app: FastAPI):
# initialize funding source
await check_funding_source()
# check extensions after restart
await check_installed_extensions(app)
except Exception as e:
logger.error(str(e))
raise ImportError("Failed to run 'startup' event.")

6
lnbits/commands.py
View File

@ -81,6 +81,12 @@ async def migrate_databases():
logger.info("✔️ All migrations done.")
async def db_versions():
async with core_db.connect() as conn:
current_versions = await get_dbversions(conn)
return current_versions
async def load_disabled_extension_list() -> None:
"""Update list of extensions that have been explicitly disabled"""
inactive_extensions = await get_inactive_extensions()