CasperSecurity
o
��Jh�
� @ s~ d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z edde
fde fd e fg�Z G d
d � d e�Z
G d d
� d
e j�ZdS )� N)� OrderedDict)�Enum)�List�
NamedTuple)�messages� HelpEntry�position�name� help_stringc @ sj e Zd ZG dd� d�Zedej�Zedej�Zedej �Z
edej �Z edej
�Zdd � Zed
d � �Zd S )
� HelpCategoryc @ s e Zd Zdedefdd�ZdS )zHelpCategory._Value�code�msgc C s || _ || _d S �N)r r
)�selfr r
� r �5/usr/lib/python3/dist-packages/uaclient/cli/parser.py�__init__ s
z HelpCategory._Value.__init__N)�__name__�
__module__� __qualname__�strr r r r r �_Value s r �
quickstart�security� troubleshoot�other�flagsc C � | j jS r )�valuer �r r r r �__str__ s zHelpCategory.__str__c C r r )r r
r r r r �header s zHelpCategory.headerN)r r r r r �CLI_HELP_HEADER_QUICK_START�
QUICKSTART�CLI_HELP_HEADER_SECURITY�SECURITY� CLI_HELP_HEADER_TROUBLESHOOT� TROUBLESHOOT�CLI_HELP_HEADER_OTHER�OTHER� CLI_FLAGS�FLAGSr �propertyr! r r r r r
s � r c
s� e Zd Zeejg fejg fejg fejg fej g fg�Z
e ddede de de
fdd��Zdd �d
ef� fd d �Zd
e fdd�Z� fdd�Z� ZS )�ProArgumentParserr �categoryr r
r c C s4 t |||d�}|| j| vr| j| �|� d S d S )N)r r r
)r � help_entries�append)�clsr. r r
r �entryr r r �add_help_entry/ s ��z ProArgumentParser.add_help_entryT)�
use_main_helpr4 c s t � j|i |�� || _d S r )�superr r4 )r r4 �args�kwargs�� __class__r r r = s
zProArgumentParser.__init__�commandc C s8 |� � }|�d� z| �|� W d S ty Y d S w )Nz--help)�splitr0 �
parse_args�
SystemExit)r r: � args_listr r r �print_help_for_commandA s
�z(ProArgumentParser.print_help_for_commandc s� | j rt� �� S | �� }| j�� D ]1\}}|d7 }|d�|j�7 }|d7 }t|dd� d�D ]}|d7 }|d�|j |j
�7 }q-|d7 }q| j rS|d7 }|| j 7 }|d7 }|S )N�
z{}:c S s | j S r )r )�itemr r r �<lambda>T s z/ProArgumentParser.format_help.<locals>.<lambda>)�keyz
{:<17}{}) r4 r5 � format_help� format_usager/ �items�formatr! �sortedr r
�epilog)r � help_outputr. rF rA r8 r r rD K s
z ProArgumentParser.format_help)r )r r r r r r# r% r' r) r+ r/ � classmethodr �intr3 �boolr r? rD �
__classcell__r r r8 r r- $ s. ��
���� �
r- )�argparse� collectionsr �enumr �typingr r �uaclientr rL r r r �ArgumentParserr- r r r r �<module> s �