File: //lib/python3.6/site-packages/rhn/__pycache__/transports.cpython-36.pyc
3
�Z�i� � @ s6 d dl Z d dlZd dljZdZG dd� de jj�ZdS )� Nz13.0.3-1.module_el8.10.0+7020+fd2ced8d.cloudlinux-c sl e Zd ZdZde Zd� fdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z ddd�Z dd� Z
dd� Z� ZS )�RequestsTransportzO
Drop in Transport for xmlrpclib that uses Requests instead of httplib
zrhn.rpclib.py/%sTNc s0 || _ || _|| _i | _|| _tt| �j� d S )N)� https_verify� use_https�timeout�_injected_headers� proxy_uri�superr �__init__)�selfr r r �proxy)� __class__� � /usr/lib/python3.6/transports.pyr s zRequestsTransport.__init__c C s@ t |�t g �t f �gkr.dd� |D �| j|< nt|�| j|< d S )Nc S s g | ] }t |��qS r
)�str)�.0�ar
r
r �
<listcomp>. s z0RequestsTransport.set_header.<locals>.<listcomp>)�typer r )r
�name�argr
r
r �
set_header, s z RequestsTransport.set_headerc C sF || j kr&| j | }t|t�s4|g}ng }| j |< |jt|�� d S )N)r �
isinstance�list�appendr )r
r r Zvlistr
r
r �
add_header2 s
z RequestsTransport.add_headerc C s | j j� d S )N)Z_headers�clear)r
r
r
r �
clear_headers; s z RequestsTransport.clear_headersc C s
|| _ d S )N)r )r
Zcertfiler
r
r �set_trusted_cert> s z"RequestsTransport.set_trusted_certc c sN xH| j j� D ]:\}}t|�tg �tf �gkr<djtt|��}||fV q W d S )N�,)r �itemsr �join�mapr )r
r r r
r
r �get_injected_headersA s z&RequestsTransport.get_injected_headersFc C s� | j ddd�}t| j� �}| j||�}d| ji}tj|f|||�| j| j| jd�d�|��} y | j � W n> tj
k
r� }
z t j j
|| jt|
�| j��W Y dd}
~
X n X | j| �S dS )z)
Make an xmlrpc request.
ztext/xmlz*/*)z
User-Agentz Content-TypeZAcceptZverify)�http�https)�data�headersr ZproxiesN)�
user_agent�dictr" �
_build_urlr �requestsZpostr r Zraise_for_statusZRequestException�xmlrpc�clientZ
ProtocolErrorZ status_coder r& �parse_response) r
�host�handlerZ request_body�verboser& Z
extra_headersZurl�kwargsZresp�er
r
r �requestR s(
zRequestsTransport.requestc C s( | j � \}}|j|j� |j� |j� S )z,
Parse the xmlrpc response.
)Z getparserZfeedZcontent�close)r
Zresponse�p�ur
r
r r- u s z RequestsTransport.parse_responsec C s& | j r
dnd}|jd�}d|||f S )zf
Build a url for our request based on the host, handler and use_http
property
r$ r# �/z
%s://%s/%s)r �lstrip)r
r. r/ �schemer
r
r r) � s
z RequestsTransport._build_url)TTNN)F)�__name__�
__module__� __qualname__�__doc__� __version__r' r r r r r r" r3 r- r) �
__classcell__r
r
)r r r s
# r )r+ r* Z
xmlrpc.clientr, Z xmlrpclibr> Z Transportr r
r
r
r �<module> s