o Vi2@sUddlmZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZdd lmZe rUdd lmZmZmZmZmZdd lmZeeZd ed<ej dkZ!ded<    d\d]ddZ"   d^d_d"d#Z#d`d(d)Z$dad,d-Z%  dbdcd1d2Z&ddd4d5Z'ded6d7Z(ded8d9Z)dfd:d;Z*dgd=d>Z+Gd?d@d@Z,dhdBdCZ-didEdFZ.gdGZ/dHedI<djdLdMZ0dkdOdPZ1dQdRdldWdXZ2GdYdZdZeZ3gd[Z4dS)m) annotationsN)suppress)Path) TYPE_CHECKINGFinal)user_data_path) fs_path_id) PythonInfo) PythonSpec)Callable GeneratorIterableMappingSequence) PyInfoCachezFinal[logging.Logger]_LOGGERwin32z Final[bool]IS_WINkeystr | Sequence[str]try_first_withIterable[str] | NonecachePyInfoCache | NoneenvMapping[str, str] | None predicate#Callable[[PythonInfo], bool] | NonereturnPythonInfo | NonecCsBt|tr|gn|}|D]}t||pd|||}r|Sq dS)u Find a Python interpreter matching *key*. Iterates over one or more specification strings and returns the first interpreter that satisfies the spec and passes the optional *predicate*. :param key: interpreter specification string(s) — an absolute path, a version (``3.12``), an implementation prefix (``cpython3.12``), or a PEP 440 specifier (``>=3.10``). When a sequence is given each entry is tried in order. :param try_first_with: executables to probe before the normal discovery search. :param cache: interpreter metadata cache; when ``None`` results are not cached. :param env: environment mapping for ``PATH`` lookup; defaults to :data:`os.environ`. :param predicate: optional callback applied after an interpreter matches the spec. Return ``True`` to accept the interpreter, ``False`` to skip it and continue searching. :return: the first matching interpreter, or ``None`` if no match is found. N) isinstancestr_find_interpreter)rrrrrspecsspec_strresultr!r!O/home/ubuntu/.local/lib/python3.10/site-packages/python_discovery/_discovery.pyget_interpreters r)r# Iterable[str]c Cst|}td|t}|durtjn|}t||||D]7\}}|dur'q|j|f} | |vr1qtd||j ||drP|dusF||rPt d||S| | qdS)Nzfind interpreter for spec %rz proposed %s)impl_must_matchz accepted %s) r from_string_specrinfosetosenvironpropose_interpreterssystem_executable satisfiesdebugadd) rrrrrspecproposed_paths interpreterr+ proposed_keyr!r!r(r$6s"      r$path tested_exesset[str] str | NonecCsbzt|Wn tyYdSwtt|}t|}||vr$dS||tt|S)zWResolve *path* to an absolute path and return it if not yet tested, otherwise ``None``.N) r/lstatOSErrorr#rresolver r5absolute)r:r;resolvedexe_idr!r!r( _check_exeQs  rDexe_rawboolcCs"t|}||vr dS||dS)zDReturn ``True`` and register *exe_raw* if it hasn't been tested yet.FT)r r5)rEr;rCr!r!r( _is_new_exe_s  rGr6r 5Generator[tuple[PythonInfo | None, bool], None, None]ccs|durtjn|}t}|jr*|jdur*t|j|}r(tj|||ddfVdSt|||||EdH|jdur?|jr?dSt ||||EdHt ||EdHdS)a Yield ``(interpreter, impl_must_match)`` candidates for *spec*. :param spec: the parsed interpreter specification to match against. :param try_first_with: executable paths to probe before the standard search. :param cache: interpreter metadata cache; when ``None`` results are not cached. :param env: environment mapping for ``PATH`` lookup; defaults to :data:`os.environ`. NrT) r/r0r.is_absr:rDr from_exe_propose_explicit_propose_from_path_propose_from_uv)r6rrrr;rEr!r!r(r1hsr1Mapping[str, str]ccs|D]}ttt||}rtj|||ddfVq|jdur:t|j|}r8tj|||ddfVdSdSt||||EdHdS)NrIT)rDr#rr@r rKr:_propose_current_and_windows)r6rrrr;py_exerEr!r!r(rLs rLccsjt|}tt|j|r|dfVtr1ddlm}||||D]}tt|j|r0|dfVq!dSdS)NTr)r1)r current_systemrGr# executabler_windowsr1)r6rrr;current_python win_proposer8r!r!r(rPs    rPc cst|}tt|D]B\}}tt|||||D]0\}}t|} t| |} r4td| | | } t| |s:qt j | |d|d} | durL| |fVqq dS)Nzresolved shim %s to %sFraise_on_errorr) path_exe_finder enumerate get_pathsrr4 LazyPathDumpr# _resolve_shimrGPathPythonInforK) r6rrr;find_candidatesposr:exer+rErBr8r!r!r(rMs"  rMccstd}rt|}ntd}r!t|dd}ntdd}|dD]}tjt||d|d}|durB|dfVq,dS) NUV_PYTHON_INSTALL_DIR XDG_DATA_HOMEuvpythonz */bin/pythonFrWT) r/getenvr expanduserrglobr^rKr#)rr uv_python_diruv_python_path xdg_data_homeexe_pathr8r!r!r(rNs  rNGenerator[Path, None, None]c cs|dd}|dur!ztd}Wnttfy tj}Ynw|rQtt|tj D]&}t t | rAt |drA|VWdn1sKwYq,dSdS)NPATHCS_PATH)getr/confstrAttributeError ValueErrordefpathmaprsplitpathseprr?is_dirnextiterdir)rr:entryr!r!r(r[s$   r[c@s eZdZdd d Zdd d ZdS)r\r`intr:rrrOrNonecCs||_||_||_dSN)r`r:r)selfr`r:rr!r!r(__init__s zLazyPathDump.__init__r#c sd|jd|j}|jdr[|d7}|jD]Az,r"Wqtr<|jddd}tfdd |Ds;Wqn j t j @sFWqWn t yPYnw|d 7}|j7}q|S) Nzdiscover PATH[z]=_VIRTUALENV_DEBUGz with =>PATHEXTz.COM;.EXE;.BAT;.CMD;c3s |] }j|VqdSr~)nameupperendswith).0ext file_pathr!r( sz(LazyPathDump.__repr__.. )r`r:rrprzrxrrvanystatst_moder/X_OKr?r)rcontentpathextr!rr(__repr__s*   zLazyPathDump.__repr__N)r`r|r:rrrOrr})rr#)__name__ __module__ __qualname__rrr!r!r!r(r\s r\:Callable[[Path], Generator[tuple[Path, bool], None, None]]cs@|jtjdkd|jtjdkrdd fdd }|S) z^Given a spec, return a function that can be called on a path to find all matching files in it.r)windowsz.exer:rr(Generator[tuple[Path, bool], None, None]c3sT|}|r|dfV|D]}|j}|r'||ddkfVqdS)NFimplre)existsrz fullmatchrrA)r: direct_pathramatchdirectpatr!r( path_exess   z"path_exe_finder..path_exesN)r:rrr) generate_resysplatformstr_spec)r6rr!rr(rYs    rYrlcCsntD]2\}}||}r4tj|d}tj||kr4tj|}tjj|g|R}t|||SqdS)z;Resolve a version-manager shim to the actual Python binary.shimsN)_VERSION_MANAGER_LAYOUTSrpr/r:joindirnamebasename_resolve_shim_to_binary)rlr shims_dir_env versions_pathroot shims_direxe_name versions_dirr!r!r(r] s  r])) PYENV_ROOT)versions) MISE_DATA_DIRinstallsre) ASDF_DATA_DIRrz!list[tuple[str, tuple[str, ...]]]rrrcCsFt|D]}tj||d|}t|r t|tjr |SqdS)Nbin)_active_versionsr/r:rris_fileaccessr)rrrversionrBr!r!r(rs rGenerator[str, None, None]ccs|d}r|dEdHdStt}r!|EdHdS|d}r|EdHdSdSdS)zMYield active Python version strings by reading version-manager configuration. PYENV_VERSION:NrrFsearch_parents)rprv_read_python_version_filercwdr/r:r)r pyenv_versionr pyenv_rootr!r!r(r&s rTrstart str | Pathrlist[str] | Nonecs|} t|rtj|dn|}t|rDt|jdd}fdd|D}r5|WdSWdn1s?wY|sHdSt|j}||krSdS|}q)zIRead a ``.python-version`` file, optionally searching parent directories.Tz.python-versionzutf-8)encodingcs&g|]}|rdsqS)#)strip startswith)rlinevr!r( ;s&z-_read_python_version_file..N)rrxr/r:rropenparent)rrcurrent candidatefhrrr!rr(r4s"  rc@seZdZdZdS)r^zpython info from path.N)rrr__doc__r!r!r!r(r^Esr^)r\r^r)r[r1)NNNN) rrrrrrrrrrrr )NNN) rr#rr*rrrrrrrr )r:r#r;r<rr=)rEr#r;r<rrF)NN) r6r rr*rrrrrrH) r6r rr*rrrrOr;r<rrH) r6r rrrrOr;r<rrH)rrrrOrrH)rrOrrm)r6r rr)rlr#rrOrr=)rr#rr#rrOrr=)rrOrr)rrrrFrr)5 __future__rloggingr/r contextlibrpathlibrtypingrr platformdirsr_compatr _py_infor _py_specr collections.abcr r rrr_cacher getLoggerrr__annotations__rrr)r$rDrGr1rLrPrMrNr[r\rYr]rrrrr^__all__r!r!r!r(sX