CasperSecurity
o
�K�i< �
@ s4 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 g d
�Zd ee d ee d
e eeef ee f fdd�Z ded ee d
e eeef ee f fdd�Z dded ee dededee d
ef dd�Z G dd� d�Z dS )� 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_version)zu.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
ty tj|d��w |r#|s)tj|d��||vr=|�ttj|d�jtj i d�� |||< q||fS )N�=��arg��title�code�meta)
�split�
ValueErrorr
�APIBadArgsFormat�appendr
r �format�msg�name)r r �kwargs�warnings�option�k�v� r) �2/usr/lib/python3/dist-packages/uaclient/api/api.py�_process_options+ s( � ��
r+ �datac C s� i }g }zt �| �}W n t jjy tj| d��w |�� D ] \}}||vr7|�tt j
|d�j t j i d�� |||< q ||fS )N)r, r r )
�json�loads�decoder�JSONDecodeErrorr
�APIJSONDataFormatError�itemsr r
r r! r" r# )r, r r$ r% � json_datar' r( r) r) r* �
_process_dataH s$ � ��
r4 �
endpoint_path�cfg�progress_objectc
C s | t vr
t�tj| d��S td| �}|j}g }|jr�dd� |jjD �}z |r/t||�\} }
n|r9t ||�\} }
ni g } }
|�
|
� W n tj y\ } z t�| �W Y d } ~ S d } ~ ww z|j� | �}W n t
y� } zt�tj| j| d��W Y d } ~ S d } ~ ww z|jr�|j|||d�} n|�||�} W nN ty� } z t�| �W Y d } ~ S d } ~ ww |s�|r�t�tj| d��S z|jr�|j||d�} n|�|�} W n ty� } z t�| �W Y d } ~ S d } ~ ww 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 r8 )r7 )�versionr )�type�
attributesr )�_schema_versionr% r, ) �VALID_ENDPOINTSr
� error_out�APIInvalidEndpointr r8 � options_clsr r+ r4 �extend�APIError� from_dictr �
APIMissingArgr9 �supports_progress�fn� Exception�APINoArgsForEndpointr r r
r r! r" r# r r= r% r r )r5 r r, r6 r7 �moduler8 �option_warningsr r$ r% �e�result� new_versionr) r) r* �call_apib s�
�
������ ���
�
������
��rR c @ s* e Zd Z d dedededefdd�ZdS )
� APIEndpointFr= r# rJ rI c C s" || _ || _|| _|| _|| _d S �N)r= r# rJ rD rI )�selfr= r# rJ rD rI r) r) r* �__init__� s
zAPIEndpoint.__init__N)F)�__name__�
__module__� __qualname__�strr �boolrV r) r) r) r* rS � s �����rS rT ) r- � importlibr �typingr r r r r r � uaclient.apir r
�uaclient.api.data_typesr r r
�uaclient.configr �uaclient.data_typesr �uaclient.messagesr r �uaclient.versionr rA rZ r+ r4 rR rS r) r) r) r* �<module> sN ��
� ��
� ������
�T