CasperSecurity
o
��Jh� � @ st d dl Z d dlmZmZmZmZ d dlmZ d dlm Z G dd� d�Z
G dd� d�Z G d d
� d
�Z G d d � d �Z
dS )
� N)�Callable�Iterable�Optional�Union)�messages)� HelpCategoryc @ sD e Zd Z d
dededee fdd�Zdeejej f fdd �Z
dS ) � ProArgumentN� long_name�help�
short_namec K s( |d u r|fn||f| _ || _|| _d S �N)�namesr
�additional_args)�selfr r
r �kwargs� r �7/usr/lib/python3/dist-packages/uaclient/cli/commands.py�__init__ s �
zProArgument.__init__�targetc C s |j | jd| ji| j�� d S )Nr
)� add_argumentr
r
r )r r r r r �register s ��
�zProArgument.registerr ) �__name__�
__module__� __qualname__�strr r r �argparse�ArgumentParser�_ArgumentGroupr r r r r r s ���
�
�r c @ s( e Zd Z ddedee fdd�ZdS ) �!ProArgumentMutuallyExclusiveGroupFr �required� argumentsc C s || _ || _d S r )r r )r r r r r r r s
z*ProArgumentMutuallyExclusiveGroup.__init__N)Fr )r r r �boolr r r r r r r r s ���r c
@ sP e Zd Z d dee dee dee dee fdd�Zd e j
fd
d �Z dS )
�ProArgumentGroupNr �title� descriptionr �mutually_exclusive_groupsc C s || _ || _|| _|| _d S r )r# r$ r r% )r r# r$ r r% r r r r ) s
zProArgumentGroup.__init__�parserc C sd |}| j r
|�| j | j�}| jD ]}|�|� q| jD ]}|j|jd�}|jD ]}|�|� q'qd S )N)r )r# �add_argument_groupr$ r r r% � add_mutually_exclusive_groupr )r r& r �argument�group� new_groupr r r r 7 s �
�
��zProArgumentGroup.register)NNr r ) r r r r r r r r r r r r r r r r r" ( s" �����
�r" c @ sh e Zd Zdd� dddddfdeded ed
ed ed ee d
ed dee de
fdd�Z de j
fdd�ZdS )�
ProCommandc O s d S r r )�argsr r r r �<lambda>O s zProCommand.<lambda>Fr Nr �namer
r$ �action�preserve_description�argument_groups� subcommands�
help_category�
help_positionc
C s: || _ || _|| _|| _|| _|| _|| _|| _| | _d S r ) r/ r
r$ r0 r1 r2 r3 r4 r5 )
r r/ r
r$ r0 r1 r2 r3 r4 r5 r r r r J s
zProCommand.__init__�
subparsersc C s� |j | j| j| jd�| _| jrtj| j_| j r%| jj
| j | j| j| j d� | j D ]}|�
| j� q(| jj| jd� | jrP| jjtjdd�}| jD ] }|�
|� qHd S d S )N)r
r$ )�categoryr/ � help_string�position)r0 �command)r# �dest)�
add_parserr/ r
r$ r& r1 r �RawDescriptionHelpFormatter�formatter_classr4 �add_help_entryr5 r2 r � set_defaultsr0 r3 �add_subparsersr �CLI_AVAILABLE_COMMANDS)r r6 �argument_groupr: r r r r ` s4 �
�
�
�zProCommand.register)r r r r r r! r r" r r �intr r �_SubParsersActionr r r r r r, I s6 �������� �
�r, )r �typingr r r r �uaclientr �uaclient.cli.parserr r r r" r, r r r r �<module> s
!