Skip to content

Commit bde961d

Browse files
committed
tooling with new insertion methods
1 parent 95de6cc commit bde961d

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

omnipath_metabo/schema/_main.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,11 @@ def load(self, resource) -> None:
4646
loader = Loader(resource, self.con)
4747
loader.load()
4848

49+
def test_load(self, resource) -> None:
50+
51+
loader = Loader(resource, self.con)
52+
loader.new_load()
53+
4954
def substructure_search(self, substructure):
5055

5156
query = f"select name, mol from structures where mol @>'{substructure}'"
@@ -170,6 +175,13 @@ def create(self) -> None:
170175

171176
create(self.con)
172177

173-
174-
175-
178+
def new_load(self, batch_size = 1000):
179+
180+
self.create()
181+
182+
#Creat method for populating using iterator.
183+
184+
self.session.scalars(
185+
insert(_structure.Structure).on_conflict_do_nothing()
186+
[x for x in self.resource]
187+
)

0 commit comments

Comments
 (0)