3
ܓi3 @ sX d dl mZ d dlmZ dd Zdd ZG dd dZG d d
d
eZG d d d Zd
S ) )Action) ConfigParserc C s t | }|tkp|tkS )z@Return true if val is an instance of a known sequence type.
)typelisttuple)valZval_type r /usr/lib/python3.6/options.py_is_seq
s r
c C s$ | j }|d krdS |d
kr dS d S ) NtrueonyesTfalseoffnoF)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 fddZe i fddZ
e dd Z dd Z dd Z
d!ddZd"dd Zd d Zd S )#
SoSOptionsc C s dd }t ||rt||}t| |}||r8|| sT|sT|| jkrPt| sT|rt| || |rr| jj| q| jj| nt|rt| ||| d S )Nc S s | dkp| d kS )N r )r r r r _unset s z%SoSOptions._merge_opt.._unset)hasattrgetattr _nondefaultr
setattrdiscardadd)selfoptsrc
is_default
prefer_newr ZnewvalueZoldvaluer r r
_merge_opt s
zSoSOptions._merge_optc C s6 t |tst|}x | jD ]}| j|||| qW d S )N)
isinstancedictvars arg_namesr! )r 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
N s z$SoSOptions.__str..c S s" g | ]}t |rd j|n|qS ),)r
join)r) vr r r r* Q s c S s& g | ] }t |tr d | d n|qS )')r" str)r) r- r r r r* T s )r% stripr ) r quotesepprefixsuffixargsZarg_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 dddddS )NTz, z SoSOptions())r1 r2 r3 r4 )r7 )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 itemsappend)r r; kwargsr& Zkwargr r r __init___ s
zSoSOptions.__init__c C s( t f t|d|i}|j|dd |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; TF)r r$ r' )clsr5 r; Zoptsr r r from_argsw s zSoSOptions.from_argsc sR d}d} |kr d gS |kr@ fddt dt|D S d d| gS )zConvert 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).
alloptionsallow-system-changesall-logsbatchbuilddebug experimental list-plugins list-presets
list-profiles no-report no-env-varsquietverifyverbosez--c s g | ]}d qS )z--r )r) d)r r r r* s z+SoSOptions._opt_to_args..r =)rD rE rF rG rH rI rJ rK rL rM rN rO rP rQ )rR )rangeint)rB r r Zno_valuecountr )r r _opt_to_args s zSoSOptions._opt_to_argsc C s t | j| t|r|S t | j| tr6t|jdS t | j| trpt|}|dkrltd| d| d|S t | j| trt|S |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
r+ Nz
Value of 'z' in z# must be True or False or analagous) r" r; r r splitboolr ExceptionrV )r keyr Zconfr r r _convert_to_type s
zSoSOptions._convert_to_typec s fdd}t }y(t dd}|j| W dQ R X W n< tk
rx } z td d|jd W Y dd}~X nX |d | ||| |jd
rtd rtd rx>|jd
D ]0\}}|j d
d j
krj j |d | qW dS )zRead 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 | rt|j| }d|j kr6t|jd|d< dddd}xHt|D ]<}d|krn|j|||jdd< ||krL|j|||| < qLW xn|j D ]b\}}t|t r|d kr|jd
d }|j
krt d | d
| d qj || }t
|| qW d S )NrR verbosityZlabelplugoptsZprofiles)nameZ
plugin_optionZprofile-_keywordsr( r zUnknown option 'z' in section 'r. ) has_sectionr# r> r< rV popr replacer" r/ r; printr] r )ZsectionconfigodictZ rename_optsr\ r ) config_filer r r _update_from_section s(
z9SoSOptions.update_from_conf.._update_from_sectionzutf-8)encodingNz+WARNING: Unable to read configuration file z : globalZplugin_optionsr_ skip_plugins.r rT )
r openZ read_fileOSErrorrg r5 rd r r> rY ro r_ r? ) r rj Z componentrk rh fer\ r r )rj r r update_from_conf s "*
zSoSOptions.update_from_confTc C sF x@| j D ]6}t||sqt||dk s,| r| j||d|d qW dS )a 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.
:param prefer_new: ``False`` if new default is not preferred.
NF)r )r% r r r! )r r Z skip_defaultr r& r r r merge s
zSoSOptions.mergec C s: i }x0| j D ]&}t| |}|r*|dkr*d}|||< q W |S )zReturn this ``SoSOptions`` option values as a dictionary of
argument name to value mappings.
:returns: a name:value dictionary of option values.
add_preset
del_presetdescnoteN)rw rx ry rz )r% r )r Z
preset_filterri r& valuer r r r# s
zSoSOptions.dictc sL fddfdddd t j j dd d } fd
d |D S ) zReturn command arguments for this object.
Return a list of the non-default options of this ``SoSOptions``
object in ``sos report`` 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.
FalseNone[]""''0Fr_ T)r| r} r~ r r r )r; r/ )r` r{ Z null_values)r r r has_value s
z%SoSOptions.to_args..has_valuec s* | d kr dS | d kr |d kr d
S | |S )
z8 Filter out preset and null-valued options.
rw rx ry rz Flog_sizeplugin_timeout cmd_timeoutr T)rw rx ry rz )r r r r )r` r{ )r r r
filter_opt$ s z&SoSOptions.to_args..filter_optc S s| | j dr dt|d }|S | jdd} t|r.argifyc S s | d S )Nr r )xr r r C s z$SoSOptions.to_args..)r\ c s$ g | ] \}}||r ||qS r r )r) nr- )r r r r r* D s z&SoSOptions.to_args..)sortedr# r> )r Z opt_itemsr )r r r r r to_args s
zSoSOptions.to_argsN)Fr( r r )TF)T)__name__
__module__ __qualname__r! r' r7 r8 r: rA classmethodrC rX r] ru rv r# r r r r r r s
>
r c @ s e Zd ZdZdddZdS )
SosListOptionz0Allow to specify comma delimited list of pluginsNc C s< t |jd}t|| jr*|t|| j7 }t|| j| d S )Nr+ )r rY r destr )r parser namespacevaluesZ
option_stringr> r r r __call__K s zSosListOption.__call__)N)r r r __doc__r r r r r r G s r c @ s e Zd ZdZdddZdS )
ClusterOptionz6Used to store/manipulate options for cluster profiles.Nc C s" || _ || _|| _|| _|| _d S )N)r` r{ opt_typecluster description)r r` r{ r r r r r r rA U s
zClusterOption.__init__)N)r r r r rA r r r r r R s r N) argparser Z configparserr r
r r r r r r r r s ,