관리-도구
편집 파일: http2.cpython-38.pyc
U )�e� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl m Z ddlmZ ddlmZ ddlmZ eZe�d�ZG d d � d eje �ZG d d � d e�ZG d d� de �Zdd�dd�Zdd�dd�Z dS )� )� annotationsN)�BaseHTTPResponse� )�HTTPHeaderDict)�HTTPSConnection)�HTTPSConnectionPool�Tc @ s@ e Zd ZdZdd�dd�Zdd�dd�Zd d d d d �dd�ZdS )� _LockedObjectaY A wrapper class that hides a specific object behind a lock. The goal here is to provide a simple way to protect access to an object that cannot safely be simultaneously accessed from multiple threads. The intended use of this class is simple: take hold of it with a context manager, which returns the protected object. r )�objc C s t �� | _|| _d S �N)� threading�RLock�lock�_obj)�selfr � r �;/opt/hc_python/lib/python3.8/site-packages/urllib3/http2.py�__init__" s z_LockedObject.__init__��returnc C s | j �� | jS r )r �acquirer �r r r r � __enter__& s z_LockedObject.__enter__ztype[BaseException] | NonezBaseException | Noneztypes.TracebackType | None�None)�exc_type�exc_val�exc_tbr c C s | j �� d S r )r �release)r r r r r r r �__exit__* s z_LockedObject.__exit__N)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c s� e Zd Zd"ddddd�� fdd� Zd d �d d �Zdd �� fd d� Zd#dddddd�dd�Zdddd�dd�Zdd �dd�Zddd�dd �Z d d �d d �Z dd �� fd d!� Z � Z S )$�HTTP2ConnectionN�strz int | Nonez typing.Anyr )�host�port�kwargsr c sD | � � | _d | _g | _d|ks&d|kr.td��t� j||f|� d S )N�proxy� proxy_configz$Proxies aren't supported with HTTP/2)� _new_h2_conn�_h2_conn� _h2_stream� _h2_headers�NotImplementedError�superr )r r% r& r' �� __class__r r r 4 s zHTTP2Connection.__init__z)_LockedObject[h2.connection.H2Connection]r c C s t jjdd�}tt jj|d��S )NT)Z client_side)�config)�h2r2 ZH2Configurationr � connectionZ H2Connection)r r2 r r r r* @ s z HTTP2Connection._new_h2_connc s8 t � �� | j� }|�� | j�|�� � W 5 Q R X d S r )r/ �connectr+ Zinitiate_connection�sock�sendall� data_to_send)r �h2_connr0 r r r5 D s zHTTP2Connection.connectF�bool)�method�url� skip_host�skip_accept_encodingr c C s� | j �|}|| _|�� | _d| jkr<d| j� d| jp4d� �}n| j� d| jpLd� �}| j�dd|�� fd|�� fd|�� ff� W 5 Q R X d S ) N�:�[z]:i� )s :schemes httpss :methods :authoritys :path) r+ � _request_urlZ get_next_available_stream_idr, r% r& r- �extend�encode)r r; r< r= r> r9 � authorityr r r � putrequestK s ��zHTTP2Connection.putrequest)�header�valuesr c G s. |D ]$}| j �|�d��� |�d�f� qd S )Nzutf-8)r- �appendrC �lower)r rF rG �valuer r r � putheaderd s �zHTTP2Connection.putheaderc C sB | j �2}|j| j| jdd� |�� }r4| j�|� W 5 Q R X d S )NT)� stream_id�headers� end_stream)r+ � send_headersr, r- r8 r6 r7 )r r9 r8 r r r � endheadersj s � zHTTP2Connection.endheaders�bytes)�datar c C s |sd S t d��d S )Nz Sending data isn't supported yet)r. )r rR r r r �sendt s zHTTP2Connection.send� HTTP2Responsec C s0 d }t � }| j��}d}|s�| j�d� }r�|�|�}|D ]�}t|tjj�rNq8q8t|tjj �r�t � }|j D ]6\} } | dkr�t | � � �}qh|�| � d�| � d�� qhq8t|tjj�r�||j7 }|�|j|j� q8t|tjj�r8d}q8|�� } r| j�| � qW 5 Q R X | �� |d k �st�t||| jt|�d�S )NFi�� s :status�asciiT)�statusrM � request_urlrR ) � bytearrayr+ r6 �recvZ receive_data� isinstancer3 �eventsZ InformationalResponseReceivedZResponseReceivedr rM �int�decode�addZ DataReceivedrR Zacknowledge_received_dataZflow_controlled_lengthrL Z StreamEndedr8 r7 �close�AssertionErrorrT rA rQ ) r rV rR r9 rN Z received_datar[ �eventrM rF rJ r8 r r r � getresponsey sP � � � �zHTTP2Connection.getresponsec sp | j �@}z$|�� |�� }r*| j�|� W n tk r@ Y nX W 5 Q R X | �� | _ d | _g | _t � � � d S r ) r+ Zclose_connectionr8 r6 r7 � Exceptionr* r, r- r/ r_ )r r9 rR r0 r r r_ � s zHTTP2Connection.close)N)FF) r r r! r r* r5 rE rK rP rS rb r_ � __classcell__r r r0 r r# 3 s � � 2r# c sN e Zd Zdddddddd�� fd d � Zedd �d d ��Zdd �dd�Z� ZS )rT Fr\ r r$ rQ r: r )rV rM rW rR �decode_contentr c s( t � j||dd ||d� || _d| _d S )N� )rV rM �version�reasonre rW r )r/ r �_data�length_remaining)r rV rM rW rR re r0 r r r � s � zHTTP2Response.__init__r c C s | j S r )ri r r r r rR � s zHTTP2Response.datac C s d S r r r r r r �get_redirect_location� s z#HTTP2Response.get_redirect_location)F)r r r! r �propertyrR rk rd r r r0 r rT � s � rT r r c C s t t_t tj_dgtjj_d S )Nr3 ) r# r � ConnectionCls�urllib3r4 r �util�ssl_�ALPN_PROTOCOLSr r r r �inject_into_urllib3� s rr c C s t t_t tj_dgtjj_d S )Nzhttp/1.1) �orig_HTTPSConnectionr rm rn r4 r ro rp rq r r r r �extract_from_urllib3� s rt ) � __future__r r �types�typingZ h2.configr3 Z h2.connectionZ h2.eventsZurllib3.connectionrn Zurllib3.util.ssl_Zurllib3.responser � _collectionsr r4 r �connectionpoolr rs �TypeVarr �Genericr r# rT rr rt r r r r �<module> s(