← Back
Editing: contextlib.cpython-310.pyc
o 0�ie � @ s~ d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m Z g d�Z G dd� dej �Z G d d � d ej �ZG d d � d e�ZG d d� de�ZG dd� d�ZG dd� dee e�ZG dd� deee�Zdd� Zdd� ZG dd� de �ZG dd � d e�ZG d d � d e �ZG d d � d e�ZG d!d"� d"e�ZG d#d$� d$e �Z G d%d&� d&�Z G d'd(� d(e e �Z G d)d*� d*e e�Z G d+d,� d,e e�Z dS )-z4Utilities for with-statement contexts. See PEP 343.� N)�deque��wraps)� MethodType� GenericAlias) �asynccontextmanager�contextmanager�closing� nullcontext�AbstractContextManager�AbstractAsyncContextManager�AsyncExitStack�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppress�aclosingc @ �: e Zd ZdZee�Zdd� Zej dd� �Z edd� �Z dS ) r z,An abstract base class for context managers.c C � | S �z0Return `self` upon entering the runtime context.� ��selfr r �1/opt/alt/python310/lib64/python3.10/contextlib.py� __enter__ s z AbstractContextManager.__enter__c C s dS �z9Raise any exception triggered within the runtime context.Nr �r �exc_type� exc_value� tracebackr r r �__exit__ � z AbstractContextManager.__exit__c C � | t u r t�|dd�S tS )Nr r! )r �_collections_abc�_check_methods�NotImplemented��cls�Cr r r �__subclasshook__ s z'AbstractContextManager.__subclasshook__N) �__name__� __module__� __qualname__�__doc__� classmethodr �__class_getitem__r �abc�abstractmethodr! r* r r r r r � r c @ r ) r z9An abstract base class for asynchronous context managers.c � � �| S r r r r r r � __aenter__+ s �z&AbstractAsyncContextManager.__aenter__c � s �dS r r r r r r � __aexit__/ s �z%AbstractAsyncContextManager.__aexit__c C r# )Nr5 r6 )r r$ r% r& r' r r r r* 4 s �z,AbstractAsyncContextManager.__subclasshook__N) r+ r, r- r. r/ r r0 r5 r1 r2 r6 r* r r r r r % r3 r c @ � e Zd ZdZdd� Zdd� ZdS )r zJA base class or mixin that enables context managers to work as decorators.c C r )a6 Return a recreated instance of self. Allows an otherwise one-shot context manager like _GeneratorContextManager to support use as a decorator via implicit recreation. This is a private interface just for _GeneratorContextManager. See issue #11647 for details. r r r r r � _recreate_cm? s z ContextDecorator._recreate_cmc � t � �� �fdd��}|S )Nc s: �� � � � | i |��W d � S 1 sw Y d S �N�r8 ��args�kwds��funcr r r �innerL s $�z(ContextDecorator.__call__.<locals>.innerr �r r@ rA r r? r �__call__K � zContextDecorator.__call__N�r+ r, r- r. r8 rC r r r r r < s r c @ r7 )�AsyncContextDecoratorzPA base class or mixin that enables async context managers to work as decorators.c C r )z-Return a recreated instance of self. r r r r r r8 V r" z"AsyncContextDecorator._recreate_cmc r9 )Nc � sV ��� � 4 I d H � � | i |��I d H W d �I d H S 1 I d H s$w Y d S r: r; r< r? r r rA \ s �0�z-AsyncContextDecorator.__call__.<locals>.innerr rB r r? r rC [ rD z AsyncContextDecorator.__call__NrE r r r r rF S s rF c @ r7 )� _GeneratorContextManagerBasezBShared functionality for @contextmanager and @asynccontextmanager.c C sN ||i |��| _ |||| _| _| _t|dd �}|d u r"t| �j}|| _d S )Nr. )�genr@ r= r> �getattr�typer. )r r@ r= r> �docr r r �__init__f s z%_GeneratorContextManagerBase.__init__c C s | � | j| j| j�S r: )� __class__r@ r= r> r r r r r8 t s z)_GeneratorContextManagerBase._recreate_cmN)r+ r, r- r. rL r8 r r r r rG c s rG c @ r7 )�_GeneratorContextManagerz%Helper for @contextmanager decorator.c C s2 | ` | `| `zt| j�W S ty td�d �w �Nzgenerator didn't yield)r= r> r@ �nextrH � StopIteration� RuntimeErrorr r r r r � s �z"_GeneratorContextManager.__enter__c C s |d u rz t | j� W td�� ty Y dS w |d u r |� }z | j�|||� W td�� tyC } z ||uW Y d }~S d }~w tyk } z ||u rUW Y d }~dS t|t�rf|j|u rfW Y d }~dS � d }~w ty� } z ||urw� W Y d }~dS d }~ww )NF�generator didn't stopz#generator didn't stop after throw())rP rH rQ rR �throw� isinstance� __cause__� BaseException�r �typ�valuer �excr r r r! � s< �� ��� ���z!_GeneratorContextManager.__exit__N)r+ r, r- r. r r! r r r r rN { � rN c @ r7 )� _AsyncGeneratorContextManagerz*Helper for @asynccontextmanager decorator.c � s: �| ` | `| `z t| j�I d H W S ty td�d �w rO )r= r> r@ �anextrH �StopAsyncIterationrR r r r r r5 � s � �z(_AsyncGeneratorContextManager.__aenter__c � s �|d u r z t | j�I d H W td�� ty Y dS w |d u r$|� }z| j�|||�I d H W td�� tyJ } z ||uW Y d }~S d }~w tyt } z ||u r\W Y d }~dS t|ttf�ro|j|u roW Y d }~dS � d }~w ty� } z ||ur�� W Y d }~dS d }~ww )NFrS z$generator didn't stop after athrow()) r^ rH r_ rR �athrowrU rQ rV rW rX r r r r6 � s> � �� �� � ���z'_AsyncGeneratorContextManager.__aexit__N)r+ r, r- r. r5 r6 r r r r r] � r\ r] c � t � �� fdd��}|S )a� @contextmanager decorator. Typical usage: @contextmanager def some_generator(<arguments>): <setup> try: yield <value> finally: <cleanup> This makes this: with some_generator(<arguments>) as <variable>: <body> equivalent to this: <setup> try: <variable> = <value> <body> finally: <cleanup> c � t � | |�S r: )rN r< �r@ r r �helper � z contextmanager.<locals>.helperr �r@ rd r rc r r � � r c ra )a� @asynccontextmanager decorator. Typical usage: @asynccontextmanager async def some_async_generator(<arguments>): <setup> try: yield <value> finally: <cleanup> This makes this: async with some_async_generator(<arguments>) as <variable>: <body> equivalent to this: <setup> try: <variable> = <value> <body> finally: <cleanup> c rb r: )r] r< rc r r rd 8 re z#asynccontextmanager.<locals>.helperr rf r rc r r rg r c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) r a2 Context to automatically close something at the end of a block. Code like this: with closing(<module>.open(<arguments>)) as f: <block> is equivalent to this: f = <module>.open(<arguments>) try: <block> finally: f.close() c C � || _ d S r: ��thing�r rk r r r rL O � zclosing.__init__c C � | j S r: rj r r r r r Q � zclosing.__enter__c G s | j �� d S r: )rk �close�r �exc_infor r r r! S s zclosing.__exit__N�r+ r, r- r. rL r r! r r r r r > s r c @ rh ) r a� Async context manager for safely finalizing an asynchronously cleaned-up resource such as an async generator, calling its ``aclose()`` method. Code like this: async with aclosing(<module>.fetch(<arguments>)) as agen: <block> is equivalent to this: agen = <module>.fetch(<arguments>) try: <block> finally: await agen.aclose() c C ri r: rj rl r r r rL i rm zaclosing.__init__c � � �| j S r: rj r r r r r5 k � �zaclosing.__aenter__c � s �| j �� I d H d S r: )rk �acloserq r r r r6 m s �zaclosing.__aexit__N)r+ r, r- r. rL r5 r6 r r r r r W s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS )�_RedirectStreamNc C s || _ g | _d S r: )� _new_target� _old_targets)r � new_targetr r r rL u s z_RedirectStream.__init__c C s* | j �tt| j�� tt| j| j� | jS r: )ry �appendrI �sys�_stream�setattrrx r r r r r z s z_RedirectStream.__enter__c C s t t| j| j�� � d S r: )r~ r| r} ry �pop�r �exctype�excinst�exctbr r r r! s z_RedirectStream.__exit__)r+ r, r- r} rL r r! r r r r rw q s rw c @ � e Zd ZdZdZdS )r aA Context manager for temporarily redirecting stdout to another file. # How to send help() to stderr with redirect_stdout(sys.stderr): help(dir) # How to write help() to a file with open('help.txt', 'w') as f: with redirect_stdout(f): help(pow) �stdoutN�r+ r, r- r. r} r r r r r � s r c @ r� )r zCContext manager for temporarily redirecting stderr to another file.�stderrNr� r r r r r � s r c @ rh ) r a? Context manager to suppress specified exceptions After the exception is suppressed, execution proceeds with the next statement following the with statement. with suppress(FileNotFoundError): os.remove(somefile) # Execution still resumes here if the file was already removed c G ri r: )� _exceptions)r � exceptionsr r r rL � rm zsuppress.__init__c C � d S r: r r r r r r � � zsuppress.__enter__c C s |d uo t || j�S r: )� issubclassr� r� r r r r! � s zsuppress.__exit__Nrs r r r r r � s r c @ sb e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z ddd�Z dS )�_BaseExitStackz.A base class for ExitStack and AsyncExitStack.c C � t || �S r: �r ��cm�cm_exitr r r �_create_exit_wrapper� � z#_BaseExitStack._create_exit_wrapperc � � ��fdd�}|S )Nc s �� i ��� d S r: r �r r[ �tb�r= �callbackr> r r � _exit_wrapper� s z8_BaseExitStack._create_cb_wrapper.<locals>._exit_wrapperr �r� r= r> r� r r� r �_create_cb_wrapper� � z!_BaseExitStack._create_cb_wrapperc C s t � | _d S r: )r �_exit_callbacksr r r r rL � s z_BaseExitStack.__init__c C s t | �� }| j|_t� | _|S )z@Preserve the context stack by transferring it to a new instance.)rJ r� r )r � new_stackr r r �pop_all� s z_BaseExitStack.pop_allc C sB t |�}z|j}W n ty | �|� Y |S w | �||� |S )a Registers a callback with the standard __exit__ method signature. Can suppress exceptions the same way __exit__ method can. Also accepts any object with an __exit__ method (registering a call to the method instead of the object itself). )rJ r! �AttributeError�_push_exit_callback� _push_cm_exit�r �exit�_cb_type� exit_methodr r r �push� s � z_BaseExitStack.pushc C s( t |�}|j}|�|�}| �||� |S )z�Enters the supplied context manager. If successful, also pushes its __exit__ method as a callback and returns the result of the __enter__ method. )rJ r! r r� �r r� �_cm_type�_exit�resultr r r � enter_context� s z _BaseExitStack.enter_contextc O s, | j |g|�R i |��}||_| �|� |S )z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. )r� � __wrapped__r� �r r� r= r> r� r r r r� � s z_BaseExitStack.callbackc C � | � ||�}| �|d� dS )z;Helper to correctly register callbacks to __exit__ methods.TN)r� r� �r r� r� r� r r r r� � s z _BaseExitStack._push_cm_exitTc C s | j �||f� d S r: )r� r{ )r r� �is_syncr r r r� s z"_BaseExitStack._push_exit_callbackN)T)r+ r, r- r. � staticmethodr� r� rL r� r� r� r� r� r� r r r r r� � s r� c @ rh ) r a� Context manager for dynamic management of a stack of exit callbacks. For example: with ExitStack() as stack: files = [stack.enter_context(open(fname)) for fname in filenames] # All opened files will automatically be closed at the end of # the with statement, even if attempts to open files later # in the list raise an exception. c C r r: r r r r r r r� zExitStack.__enter__c s� |d d u}t �� d � � fdd�}d}d}| jrI| j�� \}}|s$J �z ||� r/d}d}d}W n t �� }||d |d � d}|}Y | js|rbz |d j} |d � tya | |d _� w |oe|S )Nr � c �6 | j }|d u s ||u rd S |� u rn|} q|| _ d S r: �� __context__��new_exc�old_exc� exc_context�� frame_excr r �_fix_exception_context � � z2ExitStack.__exit__.<locals>._fix_exception_contextFT�NNN�r| rr r� r r� rW ) r � exc_details� received_excr� �suppressed_exc� pending_raiser� �cb�new_exc_details� fixed_ctxr r� r r! s: �� �zExitStack.__exit__c C s | � ddd� dS �z%Immediately unwind the context stack.N)r! r r r r rp F s zExitStack.closeN)r+ r, r- r. r r! rp r r r r r s 1r c @ s` e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dS )r a� Async context manager for dynamic management of a stack of exit callbacks. For example: async with AsyncExitStack() as stack: connections = [await stack.enter_async_context(get_connection()) for i in range(5)] # All opened connections will automatically be released at the # end of the async with statement, even if attempts to open a # connection later in the list raise an exception. c C r� r: r� r� r r r �_create_async_exit_wrapperY r� z)AsyncExitStack._create_async_exit_wrapperc r� )Nc � s ��� i ���I d H d S r: r r� r� r r r� _ s �z>AsyncExitStack._create_async_cb_wrapper.<locals>._exit_wrapperr r� r r� r �_create_async_cb_wrapper] r� z'AsyncExitStack._create_async_cb_wrapperc � s0 �t |�}|j}|�|�I dH }| �||� |S )z�Enters the supplied async context manager. If successful, also pushes its __aexit__ method as a callback and returns the result of the __aenter__ method. N)rJ r6 r5 �_push_async_cm_exitr� r r r �enter_async_contextc s � z"AsyncExitStack.enter_async_contextc C sD t |�}z|j}W n ty | �|d� Y |S w | �||� |S )a# Registers a coroutine function with the standard __aexit__ method signature. Can suppress exceptions the same way __aexit__ method can. Also accepts any object with an __aexit__ method (registering a call to the method instead of the object itself). F)rJ r6 r� r� r� r� r r r �push_async_exito s � z AsyncExitStack.push_async_exitc O s. | j |g|�R i |��}||_| �|d� |S )zfRegisters an arbitrary coroutine function and arguments. Cannot suppress exceptions. F)r� r� r� r� r r r �push_async_callback� s z"AsyncExitStack.push_async_callbackc � s �| � ddd�I dH dS r� )r6 r r r r rv � s �zAsyncExitStack.aclosec C r� )zLHelper to correctly register coroutine function to __aexit__ method.FN)r� r� r� r r r r� � s z"AsyncExitStack._push_async_cm_exitc � r4 r: r r r r r r5 � � �zAsyncExitStack.__aenter__c � s� �|d d u}t �� d � � fdd�}d}d}| jrR| j�� \}}z|r)||� }n||� I d H }|r8d}d}d}W n t �� } || d |d � d}| }Y | js|rkz |d j} |d � tyj | |d _� w |on|S )Nr r� c r� r: r� r� r� r r r� � r� z8AsyncExitStack.__aexit__.<locals>._fix_exception_contextFTr� r� ) r r� r� r� r� r� r� r� � cb_suppressr� r� r r� r r6 � s@ � �� �zAsyncExitStack.__aexit__N)r+ r, r- r. r� r� r� r� r� r� rv r� r5 r6 r r r r r L s r c @ s: e Zd ZdZd dd�Zdd� Zdd� Zd d � Zd d � ZdS )r aO Context manager that does no additional processing. Used as a stand-in for a normal context manager, when a particular block of code is only sometimes used with a normal context manager: cm = optional_cm if condition else nullcontext() with cm: # Perform operation, using optional_cm if condition is True Nc C ri r: �� enter_result)r r� r r r rL � rm znullcontext.__init__c C rn r: r� r r r r r � ro znullcontext.__enter__c G r� r: r �r �excinfor r r r! � r� znullcontext.__exit__c � rt r: r� r r r r r5 � ru znullcontext.__aenter__c � s �d S r: r r� r r r r6 � r� znullcontext.__aexit__r: ) r+ r, r- r. rL r r! r5 r6 r r r r r � s r )!r. r1 r| r$ � collectionsr � functoolsr �typesr r �__all__�ABCr r �objectr rF rG rN r] r r r r rw r r r r� r r r r r r r �<module> sH � @�A!! PE
Save File
Cancel