voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

@transaction intercept commit argument; commit / release here.

Browse source
This commit is contained in:
voussoir 2020-02-19 22:36:00 -08:00
parent 04f3f6f297
commit 203fb9e00f
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
etiquette/decorators.py
View file

@ -78,12 +78,22 @@ def transaction(method):
@functools.wraps(method) @functools.wraps(method)
def wrapped_transaction(self, *args, **kwargs): def wrapped_transaction(self, *args, **kwargs):
photodb = _get_relevant_photodb(self) photodb = _get_relevant_photodb(self)
commit = kwargs.pop('commit', False)
is_root = len(photodb.savepoints) == 0
savepoint_id = photodb.savepoint(message=method.__qualname__) savepoint_id = photodb.savepoint(message=method.__qualname__)
try: try:
result = method(self, *args, **kwargs) result = method(self, *args, **kwargs)
except Exception as e: except Exception as e:
photodb.rollback(savepoint=savepoint_id) photodb.rollback(savepoint=savepoint_id)
raise raise
else:
return result if commit:
photodb.commit(message=method.__qualname__)
elif not is_root:
photodb.release_savepoint(savepoint=savepoint_id)
return result
return wrapped_transaction return wrapped_transaction