Use try..finally in contextlib.contextmanager

This commit is contained in:
Erik Johnston 2015-04-15 10:25:43 +01:00
parent ded4128965
commit a971fa9d58

View file

@ -97,7 +97,9 @@ class StreamIdGenerator(object):
@contextlib.contextmanager
def manager():
try:
yield next_id
finally:
with self._lock:
self._unfinished_ids.remove(next_id)