OwlCyberSecurity - MANAGER
Edit File: serviceclient.cpython-38.pyc
U Uٵg� � @ 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