From e584e3ec86863c6e35f0d2b3ee6c1e025bfb6d76 Mon Sep 17 00:00:00 2001 From: davebshow <davebshow@gmail.com> Date: Mon, 22 Jun 2015 20:43:30 -0400 Subject: [PATCH] convert uuid to string, need to implement new tests --- aiogremlin/client.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/aiogremlin/client.py b/aiogremlin/client.py index 6e12c30..1297a25 100644 --- a/aiogremlin/client.py +++ b/aiogremlin/client.py @@ -28,6 +28,7 @@ def submit(gremlin, *, connector=connector, ws_response_class=GremlinClientWebSocketResponse) gremlin_client = GremlinClient(url=url, connector=client_session) + try: resp = yield from gremlin_client.submit( gremlin, bindings=bindings, lang=lang, op=op, processor=processor) @@ -55,6 +56,7 @@ class SimpleGremlinClient: """ """ writer = GremlinWriter(self._connection) + connection = writer.write(gremlin, bindings=bindings, lang=lang, op=op, processor=processor, session=session, binary=binary) @@ -102,6 +104,7 @@ class GremlinClient: return self._closed = True + try: yield from self._connector.close() finally: @@ -123,6 +126,7 @@ class GremlinClient: self.url, timeout=self._timeout) writer = GremlinWriter(ws) + ws = writer.write(gremlin, bindings=bindings, lang=lang, op=op, processor=processor, binary=binary, session=self._session) @@ -157,7 +161,7 @@ class GremlinClientSession(GremlinClient): verbose=verbose, connector=connector) if session is None: - session = uuid4.uuid4() + session = str(uuid4.uuid4()) self._session = session def set_session(self): -- GitLab