Commit 2bc5e073 authored by Nick Coghlan's avatar Nick Coghlan

Move the PEP 343 documentation and implementation closer to the

terminology in the alpha 1 documentation.

 - "context manager" reverts to its alpha 1 definition
 - the term "context specifier" goes away entirely
 - contextlib.GeneratorContextManager is renamed GeneratorContext

There are still a number of changes relative to alpha 1:

  - the expression in the with statement is explicitly called the
    "context expression" in the language reference
  - the terms 'with statement context', 'context object' or 'with
    statement context' are used in several places instead of a bare
    'context'. The aim of this is to avoid ambiguity in relation to the
    runtime context set up when the block is executed, and the context
    objects that already exist in various application domains (such as
    decimal.Context)
  - contextlib.contextmanager is renamed to contextfactory
    This best reflects the nature of the function resulting from the
    use of that decorator
  - decimal.ContextManager is renamed to WithStatementContext
    Simple dropping the 'Manager' part wasn't possible due to the
    fact that decimal.Context already exists and means something
    different. WithStatementContext is ugly but workable.

A technically unrelated change snuck into this commit:
contextlib.closing now avoids the overhead of creating a
generator, since it's trivial to implement that particular
context manager directly.
parent e6ca69fd
...@@ -11,18 +11,19 @@ This module provides utilities for common tasks involving the ...@@ -11,18 +11,19 @@ This module provides utilities for common tasks involving the
Functions provided: Functions provided:
\begin{funcdesc}{contextmanager}{func} \begin{funcdesc}{context}func}
This function is a decorator that can be used to define context managers This function is a decorator that can be used to define a factory
for use with the \keyword{with} statement, without needing to create a function for \keyword{with} statement context objects, without
class or separate \method{__enter__()} and \method{__exit__()} methods. needing to create a class or separate \method{__enter__()} and
\method{__exit__()} methods.
A simple example: A simple example:
\begin{verbatim} \begin{verbatim}
from __future__ import with_statement from __future__ import with_statement
from contextlib import contextmanager from contextlib import contextfactory
@contextmanager @contextfactory
def tag(name): def tag(name):
print "<%s>" % name print "<%s>" % name
yield yield
...@@ -36,9 +37,10 @@ foo ...@@ -36,9 +37,10 @@ foo
</h1> </h1>
\end{verbatim} \end{verbatim}
When called, the decorated function must return a generator-iterator. The function being decorated must return a generator-iterator when
This iterator must yield exactly one value, which will be bound to the called. This iterator must yield exactly one value, which will be
targets in the \keyword{with} statement's \keyword{as} clause, if any. bound to the targets in the \keyword{with} statement's \keyword{as}
clause, if any.
At the point where the generator yields, the block nested in the At the point where the generator yields, the block nested in the
\keyword{with} statement is executed. The generator is then resumed \keyword{with} statement is executed. The generator is then resumed
...@@ -53,20 +55,20 @@ reraise that exception. Otherwise the \keyword{with} statement will ...@@ -53,20 +55,20 @@ reraise that exception. Otherwise the \keyword{with} statement will
treat the exception as having been handled, and resume execution with treat the exception as having been handled, and resume execution with
the statement immediately following the \keyword{with} statement. the statement immediately following the \keyword{with} statement.
Note that you can use \code{@contextmanager} to define a context Note that you can use \code{@contextfactory} to define a context
specifier's \method{__context__} method. This is usually more manager's \method{__context__} method. This is usually more
convenient than creating another class just to serve as a context convenient than creating another class just to serve as a context
manager. For example: object. For example:
\begin{verbatim} \begin{verbatim}
from __future__ import with_statement from __future__ import with_statement
from contextlib import contextmanager from contextlib import contextfactory
class Tag: class Tag:
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
@contextmanager @contextfactory
def __context__(self): def __context__(self):
print "<%s>" % self.name print "<%s>" % self.name
yield self yield self
...@@ -83,7 +85,7 @@ hello from <__main__.Tag instance at 0x402ce8ec> ...@@ -83,7 +85,7 @@ hello from <__main__.Tag instance at 0x402ce8ec>
\end{funcdesc} \end{funcdesc}
\begin{funcdesc}{nested}{ctx1\optional{, ctx2\optional{, ...}}} \begin{funcdesc}{nested}{ctx1\optional{, ctx2\optional{, ...}}}
Combine multiple context specifiers into a single nested context manager. Combine multiple context managers into a single nested context manager.
Code like this: Code like this:
...@@ -104,12 +106,12 @@ with A as X: ...@@ -104,12 +106,12 @@ with A as X:
\end{verbatim} \end{verbatim}
Note that if the \method{__exit__()} method of one of the nested Note that if the \method{__exit__()} method of one of the nested
context managers indicates an exception should be suppressed, no context objects indicates an exception should be suppressed, no
exception information will be passed to any remaining outer context exception information will be passed to any remaining outer context
managers. Similarly, if the \method{__exit__()} method of one of the objects. Similarly, if the \method{__exit__()} method of one of the
nested context managers raises an exception, any previous exception nested context objects raises an exception, any previous exception
state will be lost; the new exception will be passed to the state will be lost; the new exception will be passed to the
\method{__exit__()} methods of any remaining outer context managers. \method{__exit__()} methods of any remaining outer context objects.
In general, \method{__exit__()} methods should avoid raising In general, \method{__exit__()} methods should avoid raising
exceptions, and in particular they should not re-raise a exceptions, and in particular they should not re-raise a
passed-in exception. passed-in exception.
...@@ -117,13 +119,13 @@ passed-in exception. ...@@ -117,13 +119,13 @@ passed-in exception.
\label{context-closing} \label{context-closing}
\begin{funcdesc}{closing}{thing} \begin{funcdesc}{closing}{thing}
Return a context manager that closes \var{thing} upon completion of the Return a context that closes \var{thing} upon completion of the
block. This is basically equivalent to: block. This is basically equivalent to:
\begin{verbatim} \begin{verbatim}
from contextlib import contextmanager from contextlib import contextfactory
@contextmanager @contextfactory
def closing(thing): def closing(thing):
try: try:
yield thing yield thing
...@@ -137,14 +139,33 @@ from __future__ import with_statement ...@@ -137,14 +139,33 @@ from __future__ import with_statement
from contextlib import closing from contextlib import closing
import codecs import codecs
with closing(urllib.urlopen('http://www.python.org')) as f: with closing(urllib.urlopen('http://www.python.org')) as page:
for line in f: for line in page:
print line print line
\end{verbatim} \end{verbatim}
without needing to explicitly close \code{f}. Even if an error occurs, without needing to explicitly close \code{page}. Even if an error
\code{f.close()} will be called when the \keyword{with} block is exited. occurs, \code{page.close()} will be called when the \keyword{with}
block is exited.
Context managers with a close method can use this context factory
directly without needing to implement their own
\method{__context__()} method.
\begin{verbatim}
from __future__ import with_statement
from contextlib import closing
class MyClass:
def close(self):
print "Closing", self
__context__ = closing
>>> with MyClass() as x:
... print "Hello from", x
...
Hello from <__main__.MyClass instance at 0xb7df02ec>
Closing <__main__.MyClass instance at 0xb7df02ec>
\end{verbatim}
\end{funcdesc} \end{funcdesc}
\begin{seealso} \begin{seealso}
......
...@@ -1756,59 +1756,59 @@ implemented in C will have to provide a writable ...@@ -1756,59 +1756,59 @@ implemented in C will have to provide a writable
\subsection{Context Types \label{typecontext}} \subsection{Context Types \label{typecontext}}
\versionadded{2.5} \versionadded{2.5}
\index{context specification protocol} \index{with statement context protocol}
\index{context management protocol} \index{context management protocol}
\index{protocol!context specification} \index{protocol!with statement context}
\index{protocol!context management} \index{protocol!context management}
Python's \keyword{with} statement supports the concept of a runtime Python's \keyword{with} statement supports the concept of a runtime
context defined by a context specifier. This is implemented using context defined by a context manager. This is implemented using
three distinct methods; these are used to allow user-defined three distinct methods; these are used to allow user-defined
classes to define a context. classes to define a runtime context.
The \dfn{context specification protocol} consists of a single The \dfn{context management protocol} consists of a single
method that needs to be provided for a context specifier object to method that needs to be provided for a context manager object to
define a runtime context: define a runtime context:
\begin{methoddesc}[context specifier]{__context__}{} \begin{methoddesc}[context manager]{__context__}{}
Return a context manager object. The object is required to support Return a with statement context object. The object is required to
the context management protocol described below. If an object support the with statement context protocol described below. If an
supports different kinds of runtime context, additional methods can object supports different kinds of runtime context, additional
be provided to specifically request context managers for those methods can be provided to specifically request context objects for
kinds of context. (An example of an object supporting multiple kinds those kinds of runtime context. (An example of an object supporting
of context would be a synchronisation object which supported both multiple kinds of context would be a synchronisation object which
a locked context for normal thread synchronisation and an unlocked supported both a locked context for normal thread synchronisation
context to temporarily release a held lock while performing a and an unlocked context to temporarily release a held lock while
potentially long running operation) performing a potentially long running operation)
\end{methoddesc} \end{methoddesc}
The context manager objects themselves are required to support the The with statement context objects themselves are required to support the
following three methods, which together form the following three methods, which together form the
\dfn{context management protocol}: \dfn{with statement context protocol}:
\begin{methoddesc}[context manager]{__context__}{} \begin{methoddesc}[with statement context]{__context__}{}
Return the context manager object itself. This is required to Return the context object itself. This is required to allow both
allow both context specifiers and context managers to be used with context objects and context managers to be used in a \keyword{with}
the \keyword{with} statement. statement.
\end{methoddesc} \end{methoddesc}
\begin{methoddesc}[context manager]{__enter__}{} \begin{methoddesc}[with statement context]{__enter__}{}
Enter the runtime context and return either the defining context Enter the runtime context and return either the defining context
specifier or another object related to the runtime context. The value manager or another object related to the runtime context. The value
returned by this method is bound to the identifier in the returned by this method is bound to the identifier in the
\keyword{as} clause of \keyword{with} statements using this context. \keyword{as} clause of \keyword{with} statements using this context.
(An example of a context with a context manager that returns the (An example of a context object that returns the original context
original context specifier is file objects, which are returned from manager is file objects, which are returned from __enter__() to
__enter__() to allow \function{open()} to be used directly in a with allow \function{open()} to be used directly in a with
statement. An example of a context manager that returns a related statement. An example of a context object that returns a related
object is \code{decimal.Context} which sets the active decimal object is \code{decimal.Context} which sets the active decimal
context to a copy of the context specifier and then returns the copy context to a copy of the context manager and then returns the copy.
to allow changes to be made to the current decimal context in the This allows changes to be made to the current decimal context in the
body of the \keyword{with} statement) without affecting code outside body of the \keyword{with} statement without affecting code outside
the \keyword{with} statement). the \keyword{with} statement).
\end{methoddesc} \end{methoddesc}
\begin{methoddesc}[context manager]{__exit__}{exc_type, exc_val, exc_tb} \begin{methoddesc}[with statement context]{__exit__}{exc_type, exc_val, exc_tb}
Exit the runtime context and return a Boolean flag indicating if any Exit the runtime context and return a Boolean flag indicating if any
expection that occurred should be suppressed. If an exception expection that occurred should be suppressed. If an exception
occurred while executing the body of the \keyword{with} statement, the occurred while executing the body of the \keyword{with} statement, the
...@@ -1829,19 +1829,18 @@ following three methods, which together form the ...@@ -1829,19 +1829,18 @@ following three methods, which together form the
\method{__exit__()} method has actually failed. \method{__exit__()} method has actually failed.
\end{methoddesc} \end{methoddesc}
Python defines several context specifiers and managers to support Python defines several context objects and managers to support
easy thread synchronisation, prompt closure of files or other easy thread synchronisation, prompt closure of files or other
objects, and thread-safe manipulation of the decimal arithmetic objects, and thread-safe manipulation of the decimal arithmetic
context. The specific types are not important beyond their context. The specific types are not important beyond their
implementation of the context specification and context implementation of the context management and with statement context
management protocols. protocols.
Python's generators and the \code{contextlib.contextmanager} Python's generators and the \code{contextlib.contextfactory} decorator
decorator provide a convenient way to implement the context provide a convenient way to implement these protocols. If a context
specification and context management protocols. If a context manager's \method{__context__()} method is implemented as a
specifier's \method{__context__()} method is implemented as a generator decorated with the \code{contextlib.contextfactory}
generator decorated with the \code{contextlib.contextmanager} decorator, it will automatically return a with statement context
decorator, it will automatically return a context manager
object supplying the necessary \method{__context__()}, object supplying the necessary \method{__context__()},
\method{__enter__()} and \method{__exit__()} methods. \method{__enter__()} and \method{__exit__()} methods.
......
...@@ -2112,59 +2112,60 @@ implement a \method{__coerce__()} method, for use by the built-in ...@@ -2112,59 +2112,60 @@ implement a \method{__coerce__()} method, for use by the built-in
\end{itemize} \end{itemize}
\subsection{Context Specifiers and Managers\label{context-managers}} \subsection{With Statement Contexts and Context Managers\label{context-managers}}
\versionadded{2.5} \versionadded{2.5}
A \dfn{context specifier} is an object that defines the runtime A \dfn{context manager} is an object that defines the runtime
context to be established when executing a \keyword{with} context to be established when executing a \keyword{with}
statement. The context specifier provides a \dfn{context manager} statement. The context manager provides a
which manages the entry into, and the exit from, the desired \dfn{with statement context object} which manages the entry into,
runtime context for the execution of the block of code. Context and the exit from, the desired runtime context for the execution
managers are normally invoked using the \keyword{with} statement of the block of code. Context managers are normally invoked using
(described in section~\ref{with}), but can also be used by the \keyword{with} statement (described in section~\ref{with}), but
directly invoking their methods. can also be used by directly invoking their methods.
\stindex{with} \stindex{with}
\index{context manager} \index{context manager}
\index{context specifier} \index{context (with statement)}
\index{with statement context}
Typical uses of context specifiers and managers include saving and Typical uses of context managers and contexts include saving and
restoring various kinds of global state, locking and unlocking restoring various kinds of global state, locking and unlocking
resources, closing opened files, etc. resources, closing opened files, etc.
For more information on context specifiers and context manager objects, For more information on context managers and context objects,
see ``\ulink{Context Types}{../lib/typecontext.html}'' in the see ``\ulink{Context Types}{../lib/typecontext.html}'' in the
\citetitle[../lib/lib.html]{Python Library Reference}. \citetitle[../lib/lib.html]{Python Library Reference}.
\begin{methoddesc}[context specifier]{__context__}{self} \begin{methoddesc}[context manager]{__context__}{self}
Invoked when the object is used as the context expression of a Invoked when the object is used as the context expression of a
\keyword{with} statement. The returned object must implement \keyword{with} statement. The returned object must implement
\method{__enter__()} and \method{__exit__()} methods. \method{__enter__()} and \method{__exit__()} methods.
Context specifiers written in Python can also implement this method Context managers written in Python can also implement this method
using a generator function decorated with the using a generator function decorated with the
\function{contextlib.contextmanager} decorator, as this can be simpler \function{contextlib.contextfactory} decorator, as this can be simpler
than writing individual \method{__enter__()} and \method{__exit__()} than writing individual \method{__enter__()} and \method{__exit__()}
methods on a separate object when the state to be managed is complex. methods on a separate object when the state to be managed is complex.
Context manager objects also need to implement this method; they are With statement context objects also need to implement this method; they
required to return themselves (that is, this method will simply are required to return themselves (that is, this method will simply
return \var{self}). return \var{self}).
\end{methoddesc} \end{methoddesc}
\begin{methoddesc}[context manager]{__enter__}{self} \begin{methoddesc}[with statement context]{__enter__}{self}
Enter the context managed by this object. The \keyword{with} statement Enter the runtime context related to this object. The \keyword{with}
will bind this method's return value to the target(s) specified in the statement will bind this method's return value to the target(s)
\keyword{as} clause of the statement, if any. specified in the \keyword{as} clause of the statement, if any.
\end{methoddesc} \end{methoddesc}
\begin{methoddesc}[context manager]{__exit__} \begin{methoddesc}[context manager]{__exit__}
{self, exc_type, exc_value, traceback} {self, exc_type, exc_value, traceback}
Exit the context managed by this object. The parameters describe the Exit the runtime context related to this object. The parameters
exception that caused the context to be exited. If the context was describe the exception that caused the context to be exited. If
exited without an exception, all three arguments will be the context was exited without an exception, all three arguments
\constant{None}. will be \constant{None}.
If an exception is supplied, and the method wishes to suppress the If an exception is supplied, and the method wishes to suppress the
exception (i.e., prevent it from being propagated), it should return a exception (i.e., prevent it from being propagated), it should return a
......
...@@ -315,10 +315,10 @@ statement to generate exceptions may be found in section~\ref{raise}. ...@@ -315,10 +315,10 @@ statement to generate exceptions may be found in section~\ref{raise}.
\versionadded{2.5} \versionadded{2.5}
The \keyword{with} statement is used to wrap the execution of a block The \keyword{with} statement is used to wrap the execution of a block
with methods defined by a context specifier or manager (see with methods defined by a context manager or with statement context
section~\ref{context-managers}). This allows common object (see section~\ref{context-managers}). This allows common
\keyword{try}...\keyword{except}...\keyword{finally} usage patterns to \keyword{try}...\keyword{except}...\keyword{finally} usage patterns to
be encapsulated as context specifiers or managers for convenient reuse. be encapsulated for convenient reuse.
\begin{productionlist} \begin{productionlist}
\production{with_stmt} \production{with_stmt}
...@@ -329,12 +329,12 @@ The execution of the \keyword{with} statement proceeds as follows: ...@@ -329,12 +329,12 @@ The execution of the \keyword{with} statement proceeds as follows:
\begin{enumerate} \begin{enumerate}
\item The expression is evaluated, to obtain a context specifier. \item The context expression is evaluated, to obtain a context manager.
\item The context specifier's \method{__context__()} method is \item The context manger's \method{__context__()} method is
invoked to obtain a context manager object. invoked to obtain a with statement context object.
\item The context manager's \method{__enter__()} method is invoked. \item The context object's \method{__enter__()} method is invoked.
\item If a target list was included in the \keyword{with} \item If a target list was included in the \keyword{with}
statement, the return value from \method{__enter__()} is assigned to it. statement, the return value from \method{__enter__()} is assigned to it.
...@@ -347,7 +347,7 @@ an error occurring within the suite would be. See step 6 below.} ...@@ -347,7 +347,7 @@ an error occurring within the suite would be. See step 6 below.}
\item The suite is executed. \item The suite is executed.
\item The context manager's \method{__exit__()} method is invoked. If \item The context object's \method{__exit__()} method is invoked. If
an exception caused the suite to be exited, its type, value, and an exception caused the suite to be exited, its type, value, and
traceback are passed as arguments to \method{__exit__()}. Otherwise, traceback are passed as arguments to \method{__exit__()}. Otherwise,
three \constant{None} arguments are supplied. three \constant{None} arguments are supplied.
......
...@@ -2,10 +2,10 @@ ...@@ -2,10 +2,10 @@
import sys import sys
__all__ = ["contextmanager", "nested", "closing"] __all__ = ["contextfactory", "nested", "closing"]
class GeneratorContextManager(object): class GeneratorContext(object):
"""Helper for @contextmanager decorator.""" """Helper for @contextfactory decorator."""
def __init__(self, gen): def __init__(self, gen):
self.gen = gen self.gen = gen
...@@ -48,8 +48,8 @@ class GeneratorContextManager(object): ...@@ -48,8 +48,8 @@ class GeneratorContextManager(object):
raise raise
def contextmanager(func): def contextfactory(func):
"""@contextmanager decorator. """@contextfactory decorator.
Typical usage: Typical usage:
...@@ -77,7 +77,7 @@ def contextmanager(func): ...@@ -77,7 +77,7 @@ def contextmanager(func):
""" """
def helper(*args, **kwds): def helper(*args, **kwds):
return GeneratorContextManager(func(*args, **kwds)) return GeneratorContext(func(*args, **kwds))
try: try:
helper.__name__ = func.__name__ helper.__name__ = func.__name__
helper.__doc__ = func.__doc__ helper.__doc__ = func.__doc__
...@@ -87,7 +87,7 @@ def contextmanager(func): ...@@ -87,7 +87,7 @@ def contextmanager(func):
return helper return helper
@contextmanager @contextfactory
def nested(*contexts): def nested(*contexts):
"""Support multiple context managers in a single with-statement. """Support multiple context managers in a single with-statement.
...@@ -133,9 +133,8 @@ def nested(*contexts): ...@@ -133,9 +133,8 @@ def nested(*contexts):
raise exc[0], exc[1], exc[2] raise exc[0], exc[1], exc[2]
@contextmanager class closing(object):
def closing(thing): """Context to automatically close something at the end of a block.
"""Context manager to automatically close something at the end of a block.
Code like this: Code like this:
...@@ -151,7 +150,11 @@ def closing(thing): ...@@ -151,7 +150,11 @@ def closing(thing):
f.close() f.close()
""" """
try: def __init__(self, thing):
yield thing self.thing = thing
finally: def __context__(self):
thing.close() return self
def __enter__(self):
return self.thing
def __exit__(self, *exc_info):
self.thing.close()
...@@ -2173,7 +2173,7 @@ for name in rounding_functions: ...@@ -2173,7 +2173,7 @@ for name in rounding_functions:
del name, val, globalname, rounding_functions del name, val, globalname, rounding_functions
class ContextManager(object): class WithStatementContext(object):
"""Helper class to simplify Context management. """Helper class to simplify Context management.
Sample usage: Sample usage:
...@@ -2249,7 +2249,7 @@ class Context(object): ...@@ -2249,7 +2249,7 @@ class Context(object):
return ', '.join(s) + ')' return ', '.join(s) + ')'
def __context__(self): def __context__(self):
return ContextManager(self.copy()) return WithStatementContext(self.copy())
def clear_flags(self): def clear_flags(self):
"""Reset all flags to zero""" """Reset all flags to zero"""
......
...@@ -13,9 +13,9 @@ from test.test_support import run_suite ...@@ -13,9 +13,9 @@ from test.test_support import run_suite
class ContextManagerTestCase(unittest.TestCase): class ContextManagerTestCase(unittest.TestCase):
def test_contextmanager_plain(self): def test_contextfactory_plain(self):
state = [] state = []
@contextmanager @contextfactory
def woohoo(): def woohoo():
state.append(1) state.append(1)
yield 42 yield 42
...@@ -26,9 +26,9 @@ class ContextManagerTestCase(unittest.TestCase): ...@@ -26,9 +26,9 @@ class ContextManagerTestCase(unittest.TestCase):
state.append(x) state.append(x)
self.assertEqual(state, [1, 42, 999]) self.assertEqual(state, [1, 42, 999])
def test_contextmanager_finally(self): def test_contextfactory_finally(self):
state = [] state = []
@contextmanager @contextfactory
def woohoo(): def woohoo():
state.append(1) state.append(1)
try: try:
...@@ -47,8 +47,8 @@ class ContextManagerTestCase(unittest.TestCase): ...@@ -47,8 +47,8 @@ class ContextManagerTestCase(unittest.TestCase):
self.fail("Expected ZeroDivisionError") self.fail("Expected ZeroDivisionError")
self.assertEqual(state, [1, 42, 999]) self.assertEqual(state, [1, 42, 999])
def test_contextmanager_no_reraise(self): def test_contextfactory_no_reraise(self):
@contextmanager @contextfactory
def whee(): def whee():
yield yield
ctx = whee().__context__() ctx = whee().__context__()
...@@ -56,8 +56,8 @@ class ContextManagerTestCase(unittest.TestCase): ...@@ -56,8 +56,8 @@ class ContextManagerTestCase(unittest.TestCase):
# Calling __exit__ should not result in an exception # Calling __exit__ should not result in an exception
self.failIf(ctx.__exit__(TypeError, TypeError("foo"), None)) self.failIf(ctx.__exit__(TypeError, TypeError("foo"), None))
def test_contextmanager_trap_yield_after_throw(self): def test_contextfactory_trap_yield_after_throw(self):
@contextmanager @contextfactory
def whoo(): def whoo():
try: try:
yield yield
...@@ -69,9 +69,9 @@ class ContextManagerTestCase(unittest.TestCase): ...@@ -69,9 +69,9 @@ class ContextManagerTestCase(unittest.TestCase):
RuntimeError, ctx.__exit__, TypeError, TypeError("foo"), None RuntimeError, ctx.__exit__, TypeError, TypeError("foo"), None
) )
def test_contextmanager_except(self): def test_contextfactory_except(self):
state = [] state = []
@contextmanager @contextfactory
def woohoo(): def woohoo():
state.append(1) state.append(1)
try: try:
...@@ -86,14 +86,14 @@ class ContextManagerTestCase(unittest.TestCase): ...@@ -86,14 +86,14 @@ class ContextManagerTestCase(unittest.TestCase):
raise ZeroDivisionError(999) raise ZeroDivisionError(999)
self.assertEqual(state, [1, 42, 999]) self.assertEqual(state, [1, 42, 999])
def test_contextmanager_attribs(self): def test_contextfactory_attribs(self):
def attribs(**kw): def attribs(**kw):
def decorate(func): def decorate(func):
for k,v in kw.items(): for k,v in kw.items():
setattr(func,k,v) setattr(func,k,v)
return func return func
return decorate return decorate
@contextmanager @contextfactory
@attribs(foo='bar') @attribs(foo='bar')
def baz(spam): def baz(spam):
"""Whee!""" """Whee!"""
...@@ -106,13 +106,13 @@ class NestedTestCase(unittest.TestCase): ...@@ -106,13 +106,13 @@ class NestedTestCase(unittest.TestCase):
# XXX This needs more work # XXX This needs more work
def test_nested(self): def test_nested(self):
@contextmanager @contextfactory
def a(): def a():
yield 1 yield 1
@contextmanager @contextfactory
def b(): def b():
yield 2 yield 2
@contextmanager @contextfactory
def c(): def c():
yield 3 yield 3
with nested(a(), b(), c()) as (x, y, z): with nested(a(), b(), c()) as (x, y, z):
...@@ -122,14 +122,14 @@ class NestedTestCase(unittest.TestCase): ...@@ -122,14 +122,14 @@ class NestedTestCase(unittest.TestCase):
def test_nested_cleanup(self): def test_nested_cleanup(self):
state = [] state = []
@contextmanager @contextfactory
def a(): def a():
state.append(1) state.append(1)
try: try:
yield 2 yield 2
finally: finally:
state.append(3) state.append(3)
@contextmanager @contextfactory
def b(): def b():
state.append(4) state.append(4)
try: try:
...@@ -148,7 +148,7 @@ class NestedTestCase(unittest.TestCase): ...@@ -148,7 +148,7 @@ class NestedTestCase(unittest.TestCase):
def test_nested_right_exception(self): def test_nested_right_exception(self):
state = [] state = []
@contextmanager @contextfactory
def a(): def a():
yield 1 yield 1
class b(object): class b(object):
...@@ -172,10 +172,10 @@ class NestedTestCase(unittest.TestCase): ...@@ -172,10 +172,10 @@ class NestedTestCase(unittest.TestCase):
self.fail("Didn't raise ZeroDivisionError") self.fail("Didn't raise ZeroDivisionError")
def test_nested_b_swallows(self): def test_nested_b_swallows(self):
@contextmanager @contextfactory
def a(): def a():
yield yield
@contextmanager @contextfactory
def b(): def b():
try: try:
yield yield
...@@ -189,7 +189,7 @@ class NestedTestCase(unittest.TestCase): ...@@ -189,7 +189,7 @@ class NestedTestCase(unittest.TestCase):
self.fail("Didn't swallow ZeroDivisionError") self.fail("Didn't swallow ZeroDivisionError")
def test_nested_break(self): def test_nested_break(self):
@contextmanager @contextfactory
def a(): def a():
yield yield
state = 0 state = 0
...@@ -201,7 +201,7 @@ class NestedTestCase(unittest.TestCase): ...@@ -201,7 +201,7 @@ class NestedTestCase(unittest.TestCase):
self.assertEqual(state, 1) self.assertEqual(state, 1)
def test_nested_continue(self): def test_nested_continue(self):
@contextmanager @contextfactory
def a(): def a():
yield yield
state = 0 state = 0
...@@ -213,7 +213,7 @@ class NestedTestCase(unittest.TestCase): ...@@ -213,7 +213,7 @@ class NestedTestCase(unittest.TestCase):
self.assertEqual(state, 3) self.assertEqual(state, 3)
def test_nested_return(self): def test_nested_return(self):
@contextmanager @contextfactory
def a(): def a():
try: try:
yield yield
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment