File: //lib/python3/dist-packages/uaclient/__pycache__/serviceclient.cpython-310.pyc
o
0xvc� � @ 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, |st �� dk}tj|d�| _d S || _d S )Nr )� root_mode)�os�getuidr
�UAConfigr )r 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�potentially_sensitive�timeoutc C sj |� d�}|s | �� }|�d�dkr |r t�|��d�}tt| j| j �|�}| �
|�\} }
| r3| |
fS |rHdd� t |� � �D �} |dt
| � 7 }|d urN|n| j} ztj||||| |d�\}
}W |
|fS tjy� } zDd }t|d �rw|j}n t|d
�r�|�� �d�}|r�z
tj|tjd �}W n ty� d }Y nw |r�| �||��tj |t|d d �||d
��d }~ww )N�/r" r zutf-8c S s i | ] \}}|d ur||�qS �Nr )�.0�k�vr r r �
<dictcomp>B s
��z/UAServiceClient.request_url.<locals>.<dictcomp>�?)�url�datar% �methodr' r& �body�read)�cls�code�r5 r% r/ ) �lstripr% �get�json�dumps�encode�urljoin�getattrr r �_get_fake_responses�sorted�itemsr � url_timeoutr
�readurlr �URLError�hasattrr2 r3 �decode�loads�DatetimeAwareJSONDecoder�
ValueErrorr r �UrlError)r �pathr0 r% r1 � query_paramsr& r' r/ �
fake_response� fake_headers�filtered_params�timeout_to_use�response�er2 �
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_overlayr8 r �featuresr rJ �existsr9 rF r � load_file)r r/ �response_overlay_pathr r r �_get_response_overlayd 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 r5 �� rP r% r6 �type�contract)�error_response) rZ �len�popr8 r rC r rI �ContractAPIError)r r/ � responsesrP rQ �
url_exceptionr r r r> � s*
��z#UAServiceClient._get_fake_responsesr) )NNNNTN)�__name__�
__module__� __qualname__rA rU �property�abc�abstractmethodr � Exceptionr �strr r r
r r r% �bool�intrS rZ r r r r> r r r r r s8
��
�7"�"�r )� metaclass)ri r9 r � posixpathr r< �typingr r r r r �urllibr � urllib.parser �uaclientr
r r r
r �ABCMetar r r r r �<module> s