GitHub/butterfly
1
0
Fork 0
mirror of https://github.com/paradoxxxzero/butterfly.git synced 2026-05-26 07:08:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Protect session closing. References #124

Browse Source
This commit is contained in:
Florian Mounier
2017-02-21 11:14:04 +01:00
parent ffd19b8162
commit 40496eb9d1
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
butterfly/routes.py
View File

@@ -283,14 +283,19 @@ class TermWebSocket(Route, KeptAliveWebSocketHandler):
@classmethod
def close_session(cls, session):
wsockets = (cls.sessions.get(session) +
TermCtlWebSocket.sessions.get(session))
wsockets = (cls.sessions.get(session, []) +
TermCtlWebSocket.sessions.get(session, []))
for wsocket in wsockets:
wsocket.on_close()
wsocket.close()
del cls.sessions[session]
del TermCtlWebSocket.sessions_secure_users[session]
del TermCtlWebSocket.sessions[session]
if session in cls.sessions:
del cls.sessions[session]
if session in TermCtlWebSocket.sessions_secure_users:
del TermCtlWebSocket.sessions_secure_users[session]
if session in TermCtlWebSocket.sessions:
del TermCtlWebSocket.sessions[session]
@classmethod
def broadcast(cls, session, message, emitter=None):