← Back
Editing: async_generator.cpython-311.pyc
� �Ïc� � �h � U d Z ddlmZmZmZ ddlmZ ddlmZm Z m Z m Z m Z m Z mZ ddlmZ ddlmZmZ dd gZer!dd lmZ G d � d e� � Z e d e�� � Zedd de d fd�� � ZdZee d< e d� � Z G d� d� � Z efde g e e f dede e df fd�Z dS ) z& Implementation for async generators. � )�Empty�Full�Queue)�Event)� TYPE_CHECKING�AsyncGenerator� Awaitable�Callable�Iterable�TypeVar�Union� )�asynccontextmanager)�get_event_loop� run_in_executor_with_context�aclosing� generator_to_async_generator)�Protocolc �& � e Zd Zdee fd�ZdS )�_SupportsAclose�returnc � � d S )N� )�selfs �J/usr/lib/python3/dist-packages/prompt_toolkit/eventloop/async_generator.py�aclosez_SupportsAclose.aclose s � � �C� N)�__name__� __module__� __qualname__r �objectr r r r r r s4 � � � � � � �I�f� -� � � � � � r r �_SupportsAcloseT)�bound�thingr )r" Nc � K � | W V � | � � � � d{V �� dS # | � � � � d{V �� w xY w)z1Similar to `contextlib.aclosing`, in Python 3.10.N)r )r$ s r r r $ sb � � � � �� � � � ��l�l�n�n� � � � � � � � � ��e�l�l�n�n� � � � � � � � ���s �% � Ai� �DEFAULT_BUFFER_SIZE�_Tc � � e Zd ZdS )�_DoneN)r r r r r r r) r) ? s � � � � � ��Dr r) � get_iterable� buffer_sizeNc �| � ��K � d�t |�� � �t � � }d� ��fd� }t |� � } �� � � }n1# t $ r$ |� d�j � � � d{V ��}Y nw xY wt |t � � rn|W V � �c d�|� d{V �� dS # d�|� d{V �� w xY w)a] Turn a generator or iterable into an async generator. This works by running the generator in a background thread. :param get_iterable: Function that returns a generator or iterable when called. :param buffer_size: Size of the queue between the async consumer and the synchronous generator that produces items. F)�maxsizer Nc �p �� �� � D ]�} �r= �� t � � d�� � dS # t $ r �rY dS Y �8w xY w �� | d�� � nP# t $ rC �r>Y �� t � � d�� � dS # t $ r �rY dS Y �8w xY wY �fw xY w�� �� t � � d�� � dS # t $ r �rY dS Y �8w xY w# �� t � � d�� � n# t $ r �rY Y dS Y �9w xY ww xY w)z~ Consume the generator in background thread. When items are received, they'll be pushed to the queue. Tr )�timeoutN)�putr) r )�itemr* �q�quittings ���r �runnerz,generator_to_async_generator.<locals>.runnerV s �� � �$� ��� � �� � �� ���E�E�%�'�'�1�E�-�-�-� �E�� � � � � � � ��� �H� ���� � ����d�A��.�.�.� �� � !� !� !�#� #� "�F� ���E�E�%�'�'�1�E�-�-�-� �E�� � � � � � � ��� �H� ���� !��!���� � �" ���E�E�%�'�'�1�E�-�-�-� �E�� � � � � � � ��� �H� ����� ���E�E�%�'�'�1�E�-�-�-� �� � � � � � � ����� �H� ���� � � � s� �C6 �#9 � A � A �C6 �A(�'C6 �( B5�4C6 �8#B � B/�.B/�2C6 �4B5�5C6 �<#C! �! C3�2C3�6D5�9#D � D5� D1�)D5�.D5�0D1�1D5T)r N) r r r � get_nowaitr �run_in_executor�get� isinstancer) )r* r+ �loopr4 �runner_fr1 r2 r3 s ` @@r r r C s1 ���� � � � �H�!&�{�!;�!;�!;�A� � � �D� � � � � � � � �D ,�F�3�3�H�� � ?��|�|�~�~����� ?� ?� ?� !� 1� 1�$��� >� >�>�>�>�>�>�>���� ?�����$��&�&� ��� � � � � � � �� ����������� �� �����������s/ �B/ � A �B/ �+B� B/ �B� B/ �/ B;) �__doc__�queuer r r � threadingr �typingr r r r r r r �async_context_managerr �utilsr r �__all__�typing_extensionsr r r" r r& �int�__annotations__r' r) r r r r �<module>rE s ��� � � %� $� $� $� $� $� $� $� $� $� � � � � � �� � � � � � � � � � � � � � � � � � 7� 6� 6� 6� 6� 6� ?� ?� ?� ?� ?� ?� ?� ?� �"� �� � J�*�*�*�*�*�*�� � � � �(� � � � �w� 1� �I�I�I�� � � � ��,�-� � � � �� �* � �S� � � � �W�T�]�]�� � � � � � � � � +�H� H��2�x��|�+�,�H��H� �B��H� �H� H� H� H� H� Hr
Save File
Cancel