mirrors/uv
1
0
Fork 0
mirror of https://github.com/astral-sh/uv.git synced 2025-07-07 13:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Refactor find_uv_bin and add a better error message

Browse source
This commit is contained in:
Zanie Blue 2025-06-21 06:58:32 -05:00
parent db356ab8d4
commit db969e5a5c
1 changed files with 23 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

44
python/uv/_find_uv.py
View file

@ -5,36 +5,38 @@ import sys
import sysconfig
class UvNotFound(FileNotFoundError): ...
def find_uv_bin() -> str:
"""Return the uv binary path."""
uv_exe = "uv" + sysconfig.get_config_var("EXE")
# Search in the scripts directory for the current prefix
path = os.path.join(sysconfig.get_path("scripts"), uv_exe)
if os.path.isfile(path):
return path
targets = [
# The scripts directory for the current Python
sysconfig.get_path("scripts"),
# The scripts directory for the base prefix (if different)
sysconfig.get_path("scripts", vars={"base": sys.base_prefix}),
# The user scheme scripts directory, e.g., `~/.local/bin`
sysconfig.get_path("scripts", scheme=_user_scheme()),
# Adjacent to the package root, e.g. from, `pip install --target`
os.path.join(os.path.dirname(os.path.dirname(__file__)), "bin"),
]
# If in a virtual environment, also search in the base prefix's scripts directory
if sys.prefix != sys.base_prefix:
path = os.path.join(
sysconfig.get_path("scripts", vars={"base": sys.base_prefix}), uv_exe
)
seen = set()
for target in targets:
if target in seen:
continue
seen.add(target)
path = os.path.join(target, uv_exe)
if os.path.isfile(path):
return path
# Search in the user scheme scripts directory, e.g., `~/.local/bin`
path = os.path.join(sysconfig.get_path("scripts", scheme=_user_scheme()), uv_exe)
if os.path.isfile(path):
return path
# Search in `bin` adjacent to package root (as created by `pip install --target`).
pkg_root = os.path.dirname(os.path.dirname(__file__))
target_path = os.path.join(pkg_root, "bin", uv_exe)
if os.path.isfile(target_path):
return target_path
raise FileNotFoundError(path)
raise UvNotFound(
f"Could not find the uv binary in any of the following locations:\n"
f"{os.linesep.join(f' - {target}' for target in seen)}\n"
)
def _user_scheme() -> str: