work on improved data insertion

saezlab · Oct 16, 2024 · 3e4697e · 3e4697e
1 parent 0fbffb8
commit 3e4697e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/omnipath_metabo/schema/_main.py b/omnipath_metabo/schema/_main.py
@@ -1,6 +1,7 @@
 import collections
 from sqlalchemy import text, select
 from sqlalchemy.dialects.postgresql import insert
+import psychopg2.extras
 
 from . import _structure
 from ._base import Base
@@ -106,7 +107,10 @@ def load(self):
         self.session.execute(insert_ids)
         self.session.commit()
 
-
+        query = """ 
+        INSERT INTO structures (smiles) VALUES %s
+        """
+        psychopg2.extras.execute_values(query, (), page_size = 1000)
         #self.indexer()
 
     def update_mol_column(self):