diff options
author | Tim Peters <tim.peters@gmail.com> | 2006-02-28 18:30:36 +0000 |
---|---|---|
committer | Tim Peters <tim.peters@gmail.com> | 2006-02-28 18:30:36 +0000 |
commit | 710ab3b5f8d114b9be200af135c6327117b8c676 (patch) | |
tree | 0b4d5f3eeac98a29e67ba966f26654f5fdf4b066 /Lib/test/contextmanager.py | |
parent | 08533fdad62197f4724f029c33b88975733f4045 (diff) | |
download | cpython-git-710ab3b5f8d114b9be200af135c6327117b8c676.tar.gz |
Whitespace normalization.
Diffstat (limited to 'Lib/test/contextmanager.py')
-rw-r--r-- | Lib/test/contextmanager.py | 67 |
1 files changed, 33 insertions, 34 deletions
diff --git a/Lib/test/contextmanager.py b/Lib/test/contextmanager.py index 0ebf646f91..e8b0894619 100644 --- a/Lib/test/contextmanager.py +++ b/Lib/test/contextmanager.py @@ -1,34 +1,33 @@ -class GeneratorContextManager(object): - def __init__(self, gen): - self.gen = gen - - def __context__(self): - return self - - def __enter__(self): - try: - return self.gen.next() - except StopIteration: - raise RuntimeError("generator didn't yield") - - def __exit__(self, type, value, traceback): - if type is None: - try: - self.gen.next() - except StopIteration: - return - else: - raise RuntimeError("generator didn't stop") - else: - try: - self.gen.throw(type, value, traceback) - except (type, StopIteration): - return - else: - raise RuntimeError("generator caught exception") - -def contextmanager(func): - def helper(*args, **kwds): - return GeneratorContextManager(func(*args, **kwds)) - return helper - +class GeneratorContextManager(object):
+ def __init__(self, gen):
+ self.gen = gen
+
+ def __context__(self):
+ return self
+
+ def __enter__(self):
+ try:
+ return self.gen.next()
+ except StopIteration:
+ raise RuntimeError("generator didn't yield")
+
+ def __exit__(self, type, value, traceback):
+ if type is None:
+ try:
+ self.gen.next()
+ except StopIteration:
+ return
+ else:
+ raise RuntimeError("generator didn't stop")
+ else:
+ try:
+ self.gen.throw(type, value, traceback)
+ except (type, StopIteration):
+ return
+ else:
+ raise RuntimeError("generator caught exception")
+
+def contextmanager(func):
+ def helper(*args, **kwds):
+ return GeneratorContextManager(func(*args, **kwds))
+ return helper
|