3
Q'ícá( ã @ sÚ d Z ddlmZ ddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
ddlmZmZmZmZ ddlmZmZ ejd ejZG d
d d ejZG d d
d
eZG dd dej Z G dd de Z dS )zBase option parser setupé )Úabsolute_importN)Ú strtobool)Ú string_types)Ú configparser)Úlegacy_config_fileÚconfig_basenameÚrunning_under_virtualenvÚsite_config_files)ÚappdirsÚget_terminal_sizez^PIP_c @ sR e Zd ZdZdd Zdd Zddd Zd
d Zd d
Zdd Z dd Z
dd Z dS )ÚPrettyHelpFormatterz4A prettier/less verbose help formatter for optparse.c O s: d|d< d|d< t d d |d< tjj| f|| d S )Né Úmax_help_positioné Úindent_incrementr é Úwidth)r ÚoptparseÚIndentedHelpFormatterÚ__init__)ÚselfÚargsÚkwargs© r ú-/tmp/pip-build-l_bcah40/pip/pip/baseparser.pyr s z PrettyHelpFormatter.__init__c C s | j |ddS )Nz <%s>z, )Ú_format_option_strings)r Úoptionr r r Úformat_option_strings! s z)PrettyHelpFormatter.format_option_stringsú <%s>ú, c C s| g }|j r|j|j d |jr0|j|jd t|dkrH|jd| |j rr|jp^|jj }|j||j dj |S )a
Return a comma-separated list of option strings and metavars.
:param option: tuple of (short opt, long opt), e.g: ('-f', '--format')
:param mvarfmt: metavar format string - evaluated as mvarfmt % metavar
:param optsep: separator
r r Ú )
Ú _short_optsÚappendÚ
_long_optsÚlenÚinsertÚ takes_valueÚmetavarÚdestÚlowerÚjoin)r r ÚmvarfmtÚoptsepÚoptsr' r r r r $ s z*PrettyHelpFormatter._format_option_stringsc C s |dkr dS |d S )NÚOptionsr z:
r )r Úheadingr r r Úformat_heading; s z"PrettyHelpFormatter.format_headingc C s d| j tj|d }|S )zz
Ensure there is only one newline between usage and the first heading
if there is no description.
z
Usage: %s
z )Ú indent_linesÚtextwrapÚdedent)r ÚusageÚmsgr r r Ú format_usage@ s z PrettyHelpFormatter.format_usagec C sV |rNt | jdrd}nd}|jd}|j }| jtj|d}d||f }|S dS d S )NÚmainÚCommandsÚ DescriptionÚ
z z%s:
%s
r )ÚhasattrÚparserÚlstripÚrstripr1 r2 r3 )r Ú descriptionÚlabelr r r Úformat_descriptionH s
z&PrettyHelpFormatter.format_descriptionc C s |r|S dS d S )Nr r )r Úepilogr r r Ú
format_epilogZ s z!PrettyHelpFormatter.format_epilogc s" fdd|j dD }dj|S )Nc s g | ] } | qS r r )Ú.0Úline)Úindentr r ú
<listcomp>b s z4PrettyHelpFormatter.indent_lines.<locals>.<listcomp>r: )Úsplitr* )r ÚtextrF Ú new_linesr )rF r r1 a s z PrettyHelpFormatter.indent_linesN)r r ) Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r0 r6 rA rC r1 r r r r r s
r c @ s e Zd ZdZdd ZdS )Ú UpdatingDefaultsHelpFormatterz®Custom help formatter for use in ConfigOptionParser.
This is updates the defaults before expanding them, allowing
them to show up correctly in the help listing.
c C s( | j d k r| j j| j j tjj| |S )N)r< Ú_update_defaultsÚdefaultsr r Úexpand_default)r r r r r rR m s
z,UpdatingDefaultsHelpFormatter.expand_defaultN)rK rL rM rN rR r r r r rO f s rO c @ s e Zd Zdd Zedd ZdS )ÚCustomOptionParserc O s( | j ||}| jj | jj|| |S )z*Insert an OptionGroup at a given position.)Úadd_option_groupÚ
option_groupsÚpopr% )r Úidxr r Úgroupr r r Úinsert_option_groupu s
z&CustomOptionParser.insert_option_groupc C s. | j dd
}x| jD ]}|j|j qW |S )z<Get a list of all options, including those in option groups.N)Ú option_listrU Úextend)r ÚresÚir r r Úoption_list_all~ s z"CustomOptionParser.option_list_allN)rK rL rM rY Úpropertyr^ r r r r rS s s rS c @ s\ e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dS )ÚConfigOptionParserzsCustom option parser which updates its defaults by checking the
configuration files and environmental variablesFc O sd t j | _|jd| _|jdd| _| j | _| jrB| jj| j | jsLt t
j j | f|| d S )NÚnameÚisolatedF)
r ÚRawConfigParserÚconfigrV ra rb Úget_config_filesÚfilesÚreadÚAssertionErrorr Ú OptionParserr )r r r r r r r s
zConfigOptionParser.__init__c C s t jjdd}|t jkr g S tt}| jsj|rFt jj|rF|j | n$|j t
|j t jj t j
dt t rt jj tjt}t jj|r|j | |S )NÚPIP_CONFIG_FILEFÚpip)ÚosÚenvironÚgetÚdevnullÚlistr rb ÚpathÚexistsr" r r* r
Úuser_config_dirr r ÚsysÚprefix)r Ú config_filerf Úvenv_config_filer r r re s&
z#ConfigOptionParser.get_config_filesc C sL y |j ||S tjk
rF } z td| tjd W Y d d }~X nX d S )Nz*An error occurred during configuration: %sé )Ú check_valuer ÚOptionValueErrorÚprintrt Úexit)r r ÚkeyÚvalÚexcr r r Ú
check_defaultÀ s
z ConfigOptionParser.check_defaultc sj i }x(dj fD ]}|jjj| qW jsH|jjj tjj_ t
}xâ|j D ]Ö\ }|stqfj dkrqfj
d
krt|}nj
dkrÆ|j } fdd|D }nlj
d kr$|jj j }j||}j püf }jpi }j||f|| nj |}||j< qfW x |D ] tj | < qFW d_ |S ) z¢Updates the given defaults with values from the config files and
the environ. Does a little special handling for certain types of
options (lists).ÚglobalNÚ
store_trueÚ store_falseÚcountr" c s g | ]}j |qS r )r )rD Úv)r} r r r r rG é s z7ConfigOptionParser._update_defaults.<locals>.<listcomp>Úcallback)r r r )ra ÚupdateÚnormalize_keysÚget_config_sectionrb Úget_environ_varsr ÚValuesrQ ÚvaluesÚsetÚitemsÚ
get_optionÚactionr rH Úaddr( Úget_opt_stringÚ
convert_valueÚ
callback_argsÚcallback_kwargsr r Úgetattr) r rQ rd ÚsectionÚ late_evalr~ Úopt_strr r r )r} r r r rP Ç s@
z#ConfigOptionParser._update_defaultsc C s@ i }x6|D ].\}}|j dd}|jds0d| }|||< q
W |S )zReturn a config dictionary with normalized keys regardless of
whether the keys were specified in environment variables or in config
filesÚ_ú-z--z--%s)ÚreplaceÚ
startswith)r r Ú
normalizedr} r~ r r r r ü s
z!ConfigOptionParser.normalize_keysc C s | j j|r| j j|S g S )z Get a section of a configuration)rd Ú has_sectionr )r ra r r r r s z%ConfigOptionParser.get_config_sectionc c s< x6t jj D ](\}}tj|r tjd|j |fV q W dS )z@Returns a generator with all environmental vars with prefix PIP_r N)rl rm r Ú_environ_prefix_reÚsearchÚsubr) )r r} r~ r r r r s
z#ConfigOptionParser.get_environ_varsc C sn | j stj| jS | j| jj }x@| j D ]4}|j|j}t |t
r,|j }|j ||||j< q,W tj|S )zOverriding to make updating the defaults after instantiation of
the option parser possible, _update_defaults() does the dirty work.)
Úprocess_default_valuesr r rQ rP ÚcopyÚ_get_all_optionsrn r( Ú
isinstancer r ry )r rQ r Údefaultr r r r Úget_default_values s
z%ConfigOptionParser.get_default_valuesc C s | j tj | jdd| d S )Nr z%s
)Ú print_usagert Ústderrr| )r r5 r r r Úerror# s zConfigOptionParser.errorN)rK rL rM rN rb r re r rP r r r r¨ r« r r r r r` s
(5 r` ) rN Ú
__future__r rt r rl Úrer2 Údistutils.utilr Zpip._vendor.sixr Zpip._vendor.six.movesr Ú
pip.locationsr r r r Ú pip.utilsr
r ÚcompileÚIr r r rO ri rS r` r r r r Ú<module> s O