Full Python 2+3 support

Author: Eugeny

setup.py

@@ -27,7 +27,7 @@ def run_tests(self):
     license="BSD",
     url="https://github.com/abourget/gevent-socketio",
     download_url="https://github.com/abourget/gevent-socketio",
-    install_requires=("gevent", "gevent-websocket",),
+    install_requires=("gevent", "gevent-websocket","six"),
     setup_requires=('versiontools >= 1.7'),
     cmdclass = {'test': PyTest},
     tests_require=['pytest', 'mock'],

socketio/handler.py

@@ -1,11 +1,12 @@
 import sys
 import re
 import gevent
-import urlparse
+from six.moves.urllib.parse import parse_qs
 
 from gevent.pywsgi import WSGIHandler
 from socketio import transports
 
+
 class SocketIOHandler(WSGIHandler):
     RE_REQUEST_URL = re.compile(r"""
         ^/(?P<resource>.+?)
@@ -45,23 +46,23 @@ def __init__(self, config, *args, **kwargs):
 
         super(SocketIOHandler, self).__init__(*args, **kwargs)
 
-        self.transports = self.handler_types.keys()
+        self.transports = list(self.handler_types.keys())
         if self.server.transports:
             self.transports = self.server.transports
             if not set(self.transports).issubset(set(self.handler_types)):
                 raise ValueError("transports should be elements of: %s" %
-                    (self.handler_types.keys()))
+                    list(self.handler_types.keys()))
 
     def _do_handshake(self, tokens):
         if tokens["resource"] != self.server.resource:
             self.log_error("socket.io URL mismatch")
         else:
             socket = self.server.get_socket()
             data = "%s:%s:%s:%s" % (socket.sessid,
-                                    self.config['heartbeat_timeout'] or '',
-                                    self.config['close_timeout'] or '',
-                                    ",".join(self.transports))
-            self.write_smart(data)
+                                     self.config['heartbeat_timeout'] or '',
+                                     self.config['close_timeout'] or '',
+                                     ",".join(self.transports))
+            self.write_smart(data.encode('latin-1'))
 
     def write_jsonp_result(self, data, wrapper="0"):
         self.start_response("200 OK", [
@@ -74,13 +75,13 @@ def write_plain_result(self, data):
             ("Access-Control-Allow-Origin", self.environ.get('HTTP_ORIGIN', '*')),
             ("Access-Control-Allow-Credentials", "true"),
             ("Access-Control-Allow-Methods", "POST, GET, OPTIONS"),
-            ("Access-Control-Max-Age", 3600),
+            ("Access-Control-Max-Age", "3600"),
             ("Content-Type", "text/plain"),
         ])
         self.result = [data]
 
     def write_smart(self, data):
-        args = urlparse.parse_qs(self.environ.get("QUERY_STRING"))
+        args = parse_qs(self.environ.get("QUERY_STRING"))
 
         if "jsonp" in args:
             self.write_jsonp_result(data, args["jsonp"][0])

socketio/mixins.py

@@ -5,6 +5,8 @@
 You'll likely want to create your own Mixins.
 """
 
+import six
+
 
 class RoomsMixin(object):
     def __init__(self, *args, **kwargs):
@@ -30,7 +32,7 @@ def emit_to_room(self, room, event, *args):
                    args=args,
                    endpoint=self.ns_name)
         room_name = self._get_room_name(room)
-        for sessid, socket in self.socket.server.sockets.iteritems():
+        for sessid, socket in six.iteritems(self.socket.server.sockets):
             if 'rooms' not in socket.session:
                 continue
             if room_name in socket.session['rooms'] and self.socket != socket:
@@ -55,7 +57,7 @@ def broadcast_event(self, event, *args):
                    args=args,
                    endpoint=self.ns_name)
 
-        for sessid, socket in self.socket.server.sockets.iteritems():
+        for sessid, socket in six.iteritems(self.socket.server.sockets):
             socket.send_packet(pkt)
 
     def broadcast_event_not_me(self, event, *args):
@@ -68,6 +70,6 @@ def broadcast_event_not_me(self, event, *args):
                    args=args,
                    endpoint=self.ns_name)
 
-        for sessid, socket in self.socket.server.sockets.iteritems():
+        for sessid, socket in six.iteritems(self.socket.server.sockets):
             if socket is not self.socket:
                 socket.send_packet(pkt)

socketio/namespace.py

@@ -167,14 +167,14 @@ def process_packet(self, packet):
         elif packet_type == 'ack':
             callback = self.socket._pop_ack_callback(packet['ackId'])
             if not callback:
-                print "ERROR: No such callback for ackId %s" % packet['ackId']
+                print("ERROR: No such callback for ackId %s" % packet['ackId'])
                 return
             return callback(*(packet['args']))
         elif packet_type == 'disconnect':
             # Force a disconnect on the namespace.
             return self.call_method_with_acl('recv_disconnect', packet)
         else:
-            print "Unprocessed packet", packet
+            print("Unprocessed packet", packet)
         # TODO: manage the other packet types: disconnect
 
     def process_event(self, packet):

socketio/packet.py

@@ -1,3 +1,4 @@
+import six
 from socketio.defaultjson import default_json_dumps, default_json_loads
 
 MSG_TYPES = {
@@ -12,21 +13,21 @@
     'noop': 8,
     }
 
-MSG_VALUES = dict((v, k) for k, v in MSG_TYPES.iteritems())
+MSG_VALUES = dict((v, k) for k, v in six.iteritems(MSG_TYPES))
 
 ERROR_REASONS = {
     'transport not supported': 0,
     'client not handshaken': 1,
     'unauthorized': 2
     }
 
-REASONS_VALUES = dict((v, k) for k, v in ERROR_REASONS.iteritems())
+REASONS_VALUES = dict((v, k) for k, v in six.iteritems(ERROR_REASONS))
 
 ERROR_ADVICES = {
     'reconnect': 0,
     }
 
-ADVICES_VALUES = dict((v, k) for k, v in ERROR_ADVICES.iteritems())
+ADVICES_VALUES = dict((v, k) for k, v in six.iteritems(ERROR_ADVICES))
 
 socketio_packet_attributes = ['type', 'name', 'data', 'endpoint', 'args',
                               'ackId', 'reason', 'advice', 'qs', 'id']
@@ -107,7 +108,7 @@ def decode(rawstr, json_loads=default_json_loads):
     Decode a rawstr packet arriving from the socket into a dict.
     """
     decoded_msg = {}
-    split_data = rawstr.split(":", 3)
+    split_data = rawstr.decode('utf-8').split(":", 3)
     msg_type = split_data[0]
     msg_id = split_data[1]
     endpoint = split_data[2]
@@ -153,7 +154,7 @@ def decode(rawstr, json_loads=default_json_loads):
     elif msg_type == "5":  # event
         try:
             data = json_loads(data)
-        except ValueError, e:
+        except ValueError:
             print("Invalid JSON event message", data)
             decoded_msg['args'] = []
         else:
@@ -183,6 +184,7 @@ def decode(rawstr, json_loads=default_json_loads):
                 decoded_msg['reason'] = REASONS_VALUES[int(data)]
             else:
                 decoded_msg['reason'] = ''
+
     elif msg_type == "8":  # noop
         pass
 

socketio/server.py

@@ -105,7 +105,7 @@ def start_accepting(self):
             try:
                 if not self.policy_server.started:
                     self.policy_server.start()
-            except error, ex:
+            except error as ex:
                 sys.stderr.write(
                     'FAILED to start flash policy server: %s\n' % (ex, ))
             except Exception:

socketio/transports.py

@@ -1,6 +1,6 @@
 import gevent
-import urllib
-import urlparse
+import six
+from six.moves.urllib.parse import parse_qs, unquote_plus
 from geventwebsocket import WebSocketError
 from gevent.queue import Empty
 
@@ -21,7 +21,7 @@ def __init__(self, handler, config, **kwargs):
             ("Access-Control-Allow-Origin", "*"),
             ("Access-Control-Allow-Credentials", "true"),
             ("Access-Control-Allow-Methods", "POST, GET, OPTIONS"),
-            ("Access-Control-Max-Age", 3600),
+            ("Access-Control-Max-Age", "3600"),
         ]
         self.handler = handler
         self.config = config
@@ -162,7 +162,7 @@ def __init__(self, handler, config):
     def _request_body(self):
         data = super(JSONPolling, self)._request_body()
         # resolve %20%3F's, take out wrapping d="...", etc..
-        data = urllib.unquote_plus(data)[3:-1] \
+        data = unquote_plus(data)[3:-1] \
                      .replace(r'\"', '"') \
                      .replace(r"\\", "\\")
 
@@ -175,7 +175,7 @@ def _request_body(self):
 
     def write(self, data):
         """Just quote out stuff before sending it out"""
-        args = urlparse.parse_qs(self.handler.environ.get("QUERY_STRING"))
+        args = parse_qs(self.handler.environ.get("QUERY_STRING"))
         if "i" in args:
             i = args["i"]
         else:
@@ -249,7 +249,9 @@ def send_into_ws():
                     break
                 try:
                     websocket.send(message)
-                except (WebSocketError, TypeError):
+                except (WebSocketError, TypeError) as e:
+                    print(e)
+                    raise
                     # We can't send a message on the socket
                     # it is dead, let the other sockets know
                     socket.disconnect()

socketio/virtsocket.py

@@ -10,6 +10,7 @@
 
 """
 import random
+import six
 import weakref
 import logging
 
@@ -311,7 +312,7 @@ def disconnect(self, silent=False):
                        :meth:`~socketio.namespace.BaseNamespace.disconnect`
                        calls.
         """
-        for ns_name, ns in list(self.active_ns.iteritems()):
+        for ns_name, ns in list(six.iteritems(self.active_ns)):
             ns.recv_disconnect()
 
     def remove_namespace(self, namespace):
@@ -362,7 +363,7 @@ def _receiver_loop(self):
                 continue  # or close the connection ?
             try:
                 pkt = packet.decode(rawdata, self.json_loads)
-            except (ValueError, KeyError, Exception), e:
+            except (ValueError, KeyError, Exception) as e:
                 self.error('invalid_packet',
                     "There was a decoding error when dealing with packet "
                     "with event: %s... (%s)" % (rawdata[:20], e))
@@ -436,7 +437,7 @@ def _watcher(self):
         while True:
             gevent.sleep(1.0)
             if not self.connected:
-                for ns_name, ns in list(self.active_ns.iteritems()):
+                for ns_name, ns in list(six.iteritems(self.active_ns)):
                     ns.recv_disconnect()
                 # Killing Socket-level jobs
                 gevent.killall(self.jobs)