util.py

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)