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