From 5ddc85e071fd7adec6b922a03c2e8caa4bad3c5c Mon Sep 17 00:00:00 2001 From: Jessica Tallon Date: Tue, 21 Oct 2014 11:44:11 +0100 Subject: Fix #984 - Improvements to Activity and ActivityIntermediator - Add unit tests to cover get and set methods on Activity - Rewrite the set to remove set and use Session.flush instead - Use sqlalchemy's validator instead of .save hack --- mediagoblin/db/base.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'mediagoblin/db/base.py') diff --git a/mediagoblin/db/base.py b/mediagoblin/db/base.py index c0cefdc2..38375a69 100644 --- a/mediagoblin/db/base.py +++ b/mediagoblin/db/base.py @@ -31,12 +31,15 @@ class GMGTableBase(object): # The key *has* to exist on sql. return getattr(self, key) - def save(self): + def save(self, commit=True): sess = object_session(self) if sess is None: sess = Session() sess.add(self) - sess.commit() + if commit: + sess.commit() + else: + sess.flush() def delete(self, commit=True): """Delete the object and commit the change immediately by default""" -- cgit v1.2.3