From 0cdecd893ae4919386227d2ad19afc3d326bd1cc Mon Sep 17 00:00:00 2001 From: Sven Heidemann Date: Tue, 30 Mar 2021 10:06:07 +0200 Subject: [PATCH] Improved database context --- src/cpl/database/context/database_context.py | 3 +++ src/cpl/database/context/database_context_abc.py | 6 ++++++ 2 files changed, 9 insertions(+) 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): """