U
    @vg:                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm	Z	m
Z
mZ d dlmZ d dlmZmZ e eZG dd deZdS )	    N)Command)ERRORSUCCESS)Configurationget_configuration_fileskinds)PipError)get_progwrite_outputc                       sx   e Zd ZdZdZdZ fddZdd Zdd	 Zd
d Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Z  ZS )ConfigurationCommanda9  Manage local and global configuration.

        Subcommands:

        list: List the active configuration (or from the file specified)
        edit: Edit the configuration file in an editor
        get: Get the value associated with name
        set: Set the name=value
        unset: Unset the value associated with name

        If none of --user, --global and --site are passed, a virtual
        environment configuration file is used if one is active and the file
        exists. Otherwise, all modifications happen on the to the user file by
        default.
    Tz
        %prog [<file-option>] list
        %prog [<file-option>] [--editor <editor-path>] edit

        %prog [<file-option>] get name
        %prog [<file-option>] set name value
        %prog [<file-option>] unset name
    c                    s   t t| j|| d | _| jjdddd dd | jjdddd	d
d | jjdddd	dd | jjdddd	dd | jd| j d S )Nz--editoreditorstorez\Editor to use to edit the file. Uses VISUAL or EDITOR environment variables if not provided.)destactiondefaulthelpz--globalglobal_file
store_trueFz+Use the system-wide configuration file onlyz--user	user_filez$Use the user configuration file onlyz--site	site_filez3Use the current environment configuration file onlyr   )superr   __init__configurationcmd_opts
add_optionparserinsert_option_group)selfargskwargs	__class__ H/tmp/pip-unpacked-wheel-gw11q0wt/pip/_internal/commands/configuration.pyr   0   s>    zConfigurationCommand.__init__c              
   C   s  | j | j| j| j| jd}|r*|d |krHtddt	| t
S |d }z| j||dkd}W n: tk
r } zt|jd  t
 W Y S d }~X Y nX t|j|d| _| j  z|| ||dd   W n< tk
r } zt|jd  t
 W Y S d }~X Y nX tS )	N)listeditgetsetunsetr   zNeed an action ({}) to perform.z, )r&   r'   r(   r%   )
need_value)isolated	load_only   )list_valuesopen_in_editorget_nameset_name_value
unset_nameloggererrorformatjoinsortedr   _determine_filer   r   r   isolated_moder   loadr   )r   optionsr   handlersr   r+   er"   r"   r#   runZ   s>    	 
 
zConfigurationCommand.runc                 C   s   dd t j|jft j|jft j|jffD }|s`|s8d S tdd t t j D rXt jS t jS nt	|dkrt|d S t
dd S )Nc                 S   s   g | ]\}}|r|qS r"   r"   ).0keyvaluer"   r"   r#   
<listcomp>   s    z8ConfigurationCommand._determine_file.<locals>.<listcomp>c                 s   s   | ]}t j|V  qd S )N)ospathexists)r>   site_config_filer"   r"   r#   	<genexpr>   s   z7ConfigurationCommand._determine_file.<locals>.<genexpr>r,   r   zLNeed exactly one file to operate upon (--user, --site, --global) to perform.)r   USERr   GLOBALr   SITEr   anyr   lenr   )r   r:   r)   Zfile_optionsr"   r"   r#   r7      s$    



z$ConfigurationCommand._determine_filec                 C   s8   | j |ddd t| j D ]\}}td|| qd S )Nr$   r   nz%s=%r)_get_n_argsr6   r   itemsr
   r   r:   r   r?   r@   r"   r"   r#   r-      s    z ConfigurationCommand.list_valuesc                 C   s*   | j |ddd}| j|}td| d S )Nz
get [name]r,   rL   z%s)rN   r   	get_valuer
   rP   r"   r"   r#   r/      s    zConfigurationCommand.get_namec                 C   s.   | j |ddd\}}| j|| |   d S )Nzset [name] [value]   rL   )rN   r   	set_value_save_configurationrP   r"   r"   r#   r0      s    z#ConfigurationCommand.set_name_valuec                 C   s(   | j |ddd}| j| |   d S )Nzunset [name]r,   rL   )rN   r   unset_valuerT   )r   r:   r   r?   r"   r"   r#   r1      s    zConfigurationCommand.unset_namec              
   C   sp   |  |}| j }|d kr$tdzt||g W n4 tjk
rj } ztd|jW 5 d }~X Y nX d S )Nz%Could not determine appropriate file.z*Editor Subprocess exited with exit code {})	_determine_editorr   get_file_to_editr   
subprocess
check_callCalledProcessErrorr4   
returncode)r   r:   r   r   fnamer<   r"   r"   r#   r.      s    

z#ConfigurationCommand.open_in_editorc                 C   s<   t ||kr$d|t |}t||dkr4|d S |S dS )zJHelper to make sure the command got the right number of arguments
        zJGot unexpected number of arguments, expected {}. (example: "{} config {}")r,   r   N)rK   r4   r	   r   )r   r   ZexamplerM   msgr"   r"   r#   rN      s      z ConfigurationCommand._get_n_argsc                 C   s>   z| j   W n* tk
r8   tjddd tdY nX d S )Nz:Unable to save configuration. Please report this as a bug.r,   )exc_infozInternal Error.)r   save	Exceptionr2   r3   r   )r   r"   r"   r#   rT      s    z(ConfigurationCommand._save_configurationc                 C   sD   |j d k	r|j S dtjkr$tjd S dtjkr8tjd S tdd S )NZVISUALZEDITORz"Could not determine editor to use.)r   rB   environr   )r   r:   r"   r"   r#   rV      s    




z&ConfigurationCommand._determine_editor)__name__
__module____qualname____doc__ignore_require_venvusager   r=   r7   r-   r/   r0   r1   r.   rN   rT   rV   __classcell__r"   r"   r    r#   r      s   	*+r   )loggingrB   rX   pip._internal.cli.base_commandr   pip._internal.cli.status_codesr   r   pip._internal.configurationr   r   r   pip._internal.exceptionsr   pip._internal.utils.miscr	   r
   	getLoggerrb   r2   r   r"   r"   r"   r#   <module>   s   
