Add flags support. Split operation replace, store, insert from one insert

Author: bigbes

src/tarantool/connection.py

@@ -218,30 +218,60 @@ def call(self, func_name, *args, **kwargs):
                                       default_type = default_type)
         return response
 
+    def _insert(self, space_name, values, flags):
+        assert isinstance(values, tuple)
+        assert (flags & (BOX_RETURN_TUPLE | BOX_ADD | BOX_REPLACE)) == flags
 
-    def insert(self, space_name, values, return_tuple=False, not_presented=False, presented=False):
-        '''\
-        Execute INSERT request.
-        Insert single record into a space `space_name`.
+        request = RequestInsert(self, space_name, values, flags)
+        return self._send_request(request, space_name)
+
+    def replace(self, space_name, values, return_tuple):
+        '''
+        Execute REPLACE request.
+        It will throw error if there's no tuple with this PK exists
 
         :param int space_name: space id to insert a record
         :type space_name: int or str
         :param values: record to be inserted. The tuple must contain only scalar (integer or strings) values
         :type values: tuple
+        :param return_tuple: True indicates that it is required to return the inserted tuple back
+        :type return_tuple: bool
         
+        :rtype: `Response` instance
+        '''
+        self._insert(space_name, values, (BOX_REPLACE_TUPLE if return_tuple else 0) | BOX_REPLACE)
+
+    def store(self, space_name, values, return_tuple):
+        '''
+        Execute STORE request.
+        It will overwrite tuple with the same PK, if it exists, or inserts if not
+
+        :param int space_name: space id to insert a record
+        :type space_name: int or str
+        :param values: record to be inserted. The tuple must contain only scalar (integer or strings) values
+        :type values: tuple
         :param return_tuple: True indicates that it is required to return the inserted tuple back
         :type return_tuple: bool
-        :param not_presented: True indicates that there's must be no tuple with same primary key
-        :type not_presented: bool
-        :param presented: True indicates that there's must be tuple with same primary key
-        :type presented: bool
-  
+
         :rtype: `Response` instance
         '''
-        assert isinstance(values, tuple)
+        self._insert(space_name, values, (BOX_REPLACE_TUPLE if return_tuple else 0)) 
 
-        request = RequestInsert(self, space_name, values, return_tuple, not_presented, presented)
-        return self._send_request(request, space_name)
+    def insert(self, space_name, values, return_tuple):
+        '''
+        Execute INSERT request.
+        It will throw error if there's tuple with same PK exists.
+
+        :param int space_name: space id to insert a record
+        :type space_name: int or str
+        :param values: record to be inserted. The tuple must contain only scalar (integer or strings) values
+        :type values: tuple
+        :param return_tuple: True indicates that it is required to return the inserted tuple back
+        :type return_tuple: bool
+
+        :rtype: `Response` instance
+        '''
+        self._insert(space_name, values, (BOX_REPLACE_TUPLE if return_tuple else 0) | BOX_ADD)
 
 
     def delete(self, space_name, key, return_tuple=False):

src/tarantool/const.py

@@ -28,6 +28,10 @@
 REQUEST_TYPE_CALL   = 22
 REQUEST_TYPE_PING   = 65280
 
+BOX_RETURN_TUPLE    = 1
+BOX_ADD             = 2
+BOX_REPLACE         = 4
+
 
 UPDATE_OPERATION_CODE = {'=': 0, '+': 1, '&': 2, '^': 3, '|': 4, 'splice': 5}
 

src/tarantool/request.py

@@ -144,15 +144,12 @@ class RequestInsert(Request):
     '''
     request_type = REQUEST_TYPE_INSERT
 
-    def __init__(self, conn, space_name, values, return_tuple, not_presented, presented): # pylint: disable=W0231
+    def __init__(self, conn, space_name, values, flags): # pylint: disable=W0231
         '''\
         '''
         super(RequestInsert, self).__init__(conn)
 
         assert isinstance(values, (tuple, list))
-        flags  = 1 if return_tuple  else 0
-        flags += 2 if not_presented else 0
-        flags += 4 if presented     else 0
 
         space_no = self.conn.schema.space_no(space_name)
         request_body = \