Merge remote-tracking branch 'remotes/origin/test'

pull/191/head
Chris Veilleux 2019-07-08 20:06:59 -05:00
commit 9d6e722d14
1 changed files with 6 additions and 1 deletions

View File

@ -18,8 +18,13 @@ def use_transaction(func):
if hasattr(instance, "db"):
prev_autocommit = instance.db.autocommit
instance.db.autocommit = False
with instance.db:
try:
return_value = func(*args, **kwargs)
except:
instance.db.rollback()
raise
else:
instance.db.commit()
instance.db.autocommit = prev_autocommit
return return_value