File: //usr/lib/python3/dist-packages/uaclient/api/__pycache__/api.cpython-310.pyc
o
nHJe� �
@ s d dl Z d dlmZ d dlmZmZmZmZmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ d dlmZ d dlmZmZ d d lmZ g d
�Zd ee d ee d
eeeef ee f fdd�Zded ee d
eeeef ee f fdd�Zded ee deded
e
f
dd�Z G dd� d�Z dS )� N)�
import_module)�Any�Callable�Dict�List�Tuple)�errors)�APIData� APIResponse�ErrorWarningObject)�UAConfig)�IncorrectFieldTypeError)�API_UNKNOWN_ARG�WARN_NEW_VERSION_AVAILABLE)�check_for_new_version)z,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.magic.initiate.v1z u.pro.attach.magic.revoke.v1zu.pro.attach.magic.wait.v1zu.pro.packages.summary.v1zu.pro.packages.updates.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.status.enabled_services.v1zu.pro.status.is_attached.v1zu.pro.version.v1z u.security.package_manifest.v1z u.unattended_upgrades.status.v1zu.apt_news.current_news.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 ]*\}}|r'|s1tjd�||�d��||vrE|� t
t j|d�j t j
i d�� |||< q ||fS )N)r* z{}:{}r r )�json�loads�decoder�JSONDecodeErrorr �APIJSONDataFormatError�itemsr r r r r r r! )r* r r" r# � json_datar% r& r' r' r( �
_process_dataA s( � ��
r2 �
endpoint_path�cfgc
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|�||�} W nC ty� }
z t�|
�W Y d }
~
S d }
~
ww |s�|r�t�tj| d��S z|�|�} 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>m s z call_api.<locals>.<listcomp>)r r5 )�versionr )�type�
attributesr )�_schema_versionr# r* ) �VALID_ENDPOINTSr � error_out�APIInvalidEndpointr r5 � options_clsr r) r2 �extend�APIError� from_dictr
�
APIMissingArgr6 �fn� Exception�APINoArgsForEndpointr r r r r r r! r
r: r# r r )
r3 r r* r4 �moduler5 �option_warningsr r" r# �e�result� new_versionr' r' r( �call_api^ s�
�
�������
������
��rN c @ s" e Zd Zdededefdd�ZdS )� APIEndpointr: r! rF c C s || _ || _|| _|| _d S )N)r: r! rF rA )�selfr: r! rF rA r' r' r( �__init__� s
zAPIEndpoint.__init__N)�__name__�
__module__� __qualname__�strr rQ r' r' r' r( rO � s ���rO ) r+ � importlibr �typingr r r r r � uaclient.apir �uaclient.api.data_typesr r
r �uaclient.configr �uaclient.data_typesr
�uaclient.messagesr r �uaclient.versionr r> rU r) r2 rN rO r' r' r' r( �<module> sH ��
� ��
� ����
�H