accept custom graphson_io instance in DriverRemoteConnection

Project: http://git-wip-us.apache.org/repos/asf/tinkerpop/repo
Commit: http://git-wip-us.apache.org/repos/asf/tinkerpop/commit/d296eb79
Tree: http://git-wip-us.apache.org/repos/asf/tinkerpop/tree/d296eb79
Diff: http://git-wip-us.apache.org/repos/asf/tinkerpop/diff/d296eb79

Branch: refs/heads/master
Commit: d296eb79719fc97e65339a03573d1be11afe47be
Parents: 2ce68df
Author: Adam Holmberg <adam.holmb...@datastax.com>
Authored: Mon Oct 10 16:23:12 2016 -0500
Committer: Adam Holmberg <adam.holmb...@datastax.com>
Committed: Mon Oct 10 16:23:12 2016 -0500

----------------------------------------------------------------------
 .../jython/gremlin_python/driver/driver_remote_connection.py     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/tinkerpop/blob/d296eb79/gremlin-python/src/main/jython/gremlin_python/driver/driver_remote_connection.py
----------------------------------------------------------------------
diff --git 
a/gremlin-python/src/main/jython/gremlin_python/driver/driver_remote_connection.py
 
b/gremlin-python/src/main/jython/gremlin_python/driver/driver_remote_connection.py
index b8b3d87..20129e9 100644
--- 
a/gremlin-python/src/main/jython/gremlin_python/driver/driver_remote_connection.py
+++ 
b/gremlin-python/src/main/jython/gremlin_python/driver/driver_remote_connection.py
@@ -34,14 +34,14 @@ class GremlinServerError(Exception):
 
 
 class DriverRemoteConnection(RemoteConnection):
-    def __init__(self, url, traversal_source, username="", password="", 
loop=None):
+    def __init__(self, url, traversal_source, username="", password="", 
loop=None, graphson_io=None):
         super(DriverRemoteConnection, self).__init__(url, traversal_source)
         self._url = url
         self._username = username
         self._password = password
         if loop is None: self._loop = ioloop.IOLoop.current()
         self._websocket = self._loop.run_sync(lambda: 
websocket.websocket_connect(self.url))
-        self._graphson_io = GraphSONIO()
+        self._graphson_io = graphson_io or GraphSONIO()
 
     def submit(self, bytecode):
         '''

Reply via email to