�
L�g�
� � � d dl mZ d dlZd dlmZmZmZ d dlmZm Z d dl
m Z d dl m
Z
mZ d dlmZmZmZmZ d dlmZ G d � d
� � ZdS ) � )� annotationsN)�ExceptionHandlers�StatusHandlers� wrap_app_handling_exceptions)�
HTTPException�WebSocketException)�Request)�PlainTextResponse�Response)�ASGIApp�Receive�Scope�Send)� WebSocketc �< � e Zd Z d!d"d �Zd#d�Zd$d�Zd%d �Zd&d �ZdS )'�ExceptionMiddlewareNF�appr �handlers�Rtyping.Mapping[typing.Any, typing.Callable[[Request, Exception], Response]] | None�debug�bool�return�Nonec �� � || _ || _ i | _ t | j t
| j i| _ |�0|� � � D ] \ }}| � ||� � � d S d S �N)
r r �_status_handlersr �http_exceptionr �websocket_exception�_exception_handlers�items�add_exception_handler)�selfr r r �key�values �o/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/starlette/middleware/exceptions.py�__init__z ExceptionMiddleware.__init__ s� � � �����
�02�� � �4�.� �� 8�7
�� � � � &�n�n� .� .�
7�
7�
��U��*�*�3��6�6�6�6� � �
7�
7� �exc_class_or_status_code�int | type[Exception]�handler�/typing.Callable[[Request, Exception], Response]c � � t |t � � r || j |< d S t |t � � sJ �|| j |< d S r )�
isinstance�intr �
issubclass� Exceptionr )r" r( r* s r% r! z)ExceptionMiddleware.add_exception_handler# sW � �
�.�� 4� 4� I�>E�D� !�":� ;� ;� ;� � 6� �B�B� B� B�B�AH�D� $�%=� >� >� >r' �scoper �receiver
�sendr c � �( K � |d dvr | � |||� � � d {V �� d S | j | j f|d< |d dk rt |||� � }nt |||� � } t | j |� � |||� � � d {V �� d S )N�type)�http� websocketz starlette.exception_handlersr6 )r r r r r r )r" r1 r2 r3 �conns r% �__call__z ExceptionMiddleware.__call__. s� � � � � ��=� 5� 5� 5��(�(�5�'�4�0�0� 0� 0� 0� 0� 0� 0� 0� �F�
� $� � !�1
��,�-� ��=�F� "� "��5�'�4�0�0�D�D� �U�G�T�2�2�D�:�*�4�8�T�:�:�5�'�4�P�P�P�P�P�P�P�P�P�P�Pr' �requestr �excr0 r c � � t |t � � sJ �|j dv rt |j |j �� � S t |j |j |j �� � S )N> �0 �� )� status_code�headers)r- r r? r r@ r
�detail)r" r: r; s r% r z"ExceptionMiddleware.http_exception@ sY � ��#�}�-�-�-�-�-� �?�j� (� (������M�M�M� M� ���� �RU�R]�^�^�^�^r' r7 r c � � K � t |t � � sJ �|� |j |j �� � � d {V �� d S )N)�code�reason)r- r �closerC rD )r" r7 r; s r% r z'ExceptionMiddleware.websocket_exceptionF sO � � � ��#� 1�2�2�2�2�2��o�o�3�8�C�J�o�?�?�?�?�?�?�?�?�?�?�?r' )NF)r r r r r r r r )r( r) r* r+ r r )r1 r r2 r
r3 r r r )r: r r; r0 r r )r7 r r; r0 r r )�__name__�
__module__� __qualname__r& r! r9 r r � r' r% r r s� � � � � � � hl�� 7� 7� 7� 7� 7�" I� I� I� I�Q� Q� Q� Q�$_� _� _� _� @� @� @� @� @� @r' r )�
__future__r �typing� starlette._exception_handlerr r r �starlette.exceptionsr r �starlette.requestsr �starlette.responsesr
r �starlette.typesr r
r r �starlette.websocketsr r rI r' r% �<module>rR s �� "� "� "� "� "� "�
�
�
�
�� � � � � � � � � �
C� B� B� B� B� B� B� B� &� &� &� &� &� &� ;� ;� ;� ;� ;� ;� ;� ;� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� 9� *� *� *� *� *� *�7@� 7@� 7@� 7@� 7@� 7@� 7@� 7@� 7@� 7@r'