__del__ for Connection and Database

omnipath_metabo/schema/_connection.py

@@ -38,8 +38,13 @@ def _uri(self) -> str:
         )
 
     def connect(self):
+
         self.engine = create_engine(self._uri)
         Session = sessionmaker(bind=self.engine)
         self.session = Session()
 
-
+
+    def __del__(self):
+
+        self.session.close()
+        self.engine.dispose()

omnipath_metabo/schema/_main.py

@@ -31,10 +31,17 @@ def create(self) -> None:
 
     def load(self, resource) -> None:
 
-        loader = Loader(resource, self.session)
+        loader = Loader(resource, self.con.session)
         loader.load()
 
 
+    def __del__(self):
+
+        if hasattr(self, 'con'):
+
+            del self.con
+
+
 class Loader():
     #accept scheme and resource.
     def __init__(self, resource, session):
@@ -44,10 +51,15 @@ def __init__(self, resource, 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)
+
+        self.session.commit()