File: //usr/local/CyberCP/lib/python3.10/site-packages/starlette/__pycache__/endpoints.cpython-310.pyc
o
�h� � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZ d d lmZmZmZmZ d d
lmZ G d d � d �ZG d
d� d�ZdS )� )� annotationsN)�status)�is_async_callable)�run_in_threadpool)�
HTTPException)�Request)�PlainTextResponse�Response)�Message�Receive�Scope�Send)� WebSocketc @ s4 e Zd Zdd d
�Zdd d
�Zddd�Zddd�ZdS )� HTTPEndpoint�scoper �receiver �sendr
�return�Nonec s: |d dksJ �|� _ |� _|� _� fdd�dD �� _d S )N�type�httpc s$ g | ]}t � |�� d �d ur|�qS �N)�getattr�lower)�.0�method��self� �F/usr/local/CyberCP/lib/python3.10/site-packages/starlette/endpoints.py�
<listcomp> s
�z)HTTPEndpoint.__init__.<locals>.<listcomp>)�GET�HEAD�POST�PUT�PATCH�DELETE�OPTIONS)r r r �_allowed_methods�r r r r r r r �__init__ s
�zHTTPEndpoint.__init__�(typing.Generator[typing.Any, None, None]c C � | � � �� S r ��dispatch� __await__r r r r r/ � zHTTPEndpoint.__await__c � s� �t | j| jd�}|jdkrt| d�sdn|j�� }t| || j�}t|�}|r/||�I d H }nt ||�I d H }|| j| j| j
�I d H d S )N)r r" �head�get) r r r r �hasattrr r �method_not_allowedr r r )r �request� handler_name�handler�is_async�responser r r r. s �" zHTTPEndpoint.dispatchr5 r r c � s6 �dd� | j�i}d| jv rtd|d��tdd|d�S )N�Allowz, �appi� )� status_code�headerszMethod Not Allowed)�joinr( r r r )r r5 r= r r r r4 + s
�
z HTTPEndpoint.method_not_allowedN�r r r r r r
r r �r r+ �r r )r5 r r r )�__name__�
__module__� __qualname__r* r/ r. r4 r r r r r s
r c @ s` e Zd ZU dZded< d#d d
�Zd$dd�Zd%dd�Zd&dd�Zd'dd�Z d(d d �Z
d)d!d"�Z dS )*�WebSocketEndpointNz
str | None�encodingr r r r r r
r r c C s& |d dksJ �|| _ || _|| _d S )Nr � websocket)r r r r) r r r r* 8 s
zWebSocketEndpoint.__init__r+ c C r, r r- r r r r r/ > r0 zWebSocketEndpoint.__await__c
� s� �t | j| j| jd�}| �|�I d H tj}zSz5 |�� I d H }|d dkr9| �||�I d H }| �||�I d H n|d dkrJt |�
d�pGtj�}nqW n t y] } ztj }|�d }~ww W | �
||�I d H d S | �
||�I d H w )N)r r Tr zwebsocket.receivezwebsocket.disconnect�code)r r r r �
on_connectr �WS_1000_NORMAL_CLOSURE�decode�
on_receive�intr2 � Exception�WS_1011_INTERNAL_ERROR�
on_disconnect)r rG �
close_code�message�data�excr r r r. A s, � ����*zWebSocketEndpoint.dispatchrG r rR r
�
typing.Anyc � s �| j dkr d|vr|jtjd�I d H td��|d S | j dkr7d|vr3|jtjd�I d H td��|d S | j dkrm|�d�d urH|d }n|d �d�}zt�|�W S tj j
yl |jtjd�I d H td��w | j d u szJ d | j � ���|�d�r�|d S |d S )
N�text)rH z/Expected text websocket messages, but got bytes�bytesz/Expected bytes websocket messages, but got text�jsonzutf-8z Malformed JSON data received.z!Unsupported 'encoding' attribute ) rF �closer �WS_1003_UNSUPPORTED_DATA� RuntimeErrorr2 rK rX �loads�decoder�JSONDecodeError)r rG rR rV r r r rK V s. �
�zWebSocketEndpoint.decodec � s �|� � I dH dS )z3Override to handle an incoming websocket connectionN)�accept)r rG r r r rI r s �z WebSocketEndpoint.on_connectrS c � � �dS )z0Override to handle an incoming websocket messageNr )r rG rS r r r rL v � z WebSocketEndpoint.on_receiverQ rM c � r` )z,Override to handle a disconnecting websocketNr )r rG rQ r r r rP y ra z WebSocketEndpoint.on_disconnectr? r@ rA )rG r rR r
r rU )rG r r r )rG r rS rU r r )rG r rQ rM r r ) rB rC rD rF �__annotations__r* r/ r. rK rI rL rP r r r r rE 5 s
rE )�
__future__r rX �typing� starletter �starlette._utilsr �starlette.concurrencyr �starlette.exceptionsr �starlette.requestsr �starlette.responsesr r �starlette.typesr
r r r
�starlette.websocketsr r rE r r r r �<module> s %