0x1949 Team - FAZEMRX - MANAGER
Edit File: serviceclient.cpython-310.pyc
o �ϴf� � @ 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