Current File : //lib/python3.6/site-packages/rhn/__pycache__/connections.cpython-36.pyc
3
�U�f�% � @ s& d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z y(d dl
Z
d dl
m Z m Z m
Z
d dlZW n< ek
r� d dljZ
d dlm Z m Z m
Z
d dljZY nX G dd� de
j�ZG d d
� d
e
j�ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�Zdd� Zdd� ZdS )� N)�python_version)�SSL)� nonblocking)�i18n)� _CS_REQ_SENT�_CS_IDLE�ResponseNotReadyc @ s e Zd Zdd� ZdS )� HTTPResponsec C s4 t | jtj�s tj| j�| _| jj|||||� d S )N)�
isinstance�fpr ZNonBlockingFile� set_callback)�self�rs�ws�ex� user_data�callback� r �!/usr/lib/python3.6/connections.pyr s zHTTPResponse.set_callbackN)�__name__�
__module__� __qualname__r r r r r r s r c @ s@ e Zd ZeZdejfdd�Zdd� Zdd� Z dd � Z
d
d � Z dS ) �HTTPConnectionNc C s^ t � dkr tjj| |||d� ntjj| ||� g | _g | _g | _d | _d | _d| _ || _
d S )Nz2.6.1)�timeoutz#rhn.connections $Revision$ (python)) r �httplibr �__init__�_cb_rs�_cb_ws�_cb_ex�
_cb_user_data� _cb_callback� _user_agentr )r
�host�portr r r r r ( s
zHTTPConnection.__init__c C s" || _ || _|| _|| _|| _d S )N)r r r r r )r
r r r r r r r r r 5 s
zHTTPConnection.set_callbackc C s
|| _ d S )N)r! )r
Z
user_agentr r r �set_user_agent= s z HTTPConnection.set_user_agentc C s� | j r| j j� rd| _ | jtks&| j r,t� �| jdkrH| j| j| j�}n | j| j�}| jrv|j | j
| j | j | j
| j� |j� |jtjks�t�t| _|jr�| j� n|| _ |S )z!Get the response from the server.Nr )Z_HTTPConnection__responseZisclosedZ_HTTPConnection__stater r Z
debuglevel�response_class�sockr r r r r r �begin�
will_closer Z_UNKNOWN�AssertionErrorr �close)r
�responser r r � getresponseB s"
zHTTPConnection.getresponsec C s t jj| � | jj| j� d S )N)r r �connectr& �
settimeoutr )r
r r r r- s s zHTTPConnection.connect) r r r r r% r �DEFAULT_TIMEOUTr r r$ r, r- r r r r r % s
1r c @ sB e Zd Zdddejfdd�Zdd� Zd
dd�Zd d
� Zd d � Z dS )�HTTPProxyConnectionNc C sP t j| ||d� | j| j | _| _| j||� | j| j | _| _|| _ || _
d S )N)r ) r r r" r# �_HTTPProxyConnection__proxy� _HTTPProxyConnection__proxy_port�
_set_hostport�_host�_port� _HTTPProxyConnection__username� _HTTPProxyConnection__password)r
�proxyr" r# �username�passwordr r r r r y s z HTTPProxyConnection.__init__c C s. | j | j| j� tj| � | j | j| j� d S )N)r3 r1 r2 r r- r4 r5 )r
r r r r- � s
zHTTPProxyConnection.connectr c C sN | j }| j| jkr$|d t| j� }d||f }tj| |||d� | j� d S )N�:z http://%s%s)� skip_host)r4 r5 � default_port�strr �
putrequest�_add_proxy_headers)r
�method�urlr< �hostnameZnewurlr r r r? � s z HTTPProxyConnection.putrequestc C sX | j s
d S d| j | jf }tjtj|��jtjd�tjd��}| jddtj|� � d S )Nz%s:%s�
� zProxy-AuthorizationzBasic %s) r6 r7 �base64Z encodestringr �bstr�replace� putheaderZsstr)r
ZuserpassZ enc_userpassr r r r@ � s
$z&HTTPProxyConnection._add_proxy_headersc C s | j ||�\| _| _d S )N)Z
_get_hostportr" r# )r
r" r# r r r r3 � s z!HTTPProxyConnection._set_hostport)r )
r r r r r/ r r- r? r@ r3 r r r r r0 x s
r0 c @ s2 e Zd ZeZejjZddej fdd�Z
dd� Z dS )�HTTPSConnectionNc C s$ t j| |||d� |pg }|| _d S )N)r )r r �
trusted_certs)r
r" r# rK r r r r r � s zHTTPSConnection.__init__c C s� t j| j| jt jt j�}x�|D ]�}|\}}}}}yt j |||�}W n t jk
r^ d}w Y nX y"|j| j| jf� |j| j � W n$ t jk
r� |j
� d}w Y nX P q W |dkr�t jd��t j || j
�| _| jj� dS )z'Connect to a host on a given (SSL) portNz0Unable to connect to the host and port specified)�socketZ getaddrinfor" r# Z AF_UNSPECZ SOCK_STREAM�errorr- r. r r* r � SSLSocketrK r& �init_ssl) r
�results�rZafZsocktype�protoZ canonnameZsar& r r r r- � s*
zHTTPSConnection.connect) r r r r r% r rJ r= r r/ r r- r r r r rJ � s
rJ c @ s e Zd Zdd� ZdS )�HTTPSProxyResponsec C s t j| � d| _d S )Nr )r r' r( )r
r r r r' � s
zHTTPSProxyResponse.beginN)r r r r' r r r r rS � s rS c @ sB e Zd ZejZddddejfdd�Zdd� Zd dd�Z d d
� Z
dS ) �HTTPSProxyConnectionNc C s* t j| ||||||d� |p g }|| _d S )N)r )r0 r rK )r
r8 r" r# r9 r: rK r r r r r � s
z HTTPSProxyConnection.__init__c C s� t j| � d| j| jf }tj| d|� | j� tj| � | j}t | _tj
| �}|| _|j � |j dkr�| j � t
j||j |j|j��tj| j| j�| _| jj� d S )Nz%s:%sZCONNECT�� )r0 r- r4 r5 r r? r@ Z
endheadersr% rS r, r* Zstatus� xmlrpclibZ
ProtocolError�reason�msgr rN r& rK rO )r
r" r% r+ r r r r- � s
z HTTPSProxyConnection.connectr c C s t j| |||d�S )N)r< )r r? )r
rA rB r< r r r r? � s z HTTPSProxyConnection.putrequestc C s t j| � | jd| j� d S )Nz
User-Agent)r0 r@ rI r! )r
r r r r@ � s
z'HTTPSProxyConnection._add_proxy_headers)r ) r r r rJ r= r r/ r r- r? r@ r r r r rT � s
rT c C s$ | dkr dS t j| �} | jd�S dS )zJ Convert Internationalized domain name from Punycode (RFC3492) to Unicode N�idna)r rG �decode)rC r r r �idn_puny_to_unicode� s
r[ c C s* | dkr dS t j| �} t j| jd��S dS )zZ Convert domain name to Punycode (RFC3492). Hostname can be instance of string or Unicode NrY )r Zustr�encode)rC r r r �idn_ascii_to_puny s
r] )rF Zencodings.idnaZ encodingsrL �platformr Zrhnr r r r r r r rV � ImportErrorZ http.clientZclientZ
xmlrpc.clientr r r0 rJ rS rT r[ r] r r r r �<module>
s,
S,&.
Mini Shell