� E�g'� ���ddlZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZddlmZmZdd lmZdd lmZmZmZGd �d eeeeef��ZdS) �N)�suppress)�Any�Optional�Tuple�)� BaseProtocol)� ClientOSError�ClientPayloadError�ServerDisconnectedError�SocketTimeoutError)� _EXC_SENTINEL�EMPTY_BODY_STATUS_CODES�BaseTimerContext� set_exception)�HttpResponseParser�RawResponseMessage)�HttpProcessingError)� EMPTY_PAYLOAD� DataQueue� StreamReaderc����eZdZdZdejddfd�Zedefd���Z edefd���Z d-d�Z d-d �Z defd �Z d eeddf�fd � Zd-d �Zd-�fd� Zd-�fd� Zefd ededdf�fd� Zdededdfd�Zdddddddddd� deededededeed ed!ed"ed#eddfd$�Zd-d%�Zd-d&�Zd-d'�Zedeefd(���Zej deeddfd)���Zd-d*�Z!d+e"ddfd,�Z#�xZ$S).�ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.�loop�returnNc���tj||���tj||��d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ dS)N)rF��)r�__init__r� _should_close�_payload� _skip_payload�_payload_parser�_timer�_tail� _upgraded�_parser� _read_timeout�_read_timeout_handle�_timeout_ceil_threshold)�selfrs �d/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/aiohttp/client_proto.pyrzResponseHandler.__init__s�����d��.�.�.�.���4��&�&�&�"���04�� �"���#����� ��� ����59�� �.2���CG��!�89��$�$�$rc��|jS�N)r%�r*s r+�upgradedzResponseHandler.upgraded/s ���~�rc��t|jpI|jduo|j��� p&|jp|jdup|jdup |jp|j��Sr-) �boolrr �is_eofr%� _exceptionr"�_bufferr$r.s r+� should_closezResponseHandler.should_close3s���� � � �� �T�)�H�$�-�2F�2F�2H�2H�.H� ��~� ���d�*� ��#�4�/�  � �|�  � �z� � � rc��d|_dS�NT)rr.s r+� force_closezResponseHandler.force_close?s��!����rc��|j}|�8|���d|_d|_|���dSdSr-)� transport�closer � _drop_timeout)r*r:s r+r;zResponseHandler.closeBsM���N� � � � �O�O� � � �!�D�N� �D�M� � � � � � � � � !� rc�H�|jduo|j��� Sr-)r:� is_closingr.s r+� is_connectedzResponseHandler.is_connectedJs%���~�T�)�M�$�.�2K�2K�2M�2M�.M�Mr�excc�(��|���|}|}|du}|j�Ett��5|j���ddd��n #1swxYwYd}|j�j |j���}nO#t$rB}|j�1d|��}|s|�d|��}t|jt|��|��Yd}~nd}~wwxYw|� ��s]t|t��rt|j �}|rt|��}|rtn|}|�J�|�J�|�||��d|_d|_d|_d|_d|_t%���|��dS)Nz#Response payload is not completed: z. TF)r<r"r� Exception�feed_eofr&r rr r2� isinstance�OSErrorr �argsr r r�_reading_paused�super�connection_lost) r*r@�original_connection_error� reraised_exc�connection_closed_cleanly� uncompleted�underlying_exc�client_payload_exc_msg�underlying_non_eof_exc� __class__s �r+rIzResponseHandler.connection_lostMs8��� ������$'�!�0� �$=��$E�!� � � +��)�$�$� 0� 0��$�-�-�/�/�/� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0���� 0� 0� 0� 0�� � �<� #� �"�l�3�3�5�5� � ��� � � ��=�,�Q�n�Q�Q�+�5��5�>�>�8�>�>�/�"�� �*�+A�B�B�&������������� ���� �{�{�}�}� E��3�W�=�=� N�,�.G�.L�M� �(� D�6�{�C�C� � -�/� � �.� #� *�5�5�5��+�+�+� � � �|�-C� D� D� D�!����� ��� �#���$��� ����� �-�-�-�-�-s)�A�A#�&A#�4B� C�8C�Cc�.�|���dSr-)r<r.s r+� eof_receivedzResponseHandler.eof_received�s�� �������rc�p��t�����|���dSr-)rH� pause_readingr<�r*rQs �r+rUzResponseHandler.pause_reading�s1��� �������� �������rc�p��t�����|���dSr-)rH�resume_reading�_reschedule_timeoutrVs �r+rXzResponseHandler.resume_reading�s1��� ����� � � � � � �"�"�"�"�"r� exc_causec���d|_|���t���||��dSr7)rr<rHr)r*r@rZrQs �r+rzResponseHandler.set_exception�s?��� "��� ������ �����c�9�-�-�-�-�-r�parser�payloadc��||_||_|���|jr&|jdc}|_|�|��dSdS)Nr)r r"r<r$� data_received)r*r\r]�datas r+� set_parserzResponseHandler.set_parser�sc��  �� �%��� ������ �:� %�#�z�3� �D�$�*� � � �t� $� $� $� $� $� %� %rFTiri�) �timer� skip_payload�read_until_eof�auto_decompress� read_timeout� read_bufsize�timeout_ceil_threshold� max_line_size�max_field_sizerbrcrdrerfrgrhrirjc ���||_||_||_t||j||t | |||| �� � |_|jr&|jdc} |_|�| ��dSdS)N)rb�payload_exception�response_with_bodyrdrerirjr) r!r'r)r�_loopr r&r$r_) r*rbrcrdrerfrgrhrirjr`s r+�set_response_paramsz#ResponseHandler.set_response_params�s���*���)���'=��$�)� � �J� ��0�#/�/�)�+�'�)�  �  �  �� � �:� %�#�z�3� �D�$�*� � � �t� $� $� $� $� $� %� %rc�X�|j�"|j���d|_dSdSr-)r(�cancelr.s r+r<zResponseHandler._drop_timeout�s6�� � $� 0� � %� ,� ,� .� .� .�(,�D� %� %� %� 1� 0rc��|j}|j�|j���|r'|j�||j��|_dSd|_dSr-)r'r(rqrn� call_later�_on_read_timeout)r*�timeouts r+rYz#ResponseHandler._reschedule_timeout�sh���$�� � $� 0� � %� ,� ,� .� .� .� � -�(,� �(=�(=���.�)�)�D� %� %� %�)-�D� %� %� %rc�.�|���dSr-)rYr.s r+� start_timeoutzResponseHandler.start_timeout�s�� � � �"�"�"�"�"rc��|jSr-�r'r.s r+rfzResponseHandler.read_timeout�s ���!�!rc��||_dSr-ry)r*rfs r+rfzResponseHandler.read_timeout�s��)����rc��td��}|�|��|j�t|j|��dSdS)Nz#Timeout on reading data from socket)r rr )r*r@s r+rtz ResponseHandler._on_read_timeout�sL�� �!F�G�G�� ���3���� �=� $� �$�-�� -� -� -� -� -� %� $rr`c��|���|sdS|j�F|j�|��\}}|r%d|_d|_|r|�|��dS|js|j�|xj|z c_dS |j�|��\}}}n�#t$r�}|j �|j � ��t|t��r"t|j |j|j���}nt��}|�||��Yd}~dSd}~wwxYw||_d}|D]e\} }| jrd|_||_|js| j t(vr|�| t*fd���M|�| |fd���f|�8|t*ur|�|j��n|���|r|r|�|��dSdSdS)N)�code�message�headersTr)rYr"� feed_datar r_r%r&r$� BaseExceptionr:r;rDrr}r~rrr5rr!rr�on_eofr<) r*r`�eof�tail�messagesr/rNr@r]r~s r+r_zResponseHandler.data_received�su�� � � �"�"�"�� � �F� � � +��,�6�6�t�<�<�I�C��� -� $�� �'+��$��-��&�&�t�,�,�,� �F� �>� �T�\�1� �J�J�$� �J�J� �F� �'+�|�'=�'=�d�'C�'C� $�H�h����� � � ��~�)���$�$�&�&�&��.�*=�>�>� ,�)�'�,�*�2�*�2������ *�+�+�� � � �s�N� 3� 3� 3� �F�F�F�F�F�����! ����$"���*.�� (� 6� 6� �G�W��#� *�%)��"�#�D�M��!� 6�W�\�5L�%L�%L������7��;�;�;�;������1�1�5�5�5�5� � � �m�+�+����t�1�2�2�2�2��"�"�$�$�$� � %�� %� � � �t� $� $� $� $� $� %� %� %� %s�B&�& D6�0A;D1�1D6)rN)%�__name__� __module__� __qualname__�__doc__�asyncio�AbstractEventLoopr�propertyr1r/r5r8r;r?rr�rIrSrUrXr rrrar�float�intror<rYrwrf�setterrt�bytesr_� __classcell__)rQs@r+rrsV�������B�B�:�W�6�:�4�:�:�:�:�*��$�����X���  �d�  �  �  ��X�  �"�"�"�"�!�!�!�!�N�d�N�N�N�N�6.�8�M�#:�6.�t�6.�6.�6.�6.�6.�6.�p����������#�#�#�#�#�#�$1�.�.� �.�!�.� � .�.�.�.�.�.� %�� %�s� %�t� %� %� %� %�$-1�"�$� $�(,�!�()�!�"�"%�"%�"%��(�)�"%�� "%� � "%� � "%��u�o�"%��"%�!&�"%��"%��"%� �"%�"%�"%�"%�H-�-�-�-� -� -� -� -�#�#�#�#��"�h�u�o�"�"�"��X�"���*��%��*�T�*�*�*���*�.�.�.�.� D%�%�D%�D�D%�D%�D%�D%�D%�D%�D%�D%rr)r�� contextlibr�typingrrr� base_protocolr�client_exceptionsr r r r �helpersr rrr�httprr�http_exceptionsr�streamsrrrr�rr+�<module>r�sb������������'�'�'�'�'�'�'�'�'�'�'�'�'�'�'�'������������� ������������ 9�8�8�8�8�8�8�8�0�0�0�0�0�0�;�;�;�;�;�;�;�;�;�;�\%�\%�\%�\%�\%�l�I�e�4F� �4T�.U�$V�\%�\%�\%�\%�\%r
Memory