def save(self, session=None): import saladier.db.api as db_api if session is None: session = db_api.get_session() super(SaladierBase, self).save(session)
def setUp(self): super(Database, self).setUp() if self.sql_connection == "sqlite://": conn = self.engine.connect() conn.connection.executescript(self._DB) self.addCleanup(self.engine.dispose) elif self.sql_connection.startswith('sqlite:///'): shutil.copyfile(self._state_path_rel(self.sqlite_clean_db), self._state_path_rel(self.sqlite_db)) self.addCleanup(os.unlink, self.sqlite_db) else: self.session = sqla_api.get_session(recycle=True) self.session.begin() self.addCleanup(self._rollback_close)