Rooba
/
peru
mirror of https://github.com/buildinspace/peru.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

a few more yapf tweaks

This commit is contained in:
Jack O'Connor 2018-12-19 17:36:25 +01:00
parent 5ac520e9fb
commit 7fb367bc95
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
peru/async_exit_stack.py
View File

@ -31,6 +31,7 @@ class AsyncExitStack:
def _create_cb_wrapper(callback, *args, **kwds): def _create_cb_wrapper(callback, *args, **kwds):
def _exit_wrapper(exc_type, exc, tb): def _exit_wrapper(exc_type, exc, tb):
callback(*args, **kwds) callback(*args, **kwds)
return _exit_wrapper return _exit_wrapper
### AsyncExitStack staticmethods ### AsyncExitStack staticmethods
@ -43,6 +44,7 @@ class AsyncExitStack:
def _create_async_cb_wrapper(callback, *args, **kwds): def _create_async_cb_wrapper(callback, *args, **kwds):
async def _exit_wrapper(exc_type, exc, tb): async def _exit_wrapper(exc_type, exc, tb):
await callback(*args, **kwds) await callback(*args, **kwds)
return _exit_wrapper return _exit_wrapper
### _BaseExitStack methods ### _BaseExitStack methods
@ -170,6 +172,7 @@ class AsyncExitStack:
# We manipulate the exception state so it behaves as though # We manipulate the exception state so it behaves as though
# we were actually nesting multiple with statements # we were actually nesting multiple with statements
frame_exc = sys.exc_info()[1] frame_exc = sys.exc_info()[1]
def _fix_exception_context(new_exc, old_exc): def _fix_exception_context(new_exc, old_exc):
# Context may not be correct, so find the end of the chain # Context may not be correct, so find the end of the chain
while 1: while 1: