fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

stats: use try/catch for contextmanager

Browse source
This commit is contained in:
Sean Breckenridge 2022-05-01 15:47:44 -07:00
parent 301ce5f35e
commit bba07d8265
2 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
my/core/common.py
View file

@ -427,7 +427,7 @@ def warn_if_empty(f):
# global state that turns on/off quick stats
# can use the 'with_quick_stats' contextmanager
# can use the 'quick_stats' contextmanager
# to enable/disable this in cli so that module 'stats'
# functions don't have to implement custom 'quick' logic
QUICK_STATS = False
@ -437,11 +437,14 @@ QUICK_STATS = False
# elsewhere -- can use this decorator instead of editing
# the global state directly
@contextmanager
def with_quick_stats():
def quick_stats():
global QUICK_STATS
prev, QUICK_STATS = QUICK_STATS, True
yield
QUICK_STATS = prev
prev = QUICK_STATS
try:
QUICK_STATS = True
yield
finally:
QUICK_STATS = prev
C = TypeVar('C')