Skip to content

Commit

Permalink
Created loader class for inserting into psql
Browse files Browse the repository at this point in the history
  • Loading branch information
for-hyde committed Sep 26, 2024
1 parent df72fe2 commit c5e1ce8
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions omnipath_metabo/schema/_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,20 @@
def create(con):

Base.metadata.create_all(con.engine)

class Loader():
#accept scheme and resource.
def __init__(self, scheme, resource, session):
self.scheme = scheme
self.resource = resource
self.session = session


def load(self):
for row in self.resource:
insert_statement = self.scheme.insert().values(
smiles=row['smiles'],
accession=row['accession'],
inchi=row['inchi']
)
self.session.execute(insert_statement)

0 comments on commit c5e1ce8

Please sign in to comment.