Fix #984 - Improvements to Activity and ActivityIntermediator

[mediagoblin.git] / mediagoblin / db / base.py

diff --git a/mediagoblin/db/base.py b/mediagoblin/db/base.py
index c0cefdc27024d2c9ba44c957f00c8baafc0c65ec..38375a6927b7bdb4432de3060cdadb870e882bb2 100644 (file)
--- 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"""