B Uvg3@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZmZddlmZmZddlmZerdd lmZmZmZmZmZmZmZejZed eZ e!e"Z#d d Z$d dZ%eddddddZ&Gddde'Z(dS)a Configuration management setup Some terminology: - name As written in config files. - value Value associated with a name - key Name combined with it's section (section.name) - variant A single word describing where the configuration key-value pair came from N)six) configparser)ConfigurationError!ConfigurationFileCouldNotBeLoaded)legacy_config_filenew_config_filerunning_under_virtualenvsite_config_filesvenv_config_file) ensure_direnum)MYPY_CHECK_RUNNING)AnyDictIterableListNewTypeOptionalTupleKindcCs*|dd}|dr&|dd}|S)zFMake a name consistent regardless of source (environment or file) _-z--N)lowerreplace startswith)namerhsz*Configuration.__init__..cSsi|] }i|qSrr)r1r2rrrr3ks)superr-__init__kindsr(r)r*rformatjoinmapreprisolated load_onlyr+r,_override_order_ignore_env_names_parsers_config_modified_parsers)selfr;r<_valid_load_only) __class__rrr5Rs   zConfiguration.__init__cCs||js|dS)zELoads configuration from configuration files and environment N)_load_config_filesr;_load_environment_vars)rBrrrloadpszConfiguration.loadcCs8|jdk stdy |dStk r2dSXdS)z@Returns the file with highest priority in configuration Nz)Need to be specified a file to be editingr)r<AssertionError_get_parser_to_modify IndexError)rBrrrget_file_to_editxs   zConfiguration.get_file_to_editcCs |jS)z`Returns key-value pairs like dict.items() representing the loaded configuration ) _dictionaryitems)rBrrrrMszConfiguration.itemscCs2y |j|Stk r,td|YnXdS)z,Get a value from the configuration. zNo such key - {}N)rLKeyErrorrr7)rBkeyrrr get_values zConfiguration.get_valuecCsj||\}}|dk rJt|\}}||s<||||||||j|j|<|||dS)z-Modify a value in the configuration. N) _ensure_have_load_onlyrIr# has_section add_sectionsetr@r<_mark_as_modified)rBrOvaluefnameparsersectionrrrr set_values    zConfiguration.set_valuec Cs|||j|jkr&td||\}}|dk rt|\}}d}||r`|||}|rt | |}yt |}Wnt k rd}YnX|dkr|||||ntd|j|j|=dS)z,Unset a value in the configuration. zNo such key - {}NFz4Fatal Internal error [id=1]. Please report as a bug.)rQr@r<rr7rIr#rR remove_optioniterrMrnext StopIterationremove_sectionrU) rBrOrWrXrYrmodified_something section_itervalrrr unset_values*      zConfiguration.unset_valuec Cs\|xN|jD]D\}}td|ttj|t|d}| |WdQRXqWdS)z)Save the currentin-memory state. z Writing to %swN) rQrAloggerinfor ospathdirnameopenwrite)rBrWrXfrrrsaves   zConfiguration.savecCs$|jdkrtdtd|jdS)Nz'Needed a specific file to be modifying.z$Will be working with %s variant only)r<rredebug)rBrrrrQs z$Configuration._ensure_have_load_onlycCs(i}x|jD]}||j|q W|S)zWdS)z5Loads configuration from configuration files rr!zZSkipping loading configuration files due to environment's PIP_CONFIG_FILE being os.devnullNz Skipping file '%s' (variant: %s)) dict_iter_config_filesr6r+rgdevnullrernrMr< _load_filer?append)rB config_filesr2filesrWrXrrrrEs    z Configuration._load_config_filescCsPtd||||}x2|D]&}||}|j||||q"W|S)Nz'For variant '%s', will try loading '%s')rern_construct_parsersectionsrMr@ro_normalized_keys)rBr2rWrXrYrMrrrrts   zConfiguration._load_filec Cst}tj|r|y||WnXtk rNtdt d|dYn.tj k rz}zt|dWdd}~XYnX|S)Nzcontains invalid {} charactersF)reasonrW)error) rRawConfigParserrgrhexistsreadUnicodeDecodeErrorrr7localegetpreferredencodingError)rBrWrXr|rrrrxs   zConfiguration._construct_parsercCs"|jtj|d|dS)z7Loads configuration from environment variables z:env:N)r@r6r,rorz_get_environ_vars)rBrrrrF2s z$Configuration._load_environment_varscCs2i}x(|D] \}}|dt|}|||<q W|S)zNormalizes items to construct a dictionary with normalized keys. This routine is where the names become keys and are made the same regardless of source - configuration files or environment. r )r)rBrYrM normalizedrrbrOrrrrz:s  zConfiguration._normalized_keysccsVxPtjD]B\}}|do2|dd|jk}|r |dd|fVq WdS)z@Returns a generator with all environmental vars with prefix PIP_PIP_N)rgenvironrMrrr>)rBrOrbshould_be_yieldedrrrrGs  zConfiguration._get_environ_varsccstjdd}|dk r&tj|gfVn tjgfVtjttfV|j oZ|oXtj | }|rptj t t gfVtrtjtgfVdS)zYields variant and configuration files associated with it. This should be treated like items of a dictionary. PIP_CONFIG_FILEN)rgrgetr6r+r)listr r;rhr~r(rrrr*r )rB config_fileshould_load_user_configrrrrrSs  z Configuration._iter_config_filescCs |j|j}|std|dS)Nz4Fatal Internal error [id=2]. Please report as a bug.r.)r?r<r)rBparsersrrrrIqs  z#Configuration._get_parser_to_modifycCs"||f}||jkr|j|dS)N)rAru)rBrWrXfile_parser_tuplerrrrUs zConfiguration._mark_as_modified)N)__name__ __module__ __qualname____doc__r5rGrKrMrPrZrcrmrQpropertyrLrErtrxrFrzrrrrIrU __classcell__rr)rDrr-Ds(   '    r-))rrloggingrg pip._vendorrZpip._vendor.six.movesrpip._internal.exceptionsrrpip._internal.locationsrrrr r pip._internal.utils.miscr r pip._internal.utils.typingr typingrrrrrrrr}strr getLoggerrrerr#r6objectr-rrrr s.   $