From 4585d97324aa2ce7ea244c2b5ab35293a97634f7 Mon Sep 17 00:00:00 2001
From: Tiago vasconcelos <talvasconcelos@gmail.com>
Date: Wed, 23 Jun 2021 18:50:01 +0100
Subject: [PATCH] small fix on paid_invoices_stream

---
 lnbits/wallets/eclair.py | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/lnbits/wallets/eclair.py b/lnbits/wallets/eclair.py
index 2f29501c..9c8fa540 100644
--- a/lnbits/wallets/eclair.py
+++ b/lnbits/wallets/eclair.py
@@ -152,15 +152,17 @@ class EclairWallet(Wallet):
         url = urllib.parse.urlsplit(self.url)
         ws_url = f"ws://{url.netloc}/ws"
 
-        try:
-            async with open_websocket_url(ws_url, extra_headers=[('Authorization', self.auth["Authorization"])]) as ws:
-                message = await ws.get_message()
-                if message["type"] == "payment-received":
-                    print('Received message: %s' % message)
-                    yield message["paymentHash"]
+        while True:
+            try:
+                async with open_websocket_url(ws_url, extra_headers=[('Authorization', self.auth["Authorization"])]) as ws:
+                    message = await ws.get_message()
+                    if "payment-received" in message["type"]:
+                        print('Received message: %s' % message)
+                        yield message["paymentHash"]
 
-        except OSError as ose:
-            pass
+            except OSError as ose:
+                print('OSE', ose)
+                pass
 
-        print("lost connection to eclair's websocket, retrying in 5 seconds")
-        await trio.sleep(5)
+            print("lost connection to eclair's websocket, retrying in 5 seconds")
+            await trio.sleep(5)