From 9aa2194d946cc06a22cd8435cfc822b7a4f654b5 Mon Sep 17 00:00:00 2001 From: Vlad Stan Date: Thu, 3 Jul 2025 12:04:46 +0300 Subject: [PATCH] refactor: use `while settings.lnbits_running` instead of `while True` --- lnbits/wallets/base.py | 4 +++- lnbits/wallets/boltz.py | 2 +- lnbits/wallets/breez.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/lnbits/wallets/base.py b/lnbits/wallets/base.py index 80badd26..6f5c6126 100644 --- a/lnbits/wallets/base.py +++ b/lnbits/wallets/base.py @@ -7,6 +7,8 @@ from typing import TYPE_CHECKING, NamedTuple from loguru import logger +from lnbits.settings import settings + if TYPE_CHECKING: from lnbits.nodes.base import Node @@ -139,7 +141,7 @@ class Wallet(ABC): pass async def paid_invoices_stream(self) -> AsyncGenerator[str, None]: - while True: + while settings.lnbits_running: for invoice in self.pending_invoices: try: status = await self.get_invoice_status(invoice) diff --git a/lnbits/wallets/boltz.py b/lnbits/wallets/boltz.py index fa7bf878..87d236d1 100644 --- a/lnbits/wallets/boltz.py +++ b/lnbits/wallets/boltz.py @@ -236,7 +236,7 @@ class BoltzWallet(Wallet): return PaymentFailedStatus() async def paid_invoices_stream(self) -> AsyncGenerator[str, None]: - while True: + while settings.lnbits_running: try: request = boltzrpc_pb2.GetSwapInfoRequest() info: boltzrpc_pb2.GetSwapInfoResponse diff --git a/lnbits/wallets/breez.py b/lnbits/wallets/breez.py index 363acb12..87f80a62 100644 --- a/lnbits/wallets/breez.py +++ b/lnbits/wallets/breez.py @@ -277,7 +277,7 @@ else: return PaymentPendingStatus() async def paid_invoices_stream(self) -> AsyncGenerator[str, None]: - while True: + while settings.lnbits_running: event = await breez_event_queue.get() if event.is_invoice_paid(): yield event.details.payment_hash