U
    
Ha#                     @   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mZ d dl	m
Z
 d dlmZmZ d dlmZmZmZmZ d dl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)Values)AnyListOptional)Command)ERRORSUCCESS)ConfigurationKindget_configuration_fileskinds)PipError)
indent_log)get_progwrite_outputc                   @   s*  e Zd ZdZdZdZddddZeee	 e
dd	d
Zeeee dddZeee	 ddddZeee	 ddddZeee	 ddddZeee	 ddddZeee	 ddddZeddddZddddZeee	 ddddZee	 e	e
edd d!Zddd"d#Zee	d$d%d&ZdS )'ConfigurationCommandah  
    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
    - debug: List the configuration files and values defined under them

    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.
    Ta  
        %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
        %prog [<file-option>] debug
    N)returnc                 C   sl   | 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   )cmd_opts
add_optionparserinsert_option_groupself r#   H/tmp/pip-unpacked-wheel-tx790h60/pip/_internal/commands/configuration.pyadd_options4   s:    z ConfigurationCommand.add_options)optionsargsr   c              
   C   s  | j | 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unsetdebugr   zNeed an action (%s) to perform.z, )r*   r+   r,   r)   )
need_value)isolated	load_only   )list_valuesopen_in_editorget_nameset_name_value
unset_namelist_config_valuesloggererrorjoinsortedr   _determine_filer   r'   r	   isolated_modeconfigurationloadr   )r"   r&   r'   handlersr   r0   er#   r#   r$   runZ   sB    
 
 
zConfigurationCommand.run)r&   r.   r   c                 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)rC   site_config_filer#   r#   r$   	<genexpr>   s   z7ConfigurationCommand._determine_file.<locals>.<genexpr>r1   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$   r<      s$    



z$ConfigurationCommand._determine_filec                 C   s8   | j |ddd t| j D ]\}}td|| qd S )Nr(   r   n%s=%r)_get_n_argsr;   r>   itemsr   r"   r&   r'   rD   rE   r#   r#   r$   r2      s    z ConfigurationCommand.list_valuesc                 C   s*   | j |ddd}| j|}td| d S )Nz
get [name]r1   rQ   z%s)rT   r>   	get_valuer   rV   r#   r#   r$   r4      s    zConfigurationCommand.get_namec                 C   s.   | j |ddd\}}| j|| |   d S )Nzset [name] [value]   rQ   )rT   r>   	set_value_save_configurationrV   r#   r#   r$   r5      s    z#ConfigurationCommand.set_name_valuec                 C   s(   | j |ddd}| j| |   d S )Nzunset [name]r1   rQ   )rT   r>   unset_valuerZ   )r"   r&   r'   rD   r#   r#   r$   r6      s    zConfigurationCommand.unset_namec                 C   s   | j |ddd |   t| j D ]T\}}td| |D ]<}t , tj	|}td|| |rn| 
| W 5 Q R X q<q&dS )z9List config key-value pairs across different config filesr-   r   rQ   %s:z%s, exists: %rN)rT   print_env_var_valuesr;   r>   Ziter_config_filesr   r   rG   rH   rI   print_config_file_values)r"   r&   r'   variantfilesfnameZfile_existsr#   r#   r$   r7      s    
 z'ConfigurationCommand.list_config_values)r_   r   c              
   C   s<   | j | D ]&\}}t  td|| W 5 Q R X qdS )z.Get key-value pairs from the file of a variantz%s: %sN)r>   Zget_values_in_configrU   r   r   )r"   r_   namerE   r#   r#   r$   r^      s
    z-ConfigurationCommand.print_config_file_valuesc              	   C   sR   t dd t 8 t| j D ]"\}}d|  }t d|| q W 5 Q R X dS )z5Get key-values pairs present as environment variablesr\   env_varPIP_rS   N)r   r   r;   r>   Zget_environ_varsupper)r"   rD   rE   rc   r#   r#   r$   r]      s
    
z)ConfigurationCommand.print_env_var_valuesc              
   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CalledProcessErrorformat
returncode)r"   r&   r'   r   ra   rA   r#   r#   r$   r3      s    

z#ConfigurationCommand.open_in_editor)r'   examplerR   r   c                 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 {}")r1   r   N)rP   rk   r   r   )r"   r'   rm   rR   msgr#   r#   r$   rT      s      z ConfigurationCommand._get_n_argsc                 C   s:   z| j   W n& tk
r4   td tdY nX d S )Nz:Unable to save configuration. Please report this as a bug.zInternal Error.)r>   save	Exceptionr8   	exceptionr   r!   r#   r#   r$   rZ      s    z(ConfigurationCommand._save_configuration)r&   r   c                 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   rG   environr   )r"   r&   r#   r#   r$   rf     s    




z&ConfigurationCommand._determine_editor)__name__
__module____qualname____doc__ignore_require_venvusager%   r   r   strintrB   boolr   r
   r<   r2   r4   r5   r6   r7   r^   r]   r3   r   rT   rZ   rf   r#   r#   r#   r$   r      s"   
&-r   )loggingrG   rh   optparser   typingr   r   r   pip._internal.cli.base_commandr   pip._internal.cli.status_codesr   r   pip._internal.configurationr	   r
   r   r   pip._internal.exceptionsr   pip._internal.utils.loggingr   pip._internal.utils.miscr   r   	getLoggerrs   r8   r   r#   r#   r#   r$   <module>   s   
