polish structure, update requirements

[hublib] / hub / websocket.py

diff --git a/hub/websocket.py b/hub/websocket.py
index 3d511168968e8386898623b9d70c6f84eba077fe..89f11928d97938ee9bd5be669ecafed79fd823ce 100644 (file)
--- a/hub/websocket.py
+++ b/hub/websocket.py
@@ -13,14 +13,13 @@ from .utils import get_redis_pass
 
 
 class BaseWebSocketHub:
-    first = True
     client_ids_sem = asyncio.Semaphore(0)
 
     @classmethod
-    def __class_init(cls, redis):
-        if not cls.first:
+    def _class_init(cls, redis):
+        if not hasattr(cls, "_class_init"):
             return
-        cls.first = False
+        delattr(cls, "_class_init")
         asyncio.create_task(cls.initialize_client_ids(redis))
 
     @classmethod
@@ -30,7 +29,8 @@ class BaseWebSocketHub:
 
     def __init__(self):
         self.redis = StrictRedis(password=get_redis_pass("/etc/redis/redis.conf"))
-        self.__class_init(self.redis)
+        if hasattr(BaseWebSocketHub, "_class_init"):
+            BaseWebSocketHub._class_init(self.redis)
 
     def task_done(self):
         self.task = None