florentx · February 19, 2014 17:15 · Feb 19, 2014
diff --git a/util.py b/util.py
@@ -0,0 +1,18 @@
+import re
+
+# Helper to check if the name is a valid identifier
+_isidentifier = re.compile(r'[a-zA-Z_]\w*').match
+
+
+@contextlib.contextmanager
+def savepoint(cr, name, quiet=False):
+    assert _isidentifier(name)
+    cr.execute('SAVEPOINT "%s";' % name)
+    try:
+        yield
+    except Exception:
+        cr.execute('ROLLBACK TO "%s";' % name)
+        if not quiet:
+            raise
+    finally:
+        cr.execute('RELEASE "%s";' % name)