added timeout error message

gretzke · Dec 9, 2019 · 17ead7b · 17ead7b
1 parent a092363
commit 17ead7b
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/plug_py/plug.py b/plug_py/plug.py
@@ -13,7 +13,7 @@
 # websockets
 import websockets
 import asyncio
-from socket import socket, AF_INET, SOCK_DGRAM, SOL_SOCKET, SO_REUSEADDR
+from socket import socket, AF_INET, SOCK_DGRAM, SOL_SOCKET, SO_REUSEADDR, timeout
 
 # utility libraries
 import operator
@@ -347,9 +347,13 @@ def run(self):
                 asyncio.set_event_loop(self.loop)
                 self.loop.run_until_complete(self.wsConnection())
             else:
-                window.updateInfo('No smart socket found')
+                window.updateInfo('Error while connecting to smart socket')
+                window.updateInfoCenter('')
+        except timeout:
+            window.updateInfo('No smart socket found')
+            window.updateInfoCenter('')
         except:
-            window.updateInfo('Error in connecting to smart socket')
+            window.updateInfo('Error while connecting to smart socket')
             window.updateInfoCenter('')
 
     # disconnect from websocket and display message to user
@@ -545,7 +549,7 @@ async def initialized_P(self):
                 if (transactionReceipt.status == 1):
                     window.updateInfoCenter(
                         'Successfully initialized Payment Channel')
-
+                    QThread().sleep(0.5)
                     await self.websocket.send(json.dumps(
                         {"id": self.paymentState.value, "price": self.pricePerSecond, 'nonce': self.nonce, 'maxValue': self.maxValue}))
                     self.paymentState = State.initialized_S
@@ -559,7 +563,6 @@ async def initialized_S(self):
         self.transactionCounter = 0
         self.transactionValue = 0
         window.updateInfo('Payment Channel active')
-        QThread().sleep(0.5)
         self.paymentState = State.active_P
 
     async def active_P(self):