Import queue from six.moves in the top level module

Author: g1itch

src/helper_sql.py

@@ -16,20 +16,14 @@
    or isn't thread-safe.
 """
 
-
-# import Queue
-import sys
-if sys.version_info[0] == 3:
-    import queue as Queue #python3
-else:
-    import Queue #python2
-
 import threading
 
+from six.moves import queue
+
 
-sqlSubmitQueue = Queue.Queue()
+sqlSubmitQueue = queue.Queue()
 """the queue for SQL"""
-sqlReturnQueue = Queue.Queue()
+sqlReturnQueue = queue.Queue()
 """the queue for results"""
 sql_lock = threading.Lock()
 """ lock to prevent queueing a new request until the previous response

src/multiqueue.py

@@ -3,21 +3,17 @@
 Elements are added into a random subqueue, and retrieval rotates
 """
 
-import sys
-if sys.version_info[0] == 3:
-    import queue as Queue
-else:
-    import Queue
-
 from collections import deque
 
-if sys.version_info[0] == 3:
-    from . import helper_random
-else:
+from six.moves import queue
+
+try:
     import helper_random
+except ImportError:
+    from . import helper_random
 
 
-class MultiQueue(Queue.Queue):
+class MultiQueue(queue.Queue):
     """A base queue class"""
     # pylint: disable=redefined-builtin,attribute-defined-outside-init
     defaultQueueCount = 10
@@ -27,7 +23,7 @@ def __init__(self, maxsize=0, count=0):
             self.queueCount = MultiQueue.defaultQueueCount
         else:
             self.queueCount = count
-        Queue.Queue.__init__(self, maxsize)
+        queue.Queue.__init__(self, maxsize)
 
     # Initialize the queue representation
     def _init(self, maxsize):
@@ -42,7 +38,8 @@ def _qsize(self, len=len):
     # Put a new item in the queue
     def _put(self, item):
         # self.queue.append(item)
-        self.queues[helper_random.randomrandrange(self.queueCount)].append((item))
+        self.queues[helper_random.randomrandrange(self.queueCount)].append(
+            (item))
 
     # Get an item from the queue
     def _get(self):

src/network/connectionchooser.py

@@ -9,7 +9,7 @@
 import protocol
 import state
 from bmconfigparser import BMConfigParser
-from queues import Queue, portCheckerQueue
+from queues import queue, portCheckerQueue
 
 logger = logging.getLogger('default')
 
@@ -37,7 +37,7 @@ def chooseConnection(stream):
         retval = portCheckerQueue.get(False)
         portCheckerQueue.task_done()
         return retval
-    except Queue.Empty:
+    except queue.Empty:
         pass
     # with a probability of 0.5, connect to a discovered peer
     if random.choice((False, True)) and not haveOnion:

src/queues.py

@@ -2,24 +2,22 @@
 
 import threading
 import time
-try:
-    import queue as Queue
-except ImportError:
-    import Queue
+
+from six.moves import queue
 
 try:
     from multiqueue import MultiQueue
 except ImportError:
     from .multiqueue import MultiQueue
 
 
-class ObjectProcessorQueue(Queue.Queue):
+class ObjectProcessorQueue(queue.Queue):
     """Special queue class using lock for `.threads.objectProcessor`"""
 
     maxSize = 32000000
 
     def __init__(self):
-        Queue.Queue.__init__(self)
+        queue.Queue.__init__(self)
         self.sizeLock = threading.Lock()
         #: in Bytes. We maintain this to prevent nodes from flooding us
         #: with objects which take up too much memory. If this gets
@@ -31,27 +29,27 @@ def put(self, item, block=True, timeout=None):
             time.sleep(1)
         with self.sizeLock:
             self.curSize += len(item[1])
-        Queue.Queue.put(self, item, block, timeout)
+        queue.Queue.put(self, item, block, timeout)
 
     def get(self, block=True, timeout=None):
-        item = Queue.Queue.get(self, block, timeout)
+        item = queue.Queue.get(self, block, timeout)
         with self.sizeLock:
             self.curSize -= len(item[1])
         return item
 
 
-workerQueue = Queue.Queue()
-UISignalQueue = Queue.Queue()
-addressGeneratorQueue = Queue.Queue()
+workerQueue = queue.Queue()
+UISignalQueue = queue.Queue()
+addressGeneratorQueue = queue.Queue()
 #: `.network.ReceiveQueueThread` instances dump objects they hear
 #: on the network into this queue to be processed.
 objectProcessorQueue = ObjectProcessorQueue()
 invQueue = MultiQueue()
 addrQueue = MultiQueue()
-portCheckerQueue = Queue.Queue()
-receiveDataQueue = Queue.Queue()
+portCheckerQueue = queue.Queue()
+receiveDataQueue = queue.Queue()
 #: The address generator thread uses this queue to get information back
 #: to the API thread.
-apiAddressGeneratorReturnQueue = Queue.Queue()
+apiAddressGeneratorReturnQueue = queue.Queue()
 #: for exceptions
-excQueue = Queue.Queue()
+excQueue = queue.Queue()

src/shutdown.py

@@ -1,9 +1,11 @@
 """shutdown function"""
+
 import os
-import Queue
 import threading
 import time
 
+from six.moves import queue
+
 import state
 from debug import logger
 from helper_sql import sqlQuery, sqlStoredProcedure
@@ -69,14 +71,14 @@ def doCleanShutdown():
     sqlStoredProcedure('exit')
 
     # flush queues
-    for queue in (
+    for q in (
             workerQueue, UISignalQueue, addressGeneratorQueue,
             objectProcessorQueue):
         while True:
             try:
-                queue.get(False)
-                queue.task_done()
-            except Queue.Empty:
+                q.get(False)
+                q.task_done()
+            except queue.Empty:
                 break
 
     if state.thisapp.daemon or not state.enableGUI: