diff --git a/src/epplibwrapper/__init__.py b/src/epplibwrapper/__init__.py index dd6664a3a..b8fb12bcb 100644 --- a/src/epplibwrapper/__init__.py +++ b/src/epplibwrapper/__init__.py @@ -44,6 +44,8 @@ except NameError: # Attn: these imports should NOT be at the top of the file try: + from epplibwrapper.socket import Socket + from epplibwrapper.utility.pool_error import PoolError, PoolErrorCodes from .client import CLIENT, commands from .errors import RegistryError, ErrorCode from epplib.models import common, info @@ -61,4 +63,7 @@ __all__ = [ "info", "ErrorCode", "RegistryError", + "Socket", + "PoolError", + "PoolErrorCodes" ] diff --git a/src/epplibwrapper/client.py b/src/epplibwrapper/client.py index 680a6661c..4f393aa0c 100644 --- a/src/epplibwrapper/client.py +++ b/src/epplibwrapper/client.py @@ -81,7 +81,8 @@ class EPPLibWrapper: self.pool_status = PoolStatus() if start_connection_pool: - self.start_connection_pool() + pass + #self.start_connection_pool() def _send(self, command): """Helper function used by `send`.""" @@ -207,7 +208,7 @@ class EPPLibWrapper: self._pool = self._create_pool( self._client, self._login, self.pool_options ) - + self.pool_status.pool_running = True self.pool_status.pool_hanging = False diff --git a/src/epplibwrapper/utility/pool.py b/src/epplibwrapper/utility/pool.py index 6d51e539f..7a58f7efe 100644 --- a/src/epplibwrapper/utility/pool.py +++ b/src/epplibwrapper/utility/pool.py @@ -3,12 +3,12 @@ import gevent from geventconnpool import ConnectionPool from epplibwrapper.socket import Socket from epplibwrapper.utility.pool_error import PoolError, PoolErrorCodes - try: from epplib.commands import Hello except ImportError: pass + logger = logging.getLogger(__name__)