mirror of
https://github.com/kennethreitz/pipenv.git
synced 2026-06-05 22:50:18 +00:00
Dan Ryan
119 lines
3.0 KiB
Python
119 lines
3.0 KiB
Python
# -*- coding=utf-8 -*-
|
|
from __future__ import absolute_import, print_function
|
|
|
|
import locale
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from fnmatch import fnmatch
|
|
from itertools import chain
|
|
|
|
import attr
|
|
import six
|
|
|
|
import vistir
|
|
|
|
from vistir.compat import Path
|
|
|
|
from .exceptions import InvalidPythonVersion
|
|
|
|
|
|
PYTHON_IMPLEMENTATIONS = ("python", "ironpython", "jython", "pypy")
|
|
|
|
KNOWN_EXTS = {"exe", "py", "fish", "sh", ""}
|
|
KNOWN_EXTS = KNOWN_EXTS | set(
|
|
filter(None, os.environ.get("PATHEXT", "").split(os.pathsep))
|
|
)
|
|
|
|
|
|
def get_python_version(path):
|
|
"""Get python version string using subprocess from a given path."""
|
|
version_cmd = [path, "-c", "import sys; print(sys.version.split()[0])"]
|
|
try:
|
|
out, _ = vistir.misc.run(version_cmd)
|
|
except OSError:
|
|
raise InvalidPythonVersion("%s is not a valid python path" % path)
|
|
if not out:
|
|
raise InvalidPythonVersion("%s is not a valid python path" % path)
|
|
return out
|
|
|
|
|
|
def optional_instance_of(cls):
|
|
return attr.validators.optional(attr.validators.instance_of(cls))
|
|
|
|
|
|
def path_and_exists(path):
|
|
return attr.validators.instance_of(Path) and path.exists()
|
|
|
|
|
|
def path_is_executable(path):
|
|
return os.access(str(path), os.X_OK)
|
|
|
|
|
|
def path_is_known_executable(path):
|
|
return (
|
|
path_is_executable(path)
|
|
or os.access(str(path), os.R_OK)
|
|
and path.suffix in KNOWN_EXTS
|
|
)
|
|
|
|
|
|
def looks_like_python(name):
|
|
rules = ["*python", "*python?", "*python?.?", "*python?.?m"]
|
|
match_rules = []
|
|
for rule in rules:
|
|
match_rules.extend(
|
|
[
|
|
"{0}.{1}".format(rule, ext) if ext else "{0}".format(rule)
|
|
for ext in KNOWN_EXTS
|
|
]
|
|
)
|
|
if not any(name.lower().startswith(py_name) for py_name in PYTHON_IMPLEMENTATIONS):
|
|
return False
|
|
return any(fnmatch(name, rule) for rule in match_rules)
|
|
|
|
|
|
def path_is_python(path):
|
|
return path_is_executable(path) and looks_like_python(path.name)
|
|
|
|
|
|
def ensure_path(path):
|
|
"""Given a path (either a string or a Path object), expand variables and return a Path object.
|
|
|
|
:param path: A string or a :class:`~pathlib.Path` object.
|
|
:type path: str or :class:`~pathlib.Path`
|
|
:return: A fully expanded Path object.
|
|
:rtype: :class:`~pathlib.Path`
|
|
"""
|
|
|
|
if isinstance(path, Path):
|
|
path = path.as_posix()
|
|
path = Path(os.path.expandvars(path))
|
|
try:
|
|
path = path.resolve()
|
|
except OSError:
|
|
path = path.absolute()
|
|
return path
|
|
|
|
|
|
def _filter_none(k, v):
|
|
if v:
|
|
return True
|
|
return False
|
|
|
|
|
|
def filter_pythons(path):
|
|
"""Return all valid pythons in a given path"""
|
|
if not isinstance(path, Path):
|
|
path = Path(str(path))
|
|
if not path.is_dir():
|
|
return path if path_is_python(path) else None
|
|
return filter(lambda x: path_is_python(x), path.iterdir())
|
|
|
|
|
|
def unnest(item):
|
|
if isinstance(next((i for i in item), None), (list, tuple)):
|
|
return chain(*filter(None, item))
|
|
return chain(filter(None, item))
|