File: //lib/python3/dist-packages/uaclient/http/__pycache__/serviceclient.cpython-38.pyc
U
��Jh� � @ st d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm Z d dl
m Z m Z m
Z
mZmZ G dd� de jd�ZdS )� N)�Any�Dict�Optional)� urlencode)�config�http�system�util�versionc @ s� e Zd ZdZdZeejed�dd���Z de
e j dd�dd�Z
d d
� Zdee
e ejd �d
d�Zed�dd�Zee
ej d�dd�ZdS )�UAServiceClient� N)�returnc C s dS )z@String in subclasses, the UAConfig attribute containing base urlN� ��selfr r �=/usr/lib/python3/dist-packages/uaclient/http/serviceclient.py�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�timeoutr
c C s� |� d�}|s| �� }|�d�dkr>|r>tj|tjd��d�}t� t
| j | j �|�}| �
|�} | rf| S |r�dd� t|�� �D �}
|dt|
� 7 }|d k r�|n| j} tj||||| |d �S )
N�/r r )�clszutf-8c S s i | ]\}}|d k r||�qS r r )�.0�k�vr r r �
<dictcomp>> s z/UAServiceClient.request_url.<locals>.<dictcomp>�?)�url�datar �methodr r )�lstripr �get�json�dumpsr ZDatetimeAwareJSONEncoder�encode� posixpath�join�getattrr r �_get_fake_responses�sorted�itemsr � url_timeoutr Zreadurl) r �pathr% r r& Z query_paramsr r r$ Z
fake_responseZfiltered_paramsZtimeout_to_user r r � request_url% s2
�
��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_overlayr( r Zfeatures�osr3 �existsr) �loadsr Z load_file)r r$ Zresponse_overlay_pathr r r �_get_response_overlayM s
� �z%UAServiceClient._get_response_overlay)r$ r
c C s� | � |�}|sdS t|�dkr(|d }n
|�d�}i }g }|d }t|t�rR|}nt|t�r`|}tj|d |�di �t j
|d dd�||d �S )
z7Return response if faked for this URL in uaclient.conf.N� r �response�coder T)Z sort_keys)r< r Zbody� json_dict� json_list) r9 �len�pop�
isinstance�dict�listr � HTTPResponser( r) r* )r r$ Z responsesr; r= r> Zrespr r r r/ o s(
�z#UAServiceClient._get_fake_responses)N)NNNNTN)�__name__�
__module__� __qualname__r2 r5 �property�abc�abstractmethod�strr r r r r r �bool�intr rD r4 r9 r/ r r r r r s(
�� ("r )� metaclass)rI r) r6 r, �typingr r r Z urllib.parser Zuaclientr r r r r
�ABCMetar r r r r �<module> s