From e32b7b41ccd69ed411e7a2b9d8570c729c5e28cf Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Mon, 30 Sep 2013 21:15:37 -0400
Subject: [PATCH] don't hard-depend on gevent

---
 flask_sockets.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/flask_sockets.py b/flask_sockets.py
index dfac23f..2c8bfed 100644
--- a/flask_sockets.py
+++ b/flask_sockets.py
@@ -1,8 +1,12 @@
 # -*- coding: utf-8 -*-
 
-import gevent
-from flask import request
-from websocket import handle_websocket
+try:
+    import gevent
+    from geventwebsocket.gunicorn.workers import GeventWebSocketWorker as Worker
+except ImportError:
+    # Don't fail, in case someone is using somethign else that provides
+    # "wsgi.websocket".
+    pass
 
 
 def log_request(self):
@@ -40,8 +44,9 @@ class Sockets(object):
     def init_app(self, app, patch=True):
         if patch:
             # Monkey-patch log_request handler for Gevent/Gunicorn compatability.
-            if hasattr(gevent, 'pywsgi'):
-                gevent.pywsgi.WSGIHandler.log_request = log_request
+            if 'gevent' in locals():
+                if hasattr(gevent, 'pywsgi'):
+                    gevent.pywsgi.WSGIHandler.log_request = log_request
 
         app.wsgi_app = SocketMiddleware(app.wsgi_app, self)
 
@@ -54,4 +59,8 @@ class Sockets(object):
         return decorator
 
     def add_url_rule(self, rule, _, f, **options):
-        self.url_map[rule] = f
\ No newline at end of file
+        self.url_map[rule] = f
+
+# CLI sugar.
+if 'Worker' in locals():
+    worker = Worker
\ No newline at end of file