Your IP : 216.73.216.14
U
8�-d� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
Z
m Z d dl m
Z
d dlmZ d dlmZmZmZmZmZ G dd� de jd �ZdS )
� N)�join)�Any�Dict�Optional�Tuple�Type)�error)� urlencode)�config�
exceptions�system�util�versionc @ s� e Zd ZdZdZeejee d�dd���Z
eeje d�dd���Z de
ej dd�d d
�Zd d � Zdee
e d�dd�Ze d�dd�Ze ee
ee ef e
ee e f f d�dd�ZdS )�UAServiceClient� N)�returnc C s dS )z1Set in subclasses to the type of API error raisedN� ��selfr r �8/usr/lib/python3/dist-packages/uaclient/serviceclient.py�
api_error_cls s z UAServiceClient.api_error_clsc C s dS )z@String in subclasses, the UAConfig attribute containing base urlNr r r r r �cfg_url_base_attr s z!UAServiceClient.cfg_url_base_attr)�cfgr c C s |st �� | _n|| _d S �N)r
�UAConfigr )r r r r r �__init__ s zUAServiceClient.__init__c C s d� t�� �ddd�S )Nz UA-Client/{}�application/json)z
user-agentZaccept� content-type)�formatr Z get_versionr r r r �headers% s �zUAServiceClient.headersT)�log_response_body�timeoutc C s� |� d�}|s| �� }|�d�dkr>|r>tj|tjd��d�}tt | j
| j �|�}| � |�\} }
| rl| |
fS |r�dd� t
|�� �D �} |dt| � 7 }|d k r�|n| j} z tj||||| |d �\}
}W n� tjk
�r~ } z�d }t|d
�r�|j}nt|d ��r |�� �d�}|�rTztj|tjd�}W n tk
�r@ d }Y nX |�rT| �||��t j |t |d d �||d
��W 5 d }~X Y nX |
|fS )N�/r r )�clszutf-8c S s i | ]\}}|d k r||�qS r r )�.0�k�vr r r �
<dictcomp>C s z/UAServiceClient.request_url.<locals>.<dictcomp>�?)�url�datar �methodr! r �body�read�code�r. r r) ) �lstripr �get�json�dumpsr
ZDatetimeAwareJSONEncoder�encode�urljoin�getattrr r �_get_fake_responses�sorted�itemsr � url_timeoutZreadurlr �URLError�hasattrr, r- �decode�loadsZDatetimeAwareJSONDecoder�
ValueErrorr r �UrlError)r �pathr* r r+ Z query_paramsr r! r) Z
fake_responseZ fake_headersZfiltered_paramsZtimeout_to_use�response�er, Z
error_detailsr r r � request_url, sb
�
��
�
�zUAServiceClient.request_url)r) c C sf | j dk r| j �|g �S | jj�d�}|s2i | _ n&tj�|�sFi | _ nt�t �
|��| _ | j �|g �S )a. Return a list of fake response dicts for a given URL.
serviceclient_url_responses in uaclient.conf should be a path
to a json file which contains a dictionary keyed by full URL path.
Each value will be a list of dicts representing each faked response
for the given URL.
The response dict item will have a code: <HTTP_STATUS_CODE> and
response: "some string of content".
The JSON string below fakes the available_resources URL on the
contract server:
'{"https://contracts.canonical.com/v1/resources": [{"code": 200, "response": {"key": "val1", "key2": "val2"}}]}'
:return: List of dicts for each faked response matching the url, or
and empty list when no matching url found.
NZserviceclient_url_responses) �_response_overlayr1 r Zfeatures�osrA �existsr2 r> r Z load_file)r r) Zresponse_overlay_pathr r r �_get_response_overlaye s
� �z%UAServiceClient._get_response_overlay)r) r c C s� | � |�}|sdi fS t|�dkr,|d }n
|�d�}|d dkrV|d |�di �fS t�|d �}tj||d |�di �|d�}|�d �d
kr�tj||d d ��n|�dS ) a� Return response and headers if faked for this URL in uaclient.conf.
:return: A tuple of response and header dicts if the URL has an overlay
response defined. Return (None, {}) otherwise.
:raises exceptions.URLError: When faked response "code" is != 200.
URLError reason will be "response" value and any optional
"headers" provided.
N� r r. �� rB r r/ �typeZcontract)Zerror_response) rH �len�popr1 r r; r r@ ZContractAPIError)r r) Z responsesrB rC Z
url_exceptionr r r r7 � s*
��z#UAServiceClient._get_fake_responses)N)NNNNTN)�__name__�
__module__� __qualname__r: rE �property�abc�abstractmethodr � Exceptionr �strr r r
r r r �bool�intrD rH r r r r7 r r r r r s0
�� 9#"�r )� metaclass)rR r2 rF � posixpathr r5 �typingr r r r r Zurllibr Z urllib.parser Zuaclientr
r r r
r �ABCMetar r r r r �<module> s