fix: only apply replace_keys flag on first batch during segmented key import

Author: kleinjoshuaa

CHANGES.txt

@@ -48,4 +48,6 @@
 3.5.3 (Jul 10, 2025)
 - fixed bug in listing large segments
 3.5.4 (Aug 12, 2025)
-- added method to retrieve segment definition key counts for standard segments
+- added method to retrieve segment definition key counts for standard segments
+3.5.5 (Sep 8, 2025)
+- fixed method to import segment keys of larger (10k+) segment keys appropriately

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "splitapiclient"
-version = "3.5.4"
+version = "3.5.5"
 description = "This Python Library provide full support for Split REST Admin API, allow creating, deleting and editing Environments, Splits, Split Definitions, Segments, Segment Keys, Users, Groups, API Keys, Change Requests, Attributes and Identities"
 classifiers = [
     "Programming Language :: Python :: 3",

splitapiclient/resources/segment_definition.py

@@ -119,14 +119,18 @@ def import_keys_from_json(self, replace_keys, json_data, apiclient=None):
             key_batches = [keys[i:i + 10000] for i in range(0, len(keys), 10000)]
             success = True
             # Process each batch
+            first_batch = True
             for key_batch in key_batches:
                 # Make a copy of the json_data to avoid modifying the original
                 batch_data = json_data.copy()
                 batch_data['keys'] = key_batch
                 # If any batch fails, mark the entire operation as failed
-                batch_result = imc.import_keys_from_json(self._name, self._environment['id'], replace_keys, batch_data)
+                # First batch uses original replace_keys value, subsequent batches use False
+                batch_replace_keys = replace_keys if first_batch else False
+                batch_result = imc.import_keys_from_json(self._name, self._environment['id'], batch_replace_keys, batch_data)
                 if not batch_result:
                     success = False
+                first_batch = False
             return success
         else:
             return imc.import_keys_from_json(self._name, self._environment['id'], replace_keys, json_data)

splitapiclient/version.py

@@ -1 +1 @@
-__version__ = '3.5.4'
+__version__ = '3.5.5'