diff --git a/happybase/pool.py b/happybase/pool.py index aa0da99..1ffff5a 100644 --- a/happybase/pool.py +++ b/happybase/pool.py @@ -4,10 +4,11 @@ HappyBase connection pool module. import contextlib import logging -import Queue import socket import threading +from six.moves import queue + from thriftpy.thrift import TException from .connection import Connection @@ -61,7 +62,7 @@ class ConnectionPool(object): "Initializing connection pool with %d connections", size) self._lock = threading.Lock() - self._queue = Queue.LifoQueue(maxsize=size) + self._queue = queue.LifoQueue(maxsize=size) self._thread_connections = threading.local() connection_kwargs = kwargs @@ -81,7 +82,7 @@ class ConnectionPool(object): """Acquire a connection from the pool.""" try: return self._queue.get(True, timeout) - except Queue.Empty: + except queue.Empty: raise NoConnectionsAvailable( "No connection available from pool within specified " "timeout")