diff --git a/app/utils/websocket_tool.py b/app/utils/websocket_tool.py index 708d8eb..64c9b39 100644 --- a/app/utils/websocket_tool.py +++ b/app/utils/websocket_tool.py @@ -102,5 +102,8 @@ class WsFile: pickled = {"id": self.id} return pickled + def getWs(self): + return self.ws + def __setstate(self, pickled_dict): self.__init__(pickled_dict['id'])