瀏覽代碼

python 2.x/3.x compatibility change

Sakis Kasampalis 13 年之前
父節點
當前提交
2f0ae88eb1
共有 1 個文件被更改,包括 6 次插入3 次删除
  1. 6 3
      pool.py

+ 6 - 3
pool.py

@@ -25,14 +25,17 @@ class qObj():
             self._q.put(self.o)           
             self._q.put(self.o)           
             self.o = None   
             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):       
     def testObj(Q):       
         someObj = qObj(Q, True)        
         someObj = qObj(Q, True)        
         print('Inside func: {}'.format(someObj.o))    
         print('Inside func: {}'.format(someObj.o))    
 
 
-    aQ = Queue.Queue()    
+    aQ = Queue.Queue()
     aQ.put("yam")    
     aQ.put("yam")    
 
 
     with qObj(aQ) as obj:       
     with qObj(aQ) as obj: