� c��j�k��"�dZddlZddlZddlZddlZddlmZddlmZddl m Z m Z gd�Z Gd�dej�ZGd �d ej�ZGd �d e�ZGd �de�ZGd�d�ZGd�deee�ZGd�deee�Zd�Zd�ZGd�de�ZGd�de�ZGd�de�ZGd�de�ZGd�d e�ZGd!�d"e�ZGd#�d$�ZGd%�d&ee�ZGd'�d(ee�Z Gd)�d*ee�Z!Gd+�d,e�Z"y)-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�aclosing�chdirc�Z�eZdZdZee�Zd�Zejd��Z ed��Z y)r z,An abstract base class for context managers.c��|S�z0Return `self` upon entering the runtime context.���selfs �#/usr/lib64/python3.12/contextlib.py� __enter__z AbstractContextManager.__enter__s��� �c��y�z9Raise any exception triggered within the runtime context.Nr�r�exc_type� exc_value� tracebacks r�__exit__zAbstractContextManager.__exit__s��rc�L�|turtj|dd�StS)Nrr%)r �_collections_abc�_check_methods�NotImplemented��cls�Cs r�__subclasshook__z'AbstractContextManager.__subclasshook__ s&�� �(� (�#�2�2�1�k�:�N� N��rN) �__name__� __module__� __qualname__�__doc__� classmethodr�__class_getitem__r�abc�abstractmethodr%r-rrrr r sC��6�#�L�1��� ���������rr c�Z�eZdZdZee�Zd�Zejd��Z ed��Z y)r z9An abstract base class for asynchronous context managers.c��K�|S�wrrrs r� __aenter__z&AbstractAsyncContextManager.__aenter__-s ����� ���c�� K�y�wr rr!s r� __aexit__z%AbstractAsyncContextManager.__aexit__1s �������c�L�|turtj|dd�StS)Nr8r;)r r'r(r)r*s rr-z,AbstractAsyncContextManager.__subclasshook__6s-�� �-� -�#�2�2�1�l�3>�@� @��rN) r.r/r0r1r2rr3r8r4r5r;r-rrrr r 'sC��C�#�L�1��� ���������rr c��eZdZdZd�Zd�Zy)rzJA base class or mixin that enables context managers to work as decorators.c��|S)a6Return 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. rrs r� _recreate_cmzContextDecorator._recreate_cmAs ��� rc�2���t����fd��}|S)Nc�`���j�5�|i|��cddd�S#1swYyxYw�N�r@��args�kwds�funcrs ��r�innerz(ContextDecorator.__call__..innerNs(����"�"�$��T�*�T�*�%�$�$�s�$�-r�rrHrIs`` r�__call__zContextDecorator.__call__Ms ��� �t�� +� � +�� rN�r.r/r0r1r@rKrrrrr>s��P� �rrc��eZdZdZd�Zd�Zy)�AsyncContextDecoratorzPA base class or mixin that enables async context managers to work as decorators.c��|S)z-Return a recreated instance of self. rrs rr@z"AsyncContextDecorator._recreate_cmXs ��� rc�2���t����fd��}|S)Nc���K��j�4�d{����|i|���d{���cddd��d{���S7�%7�7� #1�d{���7swYyxYw�wrCrDrEs ��rrIz-AsyncContextDecorator.__call__..inner^sE������(�(�*�*�!�4�0�4�0�0�+�*�*��0��+��*�*�*�sP�A�>�A� A�A�A� A�A�A�A�A�A� A � A�ArrJs`` rrKzAsyncContextDecorator.__call__]s ��� �t�� 1� � 1�� rNrLrrrrNrNUs��V�� rrNc��eZdZdZd�Zd�Zy)�_GeneratorContextManagerBasezBShared functionality for @contextmanager and @asynccontextmanager.c��||i|��|_|||c|_|_|_t |dd�}|�t |�j }||_y)Nr1)�genrHrFrG�getattr�typer1)rrHrFrG�docs r�__init__z%_GeneratorContextManagerBase.__init__hsR����&��&���*.��d�'�� �4�9�d�i��d�I�t�,�� �;��t�*�$�$�C��� rc�d�|j|j|j|j�SrC)� __class__rHrFrGrs rr@z)_GeneratorContextManagerBase._recreate_cmvs#���~�~�d�i�i����D�I�I�>�>rN)r.r/r0r1rYr@rrrrSrSes��L��?rrSc��eZdZdZd�Zd�Zy)�_GeneratorContextManagerz%Helper for @contextmanager decorator.c�l�|`|`|` t|j�S#t $r t d�d�wxYw�Nzgenerator didn't yield)rFrGrH�nextrU� StopIteration� RuntimeErrorrs rrz"_GeneratorContextManager.__enter__�sD�� �I�t�y�$�)� C�����>� !��� C��7�8�d� B� C�s��3c�B�|�" t|j� td��|�|�} |jj |� td��#|jj�wxYw#t$rYywxYw#t$r}||ucYd}~Sd}~wt$r@}||ur ||_Yd}~yt|t�r|j|ur ||_Yd}~y�d}~wt$r}||ur�||_Yd}~yd}~wwxYw#|jj�wxYw)N�generator didn't stopFz#generator didn't stop after throw()) r`rUrb�closera�throw� __traceback__� isinstance� __cause__� BaseException�r�typ�valuer$�excs rr%z!_GeneratorContextManager.__exit__�s�� �;� %��T�X�X��%�&�'>�?�?��}����# ������u�%�F !�"�#H�I�I��W�H�H�N�N�$�� !� �� ��!� (��%�'�'��� ��%�<�(1�C�%� ��u�m�4�� � ��.�*3�E�'� ��� � ��e�#��$-��!��� ������� �si�A5� A�B� D�A2�5 B�B� C?� B�C?� C?�" C�2%C�C� C?�) C:�:C?�DN)r.r/r0r1rr%rrrr]r]}s�� 0�C�7!rr]c��eZdZdZd�Zd�Zy)�_AsyncGeneratorContextManagerz*Helper for @asynccontextmanager decorator.c��K�|`|`|` t|j��d{���S7�#t $r t d�d�wxYw�wr_)rFrGrH�anextrU�StopAsyncIterationrbrs rr8z(_AsyncGeneratorContextManager.__aenter__�sM���� �I�t�y�$�)� C��t�x�x��(� (�(��!� C��7�8�d� B� C�s#�A�)�'�)�A�)�?�Ac��K�|�* t|j��d{��� td��|�|�} |jj |��d{��� td��7�I#|jj��d{���7wxYw#t$rYywxYw7�I#t$r}||ucYd}~Sd}~wt$rF}||ur ||_Yd}~yt|ttf�r|j|ur ||_Yd}~y�d}~wt$r}||ur�||_Yd}~yd}~wwxYw#|jj��d{���7wxYw�w)NrdFz$generator didn't stop after athrow()) rrrUrb�aclosers�athrowrgrhrarirjrks rr;z'_AsyncGeneratorContextManager.__aexit__�s@���� �;� ,��D�H�H�o�%�%�,�&�'>�?�?��}����# ��h�h�o�o�e�,�,�,�F (�"�#I�J�J�e&���(�(�/�/�+�+�+�� &� �� ��-��%� (��%�'�'��� ��%�<�(1�C�%� ��u�}�6H�&I�J�� � ��.�*3�E�'� ��� � ��e�#��$-��!��� ���h�h�o�o�'�'�'�s��E�B�A(�B� A*� E�B#�B!�B#� D'�(B�*B�B � B�E� B�E�B�E�!B#�# D$�,B5�/D$�0E�5 D$� D� E�+D�<E�D� D$� D�E�D$�$E�'E �E�E � EN)r.r/r0r1r8r;rrrrprp�s�� 5�C�7(rrpc�.��t���fd��}|S)a�@contextmanager decorator. Typical usage: @contextmanager def some_generator(): try: yield finally: This makes this: with some_generator() as : equivalent to this: try: = finally: c���t�||�SrC)r]�rFrGrHs �r�helperzcontextmanager..helper+s���'��d�D�9�9rr�rHrzs` rr r s ���6 �4�[�:��:� �Mrc�.��t���fd��}|S)a�@asynccontextmanager decorator. Typical usage: @asynccontextmanager async def some_async_generator(): try: yield finally: This makes this: async with some_async_generator() as : equivalent to this: try: = finally: c���t�||�SrC)rprys �rrzz#asynccontextmanager..helperLs���,�T�4��>�>rrr{s` rrr1s ���6 �4�[�?��?� �Mrc�"�eZdZdZd�Zd�Zd�Zy)r a2Context to automatically close something at the end of a block. Code like this: with closing(.open()) as f: is equivalent to this: f = .open() try: finally: f.close() c��||_yrC��thing�rr�s rrYzclosing.__init__c� ���� rc��|jSrCr�rs rrzclosing.__enter__es ���z�z�rc�8�|jj�yrC)r�re�r�exc_infos rr%zclosing.__exit__gs�� � � ���rN�r.r/r0r1rYrr%rrrr r Rs��� ��rr c�"�eZdZdZd�Zd�Zd�Zy)ra�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(.fetch()) as agen: is equivalent to this: agen = .fetch() try: finally: await agen.aclose() c��||_yrCr�r�s rrYzaclosing.__init__}r�rc��"K�|jS�wrCr�rs rr8zaclosing.__aenter__s�����z�z���� c��TK�|jj��d{���y7��wrC)r�rur�s rr;zaclosing.__aexit__�s�����j�j���!�!�!�s �(�&�(N)r.r/r0r1rYr8r;rrrrrks���"��"rrc�"�eZdZdZd�Zd�Zd�Zy)�_RedirectStreamNc� �||_g|_yrC)� _new_target� _old_targets)r� new_targets rrYz_RedirectStream.__init__�s��%�����rc���|jjtt|j��t t|j|j �|j SrC)r��appendrV�sys�_stream�setattrr�rs rrz_RedirectStream.__enter__�sC�� ��� � ���d�l�l�!;�<���T�\�\�4�#3�#3�4����rc�j�tt|j|jj ��yrC)r�r�r�r��pop)r�exctype�excinst�exctbs rr%z_RedirectStream.__exit__�s!����T�\�\�4�#4�#4�#8�#8�#:�;r)r.r/r0r�rYrr%rrrr�r��s���G��  � ._exit_wrapper�s��� �d� #�d� #rr�r�rFrGr�s``` r�_create_cb_wrapperz!_BaseExitStack._create_cb_wrapper�s��� $��rc�"�t�|_yrC)r�_exit_callbacksrs rrYz_BaseExitStack.__init__�s ��$�w��rc�f�t|��}|j|_t�|_|S)z@Preserve the context stack by transferring it to a new instance.)rWr�r)r� new_stacks r�pop_allz_BaseExitStack.pop_all�s-���D��J�L� �$(�$8�$8� �!�$�w����rc��t|�} |j}|j||�|S#t$r|j |�Y|SwxYw)aRegisters 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). )rWr%� _push_cm_exit�AttributeError�_push_exit_callback�r�exit�_cb_type� exit_methods r�pushz_BaseExitStack.push�sZ����:�� 2�"�+�+�K� � � �t�[� 1�� �� � +� � $� $�T� *�� �  +�s� -�A � A c���t|�} |j}|j}||�}|j||�|S#t$r(t d|j �d|j �d��d�wxYw)z�Enters the supplied context manager. If successful, also pushes its __exit__ method as a callback and returns the result of the __enter__ method. �'�.z6' object does not support the context manager protocolN)rWrr%r�� TypeErrorr/r0r��rr�r+�_enter�_exit�results r� enter_contextz_BaseExitStack.enter_context�s����2�h�� S��]�]�F��L�L�E����� ���2�u�%�� �� � S��a����/�q��1A�1A�0B�CG�H�I�NR� S� S�s �A�1A2c�b�|j|g|��i|��}||_|j|�|S)z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. )r�� __wrapped__r��rr�rFrGr�s rr�z_BaseExitStack.callbacks=�� 0��/�/��H�4�H�4�H� �%-� �!� � � ��/��rc�L�|j||�}|j|d�y)z;Helper to correctly register callbacks to __exit__ methods.TN)r�r��rr�r�r�s rr�z_BaseExitStack._push_cm_exits$���1�1�"�g�>� � � � ���5rc�>�|jj||f�yrC)r�r�)rr��is_syncs rr�z"_BaseExitStack._push_exit_callback$s�� ���#�#�W�h�$7�8rN)T)r.r/r0r1� staticmethodr�r�rYr�r�r�r�r�r�rrrr�r��sN��8��'��'����� '���(�& �6� 9rr�c�"�eZdZdZd�Zd�Zd�Zy)ra�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��|SrCrrs rrzExitStack.__enter__4s��� rc�� �|ddu}tj�d� � fd�}d}d}|jr6|jj�\}} ||�rd}d}d}|jr�6|r |dj} |d�|xr|S#tj�}||d|d�d}|}Y�UxYw#t $r  |d_�wxYw)Nr�c�L�� |j}|�||ury|�ur ||_y|}�#rC�� __context__��new_exc�old_exc� exc_context� frame_excs �r�_fix_exception_contextz2ExitStack.__exit__.._fix_exception_context=�B����%�1�1� ��&�+��*@���)�+��#*�G� �&��rFT�NNN�r�r�r�r�r�rj) r� exc_details� received_excr��suppressed_exc� pending_raiser��cb�new_exc_details� fixed_ctxr�s @rr%zExitStack.__exit__7s����"�1�~�T�1� ��L�L�N�1�%� � *� ��� ��"�"��.�.�2�2�4�K�G�R� .��{�#�%)�N�$)�M�"4�K��"�"� � �(��N�6�6� �!�!�n�$��.��.�� .�"%�,�,�.��&��q�'9�;�q�>�J� $� �-� ��!� �-6� �A��*�� �s� B�.B6�)B3�6C c�*�|jddd�y�z%Immediately unwind the context stack.N)r%rs rrezExitStack.closehs�� � � �d�D�$�'rN)r.r/r0r1rr%rerrrrr)s����//�b(rrc�Z�eZdZdZed��Zed��Zd�Zd�Zd�Z d�Z d�Z d �Z d �Z y ) ra�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��t||�SrCr�r�s r�_create_async_exit_wrapperz)AsyncExitStack._create_async_exit_wrapper{r�rc��������fd�}|S)Nc��2�K���i����d{���y7��wrCrr�s ���rr�z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapper�s������D�)�D�)� )� )�s � ��rr�s``` r�_create_async_cb_wrapperz'AsyncExitStack._create_async_cb_wrappers��� *��rc��K�t|�} |j}|j}||��d{���}|j||�|S#t$r(t d|j �d|j �d��d�wxYw7�L�w)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. r�r�zC' object does not support the asynchronous context manager protocolN)rWr8r;r�r�r/r0�_push_async_cm_exitr�s r�enter_async_contextz"AsyncExitStack.enter_async_context�s����� �2�h�� '��^�^�F��M�M�E� �b�z�!�� � � ��U�+�� �� � '��a����/�q��1A�1A�0B�CT�U��"&� '� '��"�s%� B�A � B�A?�B� 1A<�<Bc��t|�} |j}|j||�|S#t$r|j |d�Y|SwxYw)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)rWr;r�r�r�r�s r�push_async_exitzAsyncExitStack.push_async_exit�s\����:�� 8�"�,�,�K� � $� $�T�;� 7�� �� � 2� � $� $�T�5� 1�� �  2�s� -�A � A c�d�|j|g|��i|��}||_|j|d�|S)zfRegisters an arbitrary coroutine function and arguments. Cannot suppress exceptions. F)r�r�r�r�s r�push_async_callbackz"AsyncExitStack.push_async_callback�s?�� 6��5�5�h�N��N��N� �%-� �!� � � ���6��rc��FK�|jddd��d{���y7��wr�)r;rs rruzAsyncExitStack.aclose�s�����n�n�T�4��.�.�.�s �!��!c�L�|j||�}|j|d�y)zLHelper to correctly register coroutine function to __aexit__ method.FN)r�r�r�s rr�z"AsyncExitStack._push_async_cm_exit�s&���7�7��G�D� � � � ���6rc��K�|S�wrCrrs rr8zAsyncExitStack.__aenter__�s ����� �r9c���� K�|ddu}tj�d� � fd�}d}d}|jrH|jj�\}} |r||�}n ||��d{���}|rd}d}d}|jr�H|r |dj} |d�|xr|S7�6#tj�} || d|d�d}| }Y�WxYw#t $r  |d_�wxYw�w)Nrr�c�L�� |j}|�||ury|�ur ||_y|}�#rCr�r�s �rr�z8AsyncExitStack.__aexit__.._fix_exception_context�r�rFTr�r�) rr�r�r�r�r�r�r�� cb_suppressr�r�r�s @rr;zAsyncExitStack.__aexit__�s�����"�1�~�T�1� ��L�L�N�1�%� � *� ��� ��"�"��.�.�2�2�4�K�G�R� .��"$�k�"2�K�(*�K�(8�"8�K��%)�N�$)�M�"4�K��"�"�$ � �(��N�6�6� �!�!�n�$��.��.�+#9��  .�"%�,�,�.��&��q�'9�;�q�>�J� $� �-� ��!� �-6� �A��*�� �sN�AC$�B�%B�& B�2 C$�?C$�C �C$�B�)C �C$� C!�!C$N)r.r/r0r1r�r�r�rrrrur�r8r;rrrrrnsR�� ��'��'����� �$�$ �/�7� �3/rrc�0�eZdZdZdd�Zd�Zd�Zd�Zd�Zy) r aOContext 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��||_yrC�� enter_result)rrs rrYznullcontext.__init__s ��(��rc��|jSrCr rs rrznullcontext.__enter__s��� � � rc��yrCr�r�excinfos rr%znullcontext.__exit__ r�rc��"K�|jS�wrCr rs rr8znullcontext.__aenter__ s����� � � �r�c�� K�y�wrCrrs rr;znullcontext.__aexit__s ���� �r<rC) r.r/r0r1rYrr%r8r;rrrr r �s ���)�!� �!� rr c�"�eZdZdZd�Zd�Zd�Zy)rzHNon thread-safe context manager to change the current working directory.c� �||_g|_yrC)�path�_old_cwd)rrs rrYzchdir.__init__s���� ��� rc��|jjtj��tj|j �yrC)rr��os�getcwdrrrs rrzchdir.__enter__s*�� � � ���R�Y�Y�[�)� ������rc�^�tj|jj��yrC)rrrr�rs rr%zchdir.__exit__s�� ������"�"�$�%rNr�rrrrrs��R���&rr)#r1r4rr�r'� collectionsr� functoolsr�typesrr�__all__�ABCr r �objectrrNrSr]rpr rr rr�rrrr�rrr rrrr�r#s9��:� � � ����*� ���S�W�W��,�#�'�'��.�v��. �F� � ?�?�0G!� ���G!�RG(� ���G(�T�B�B�$��2"�*�"�4<�,�<�$ �o� � �o�� $�%�$�NQ9�Q9�jA(�� 6�A(�JH/�^�%@�H/�V �(�*E� �6 &� "� &r