#48: Allow use of with keyword with Queries

common/database_helpers.py

@@ -20,6 +20,13 @@ def __init__(self, table):
         self.table = table
         self.base_query = self.session.query(table)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        log.info("Closing DB session")
+        self.session.close()
+
     @abstractmethod
     def execute_query(self):
         pass