3

Yd�i�
�@s:dd
lZddl
mZ
ddlmZ
Gdd�d�Zdd�Zd
S)�N)
�request)
�SubparsersActionc@s4eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�ZdS)�AnalyzercCs*|
j||d
�}|j
||�
|j|d�

dS)a4

        Add subcommand to existing subcommand group

        Args:
            name(str): Subcommand name
            help_msg(str): Help message for subcommand
            func(function): Function to call on execution
            opts(list of Object()): List of Option objects to add to subcommand
        )
�help)
�funcN)Z
add_parser�_add_subcommand_optionsZset_defaults)�selfZ
subcmd_grp�name�help_msgr�optsZ
req_parser�r�!/usr/lib/python3.6/sss_analyze.py�add_subcommandszAnalyzer.add_subcommandcCsrxl|D]d}|jt
krL|jd
kr4|
j|j|jdd�
n|
j|j|j|jdd�
|jtkr|
j|j|jtd�
qWd
S)z�
        Add subcommand options to subcommand parser

        Args:
            parser(str): Subcommand group parser
            opts(list of Object()): List of Option objects to add to subcommand
        N�
store_true)r�action)r�type)Zopt_type�boolZ	short_opt�add_argumentr	r
�int)r�parserr�optrrr
rs












z Analyzer._add_subcommand_optionscCstj
�}t�}|j||�
d
S)z�
        Initialize analyzer modules from modules/*

        Args:
            parser (ArgumentParser): Base parser object
            parser_grp (argparse.Action): Parser group that can have
                additional parsers attached.
        N)rZRequestAnalyzerr�
setup_args)rr�
parser_grpZreqZclirrr
�load_modules/s

zAnalyzer.load_modulesc
Csbtj
}
tjd
|
d�}|jddddgd�
|jddd	d
�
|jdtdd
�}|jd�
}|j||�
|S)z�
        Top-level argument setup function.
        Setup analyzer argument parsers and subcommand parser/options.

        Returns:
            parser (ArgumentParser): Base parser object
        z-Analyzer tool to assist with SSSD log parsing)�descriptionZformatter_classz--source�filesZjournald)�default�choicesz--logdirz/var/log/sssd/z*SSSD Log directory to parse log files from)rrNZCOMMANDS)�titler�metavar�Modules)�argparseZRawTextHelpFormatter�ArgumentParserrZadd_subparsersrZadd_parser_groupr)rZ	formatterrZ	subparserrrrr
r>s	










zAnalyzer.setup_argsc
Cs4|j�}
|
j
�}t|d
�s&|
j�
dS|j|�

dS)Nrr
)r�
parse_args�hasattrZ
print_helpr)rr�argsrrr
�main[s




z
Analyzer.mainN)�__name__�
__module__�__qualname__rrrrr&rrrr
rs

rc

Cst�}|j
�
dS)
N)rr&)
Zanalyzerrrr
�runfs

r*)r!Zsssd.modulesrZsssd.parserrrr*rrrr
�<module>
s
_