�
!V1f � � � d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ d dlmZ d dlmZmZmZmZ G d� d e� Z G d
� d e� Z G d � d
ee� Z de de fd�Z de dede fd�Z edde e �� Z!y)� )�List)� APIEndpoint)�AdditionalInfo)�get_expected_overall_status) �AptUpgradeData�
AttachData�
EnableData� FixPlanError�
FixPlanResult� FixPlanStep�FixPlanUSNResult�FixPlanWarning�NoOpData� PackageCannotBeInstalledData�SecurityIssueNotFixedData� fix_plan_usn)�UAConfig)�
DataObject�Field�StringDataValue� data_listc �> � e Zd Z ed ee� � gZdee fd�Z y)�USNFixPlanOptions�usnsc � � || _ y �N)r )�selfr s �M/usr/lib/python3/dist-packages/uaclient/api/u/pro/security/fix/usn/plan/v1.py�__init__zUSNFixPlanOptions.__init__! s � ��� � N)
�__name__�
__module__� __qualname__r r r �fieldsr �strr � r r r r s( � �
�f�i��0�1��F��T�#�Y� r r c �R � e Zd Z ede� ed ee� � gZdede e fd�Z
y)�USNFixPlanResult�expected_statusr c � � || _ || _ y r �r) r )r r) r s r r zUSNFixPlanResult.__init__+ s � � .�� ��� r N) r! r" r# r r r r
r$ r% r r r& r r r( r( % s= � �
� � �1�
�f�i� 0�1�2��F�
�3� �d�;K�6L� r r( c �, � e Zd Z ede� gZdefd�Zy)�USNSFixPlanResult� usns_datac � � || _ y r �r. )r r. s r r zUSNSFixPlanResult.__init__5 s � �"��r N)r! r" r# r r( r$ r r& r r r- r- 0 s! � �
�k�+�,��F�#�%5� #r r- �options�returnc �* � t | t � � S r )�_planr )r1 s r �planr5 9 s � � ��(�*� %�%r �cfgc �� � g }d}| j D ]@ }t ||�� }t ||j j � }|j |� �B t
t ||�� �� S )N� )r6 r+ r0 )r r r �target_usn_planr) �appendr- r( )r1 r6 r r) �usn�usn_plans r r4 r4 = sq � �
�D��O��|�|� �� ���-��5� �X� 5� 5� E� E�
��
� � �H� � � �"� +��
� � r �v1�
USNFixPlan)�version�name�fn� options_clsN)"�typingr �uaclient.api.apir �uaclient.api.data_typesr �'uaclient.api.u.pro.security.fix._commonr �/uaclient.api.u.pro.security.fix._common.plan.v1r r r r
r r r
r r r r r �uaclient.configr �uaclient.data_typesr r r r r r( r- r5 r4 �endpointr& r r �<module>rK s� �� � (� 2� O�
�
�
�
� %� M� M��
� ��z� �#�
�N� #�&�#� &�(9� &��$� �8� �8I� �$ � � � �!� �r |