Commit d2e825d8 authored by davebshow's avatar davebshow
Browse files

more params fixes

parent a29147b3
......@@ -70,15 +70,16 @@ class GremlinClient:
self.timeout = timeout
self._pool = pool
self._connector = connector
self._factory = factory or GremlinFactory(connector=self._connector)
self._conn = connection
if pool is not None:
factory = pool._factory
self._connected = self._pool._connected
elif self._conn and not getattr(connection, "closed", True):
self._connected = True
else:
self._connected = False
self._conn = asyncio.async(self._connect(), loop=self._loop)
self._factory = factory or GremlinFactory(connector=self._connector)
if verbose:
logger.setLevel(INFO)
......
......@@ -16,11 +16,11 @@ class WebSocketPool:
"""
"""
self.url = url
self._ws_response_class = (ws_response_class or
GremlinClientWebSocketResponse)
if ws_response_class is None:
ws_response_class = GremlinClientWebSocketResponse
self._factory = factory or GremlinFactory(
connector=connector,
ws_response_class=self._ws_response_class)
ws_response_class=ws_response_class)
self.poolsize = poolsize
self.max_retries = max_retries
self.timeout = timeout
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment