mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-16 12:39:16 +08:00
173 lines
6.1 KiB
Python
173 lines
6.1 KiB
Python
|
"""Locations where we look for configs, install stuff, etc"""
|
||
|
|
||
|
import sys
|
||
|
import site
|
||
|
import os
|
||
|
import tempfile
|
||
|
from distutils.command.install import install, SCHEME_KEYS
|
||
|
import getpass
|
||
|
from pip.backwardcompat import get_python_lib, get_path_uid, user_site
|
||
|
import pip.exceptions
|
||
|
|
||
|
|
||
|
DELETE_MARKER_MESSAGE = '''\
|
||
|
This file is placed here by pip to indicate the source was put
|
||
|
here by pip.
|
||
|
|
||
|
Once this package is successfully installed this source code will be
|
||
|
deleted (unless you remove this file).
|
||
|
'''
|
||
|
PIP_DELETE_MARKER_FILENAME = 'pip-delete-this-directory.txt'
|
||
|
|
||
|
def write_delete_marker_file(directory):
|
||
|
"""
|
||
|
Write the pip delete marker file into this directory.
|
||
|
"""
|
||
|
filepath = os.path.join(directory, PIP_DELETE_MARKER_FILENAME)
|
||
|
marker_fp = open(filepath, 'w')
|
||
|
marker_fp.write(DELETE_MARKER_MESSAGE)
|
||
|
marker_fp.close()
|
||
|
|
||
|
|
||
|
def running_under_virtualenv():
|
||
|
"""
|
||
|
Return True if we're running inside a virtualenv, False otherwise.
|
||
|
|
||
|
"""
|
||
|
if hasattr(sys, 'real_prefix'):
|
||
|
return True
|
||
|
elif sys.prefix != getattr(sys, "base_prefix", sys.prefix):
|
||
|
return True
|
||
|
|
||
|
return False
|
||
|
|
||
|
|
||
|
def virtualenv_no_global():
|
||
|
"""
|
||
|
Return True if in a venv and no system site packages.
|
||
|
"""
|
||
|
#this mirrors the logic in virtualenv.py for locating the no-global-site-packages.txt file
|
||
|
site_mod_dir = os.path.dirname(os.path.abspath(site.__file__))
|
||
|
no_global_file = os.path.join(site_mod_dir, 'no-global-site-packages.txt')
|
||
|
if running_under_virtualenv() and os.path.isfile(no_global_file):
|
||
|
return True
|
||
|
|
||
|
def __get_username():
|
||
|
""" Returns the effective username of the current process. """
|
||
|
if sys.platform == 'win32':
|
||
|
return getpass.getuser()
|
||
|
import pwd
|
||
|
return pwd.getpwuid(os.geteuid()).pw_name
|
||
|
|
||
|
def _get_build_prefix():
|
||
|
""" Returns a safe build_prefix """
|
||
|
path = os.path.join(tempfile.gettempdir(), 'pip_build_%s' %
|
||
|
__get_username())
|
||
|
if sys.platform == 'win32':
|
||
|
""" on windows(tested on 7) temp dirs are isolated """
|
||
|
return path
|
||
|
try:
|
||
|
os.mkdir(path)
|
||
|
write_delete_marker_file(path)
|
||
|
except OSError:
|
||
|
file_uid = None
|
||
|
try:
|
||
|
# raises OSError for symlinks
|
||
|
# https://github.com/pypa/pip/pull/935#discussion_r5307003
|
||
|
file_uid = get_path_uid(path)
|
||
|
except OSError:
|
||
|
file_uid = None
|
||
|
|
||
|
if file_uid != os.geteuid():
|
||
|
msg = "The temporary folder for building (%s) is either not owned by you, or is a symlink." \
|
||
|
% path
|
||
|
print (msg)
|
||
|
print("pip will not work until the temporary folder is " + \
|
||
|
"either deleted or is a real directory owned by your user account.")
|
||
|
raise pip.exceptions.InstallationError(msg)
|
||
|
return path
|
||
|
|
||
|
if running_under_virtualenv():
|
||
|
build_prefix = os.path.join(sys.prefix, 'build')
|
||
|
src_prefix = os.path.join(sys.prefix, 'src')
|
||
|
else:
|
||
|
# Note: intentionally NOT using mkdtemp
|
||
|
# See https://github.com/pypa/pip/issues/906 for plan to move to mkdtemp
|
||
|
build_prefix = _get_build_prefix()
|
||
|
|
||
|
## FIXME: keep src in cwd for now (it is not a temporary folder)
|
||
|
try:
|
||
|
src_prefix = os.path.join(os.getcwd(), 'src')
|
||
|
except OSError:
|
||
|
# In case the current working directory has been renamed or deleted
|
||
|
sys.exit("The folder you are executing pip from can no longer be found.")
|
||
|
|
||
|
# under Mac OS X + virtualenv sys.prefix is not properly resolved
|
||
|
# it is something like /path/to/python/bin/..
|
||
|
# Note: using realpath due to tmp dirs on OSX being symlinks
|
||
|
build_prefix = os.path.abspath(os.path.realpath(build_prefix))
|
||
|
src_prefix = os.path.abspath(src_prefix)
|
||
|
|
||
|
# FIXME doesn't account for venv linked to global site-packages
|
||
|
|
||
|
site_packages = get_python_lib()
|
||
|
user_dir = os.path.expanduser('~')
|
||
|
if sys.platform == 'win32':
|
||
|
bin_py = os.path.join(sys.prefix, 'Scripts')
|
||
|
bin_user = os.path.join(user_site, 'Scripts') if user_site else None
|
||
|
# buildout uses 'bin' on Windows too?
|
||
|
if not os.path.exists(bin_py):
|
||
|
bin_py = os.path.join(sys.prefix, 'bin')
|
||
|
bin_user = os.path.join(user_site, 'bin') if user_site else None
|
||
|
default_storage_dir = os.path.join(user_dir, 'pip')
|
||
|
default_config_file = os.path.join(default_storage_dir, 'pip.ini')
|
||
|
default_log_file = os.path.join(default_storage_dir, 'pip.log')
|
||
|
else:
|
||
|
bin_py = os.path.join(sys.prefix, 'bin')
|
||
|
bin_user = os.path.join(user_site, 'bin') if user_site else None
|
||
|
default_storage_dir = os.path.join(user_dir, '.pip')
|
||
|
default_config_file = os.path.join(default_storage_dir, 'pip.conf')
|
||
|
default_log_file = os.path.join(default_storage_dir, 'pip.log')
|
||
|
|
||
|
# Forcing to use /usr/local/bin for standard Mac OS X framework installs
|
||
|
# Also log to ~/Library/Logs/ for use with the Console.app log viewer
|
||
|
if sys.platform[:6] == 'darwin' and sys.prefix[:16] == '/System/Library/':
|
||
|
bin_py = '/usr/local/bin'
|
||
|
default_log_file = os.path.join(user_dir, 'Library/Logs/pip.log')
|
||
|
|
||
|
|
||
|
def distutils_scheme(dist_name, user=False, home=None, root=None):
|
||
|
"""
|
||
|
Return a distutils install scheme
|
||
|
"""
|
||
|
from distutils.dist import Distribution
|
||
|
|
||
|
scheme = {}
|
||
|
d = Distribution({'name': dist_name})
|
||
|
d.parse_config_files()
|
||
|
i = d.get_command_obj('install', create=True)
|
||
|
# NOTE: setting user or home has the side-effect of creating the home dir or
|
||
|
# user base for installations during finalize_options()
|
||
|
# ideally, we'd prefer a scheme class that has no side-effects.
|
||
|
i.user = user or i.user
|
||
|
i.home = home or i.home
|
||
|
i.root = root or i.root
|
||
|
i.finalize_options()
|
||
|
for key in SCHEME_KEYS:
|
||
|
scheme[key] = getattr(i, 'install_'+key)
|
||
|
|
||
|
if running_under_virtualenv():
|
||
|
scheme['headers'] = os.path.join(sys.prefix,
|
||
|
'include',
|
||
|
'site',
|
||
|
'python' + sys.version[:3],
|
||
|
dist_name)
|
||
|
|
||
|
if root is not None:
|
||
|
scheme["headers"] = os.path.join(
|
||
|
root,
|
||
|
os.path.abspath(scheme["headers"])[1:],
|
||
|
)
|
||
|
|
||
|
return scheme
|