OwlCyberSecurity - MANAGER
Edit File: commands.cpython-38.pyc
U Uٵg� � @ 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 !