aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions 1

refactor: class rename and code format

Browse source
This commit is contained in:
Vlad Stan 2022-11-29 18:19:33 +02:00
parent 598d2a6939
commit 8f8ebbeeaa
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
lnbits/app.py
View file

@ -27,8 +27,8 @@ from .core import core_app, core_app_extra
from .core.services import check_admin_settings from .core.services import check_admin_settings
from .core.views.generic import core_html_routes from .core.views.generic import core_html_routes
from .helpers import ( from .helpers import (
EnabledExtensionMiddleware,
Extension, Extension,
InstalledExtensionMiddleware,
get_css_vendored, get_css_vendored,
get_js_vendored, get_js_vendored,
get_valid_extensions, get_valid_extensions,
@ -72,7 +72,7 @@ def create_app() -> FastAPI:
) )
app.add_middleware(GZipMiddleware, minimum_size=1000) app.add_middleware(GZipMiddleware, minimum_size=1000)
app.add_middleware(EnabledExtensionMiddleware) app.add_middleware(InstalledExtensionMiddleware)
register_startup(app) register_startup(app)
register_assets(app) register_assets(app)

7
lnbits/helpers.py
View file

@ -80,7 +80,7 @@ class ExtensionManager:
return output return output
class EnabledExtensionMiddleware: class InstalledExtensionMiddleware:
def __init__(self, app: ASGIApp) -> None: def __init__(self, app: ASGIApp) -> None:
self.app = app self.app = app
@ -97,14 +97,15 @@ class EnabledExtensionMiddleware:
# re-route trafic if the extension has been upgraded # re-route trafic if the extension has been upgraded
upgraded_extensions = list( upgraded_extensions = list(
filter( filter(
lambda ext: ext.endswith(f"/{pathname}"), g().config.LNBITS_UPGRADED_EXTENSIONS) lambda ext: ext.endswith(f"/{pathname}"),
g().config.LNBITS_UPGRADED_EXTENSIONS,
)
) )
if len(upgraded_extensions) != 0: if len(upgraded_extensions) != 0:
upgrade_path = upgraded_extensions[0] upgrade_path = upgraded_extensions[0]
tail = "/".join(rest) tail = "/".join(rest)
scope["path"] = f"/upgrades/{upgrade_path}/{tail}" scope["path"] = f"/upgrades/{upgrade_path}/{tail}"
await self.app(scope, receive, send) await self.app(scope, receive, send)