File: //proc/self/root/lib/python3/dist-packages/uaclient/cli/__pycache__/commands.cpython-38.pyc
U
��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 @ s> e Zd Zdeeee d�dd�Zeejej f d�dd�Z
dS ) � ProArgumentN)� long_name�help�
short_namec K s( |d kr|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.register)N) �__name__�
__module__� __qualname__�strr r r �argparse�ArgumentParserZ_ArgumentGroupr r r r r r s �� �r c @ s" e Zd Zdeee d�dd�ZdS )�!ProArgumentMutuallyExclusiveGroupFr ��required� argumentsc C s || _ || _d S 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 @ sB e Zd Zd ee ee ee ee d�dd�Ze j
d�dd�Z dS )
�ProArgumentGroupNr ��title� descriptionr �mutually_exclusive_groupsc C s || _ || _|| _|| _d S 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 ]}|�|� qNq6d S )N)r )r$ Zadd_argument_groupr% r r r& Z add_mutually_exclusive_groupr )r r' r Zargument�groupZ 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 @ sX e Zd Zdd� dddddfeeeeeee ed ee e
d� dd �Z e j
d
�d d �ZdS )
�
ProCommandc O s d S r r )�argsr r r r �<lambda>O � zProCommand.<lambda>Fr Nr � �namer
r% �action�preserve_description�argument_groups� subcommands�
help_category�
help_positionc
C s: || _ || _|| _|| _|| _|| _|| _|| _| | _d S r r- )
r r. r
r% r/ r0 r1 r2 r3 r4 r r r r J s zProCommand.__init__)�
subparsersc C s� |j | j| j| jd�| _| jr(tj| j_| j rJ| jj
| j | j| j| j d� | j D ]}|�
| j� qP| jj| jd� | jr�| jjtjdd�}| jD ]}|�
|� q�d S )N)r
r% )�categoryr. Z help_stringZposition)r/ �command)r$ �dest)Z
add_parserr. r
r% r' r0 r ZRawDescriptionHelpFormatterZformatter_classr3 Zadd_help_entryr4 r1 r Z set_defaultsr/ r2 Zadd_subparsersr ZCLI_AVAILABLE_COMMANDS)r r5 Zargument_groupr7 r r r r ` s0 �
�
�
zProCommand.register)r r r r r r! r r" r r �intr r Z_SubParsersActionr r r r r r) I s$ �� r) )r �typingr r r r Zuaclientr Zuaclient.cli.parserr r r r" r) r r r r �<module> s
!