connection creates session

omnipath_metabo/schema/_connection.py

@@ -39,5 +39,7 @@ def _uri(self) -> str:
 
     def connect(self):
         self.engine = create_engine(self._uri)
+        Session = sessionmaker(bind=self.engine)
+        self.session = Session()
 
 

omnipath_metabo/schema/_main.py

@@ -31,14 +31,14 @@ def create(self) -> None:
 
     def load(self, resource) -> None:
 
-        loader = Loader(resource, self.con.engine)
+        loader = Loader(resource, self.session)
         loader.load()
 
 
 class Loader():
     #accept scheme and resource.
-    def __init__(self, scheme, resource, session):
-        self.scheme = scheme
+    def __init__(self, resource, session):
+        self.scheme = resource.scheme
         self.resource = resource
         self.session = session
 

omnipath_metabo/schema/_structure.py

@@ -7,3 +7,8 @@ class Structure(Base):
     id = Column(Integer, primary_key=True)
     name = Column(String)
     structure = Column('mol_structure', String)
+
+
+class Hmdb(Base):
+    scheme = Structure
+