!xj$ddlmZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZedZGd d eeZed Gd dZd(dZd(dZd)dZd*dZd(dZd+dZd,d-dZ d.d/d%Z d.d0d'ZdS)1) annotationsN) dataclass)Enum) DistributionPackageNotFoundError distribution)Path)Literal)addlockc2eZdZdZdZdZdZdZdZdZ dZ d Z d S) Installerpipzuv-pipz uv-projectzuv-toolpipxbrewcondamambaunknownN) __name__ __module__ __qualname__PIPUV_PIPUVUV_TOOLPIPXBREWCONDAMAMBAUNKNOWNIC:\Projects\Historical\venv\Lib\site-packages\detect_installer/_detect.pyrrs; C F BG D D E EGGGr"rT)frozenc$eZdZUded<ded<dS) InstallerInfor installer str | None upgrade_cmdN)rrr__annotations__r!r"r#r&r&s*r"r&returnboolc:dtjvp dtjvS)zDCheck whether the current environment is inside a pipx-managed venv.z pipx/venvsz pipx\venvssysprefixr!r"r#_is_pipx_environmentr1 s 3: % D#*)DDr"c:dtjvp dtjvS)zGCheck whether the current environment is inside a uv tool-managed venv.z /uv/tools/z \uv\tools\r.r!r"r#_is_uv_tool_environmentr3&s 3: % FCJ)FFr"cptjdr tjn tjS)zDReturn MAMBA if the Mamba executable is configured, otherwise CONDA. MAMBA_EXE)osenvirongetrrrr!r"r#_detect_conda_variantr9,s%!jnn[99 N9??yNr"Installer | Nonectjdd}|rgtjt jtj|rtSt j }| dd d}dtfd|DrtSdS)a7Detect whether the current environment is a Conda (or Mamba) environment. Checks in two ways: - CONDA_PREFIX env var being set (i.e. a conda env is activated) - sys.prefix path containing a known conda distribution directory name Returns the installer, or None if not a conda environment. CONDA_PREFIX\/)r miniconda miniforge mambaforgec3BK|]}|VdSN) startswith).0partconda_prefixess r# z,_detect_conda_environment..Fs/ = =t4??> * * = = = = = =r"N) r6r7r8pathnormcaser/r0rEr9lowerreplacesplitany) conda_prefix prefix_lowerpartsrHs @r#_detect_conda_environmentrS2s:>>."55L'((44?? &&'%&&&:##%%L  s + + 1 1# 6 6EFN = = = =u = = ==='$&&& 4r"cvtjtfddDS)zBCheck whether the Python executable lives under a Homebrew prefix.c3 K|]}|vV dSrDr!)rFpexes r#rIz'_is_brew_environment..Os8Sr")z/opt/homebrew/z/usr/local/cellar/z/home/linuxbrew/)r/ executablerLrO)rWs@r#_is_brew_environmentrYLsN .   C V  r"distrr(c|dx}dS|}|r|ndS)zARead the INSTALLER metadata file from a distribution, if present. INSTALLERN) read_textstriprL)rZvalues r#_get_installer_metadatar`TsH ,,,5t KKMM   ! !E #55t#r" max_depthintcttj}t|D]#}|j}|dz rdS$dS)zHCheck whether a uv.lock file exists in a parent directory of sys.prefix.zuv.lockTF)r r/r0rangeparentexists)rbrf_s r# _has_uv_lockri_s]#*  F 9   Y  & & ( ( 44  5r"r r' package_namestruv_upgrade_strategyUvUpgradeStrategycH|tjkr|dkrd|Sd|d|Stjd|tjd|tjd|tjd|tjd |tjd |tjd |i}| |S) zFReturn the shell command to upgrade a package for the given installer.r zuv lock --upgrade-package zuv add z --upgrade-package zpip install -U zuv pip install --upgrade zuv tool upgrade z pipx upgrade z brew upgrade z conda update z mamba update ) rrrrrrrrrr8)r'rjrlcommandss r#_get_upgrade_cmdrpmsIL  & ( (> >> >HHH,HHH  777DlDD._results(  Y 6I J J   r"Nuvr)r'rr+r&)rrr1rrr3rrSrYrr`rirrrr&r )rjrlrtrZ conda_resultmetadata_values`` r#detect_installerrxs`       L)) tt'wy~&&&  *wy()))0222|%w|$$$'wy~&&&-T22N >> )79<(( (wy'(((wy}%%% *D 1 11s  )))r+r,)r+r)r+r:)rZrr+r()ra)rbrcr+r,)r )r'rrjrkrlrmr+r()rjrkrlrmr+rq) __future__rr6r/ dataclassesrenumrimportlib.metadatarrrpathlibr typingr rmrkrr&r1r3r9rSrYr`rirprxr!r"r#rs"""""" !!!!!!OOOOOOOOOOM*     T    $ EEEE GGGG OOOO 4$$$$     ".3#####8.3-2-2-2-2-2-2-2r"