socket connection

Author: raimdev

tarantool/connection.py

@@ -84,6 +84,7 @@ class Connection(object):
     def __init__(self, host, port,
                  user=None,
                  password=None,
+                 socket=None,
                  socket_timeout=SOCKET_TIMEOUT,
                  reconnect_max_attempts=RECONNECT_MAX_ATTEMPTS,
                  reconnect_delay=RECONNECT_DELAY,
@@ -113,6 +114,7 @@ def __init__(self, host, port,
         self.port = port
         self.user = user
         self.password = password
+        self.socket = socket
         self.socket_timeout = socket_timeout
         self.reconnect_delay = reconnect_delay
         self.reconnect_max_attempts = reconnect_max_attempts
@@ -134,7 +136,7 @@ def close(self):
         self._socket = None
 
     def connect_basic(self):
-        if self.host == None:
+        if self.socket:
             self.connect_unix()
         else:
             self.connect_tcp()
@@ -168,7 +170,7 @@ def connect_unix(self):
             if self._socket:
                 self._socket.close()
             self._socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
-            self._socket.connect(self.port)
+            self._socket.connect(self.socket)
         except socket.error as e:
             self.connected = False
             raise NetworkError(e)