FEAT: add funding source fallback and VoidWallet warning for frontend

This commit is contained in:
dni ⚡ 2023-03-02 12:03:58 +01:00
parent c81d3e06c8
commit e93fbd24be
No known key found for this signature in database
GPG Key ID: 886317704CC4E618
4 changed files with 32 additions and 5 deletions

View File

@ -105,20 +105,42 @@ async def check_funding_source() -> None:
signal.signal(signal.SIGINT, signal_handler)
WALLET = get_wallet_class()
# fallback to void after 30 seconds of failures
sleep_time = 5
timeout = int(30 / sleep_time)
balance = 0
retry_counter = 0
while True:
try:
error_message, balance = await WALLET.status()
if not error_message:
retry_counter = 0
break
logger.error(
f"The backend for {WALLET.__class__.__name__} isn't working properly: '{error_message}'",
RuntimeWarning,
)
except:
pass
logger.info("Retrying connection to backend in 5 seconds...")
await asyncio.sleep(5)
if retry_counter == timeout:
logger.warning(
f"Fallback to VoidWallet, because the backend for {WALLET.__class__.__name__} isn't working properly"
)
set_wallet_class("VoidWallet")
WALLET = get_wallet_class()
break
else:
logger.warning(f"Retrying connection to backend in {sleep_time} seconds...")
retry_counter += 1
await asyncio.sleep(sleep_time)
signal.signal(signal.SIGINT, original_sigint_handler)
logger.info(
f"✔️ Backend {WALLET.__class__.__name__} connected and with a balance of {balance} msat."
)

View File

@ -317,8 +317,9 @@ def set_cli_settings(**kwargs):
# set wallet class after settings are loaded
def set_wallet_class():
wallet_class = getattr(wallets_module, settings.lnbits_backend_wallet_class)
def set_wallet_class(class_name: Optional[str] = None):
backend_wallet_class = class_name or settings.lnbits_backend_wallet_class
wallet_class = getattr(wallets_module, backend_wallet_class)
global WALLET
WALLET = wallet_class()

View File

@ -45,6 +45,9 @@
</q-btn>
</q-toolbar-title>
{% block beta %}
<q-badge color="red" text-color="black" class="q-mr-md">
<span>VoidWallet is active! Payments disabled</span>
</q-badge>
<q-badge color="yellow" text-color="black" class="q-mr-md">
<span
><span v-show="$q.screen.gt.sm"

View File

@ -72,12 +72,14 @@ exclude = [
"lnbits/wallets",
"lnbits/core",
"lnbits/*.py",
"lnbits/extensions",
]
[tool.mypy]
files = "lnbits"
exclude = """(?x)(
^lnbits/wallets/lnd_grpc_files.
| ^lnbits/extensions.
)"""
[[tool.mypy.overrides]]
@ -90,7 +92,6 @@ module = [
"websocket.*",
"websockets.*",
"pyqrcode.*",
"cashu.*",
"shortuuid.*",
"grpc.*",
"lnurl.*",