Polling fixes and session code cleanup (thanks Enki)

liuwen-lvtu · Aug 1, 2011 · bf8afc6 · bf8afc6
1 parent edc8ad1
commit bf8afc6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tornadio/polling.py b/tornadio/polling.py
@@ -253,7 +253,6 @@ class TornadioHtmlFileSocketHandler(TornadioPollingHandlerBase):
     @asynchronous
     def get(self, *args, **kwargs):
         if not self.session.set_handler(self):
-            # TODO: Error logging
             raise HTTPError(401, 'Forbidden')
 
         self.set_header('Content-Type', 'text/html')
@@ -311,6 +310,11 @@ def post(self, *args, **kwargs):
         super(TornadioJSONPSocketHandler, self).post(*args, **kwargs)
 
     def data_available(self, raw_data):
+        if not self._index:
+
+
+            raise HTTPError(401, 'unauthorized')
+
         message = 'io.JSONP[%s]._(%s);' % (
             self._index,
             json.dumps(raw_data)

diff --git a/tornadio/session.py b/tornadio/session.py
@@ -81,9 +81,8 @@ def remove(self, session_id):
         session = self._items.get(session_id, None)
 
         if session is not None:
-            self._items[session].promoted = -1
+            session.promoted = -1
             session.on_delete(True)
-            del self._items[session_id]
             return True
 
         return False