Add a .save method on the sql db objects

This is a shortcut to adding the object to a session (if
needed) and giving a commit on the session.

In reality, calling code should probably utilize the
session on its own and call commit in an appropiate place.

1 files changed, 9 insertions, 1 deletions

diff --git a/mediagoblin/db/sql/base.py b/mediagoblin/db/sql/base.py
index 1249bace..40140327 100644
--- a/mediagoblin/db/sql/base.py
+++ b/mediagoblin/db/sql/base.py
@@ -1,4 +1,4 @@
-from sqlalchemy.orm import scoped_session, sessionmaker
+from sqlalchemy.orm import scoped_session, sessionmaker, object_session
 
 
 Session = scoped_session(sessionmaker())
@@ -28,3 +28,11 @@ class GMGTableBase(object):
 
     def get(self, key):
         return getattr(self, key)
+
+    def save(self, validate = True):
+        assert validate
+        sess = object_session(self)
+        if sess is None:
+            sess = Session()
+        sess.add(self)
+        sess.commit()