python 2.x/3.x compatibility change

Sakis Kasampalis · Sakis Kasampalis · commit 2f0ae88eb123 · 2012-09-02T20:58:04.000+02:00
diff --git a/pool.py b/pool.py
@@ -25,14 +25,17 @@ def __del__(self):
             self._q.put(self.o)           
             self.o = None   
 
-if __name__ == "__main__":   
-    import Queue    
+if __name__ == "__main__":
+    try:
+        import queue as Queue
+    except:                     # python 2.x compatibility
+        import Queue
 
     def testObj(Q):       
         someObj = qObj(Q, True)        
         print('Inside func: {}'.format(someObj.o))    
 
-    aQ = Queue.Queue()    
+    aQ = Queue.Queue()
     aQ.put("yam")    
 
     with qObj(aQ) as obj:       
