Fix flake8 errors related to whitespace

tests/echo.py

@@ -11,7 +11,7 @@ as taken from http://docs.python.org/dev/library/ssl.html#certificates
 '''
 
 import os, sys, select, optparse, logging
-sys.path.insert(0,os.path.join(os.path.dirname(__file__), ".."))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 from websockify.websockifyserver import WebSockifyServer, WebSockifyRequestHandler
 
 

tests/echo_client.py

@@ -5,7 +5,7 @@ import sys
 import optparse
 import select
 
-sys.path.insert(0,os.path.join(os.path.dirname(__file__), ".."))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 from websockify.websocket import WebSocket, \
     WebSocketWantReadError, WebSocketWantWriteError
 

tests/load.py

@@ -7,7 +7,7 @@ given a sequence number. Any errors are reported and counted.
 '''
 
 import sys, os, select, random, time, optparse, logging
-sys.path.insert(0,os.path.join(os.path.dirname(__file__), ".."))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 from websockify.websockifyserver import WebSockifyServer, WebSockifyRequestHandler
 
 
@@ -74,13 +74,13 @@ class WebSocketLoad(WebSockifyRequestHandler):
 
     def generate(self):
         length = random.randint(10, self.max_packet_size)
-        numlist = self.rand_array[self.max_packet_size-length:]
+        numlist = self.rand_array[self.max_packet_size - length:]
         # Error in length
         #numlist.append(5)
         chksum = sum(numlist)
         # Error in checksum
         #numlist[0] = 5
-        nums = "".join( [str(n) for n in numlist] )
+        nums = "".join([str(n) for n in numlist])
         data = "^%d:%d:%d:%s$" % (self.send_cnt, length, chksum, nums)
         self.send_cnt += 1
 
@@ -151,7 +151,7 @@ if __name__ == '__main__':
         if len(args) != 1: raise ValueError
         opts.listen_port = int(args[0])
 
-        if len(args) not in [1,2]: raise ValueError
+        if len(args) not in [1, 2]: raise ValueError
         opts.listen_port = int(args[0])
         if len(args) == 2:
             opts.delay = int(args[1])

tests/test_token_plugins.py

@@ -114,7 +114,7 @@ class JWSTokenTestCase(unittest.TestCase):
         key = jwk.JWK()
         private_key = open("./tests/fixtures/private.pem", "rb").read()
         key.import_from_pem(private_key)
-        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200 })
+        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200})
         jwt_token.make_signed_token(key)
         mock_time.return_value = 150
 
@@ -131,7 +131,7 @@ class JWSTokenTestCase(unittest.TestCase):
         key = jwk.JWK()
         private_key = open("./tests/fixtures/private.pem", "rb").read()
         key.import_from_pem(private_key)
-        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200 })
+        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200})
         jwt_token.make_signed_token(key)
         mock_time.return_value = 50
 
@@ -146,7 +146,7 @@ class JWSTokenTestCase(unittest.TestCase):
         key = jwk.JWK()
         private_key = open("./tests/fixtures/private.pem", "rb").read()
         key.import_from_pem(private_key)
-        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200 })
+        jwt_token = jwt.JWT({"alg": "RS256"}, {'host': "remote_host", 'port': "remote_port", 'nbf': 100, 'exp': 200})
         jwt_token.make_signed_token(key)
         mock_time.return_value = 250
 
@@ -159,7 +159,7 @@ class JWSTokenTestCase(unittest.TestCase):
 
         secret = open("./tests/fixtures/symmetric.key").read()
         key = jwk.JWK()
-        key.import_key(kty="oct",k=secret)
+        key.import_key(kty="oct", k=secret)
         jwt_token = jwt.JWT({"alg": "HS256"}, {'host': "remote_host", 'port': "remote_port"})
         jwt_token.make_signed_token(key)
 
@@ -174,7 +174,7 @@ class JWSTokenTestCase(unittest.TestCase):
 
         secret = open("./tests/fixtures/symmetric.key").read()
         key = jwk.JWK()
-        key.import_key(kty="oct",k=secret)
+        key.import_key(kty="oct", k=secret)
         jwt_token = jwt.JWT({"alg": "HS256"}, {'host': "remote_host", 'port': "remote_port"})
         jwt_token.make_signed_token(key)
 

websockify/sysloghandler.py

@@ -13,7 +13,7 @@ class WebsockifySysLogHandler(handlers.SysLogHandler):
     _legacy = False
     _timestamp_fmt = '%Y-%m-%dT%H:%M:%SZ'
     _max_hostname = 255
-    _max_ident = 24 #safer for old daemons
+    _max_ident = 24  # safer for old daemons
     _send_length = False
     _tail = '\n'
     ident = None

websockify/token_plugins.py

@@ -114,8 +114,8 @@ class BaseTokenAPI(BasePlugin):
 
     def process_result(self, resp):
         host, port = resp.text.split(':')
-        port = port.encode('ascii','ignore')
+        port = port.encode('ascii', 'ignore')
-        return [ host, port ]
+        return [host, port]
 
     def lookup(self, token):
         import requests
@@ -159,7 +159,7 @@ class JWTTokenApi(BasePlugin):
                 key.import_from_pem(key_data)
             except:
                 try:
-                    key.import_key(k=key_data.decode('utf-8'),kty='oct')
+                    key.import_key(k=key_data.decode('utf-8'), kty='oct')
                 except:
                     logger.error('Failed to correctly parse key data!')
                     return None
@@ -373,7 +373,7 @@ class UnixDomainSocketDirectory(BasePlugin):
             if not stat.S_ISSOCK(uds_path.stat().st_mode):
                 return None
 
-            return [ 'unix_socket', uds_path ]
+            return ['unix_socket', uds_path]
         except Exception as e:
                 logger.error("Error finding unix domain socket: %s" % str(e))
                 return None

websockify/websocket.py

@@ -761,7 +761,7 @@ class WebSocket:
             pend = plen
             b = c = b''
             if plen >= 4:
-                dtype=numpy.dtype('<u4')
+                dtype = numpy.dtype('<u4')
                 if sys.byteorder == 'big':
                     dtype = dtype.newbyteorder('>')
                 mask = numpy.frombuffer(mask, dtype, count=1)
@@ -770,7 +770,7 @@ class WebSocket:
                 b = numpy.bitwise_xor(data, mask).tobytes()
 
             if plen % 4:
-                dtype=numpy.dtype('B')
+                dtype = numpy.dtype('B')
                 if sys.byteorder == 'big':
                     dtype = dtype.newbyteorder('>')
                 mask = numpy.frombuffer(mask, dtype, count=(plen % 4))
@@ -829,11 +829,11 @@ class WebSocket:
              'payload'      : decoded_buffer}
         """
 
-        f = {'fin'          : 0,
+        f = {'fin': 0,
-             'opcode'       : 0,
+             'opcode': 0,
-             'masked'       : False,
+             'masked': False,
-             'length'       : 0,
+             'length': 0,
-             'payload'      : None}
+             'payload': None}
 
         blen = len(buf)
         hlen = 2
@@ -871,9 +871,9 @@ class WebSocket:
 
         if f['masked']:
             # unmask payload
-            mask_key = buf[hlen-4:hlen]
+            mask_key = buf[hlen - 4:hlen]
-            f['payload'] = self._unmask(buf[hlen:(hlen+length)], mask_key)
+            f['payload'] = self._unmask(buf[hlen:(hlen + length)], mask_key)
         else:
-            f['payload'] = buf[hlen:(hlen+length)]
+            f['payload'] = buf[hlen:(hlen + length)]
 
         return f

websockify/websocketproxy.py

@@ -63,7 +63,7 @@ Traffic Legend:
 
         # clear out any existing SSL_ headers that the client might
         # have maliciously set
-        ssl_headers = [ h for h in self.headers if h.startswith('SSL_') ]
+        ssl_headers = [h for h in self.headers if h.startswith('SSL_')]
         for h in ssl_headers:
             del self.headers[h]
 
@@ -269,7 +269,7 @@ Traffic Legend:
                     # Send queued target data to the client
                     if len(cqueue) != 0:
                         c_pend = True
-                        while(c_pend):
+                        while (c_pend):
                             c_pend = self.send_frames(cqueue)
 
                         cqueue = []
@@ -334,7 +334,7 @@ class WebSocketProxy(websockifyserver.WebSockifyServer):
             sock.close()
 
             # Insert rebinder at the head of the (possibly empty) LD_PRELOAD pathlist
-            ld_preloads = filter(None, [ self.rebinder, os.environ.get("LD_PRELOAD", None) ])
+            ld_preloads = filter(None, [self.rebinder, os.environ.get("LD_PRELOAD", None)])
 
             os.environ.update({
                 "LD_PRELOAD": os.pathsep.join(ld_preloads),
@@ -401,7 +401,7 @@ class WebSocketProxy(websockifyserver.WebSockifyServer):
                 sys.exit(ret)
             elif self.wrap_mode == "respawn":
                 now = time.time()
-                avg = sum(self.wrap_times)/len(self.wrap_times)
+                avg = sum(self.wrap_times) / len(self.wrap_times)
                 if (now - avg) < 10:
                     # 3 times in the last 10 seconds
                     if self.spawn_message:

websockify/websockifyserver.py

@@ -126,7 +126,7 @@ class WebSockifyRequestHandler(WebSocketRequestHandlerMixIn, SimpleHTTPRequestHa
         fully sent, in which case the caller should call again when
         the socket is ready. """
 
-        tdelta = int(time.time()*1000) - self.start_time
+        tdelta = int(time.time() * 1000) - self.start_time
 
         if bufs:
             for buf in bufs:
@@ -157,7 +157,7 @@ class WebSockifyRequestHandler(WebSocketRequestHandlerMixIn, SimpleHTTPRequestHa
 
         closed = False
         bufs = []
-        tdelta = int(time.time()*1000) - self.start_time
+        tdelta = int(time.time() * 1000) - self.start_time
 
         while True:
             try:
@@ -210,7 +210,7 @@ class WebSockifyRequestHandler(WebSocketRequestHandlerMixIn, SimpleHTTPRequestHa
         # Initialize per client settings
         self.send_parts = []
         self.recv_part = None
-        self.start_time = int(time.time()*1000)
+        self.start_time = int(time.time() * 1000)
 
         # client_address is empty with, say, UNIX domain sockets
         client_addr = ""
@@ -816,7 +816,7 @@ class WebSockifyServer():
                                 startsock, address = lsock.accept()
                                 # Unix Socket will not report address (empty string), but address[0] is logged a bunch
                                 if self.unix_listen != None:
-                                    address = [ self.unix_listen ]
+                                    address = [self.unix_listen]
                             else:
                                 continue
                         except self.Terminate:
@@ -838,7 +838,7 @@ class WebSockifyServer():
                         if self.run_once:
                             # Run in same process if run_once
                             self.top_new_client(startsock, address)
-                            if self.ws_connection :
+                            if self.ws_connection:
                                 self.msg('%s: exiting due to --run-once'
                                         % address[0])
                                 break