fix tarantool connector for python 3.4 with six library usage

Author: shveenkov

requirements.txt

@@ -1,2 +1,3 @@
 msgpack-python>=0.4.0
 pyyaml>=3.10
+six

setup.py

@@ -83,5 +83,6 @@ def find_version(*file_paths):
     command_options=command_options,
     install_requires=[
         'msgpack-python>=0.4',
+        'six',
     ]
 )

tarantool/connection.py

@@ -4,6 +4,7 @@
 This module provides low-level API for Tarantool
 '''
 
+import six
 import time
 import errno
 import ctypes
@@ -154,7 +155,7 @@ def connect(self):
             raise NetworkError(e)
 
     def _recv(self, to_read):
-        buf = ''
+        buf = b""
         while to_read > 0:
             try:
                 tmp = self._socket.recv(to_read)
@@ -188,7 +189,7 @@ def _send_request_wo_reconnect(self, request):
 
         # Repeat request in a loop if the server returns completion_status == 1
         # (try again)
-        for attempt in xrange(RETRY_MAX_ATTEMPTS):    # pylint: disable=W0612
+        for attempt in range(RETRY_MAX_ATTEMPTS):    # pylint: disable=W0612
             self._socket.sendall(bytes(request))
             response = Response(self, self._read_response())
 
@@ -328,7 +329,7 @@ def replace(self, space_name, values):
 
         :rtype: `Response` instance
         '''
-        if isinstance(space_name, basestring):
+        if isinstance(space_name, six.string_types):
             space_name = self.schema.get_space(space_name).sid
         request = RequestReplace(self, space_name, values)
         return self._send_request(request)
@@ -377,7 +378,7 @@ def insert(self, space_name, values):
 
         :rtype: `Response` instance
         '''
-        if isinstance(space_name, basestring):
+        if isinstance(space_name, six.string_types):
             space_name = self.schema.get_space(space_name).sid
         request = RequestInsert(self, space_name, values)
         return self._send_request(request)
@@ -397,9 +398,9 @@ def delete(self, space_name, key, **kwargs):
         index_name = kwargs.get("index", 0)
 
         key = check_key(key)
-        if isinstance(space_name, basestring):
+        if isinstance(space_name, six.string_types):
             space_name = self.schema.get_space(space_name).sid
-        if isinstance(index_name, basestring):
+        if isinstance(index_name, six.string_types):
             index_name = self.schema.get_index(space_name, index_name).iid
         request = RequestDelete(self, space_name, index_name, key)
         return self._send_request(request)
@@ -427,9 +428,9 @@ def update(self, space_name, key, op_list, **kwargs):
         index_name = kwargs.get("index", 0)
 
         key = check_key(key)
-        if isinstance(space_name, basestring):
+        if isinstance(space_name, six.string_types):
             space_name = self.schema.get_space(space_name).sid
-        if isinstance(index_name, basestring):
+        if isinstance(index_name, six.string_types):
             index_name = self.schema.get_index(space_name, index_name).iid
         request = RequestUpdate(self, space_name, index_name, key, op_list)
         return self._send_request(request)
@@ -503,9 +504,9 @@ def select(self, space_name, key=None, **kwargs):
         # tuples)
         key = check_key(key, select=True)
 
-        if isinstance(space_name, basestring):
+        if isinstance(space_name, six.string_types):
             space_name = self.schema.get_space(space_name).sid
-        if isinstance(index_name, basestring):
+        if isinstance(index_name, six.string_types):
             index_name = self.schema.get_index(space_name, index_name).iid
         request = RequestSelect(self, space_name, index_name, key, offset,
                                 limit, iterator_type)

tarantool/error.py

@@ -24,10 +24,12 @@
 import sys
 import warnings
 
-
-class Error(StandardError):
-
-    '''Base class for error exceptions'''
+try:
+    class Error(StandardError):
+        '''Base class for error exceptions'''
+except NameError:
+    class Error(Exception):
+        '''Base class for error exceptions'''
 
 
 class DatabaseError(Error):

tarantool/schema.py

@@ -5,18 +5,20 @@
 It is a Tarantool schema description.
 '''
 
+import six
+
 from tarantool.error import SchemaError
 import tarantool.const as const
 
 
 class SchemaIndex(object):
     def __init__(self, array, space):
         self.iid = array[1]
-        self.name = array[2]
+        self.name = array[2].decode()
         self.index = array[3]
         self.unique = array[4]
         self.parts = []
-        for i in xrange(array[5]):
+        for i in range(array[5]):
             self.parts.append((array[5 + 1 + i * 2], array[5 + 2 + i * 2]))
         self.space = space
         self.space.indexes[self.iid] = self
@@ -33,7 +35,7 @@ class SchemaSpace(object):
     def __init__(self, array, schema):
         self.sid = array[0]
         self.arity = array[1]
-        self.name = array[2]
+        self.name = array[2].decode()
         self.indexes = {}
         self.schema = schema
         self.schema[self.sid] = self
@@ -57,14 +59,14 @@ def get_space(self, space):
         except KeyError:
             pass
         _index = (const.INDEX_SPACE_NAME
-                  if isinstance(space, basestring)
+                  if isinstance(space, six.string_types)
                   else const.INDEX_SPACE_PRIMARY)
 
         array = self.con.select(const.SPACE_SPACE, space, index=_index)
         if len(array) > 1:
             raise SchemaError('Some strange output from server: \n' + array)
         elif len(array) == 0 or not len(array[0]):
-            temp_name = ('name' if isinstance(space, basestring) else 'id')
+            temp_name = ('name' if isinstance(space, six.string_types) else 'id')
             raise SchemaError(
                 "There's no space with {1} '{0}'".format(space, temp_name))
         array = array[0]
@@ -77,15 +79,15 @@ def get_index(self, space, index):
         except KeyError:
             pass
         _index = (const.INDEX_INDEX_NAME
-                  if isinstance(index, basestring)
+                  if isinstance(index, six.string_types)
                   else const.INDEX_INDEX_PRIMARY)
 
         array = self.con.select(const.SPACE_INDEX, [_space.sid, index],
                                 index=_index)
         if len(array) > 1:
             raise SchemaError('Some strange output from server: \n' + array)
         elif len(array) == 0 or not len(array[0]):
-            temp_name = ('name' if isinstance(index, basestring) else 'id')
+            temp_name = ('name' if isinstance(index, six.string_types) else 'id')
             raise SchemaError(
                 "There's no index with {2} '{0}' in space '{1}'".format(
                     index, _space.name, temp_name))

tarantool/utils.py

@@ -1,3 +1,7 @@
+# -*- coding: utf-8 -*-
+
+import six
+
 def check_key(*args, **kwargs):
     if 'first' not in kwargs:
         kwargs['first'] = True
@@ -12,5 +16,5 @@ def check_key(*args, **kwargs):
         elif args[0] is None and kwargs['select']:
             return []
     for key in args:
-        assert isinstance(key, (int, long, basestring))
+        assert isinstance(key, six.integer_types + six.string_types)
     return list(args)

tests/suites/lib/tarantool_server.py

@@ -193,7 +193,7 @@ def find_exe(self):
             exe = os.path.join(_dir, self.default_tarantool["bin"])
             if os.access(exe, os.X_OK):
                 return os.path.abspath(exe)
-        raise RuntimeError("Can't find server executable in " + path)
+        raise RuntimeError("Can't find server executable in " + os.environ["PATH"])
 
     def generate_configuration(self):
         os.putenv("PRIMARY_PORT", str(self.args['primary']))
@@ -240,7 +240,7 @@ def start(self):
         self.generate_configuration()
         if self.script:
             shutil.copy(self.script, self.script_dst)
-            os.chmod(self.script_dst, 0777)
+            os.chmod(self.script_dst, 0o777)
         args = self.prepare_args()
         self.process = subprocess.Popen(args,
                 cwd = self.vardir,

tests/suites/test_dml.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+
+import six
 import yaml
 import unittest
 import tarantool
@@ -7,8 +9,8 @@
 class Request(unittest.TestCase):
     @classmethod
     def setUpClass(self):
-        print ' DML '.center(70, '=')
-        print '-' * 70
+        print(' DML '.center(70, '='))
+        print('-' * 70)
         self.srv = TarantoolServer()
         self.srv.script = 'tests/suites/box.lua'
         self.srv.start()
@@ -32,7 +34,7 @@ def test_00_01_space_created(self):
 
     def test_00_02_fill_space(self):
         # Fill space with values
-        for i in xrange(1, 500):
+        for i in range(1, 500):
             self.assertEqual(
                     self.con.insert('space_1', [i, i%5, 'tuple_'+str(i)])[0],
                     [i, i%5, 'tuple_'+str(i)]
@@ -145,7 +147,7 @@ def test_07_call(self):
         ans = self.con.call('fiber.time64')
         self.assertEqual(len(ans), 1)
         self.assertEqual(len(ans[0]), 1)
-        self.assertIsInstance(ans[0][0], (int, long))
+        self.assertIsInstance(ans[0][0], six.integer_types)
         ans = self.con.call('uuid.str')
         self.assertEqual(len(ans), 1)
         self.assertEqual(len(ans[0]), 1)

tests/suites/test_schema.py

@@ -7,8 +7,8 @@
 class TestSuite_Schema(unittest.TestCase):
     @classmethod
     def setUpClass(self):
-        print ' SCHEMA '.center(70, '=')
-        print '-' * 70
+        print(' SCHEMA '.center(70, '='))
+        print('-' * 70)
         self.srv = TarantoolServer()
         self.srv.script = 'tests/suites/box.lua'
         self.srv.start()