server/repository.py
changeset 9008 e0aa7cf8abf8
parent 9007 e27337dfec8c
child 9020 cb87e831c183
child 9132 b6bcf99504c9
--- a/server/repository.py	Wed Jun 12 18:33:50 2013 +0200
+++ b/server/repository.py	Fri Jun 14 11:53:40 2013 +0200
@@ -885,6 +885,8 @@
         # during `session_close` hooks
         session.commit()
         session.close()
+        if threading.currentThread() in self._pyro_sessions:
+            self._pyro_sessions[threading.currentThread()] = None
         del self._sessions[sessionid]
         self.info('closed session %s for user %s', sessionid, session.user.login)
 
@@ -1654,10 +1656,7 @@
         daemon.getAdapter().handleConnection = handleConnection
         def removeConnection(conn, sessions=pyro_sessions):
             daemon.__class__.removeConnection(daemon, conn)
-            try:
-                session = sessions[threading.currentThread()]
-            except KeyError:
-                return
+            session = sessions.pop(threading.currentThread(), None)
             if session is None:
                 # client was not yet connected to the repo
                 return