File: //usr/lib/python3/dist-packages/uaclient/api/__pycache__/api.cpython-38.pyc
U
��Jh< � @ sR d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl
m Z m Z d dl
mZmZmZ d dlmZ d dlmZ d dlmZmZ d d lmZ d
d d d
ddddddddddddddd d d d d d!d"d#d$gZee ee e eeef ee f d%�d&d'�Z eee e eeef ee f d(�d)d*�Z d0eee eeee ed+�d,d-�Z G d.d/� d/�Z dS )1� N)�
import_module)�Any�Callable�Dict�List�Optional�Tuple)�AbstractProgress�errors)�APIData� APIResponse�ErrorWarningObject)�UAConfig)�IncorrectFieldTypeError)�API_UNKNOWN_ARG�WARN_NEW_VERSION_AVAILABLE)�check_for_new_versionzu.pro.version.v1z,u.pro.attach.auto.configure_retry_service.v1z%u.pro.attach.auto.full_auto_attach.v1z'u.pro.attach.auto.should_auto_attach.v1z%u.pro.attach.guest.get_guest_token.v1z u.pro.attach.magic.initiate.v1z u.pro.attach.magic.revoke.v1zu.pro.attach.magic.wait.v1z'u.pro.attach.token.full_token_attach.v1zu.pro.detach.v1zu.pro.packages.summary.v1zu.pro.packages.updates.v1zu.pro.security.cves.v1z!u.pro.security.fix.cve.execute.v1z u.pro.security.fix.cve.plan.v1z!u.pro.security.fix.usn.execute.v1z u.pro.security.fix.usn.plan.v1z'u.pro.security.status.livepatch_cves.v1z(u.pro.security.status.reboot_required.v1z u.pro.services.dependencies.v1zu.pro.services.disable.v1zu.pro.services.enable.v1z u.pro.status.enabled_services.v1zu.pro.status.is_attached.v1zu.apt_news.current_news.v1z u.security.package_manifest.v1z u.unattended_upgrades.status.v1)�options�fields�returnc C s� i }g }| D ]|}z|� d�\}}W n tk
rB tj|d��Y nX |rL|sXtj|d��||kr�|�ttj|d�jtj i d�� |||< q ||fS )N�=��arg��title�code�meta)
�split�
ValueErrorr
ZAPIBadArgsFormat�appendr
r �format�msg�name)r r �kwargs�warningsZoption�k�v� r' �2/usr/lib/python3/dist-packages/uaclient/api/api.py�_process_options+ s& ��
r) )�datar r c C s� i }g }zt �| �}W n$ t jjk
r: tj| d��Y nX |�� D ]8\}}||krt|�tt j
|d�j t j i d�� |||< qD||fS )N)r* r r )
�json�loads�decoderZJSONDecodeErrorr
ZAPIJSONDataFormatError�itemsr r
r r r! r" )r* r r# r$ Z json_datar% r& r' r' r( �
_process_dataH s" ��
r/ )�
endpoint_pathr r* �cfg�progress_objectr c
C s6 | t krt�tj| d��S td| �}|j}g }|j�rndd� |jjD �}z@|r`t||�\} }
n |rtt ||�\} }
n
i g } }
|�
|
� W n2 tj k
r� } zt�| � W Y �S d } ~ X Y nX z|j� | �}W n> t
k
�r } z t�tj| j| d�� W Y �S d } ~ X Y nX z*|j�r*|j|||d�} n |�||�} W n2 tk
�rj } zt�| � W Y �S d } ~ X Y nX nx|�sz|�r�t�tj| d��S z&|j�r�|j||d�} n
|�|�} W n2 tk
�r� } zt�| � W Y �S d } ~ X Y nX t� }
|
�r|�ttj|
d�jtji d�� t|j | j | t |j| | j d �d
�S ) N)�endpointz
uaclient.api.c S s g | ]
}|j �qS r' )�key)�.0�fr' r' r( �
<listcomp>u s z call_api.<locals>.<listcomp>)r r3 )r2 )�versionr )�typeZ
attributesr )Z_schema_versionr$ r* ) �VALID_ENDPOINTSr
Z error_outZAPIInvalidEndpointr r3 � options_clsr r) r/ �extendZAPIErrorZ from_dictr Z
APIMissingArgr4 �supports_progress�fn� ExceptionZAPINoArgsForEndpointr r r
r r r! r" r r8 r$ r r )r0 r r* r1 r2 �moduler3 Zoption_warningsr r# r$ �e�resultZ new_versionr' r' r( �call_apib s�
�
� �"
� ���
��rC c @ s" e Zd Zdeeeed�dd�ZdS )� APIEndpointF)r8 r" r> r= c C s" || _ || _|| _|| _|| _d S )N)r8 r" r> r; r= )�selfr8 r" r> r; r= r' r' r( �__init__� s
zAPIEndpoint.__init__N)F)�__name__�
__module__� __qualname__�strr �boolrF r' r' r' r( rD � s ��rD )N) r+ � importlibr �typingr r r r r r Z uaclient.apir r
Zuaclient.api.data_typesr r r
Zuaclient.configr Zuaclient.data_typesr Zuaclient.messagesr r Zuaclient.versionr r: rJ r) r/ rC rD r' r' r' r( �<module> sl � � � �� T