U
��.e% � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZmZ ddlmZ e�e�ZG d d
� d
ej�ZG d d � d e�ZG d
d� dej�ZG dd� de�Zdd� ZdS )zBase option parser setup� )�absolute_importN)� strtobool)� string_types)�
UNKNOWN_ERROR)�
Configuration�ConfigurationError)�get_terminal_sizec @ 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� Zmax_help_position� Zindent_incrementr � �width)r �optparse�IndentedHelpFormatter�__init__)�self�args�kwargs� r ��, )�_format_option_strings�r �optionr r r �format_option_strings! s z)PrettyHelpFormatter.format_option_stringsr r c C s| g }|j r|�|j d � |jr0|�|jd � t|�dkrH|�d|� |�� rr|jp^|j�� }|�||�� � d� |�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 � )
Z _short_opts�appendZ
_long_opts�len�insertZ takes_value�metavar�dest�lower�join)r r ZmvarfmtZoptsepZoptsr r r r r $ s z*PrettyHelpFormatter._format_option_stringsc C s |dkr dS |d S )NZOptionsr z:
r )r Zheadingr r r �format_heading; s z"PrettyHelpFormatter.format_headingc C s d| � t�|�d� }|S )zz
Ensure there is only one newline between usage and the first heading
if there is no description.
z
Usage: %s
� )� indent_lines�textwrap�dedent)r Zusage�msgr r r � format_usage@ s z PrettyHelpFormatter.format_usagec C sV |rNt | jd�rd}nd}|�d�}|�� }| �t�|�d�}d||f }|S dS d S )N�mainZCommandsZ Description�
r% z%s:
%s
r )�hasattr�parser�lstrip�rstripr& r' r( )r � descriptionZlabelr r r �format_descriptionH s
z&PrettyHelpFormatter.format_descriptionc C s |r|S dS d S )Nr r )r Zepilogr r r �
format_epilogZ s z!PrettyHelpFormatter.format_epilogc s" � fdd�|� d�D �}d�|�S )Nc s g | ] }� | �qS r r )�.0�line��indentr r �
b s z4PrettyHelpFormatter.indent_lines..r, )�splitr# )r �textr7 Z new_linesr r6 r r& a s z PrettyHelpFormatter.indent_linesN)r r ) �__name__�
__module__� __qualname__�__doc__r r r r$ r* r2 r3 r& 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� tj�| |�S )N)r. �_update_defaults�defaultsr r �expand_defaultr r r r rB m s
z,UpdatingDefaultsHelpFormatter.expand_defaultN)r; r< r= r> rB r r r r r? f s r? c @ s e Zd Zdd� Zedd� �ZdS )�CustomOptionParserc O s( | j ||�}| j�� | j�||� |S )z*Insert an OptionGroup at a given position.)Zadd_option_group�
option_groups�popr )r �idxr r �groupr r r �insert_option_groupu s
z&CustomOptionParser.insert_option_groupc C s* | j dd� }| jD ]}|�|j � q|S )zt�d|� q$|�dd�\}}||kr$|| �||f� q$|D ] }|| D ]\}}||fV qzqnd S )N�globalz:env:c S s i | ]
}|g �qS r r )r4 rO r r r �
� s zGConfigOptionParser._get_ordered_configuration_items..z7Ignoring configuration key '%s' as it's value is empty.�.r )rO rQ �items�logger�debugr9 r )r Zoverride_orderZ
section_itemsZ section_keyrZ ZsectionrY r r r � _get_ordered_configuration_items� s �z3ConfigOptionParser._get_ordered_configuration_itemsc sH t ��j��_t� }��� D �]\� }��d� ���dkr>q �jdkr�z t|�}W n, t k
r� t
�j� |�}�� |� Y nX n��jdkr�|� � }� ��fdd�|D �}nh�jdk�r|�
�j� ��� }��||�}�jp�d}�jp�i }�j�||�f|�|� n���� |�}||�j< q |D ]� t�j� �|� < �q&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).z--N)�
store_true� store_false�countr c s g | ]}�� �� |��qS r )r\ )r4 �v�rY r r r r r8 � s z7ConfigOptionParser._update_defaults..�callbackr )r �ValuesrA �values�setrc Z
get_option�actionr �
ValueError� invalid_config_error_message�errorr9 �addr! �get_opt_stringZ
convert_valueZ
callback_argsZcallback_kwargsri r\ �getattr)r rA Z late_evalrZ Z error_msg�opt_strr r r rh r r@ � s@
�
z#ConfigOptionParser._update_defaultsc
C s� | j st�| j�S z| j�� W n2 tk
rR } z| �tt |�� W 5 d}~X Y nX | �
| j� � �}| � � D ]4}|�
|j�}t|t�rl|�� }|�||�||j< qlt�|�S )z�Overriding to make updating the defaults after instantiation of
the option parser possible, _update_defaults() does the dirty work.N)Zprocess_default_valuesr rj rA rQ �loadr rX r �strr@ �copyZ_get_all_options�getr! �
isinstancer rr rU )r �errrA r �defaultrt r r r �get_default_values� s "
z%ConfigOptionParser.get_default_valuesc C s | � tj� | �td| � d S )Nz%s
)Z print_usagerW �stderrrX r )r r) r r r rp � s zConfigOptionParser.errorN)
r; r< r= r> r r\ rc r@ r| rp r r r r rN � s 1rN c C s | dkrd� ||�S d� ||�S )zQReturns a better error message when invalid configuration option
is provided.)rd re zo{0} is not a valid value for {1} option, please specify a boolean value like yes/no, true/false or 1/0 instead.z[{0} is not a valid value for {1} option, please specify a numerical value like 1/0 instead.)�format)rm rY rZ r r r ro � s � �ro ) r> Z
__future__r Zloggingr rW r' Zdistutils.utilr Zpip._vendor.sixr Z pip._internal.cli.status_codesr Zpip._internal.configurationr r Zpip._internal.utils.compatr Z getLoggerr; ra r r r? rS rC rN ro r r r r � s
O
w