mirror of
https://github.com/python/cpython.git
synced 2025-07-29 06:05:00 +00:00
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.
This commit is contained in:
parent
327ea38cc4
commit
a7e820a408
7 changed files with 165 additions and 141 deletions
|
@ -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
|
||||||
</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
|
||||||
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>
|
||||||
\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:
|
||||||
\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.
|
||||||
|
|
||||||
\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
|
||||||
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
|
||||||
\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__}{}
|
|
||||||
Return a context manager object. The object is required to support
|
|
||||||
the context management protocol described below. If an object
|
|
||||||
supports different kinds of runtime context, additional methods can
|
|
||||||
be provided to specifically request context managers for those
|
|
||||||
kinds of context. (An example of an object supporting multiple kinds
|
|
||||||
of context would be a synchronisation object which supported both
|
|
||||||
a locked context for normal thread synchronisation and an unlocked
|
|
||||||
context to temporarily release a held lock while performing a
|
|
||||||
potentially long running operation)
|
|
||||||
\end{methoddesc}
|
|
||||||
|
|
||||||
The context manager objects themselves are required to support the
|
|
||||||
following three methods, which together form the
|
|
||||||
\dfn{context management protocol}:
|
|
||||||
|
|
||||||
\begin{methoddesc}[context manager]{__context__}{}
|
\begin{methoddesc}[context manager]{__context__}{}
|
||||||
Return the context manager object itself. This is required to
|
Return a with statement context object. The object is required to
|
||||||
allow both context specifiers and context managers to be used with
|
support the with statement context protocol described below. If an
|
||||||
the \keyword{with} statement.
|
object supports different kinds of runtime context, additional
|
||||||
|
methods can be provided to specifically request context objects for
|
||||||
|
those kinds of runtime context. (An example of an object supporting
|
||||||
|
multiple kinds of context would be a synchronisation object which
|
||||||
|
supported both a locked context for normal thread synchronisation
|
||||||
|
and an unlocked context to temporarily release a held lock while
|
||||||
|
performing a potentially long running operation)
|
||||||
\end{methoddesc}
|
\end{methoddesc}
|
||||||
|
|
||||||
\begin{methoddesc}[context manager]{__enter__}{}
|
The with statement context objects themselves are required to support the
|
||||||
|
following three methods, which together form the
|
||||||
|
\dfn{with statement context protocol}:
|
||||||
|
|
||||||
|
\begin{methoddesc}[with statement context]{__context__}{}
|
||||||
|
Return the context object itself. This is required to allow both
|
||||||
|
context objects and context managers to be used in a \keyword{with}
|
||||||
|
statement.
|
||||||
|
\end{methoddesc}
|
||||||
|
|
||||||
|
\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
|
||||||
\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
|
||||||
|
|
||||||
\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}.
|
||||||
\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:
|
||||||
|
|
||||||
\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.}
|
||||||
|
|
||||||
\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 @@
|
||||||
|
|
||||||
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):
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
def contextmanager(func):
|
def contextfactory(func):
|
||||||
"""@contextmanager decorator.
|
"""@contextfactory decorator.
|
||||||
|
|
||||||
Typical usage:
|
Typical usage:
|
||||||
|
|
||||||
|
@ -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):
|
||||||
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):
|
||||||
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):
|
||||||
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:
|
||||||
|
|
||||||
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):
|
||||||
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
|
||||||
|
|
||||||
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):
|
||||||
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):
|
||||||
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):
|
||||||
# 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):
|
||||||
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):
|
||||||
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):
|
||||||
# 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):
|
||||||
|
|
||||||
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):
|
||||||
|
|
||||||
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):
|
||||||
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):
|
||||||
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):
|
||||||
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):
|
||||||
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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue