mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-02 06:42:02 +00:00
6683ed49bc
Prefer the version from a currently active virtualenv over a version from `pip install --user`. Add the .exe extension on Windows, and find the path for `pip install --user` correctly on Windows. Signed-off-by: Anders Kaseorg <andersk@mit.edu>
34 lines
869 B
Python
34 lines
869 B
Python
import os
|
|
import sys
|
|
import sysconfig
|
|
from pathlib import Path
|
|
|
|
|
|
def find_ruff_bin() -> Path:
|
|
"""Return the ruff binary path."""
|
|
|
|
ruff_exe = "ruff" + sysconfig.get_config_var("EXE")
|
|
|
|
path = Path(sysconfig.get_path("scripts")) / ruff_exe
|
|
if path.is_file():
|
|
return path
|
|
|
|
if sys.version_info >= (3, 10):
|
|
user_scheme = sysconfig.get_preferred_scheme("user")
|
|
elif os.name == "nt":
|
|
user_scheme = "nt_user"
|
|
elif sys.platform == "darwin" and sys._framework:
|
|
user_scheme = "osx_framework_user"
|
|
else:
|
|
user_scheme = "posix_user"
|
|
|
|
path = Path(sysconfig.get_path("scripts", scheme=user_scheme)) / ruff_exe
|
|
if path.is_file():
|
|
return path
|
|
|
|
raise FileNotFoundError(path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ruff = find_ruff_bin()
|
|
sys.exit(os.spawnv(os.P_WAIT, ruff, ["ruff", *sys.argv[1:]]))
|