U b|K@sddlmZmZmZmZddlmZddlmZm Z ddl Z ddl m Z m Z mZmZmZddlmZddlmZddlZddlZdd lmZdd lmZmZdd lmZmZdd lm Z m!Z!dd l"m#Z#e $e%Z&ddZ'dlddZ(ddZ)ddZ*ddZ+ddZ,e-dddde'e'dd d!d"d#d$d%d&d'gdd(d)dd*d+Z.e-e.e(e dd,d-Z/ere-d.d/d0d1dd2d3d4d4d5d6d7d8d9d:d;ge'e'dd?e-e.d@dAe*e)dBe-e/d@dCe-e/dDdCe-e/dEdCe-e/dFdCe-e/dGdCdHZ0n2e-e.dDdCe-e.dIdJdKe-e.dGdCe-e.dLdMdNdOZ0e#Z1Z2dPdQZ3dmdSdTZ4eddUdVdWZ5dXdYZ6erTe7dZZ8e7d[Z9d\d]Z:n0zdd^l;m Z:WnedfdgZ?dhdiZ@edjdkZAdS)p)absolute_importdivisionprint_functionunicode_literals)contextmanager) lru_cachewrapsN)abspathjoinisfilebasenamedirname)environ)Path) CondaError)shlex_split_unicodeUtf8NamedTemporaryFile)on_win isiterable)win_path_to_unixwhich) path_to_urlcCs|S)z])*)cs:|d}d|td|tddddS)Nrz{0}:{1}rr!r")groupformatlenreplace)Z found_pathr$ root_prefixrr _translation&s z&unix_path_to_win.._translationz:([a-zA-Z]):\\cSsd|dddS)Nrrrz:\)r$)matchrrr,z"unix_path_to_win..)r&countr'resub)rr)Zpath_rer*Z translationrr(runix_path_to_wins.  r1cCs t|dSNz /cygdrive)rrrrrwin_path_to_cygwin2sr3cCs t|dSr2)r1rrrrcygwin_path_to_win6sr4csdfdd|dDS)N c3s|]}|VqdSNr).0line translatorrr ;sz#translate_stream..)r split)streamr:rr9rtranslate_stream:sr>cCsT|dkrd|S|d}|dkr,dt|S|d}|dkrDd|S|d}d|S)a Return the number of bytes n in more human readable form. Examples: >>> human_bytes(42) '42 B' >>> human_bytes(1042) '1 KB' >>> human_bytes(10004242) '9.5 MB' >>> human_bytes(100000004242) '93.13 GB' iz%d Bz%d KBz%.1f MBz%.2f GB)round)nkmgrrr human_bytes>s rDz/bin/echoz.shz 2>/dev/nullr zecho $CONDA_DEFAULT_ENVz echo $PATHzecho $CONDA_PROMPT_MODIFIERZPS1r!zexport z-lz-c)r"r!sourcez${})binpathrEenv_script_suffixnul path_frompath_topathsepprintdefaultenv printpathprintps1 promptvarsepset_var shell_args shell_suffix slash_convert source_setuptest_echo_extra var_formatz_python -c "import os; print(';'.join(os.environ['PATH'].split(';')[1:]))" | cygpath --path -f -)rJrKrGrNz@echoz%{}%z \Scripts\Zcallz 1>NUL 2>&1zset .batz@echo %PROMPT%ZPROMPTzMIF NOT "%CONDA_DEFAULT_ENV%" == "" ( echo %CONDA_DEFAULT_ENV% ) ELSE ( echo()z @echo %PATH%cmd.exe/d/c)r!r"r"r)rErXrGrVrWrIrRrTrHrOrPrMrNexerSrJrKrUrQrLbash.exez /Scripts/)r]rGrJrK)r]bashsh.exezsh.exezsh)rZcygwinr^r_r`rarbdash.)r]rVfish )r]rL)r_rdrbrfcCsddlm}||S)Nr)compute_md5sum)Zgateways.disk.readrh)rrhrrrmd5_files rimd5c CsLddl}||}t|d"}|d}|s.q:||qW5QRX|S)Nrrbi)hashlibnewopenreadupdateZ hexdigest)rmoderlhZfichunkrrr hashsum_files   rt)maxsizecCs^z>ttt}|jr$|j}q|j}|j}tt|}Wnt k rXtj YSX|S)aSince conda is installed into non-root environments as a symlink only and because sys.prefix follows symlinks, this function can be used to get the 'unfollowed' sys.prefix. This value is usually the same as the prefix of the environment into which conda has been symlinked. An example of when this is necessary is when conda looks for external sub-commands in find_commands.py ) nextitersys_current_framesvaluesf_backf_code co_filenamer Exceptionprefix)framecodefilenameZ unfollowedrrrsys_prefix_unfolloweds  rcGs(t|dkr t|dr |d}t|S)a$Properly quote arguments for command line passing. For POSIX uses `shlex.join`, for Windows uses a custom implementation to properly escape metacharacters. :param arguments: Arguments to quote. :type arguments: list of str :return: Quoted arguments. :rtype: str rr)r&r _args_join) argumentsrrrquote_for_shells rz ["%\s^<>&|]z(["%])cs dddfdd|DS)*Return a shell-escaped string from *args*.cSs.|sdSt|s|Std|}d|dS)Nz""z\1\1") _RE_UNSAFEsearch_RE_DBLr0)srrrquote s   z_args_join..quotergc3s|]}|VqdSr6rr7argrrrr;,s_args_join..r argsrrrrs rrcs$ddlmdfdd|DS)rrrrgc3s|]}|VqdSr6rrrrrr;6sr)shlexrr rrrrr2s assertcCsft|tr2|dkr ds2tdnt|}tdt|s@|f}tdd|DrZtdt|}|S)NrFz'Please ensure arguments are not stringszKPlease ensure arguments is not a string; used `shlex_split_unicode()` on itcSsg|] }t|qSr)rrrrr Tsz%massage_arguments..z)Individual arguments must not be iterable) isinstancestrAssertionErrorrlogZwarningranylist)rerrorsrrrmassage_arguments<s  rFc Cst|}|stt|d}nd}d}d}t|dkrDd|dkrDd}trFt} |rpddlm} t| dd d } nt d tt|d d } t d |d dd} |rdnd} | d | | d | | d | | d | | d | |r`ddl m}| d | | d | || d | tj| d | | d | |rz| d| d| d | | || d | |r| d | d|r| d! |dn2td"d#|Drtd$| d% | t|| d | | d& | | j}W5QRX| d'd(|g}nbtd)pVtd*}|dkrjtd+|rtt|d,d-d.d/g}d0}|g}n"t d1tt|d,d/g}d}g}t d |dd2} |rddl m}| d3|dt|d=|}|r| d6| d7 || d8 || d9 |t||rL| d:|rj| d! d;|n| d! t|| j}W5QRX|r|d<|g}n||g}||fS)>Nz.tmpFrr5rT)CONDA_PACKAGE_ROOTshellZcondabinz conda.batZ CONDA_BATwrY)rqrsuffixdeleter@z {}ECHO OFF z{}SET PYTHONIOENCODING=utf-8 z{}SET PYTHONUTF8=1 z_{}FOR /F "tokens=2 delims=:." %%A in ('chcp') do for %%B in (%%A) do set "_CONDA_OLD_CHCP=%%B" z{}chcp 65001 > NUL )CONDA_SOURCE_ROOTz{}SET CONDA_DEV=1 z{}SET PYTHONPATH={} z{}SET CONDA_EXE={} z{}SET _CE_M=-m z{}SET _CE_CONDA=conda z%echo *** environment before *** 1>&2 z SET 1>&2 z{0}CALL "{1}" activate "{2}" z-{}IF %ERRORLEVEL% NEQ 0 EXIT /b %ERRORLEVEL% z$echo *** environment after *** 1>&2 z{0} css|]}d|kVqdS)r5Nrrrrrr;sz'wrap_subprocess_call..aWSupport for scripts where arguments contain newlines not implemented. .. requires writing the script to an external file and knowing how to transform the command-line (e.g. `python -c args` => `python file`) in a tool dependent way, or attempting something like: .. https://stackoverflow.com/a/15032476 (adds unacceptable escapingrequirements)z{0}{1} z{}chcp %_CONDA_OLD_CHCP%>NUL r[r\r_ZshzNo compatible shell found!binZpythonz-mcondaz--devZ CONDA_EXE)rqrrz>&2 export PYTHONPATH= shell.posixhookz.>&2 echo '*** environment before ***' >&2 env z>&2 echo "$({0})" zeval "$({0})" zconda activate {0} {1} z->&2 echo '*** environment after ***' >&2 env rgz-x)rr)rr r r&r get_comspecrrrgetrwriter%rrrx executablerrrnamerr~)r)rdev_modeZdebug_wrapper_scriptsrZuse_system_tmp_pathZ tmp_prefixZ script_callerZ multilinecomspecrZ conda_batZfhZsilencerrZ command_argsZ shell_pathZ conda_exeZdev_argZdev_argsZ hook_quotedrrrwrap_subprocess_callZs            rcCs|ttdddkrttdo0ttdddtdoHttdddfD]}|rLt|rL|td<qtqLtdtdS)zReturns COMSPEC from envvars. Ensures COMSPEC envvar is set to cmd.exe, if not attempt to find it. :raises KeyError: COMSPEC is undefined and cannot be found. :returns: COMSPEC value. :rtype: str ZCOMSPECrrZZ SystemRootZSystem32ZwindirzFcmd.exe could not be found. Looked in SystemRoot and windir env vars. )r rrlowerr r rwarn)rrrrrs   rcstfdd}|S)zl Ensures that the directory exists for functions returning a Path object containing a directory c sb||}t|tr^z|jdddWn6tk r\}ztd|d|W5d}~XYnX|S)NT)parentsexist_okzKError encountered while attempting to create cache directory. Directory:  Exception: )rrmkdirOSErrorr)rkwargsresultexcfuncrrwrappers  z"ensure_dir_exists..wrapper)r)rrrrrensure_dir_existssrc os\zt||}|VWn:tk rN}ztd|dd|W5d}~XYnX|dS)z Allows us to open files while catching any exceptions and raise them as CondaErrors instead. We do this to provide a more informative/actionable error output. z?Error encountered while reading or writing from cache. File: rrN)rnrrclose)rrfprrrr safe_opens  r)r)rj)r)F)BZ __future__rrrr contextlibr functoolsrrZloggingos.pathr r r r r osrpathlibrr/rxrrZ auxlib.compatrrZ common.compatrrZ common.pathrrZ common.urlrZ getLogger__name__rrr1r3r4r>rDdictZunix_shell_baseZmsys2_shell_baseZshellsZurlpathZurl_pathrirtrrcompilerrrr ImportErrorrrrrrrrrrs        I      $ t