Keep track of websocket connections

This commit is contained in:
shortcutme 2017-11-05 23:40:59 +01:00
parent b39d86d020
commit 2781e31be0
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
2 changed files with 3 additions and 0 deletions

View file

@ -605,7 +605,9 @@ class UiRequest(object):
return self.error403()
ui_websocket = UiWebsocket(ws, site, self.server, user, self)
site.websockets.append(ui_websocket) # Add to site websockets to allow notify on events
self.server.websockets.append(ui_websocket)
ui_websocket.start()
self.server.websockets.remove(ui_websocket)
for site_check in self.server.sites.values():
# Remove websocket from every site (admin sites allowed to join other sites event channels)
if ui_websocket in site_check.websockets:

View file

@ -70,6 +70,7 @@ class UiServer:
self.wrapper_nonces = []
self.add_nonces = []
self.websockets = []
self.site_manager = SiteManager.site_manager
self.sites = SiteManager.site_manager.list()
self.log = logging.getLogger(__name__)