�
Uٵg� � � � S SK r S SKrS SKrS SKrS SKJrJrJr S SKJ r S SK
J r J r J
r
JrJr " S S\ R S9rg)� N)�Any�Dict�Optional)� urlencode)�config�http�system�util�versionc � � \ rS rSrSrSr\\R S\ 4S j5 5 r
SS\ \ R SS4S jjrS r SS \S
\ \ S\R& 4S jjrS \ 4S
jrS \ S\ \R& 4S jrSrg)�UAServiceClient� � N�returnc � � g)z@String in subclasses, the UAConfig attribute containing base urlN� ��selfs �=/usr/lib/python3/dist-packages/uaclient/http/serviceclient.py�cfg_url_base_attr�!UAServiceClient.cfg_url_base_attr s � �
� �cfgc �T � U( d [ R " 5 U l g Xl g �N)r �UAConfigr )r r s r �__init__�UAServiceClient.__init__ s � �� ���(�D�H��Hr c �R � SR [ R " 5 5 SSS.$ )Nz UA-Client/{}�application/json)z
user-agent�accept� content-type)�formatr � get_versionr s r �headers�UAServiceClient.headers s) � �(�/�/��0C�0C�0E�F�(� .�
�
r �log_response_body�timeoutc � � UR S5 nU( d U R 5 nUR S5 S:X a9 U( a2 [ R " U[
R S9R S5 n[ R " [ U R U R 5 U5 nU R U5 n U ( a U $ U( aC [ UR 5 5 V
V s0 s H u p�U c M
X�_M n n
n US[! U 5 - -
nUb UO U R" n
[$ R& " UUUUU
US9$ s sn n
f )N�/r" r )�clszutf-8�?)�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�k�v�filtered_params�timeout_to_uses r � request_url�UAServiceClient.request_url% s � � �{�{�3� ����l�l�n�G� �;�;�~� &�*<� <���:�:�d��(E�(E�F�M�M���D� �n�n�W�T�X�X�t�/E�/E� F��M�� �0�0��5�
� � � � � "(� �(:�(:�(<�!=� �!=�������!=�
� �
�3��?�3�3� 3�C�$+�$7��T�=M�=M���|�|�����"� /�
�
��
s �) D:�6D:r- c � � U R b U R R U/ 5 $ U R R R S5 nU( d 0 U l O[[ R
R
U5 ( d 0 U l O/[ R " [ R " U5 5 U l U R R U/ 5 $ )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.
�serviceclient_url_responses) �_response_overlayr1 r �features�osr> �existsr2 �loadsr � load_file)r r- �response_overlay_paths r �_get_response_overlay�%UAServiceClient._get_response_overlayM s� � �$ � !� !� -��)�)�-�-�c�2�6� 6� $��� 1� 1� 5� 5� )�!
� � %�%'�D� "���� � � 5�6�6�%'�D� "�%)�Z�Z�� � �!6�7�&�D� "� �%�%�)�)�#�r�2�2r c �r � U R U5 nU( d g[ U5 S:X a US nOUR S5 n0 n/ nUS n[ U[ 5 ( a UnO[ U[
5 ( a Un[ R " US UR S0 5 [ R " US SS9UUS 9$ )
z7Return response if faked for this URL in uaclient.conf.N� r �response�coder% T)� sort_keys)rU r% �body� json_dict� json_list) rP �len�pop�
isinstance�dict�listr � HTTPResponser1 r2 r3 )r r- � responsesrT rX rY �resps r r9 �#UAServiceClient._get_fake_responseso s� � ��.�.�s�3� ��� �y�>�Q� � ��|�H� !�}�}�Q�'�H�� �� ��
�#�� �d�D� !� !� �I�
��d�
#�
#� �I�� � ��&�!� �L�L��B�/����H�Z� 0�D�A� � �
�
r )rI r r )NNNNTN)�__name__�
__module__� __qualname__�__firstlineno__r< rI �property�abc�abstractmethod�strr r r r r r% �bool�intr r_ rE rP r9 �__static_attributes__r r r r
r
s� � ��K� ��
����
�3�
� � �
��H�V�_�_� 5� �� �
� ����"&�!%�&
� �&
� �#� �&
�
� � �&
�P 3�� 3�D
�s�
�x��8I�8I�/J�
r r
)� metaclass)rh r2 rK r6 �typingr r r � urllib.parser �uaclientr r r r
r �ABCMetar
r r r �<module>rs s2 ��
� � � � &� &� "� 8� 8�
�� � �
r