3
�f,5 � @ sd d dl mZ d dlmZmZmZmZ dd� Zdd� ZG dd� d�Z G d d
� d
e�Z
G d d � d �Z d
S )� )�Action)� ConfigParser� ParsingError�Error�DuplicateOptionErrorc C s t | �}|tkp|tkS )z@Return true if val is an instance of a known sequence type.
)�type�list�tuple)�valZval_type� r � /usr/lib/python3.6/options.py�_is_seq s r
c C s( | j � }|d krdS |d
kr dS d S d S ) N�true�on�yesT�false�off�noF)r r r )r r r )�lower)r
Z_valr r r � str_to_bool s r c @ s� e Zd Zdd� Zdd� Zd dd �Zd
d � Zd d
� Zi fdd�Ze i fdd��Z
e dd� �Z dd� Z dd� Z
d!dd�Zd"dd �Zd d � Zd S )#�
SoSOptionsc C s� dd� }t ||�r�t||�}t| |�}||�r8||� sP|sP|| jkr|t|� r|t| ||� |rn| jj|� q�| jj|� nt|�r�t| ||| � d S )Nc S s | dkp| d kS )N� r )r
r r r �_unset" s z%SoSOptions._merge_opt.<locals>._unset)�hasattr�getattr� _nondefaultr
�setattr�discard�add)�self�opt�src�
is_defaultr ZnewvalueZoldvaluer r r �
_merge_opt! s
zSoSOptions._merge_optc C s4 t |t�st|�}x | jD ]}| j|||� qW d S )N)�
isinstance�dict�vars� arg_namesr# )r r! r" �argr r r � _merge_opts9 s
zSoSOptions._merge_optsF� r c s~ |}d}x � j D ]}||| | 7 }qW |j|� � fdd�� j D �}|sZdd� |D �}ndd� |D �}|t|� j|�| S )aI Format a SoSOptions object as a human or machine readable string.
:param quote: quote option values
:param sep: list separator string
:param prefix: arbitrary prefix string
:param suffix: arbitrary suffix string
:param literal: print values as Python literals
z=%sc s g | ]}t � |��qS r )r )�.0r( )r r r �
<listcomp>N s z$SoSOptions.__str.<locals>.<listcomp>c S s" g | ]}t |�rd j|�n|�qS )�,)r
�join)r+ �vr r r r, Q s c S s" g | ]}t |t�rd | n|�qS )z'%s')r$ �str)r+ r/ r r r r, T s )r' �stripr ) r �quote�sep�prefix�suffix�argsZarg_fmtr( �valsr )r r Z__str? s
zSoSOptions.__strc C s | j � S )N)�_SoSOptions__str)r r r r �__str__X s zSoSOptions.__str__c C s | j ddddd�S )NTz, z SoSOptions(�))r2 r3 r4 r5 )r8 )r r r r �__repr__[ s
zSoSOptions.__repr__c K sp || _ t|j� �| _t� | _x | j D ]}t| || j | � q$W x,|j� D ] }| jj|� t| ||| � qHW dS )a� Initialise a new ``SoSOptions`` object from keyword arguments.
Initialises the new object with values taken from keyword
arguments matching the names of ``SoSOptions`` attributes.
A ``ValueError`` is raised is any of the supplied keyword
arguments does not correspond to a known ``SoSOptions`
attribute name.
:param *kwargs: a list of ``SoSOptions`` keyword args.
:returns: the new ``SoSOptions`` object.
N)� arg_defaultsr �keysr' �setr r �append)r r< �kwargsr( r r r �__init___ s
zSoSOptions.__init__c C s& t f t|�d|i��}|j|d� |S )a Initialise a new SoSOptions object from a ``Namespace``
obtained by parsing command line arguments.
:param args: parsed command line arguments
:returns: an initialised SoSOptions object
:returntype: SoSOptions
r< T)r r& r) )�clsr6 r< Zoptsr r r � from_argsw s zSoSOptions.from_argsc sP d}d}� |krd� gS � |kr>� fdd�t dt|��D �S d� d | gS )z�Convert a named option and optional value to command line
argument notation, correctly handling options that take
no value or that have special representations (e.g. verify
and verbose).
�
alloptions�allow-system-changes�all-logs�batch�build�debug� experimental� list-plugins� list-presets�
list-profiles� no-report� no-env-vars�quiet�verify�verbosez--%sc s g | ] }d � �qS )z--%sr )r+ �d)r r r r, � s z+SoSOptions._opt_to_args.<locals>.<listcomp>r z--�=)rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ )rR )�range�int)rB r r
Zno_value�countr )r r � _opt_to_args� s
zSoSOptions._opt_to_argsc C s� t | j| t|��r|S t | j| t�r<dd� |jd�D �S t | j| t�rrt|�}|dkrntd||f ��n|S t | j| t�r�yt|�S t k
r� td||f ��Y nX |S )a� Ensure that the value read from a config file is the proper type
for consumption by the component, as defined by arg_defaults.
Params:
:param key: The key in arg_defaults we need to match the
type of
:param val: The value to be converted to a particular type
:param conf: File values are being loaded from
c S s g | ]}|�qS r r )r+ r/ r r r r, � s z/SoSOptions._convert_to_type.<locals>.<listcomp>r- Nz6Value of '%s' in %s must be True or False or analagousz#Value of '%s' in %s must be integer)
r$ r< r r �split�boolr � ExceptionrV �
ValueError)r �keyr
Zconfr r r �_convert_to_type� s$
zSoSOptions._convert_to_typec
sH � �fdd�}t � }y�y$t� ��}|j|� � W dQ R X W nZ tk
rr } z td|j|j� f ��W Y dd}~X n$ ttfk
r� td� ��Y nX W n< t t
fk
r� } z t d� |j d f � W Y dd}~X nX |d|� |||� |j
d �o�t�d
��rDxB|jd �D ]4\}} |jd �d �jk�r �jj|d
| � �q W dS )z�Read the provided config_file and update options from that.
Positional arguments:
:param config_file: Filepath to the config file
:param component: Which component (section) to load
c s� |j | �r�t|j| ��}d|j� kr6t|jd��|d< dddd�}xHt|�D ]<}d|krn|j|�||jdd�< ||krL|j|�||| < qLW xh|j� D ]\\}}t|t �r�|d kr�|jd
d �}|�j
kr�t d || f � q��j ||� �}t
�||� q�W d S )
NrR � verbosityZlabel�plugoptsZprofiles)�nameZ
plugin_optionZprofile�-�_�keywordsr* r z#Unknown option '%s' in section '%s')� has_sectionr% �itemsr= rV �popr �replacer$ r0 r< �printr^ r )�section�config�odictZ rename_optsr] r
)� config_filer r r �_update_from_section� s*
z9SoSOptions.update_from_conf.<locals>._update_from_sectionNz0Duplicate option '%s' in section '%s' in file %sz%Failed to parse configuration file %sz2WARNING: Unable to read configuration file %s : %s� �globalZplugin_optionsr` �.r rT )r �openZ read_filer �exitZoptionrj r r �OSError�IOErrorri r6 re r rf rY Z skip_pluginsr` r? )
r rm Z componentrn rk �f�err�er] r
r )rm r r �update_from_conf� s, #
"$
zSoSOptions.update_from_confTc C sB x<| j D ]2}t||�sqt||�dk s,| r| j||d� qW dS )aN Merge another set of ``SoSOptions`` into this object.
Merge two ``SoSOptions`` objects by setting unset or default
values to their value in the ``src`` object.
:param src: the ``SoSOptions`` object to copy from
:param is_default: ``True`` if new default values are to be set.
NF)r' r r r# )r r! Z skip_defaultr( r r r �merge� s
zSoSOptions.mergec C s: i }x0| j D ]&}t| |�}|r*|dkr*d}|||< q W |S )z�Return this ``SoSOptions`` option values as a dictionary of
argument name to value mappings.
:returns: a name:value dictionary of option values.
�
add_preset�
del_preset�desc�noteN)r{ r| r} r~ )r' r )r Z
preset_filterrl r( �valuer r r r% s
zSoSOptions.dictc sL �fdd���fdd��dd� � t �j� j� dd� d �}� �fd
d �|D �S ) z�Return command arguments for this object.
Return a list of the non-default options of this ``SoSOptions``
object in ``sosreport`` command line argument notation:
``["--all-logs", "-vvv"]``
c sN d
}| s||krdS | dkr&|r&d S | � j krJt|�t� j | �krJdS d S ) z. Test for non-null option values.
�False�None�[]�""�''�0Fr` T)r� r� r� r� r� r� )r< r0 )ra r Z null_values)r r r � has_value! s
z%SoSOptions.to_args.<locals>.has_valuec s* | d kr dS | d kr |d kr d
S � | |�S )
z8 Filter out preset and null-valued options.
r{ r| r} r~ F�log_size�plugin_timeout� cmd_timeoutr T)r{ r| r} r~ )r� r� r� r )ra r )r� r r �
filter_opt. s z&SoSOptions.to_args.<locals>.filter_optc S sz | j d�r dt|�d }|S | jdd�} t|�r<dj|�n|}|dk rVd| |f }n| }t|�dkrnd | nd| }|S )
zD Convert sos option notation to command line arguments.
r_ rb r/ rc r- Tz%s %sro z--)�
startswithrV rh r
r. �len)ra r r( r r r r �argify9 s
z"SoSOptions.to_args.<locals>.argifyc S s | d S )Nr r )�xr r r �<lambda>M s z$SoSOptions.to_args.<locals>.<lambda>)r] c s$ g | ] \}}�||�r� ||��qS r r )r+ �nr/ )r� r� r r r, N s z&SoSOptions.to_args.<locals>.<listcomp>)�sortedr% rf )r Z opt_itemsr )r� r� r� r r �to_args s
zSoSOptions.to_argsN)Fr* r r )T)T)�__name__�
__module__� __qualname__r# r) r8 r9 r; rA � classmethodrC rX r^ ry rz r% r� r r r r r s
D
r c @ s e Zd ZdZddd�ZdS )�
SosListOptionz0Allow to specify comma delimited list of pluginsNc C sB dd� |j d�D �}t|| j�r0|t|| j�7 }t|| j|� d S )Nc S s g | ]}|�qS r r )r+ r r r r r, V s z*SosListOption.__call__.<locals>.<listcomp>r- )rY r �destr )r �parser� namespace�valuesZ
option_stringrf r r r �__call__U s zSosListOption.__call__)N)r� r� r� �__doc__r� r r r r r� Q s r� c @ s e Zd ZdZddd�ZdS )�
ClusterOptionz6Used to store/manipulate options for cluster profiles.Nc C s" || _ || _|| _|| _|| _d S )N)ra r �opt_type�cluster� description)r ra r r� r� r� r r r rA _ s
zClusterOption.__init__)N)r� r� r� r� rA r r r r r� \ s r� N) �argparser Z configparserr r r r r
r r r� r� r r r r �<module> s
4