diff --git a/src/cpl/database/context/database_context.py b/src/cpl/database/context/database_context.py index fe77f013..c07f1acd 100644 --- a/src/cpl/database/context/database_context.py +++ b/src/cpl/database/context/database_context.py @@ -30,6 +30,9 @@ class DatabaseContext(DatabaseContextABC): self._db.connect(connection_string) self._create_tables() + def save_changes(self): + self._db.session.commit() + def _create_tables(self): try: for subclass in DatabaseModel.__subclasses__(): diff --git a/src/cpl/database/context/database_context_abc.py b/src/cpl/database/context/database_context_abc.py index 0bbf3dd5..d27d6da0 100644 --- a/src/cpl/database/context/database_context_abc.py +++ b/src/cpl/database/context/database_context_abc.py @@ -27,6 +27,12 @@ class DatabaseContextABC(ABC): """ pass + def save_changes(self): + """ + Saves changes of the database + """ + pass + @abstractmethod def _create_tables(self): """