update generate_inserts

PedroMTQ · Apr 16, 2022 · cc645be · cc645be
1 parent cc202df
commit cc645be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/mantis/Metadata_SQLITE_Connector.py b/mantis/Metadata_SQLITE_Connector.py
@@ -138,12 +138,10 @@ def generate_inserts(self, metadata_yielder):
         step=self.insert_step
         temp=[]
         for i in metadata_yielder:
-            if len(temp)<step:
-                temp.append(i)
-            else:
+            temp.append(i)
+            if len(temp)>=step:
                 yield temp
                 temp=[]
-                temp.append(i)
         yield temp
 
     def store_metadata(self):

diff --git a/mantis/Taxonomy_SQLITE_Connector.py b/mantis/Taxonomy_SQLITE_Connector.py
@@ -151,9 +151,8 @@ def generate_inserts(self, input_generator):
         step=self.insert_step
         temp=[]
         for i in input_generator:
-            if len(temp)<step:
-                temp.append(i)
-            elif len(temp)==step:
+            temp.append(i)
+            if len(temp)>=step:
                 yield temp
                 temp=[]
         yield temp