include engine and session to loaded etrago obj

openego · Aug 9, 2023 · 4dac3b1 · 4dac3b1
1 parent 670ff5e
commit 4dac3b1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/etrago/tools/network.py b/etrago/tools/network.py
@@ -195,6 +195,12 @@ def __init__(
 
             self.get_clustering_data(csv_folder_name)
 
+            conn = db.connection(section=self.args["db"])
+            self.engine = conn
+
+            session = sessionmaker(bind=conn)
+            self.session = session()
+
         else:
             logger.error("Set args or csv_folder_name")