diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2015-06-28 17:09:11 +0300 |
---|---|---|
committer | Serhiy Storchaka <storchaka@gmail.com> | 2015-06-28 17:09:11 +0300 |
commit | 9aad9f27407861d674570bf5c0bd957bbe13663b (patch) | |
tree | 90c1467cde02dbc8e1006123aa07b18cd5ca7495 /Lib/contextlib.py | |
parent | 2eff9e9441d76904792141b4b38b0eb649d9eb86 (diff) | |
parent | eab770404437bd49ebf897b681221784b0035795 (diff) | |
download | cpython-git-9aad9f27407861d674570bf5c0bd957bbe13663b.tar.gz |
Issue #24336: The contextmanager decorator now works with functions with
keyword arguments called "func" and "self". Patch by Martin Panter.
Diffstat (limited to 'Lib/contextlib.py')
-rw-r--r-- | Lib/contextlib.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/Lib/contextlib.py b/Lib/contextlib.py index 379c2515d5..5377987ddf 100644 --- a/Lib/contextlib.py +++ b/Lib/contextlib.py @@ -34,7 +34,7 @@ class ContextDecorator(object): class _GeneratorContextManager(ContextDecorator): """Helper for @contextmanager decorator.""" - def __init__(self, func, *args, **kwds): + def __init__(self, func, args, kwds): self.gen = func(*args, **kwds) self.func, self.args, self.kwds = func, args, kwds # Issue 19330: ensure context manager instances have good docstrings @@ -52,7 +52,7 @@ class _GeneratorContextManager(ContextDecorator): # _GCM instances are one-shot context managers, so the # CM must be recreated each time a decorated function is # called - return self.__class__(self.func, *self.args, **self.kwds) + return self.__class__(self.func, self.args, self.kwds) def __enter__(self): try: @@ -130,7 +130,7 @@ def contextmanager(func): """ @wraps(func) def helper(*args, **kwds): - return _GeneratorContextManager(func, *args, **kwds) + return _GeneratorContextManager(func, args, kwds) return helper |