Fix to maintain backwards compatibility

wireload · Nov 28, 2013 · 4544cd5 · 4544cd5
1 parent 0154a66
commit 4544cd5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/geventwebsocket/websocket.py b/geventwebsocket/websocket.py
@@ -107,7 +107,14 @@ def _is_valid_close_code(self, code):
 
     @property
     def current_app(self):
-        return self.handler.server.application.current_app
+        if hasattr(self.handler.server.application, 'current_app'):
+            return self.handler.server.application.current_app
+        else:
+            class MockApp():
+                def on_close(self, *args):
+                    pass
+
+            return MockApp()
 
     @property
     def origin(self):