server: fix error 'has no attribute last_read_time'

fossasia · Mar 20, 2023 · 8e8074d · 8e8074d
1 parent f8bb7ad
commit 8e8074d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/py/visdom/server/handlers/socket_handlers.py b/py/visdom/server/handlers/socket_handlers.py
@@ -330,10 +330,16 @@ def socket_wrap_monitor_thread(self):
         # TODO mark wrapped subs and sources separately
         if len(self.subs) > 0 or len(self.sources) > 0:
             for sub in list(self.subs.values()):
-                if time.time() - sub.last_read_time > MAX_SOCKET_WAIT:
+                if (
+                    hasattr(sub, "last_read_time")
+                    and time.time() - sub.last_read_time > MAX_SOCKET_WAIT
+                ):
                     sub.close()
             for sub in list(self.sources.values()):
-                if time.time() - sub.last_read_time > MAX_SOCKET_WAIT:
+                if (
+                    hasattr(sub, "last_read_time")
+                    and time.time() - sub.last_read_time > MAX_SOCKET_WAIT
+                ):
                     sub.close()
         else:
             self.app.socket_wrap_monitor.stop()