CasperSecurity
o
��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 ZdZdZeejdefdd���Z dde
e j ddfdd�Z
d d
� Z dd ed
e
e dejfdd�Zdefdd�Zdede
ej fdd�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�cfgc C s |s t �� | _d S || _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-agent�accept� content-type)�formatr
� get_versionr r r r �headers s �zUAServiceClient.headersT�log_response_body�timeoutc C s� |� d�}|s | �� }|�d�dkr |r tj|tjd��d�}t� t
| j | j �|�}| �
|�} | r3| S |rHdd� t|�� �D �}
|dt|
� 7 }|d urN|n| j} tj||||| |d �S )
N�/r r )�clszutf-8c S s i | ] \}}|d ur||�qS r r )�.0�k�vr r r �
<dictcomp>> s z/UAServiceClient.request_url.<locals>.<dictcomp>�?)�url�datar �methodr r )�lstripr �get�json�dumpsr �DatetimeAwareJSONEncoder�encode� posixpath�join�getattrr r �_get_fake_responses�sorted�itemsr � url_timeoutr �readurl) r �pathr' r r( � query_paramsr r r& �
fake_response�filtered_params�timeout_to_user r r � request_url% s2
�
��zUAServiceClient.request_urlr& c C sf | j dur | j �|g �S | jj�d�}|si | _ ntj�|�s#i | _ n t�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.
N�serviceclient_url_responses) �_response_overlayr* r �features�osr7 �existsr+ �loadsr � load_file)r r& �response_overlay_pathr r r �_get_response_overlayM s
� �z%UAServiceClient._get_response_overlayc C s� | � |�}|s dS t|�dkr|d }n|�d�}i }g }|d }t|t�r)|}nt|t�r0|}tj|d |�di �t j
|d dd�||d �S )
z7Return response if faked for this URL in uaclient.conf.N� r �response�coder T)� sort_keys)rH r �body� json_dict� json_list) rE �len�pop�
isinstance�dict�listr � HTTPResponser* r+ r, )r r& � responsesrG rK rL �respr r r r2 o s(
�z#UAServiceClient._get_fake_responsesr )NNNNTN)�__name__�
__module__� __qualname__r5 r> �property�abc�abstractmethod�strr r r r r r �bool�intr rR r< rE r2 r r r r r s.
���
�( "r )� metaclass)rY r+ r@ r/ �typingr r r � urllib.parser �uaclientr r r r r
�ABCMetar r r r r �<module> s