File: //lib/python3/dist-packages/uaclient/__pycache__/serviceclient.cpython-310.pyc
o
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 ZdZdZeejdee fdd���Z
eejde fdd���Z dde
ej ddfd d
�Zd d � Z
ddede
e fdd�Zde fdd�Zde dee
ee ef e
ee e f f fdd�ZdS )�UAServiceClient� N�returnc C � 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 � z UAServiceClient.api_error_clsc C r )z@String in subclasses, the UAConfig attribute containing base urlNr r r r r �cfg_url_base_attr r 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 sp |� d�}|s | �� }|�d�dkr |r tj|tjd��d�}tt | j
| j �|�}| � |�\} }
| r6| |
fS |rKdd� t
|�� �D �} |dt| � 7 }|d urQ|n| j} ztj||||| |d �\}
}W |
|fS tjy� } zDd }t|d
�rz|j}n t|d �r�|�� �d�}|r�z
tj|tjd�}W n ty� d }Y nw |r�| �||��t j |t |d d �||d
��d }~ww )N�/r r )�clszutf-8c S s i | ] \}}|d ur||�qS r r )�.0�k�vr r r �
<dictcomp>C s
��z/UAServiceClient.request_url.<locals>.<dictcomp>�?)�url�datar# �methodr% r$ �body�read�code�r2 r# r- ) �lstripr# �get�json�dumpsr
�DatetimeAwareJSONEncoder�encode�urljoin�getattrr r �_get_fake_responses�sorted�itemsr � url_timeout�readurlr �URLError�hasattrr0 r1 �decode�loads�DatetimeAwareJSONDecoder�
ValueErrorr r �UrlError)r �pathr. r# r/ � query_paramsr$ r% r- �
fake_response� fake_headers�filtered_params�timeout_to_use�response�er0 �
error_detailsr r r � request_url, s`
�
� ��
� � ���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_overlayr5 r �features�osrH �existsr6 rD r � load_file)r r- �response_overlay_pathr r r �_get_response_overlaye s
� �z%UAServiceClient._get_response_overlayc C s� | � |�}|s di fS t|�dkr|d }n|�d�}|d dkr+|d |�di �fS t�|d �}tj||d |�di �|d�}|�d �d
krQtj||d d ��|�) 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 r2 �� rN r# r3 �type�contract)�error_response) rY �len�popr5 r rA r rG �ContractAPIError)r r- � responsesrN rO �
url_exceptionr r r r< � s*
��z#UAServiceClient._get_fake_responsesr )NNNNTN)�__name__�
__module__� __qualname__r? rS �property�abc�abstractmethodr � Exceptionr �strr r r
r r r# �bool�intrQ rY r r r r< r r r r r s8
��
�9"�"�r )� metaclass)rh r6 rU � posixpathr r: �typingr r r r r �urllibr � urllib.parser �uaclientr
r r r
r �ABCMetar r r r r �<module> s