mirror of https://github.com/pypa/pip
124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
"""Stuff that differs in different Python versions and platform
|
|
distributions."""
|
|
from __future__ import absolute_import, division
|
|
|
|
import os
|
|
import sys
|
|
|
|
from pip._vendor.six import text_type
|
|
|
|
try:
|
|
import ipaddress
|
|
except ImportError:
|
|
try:
|
|
from pip._vendor import ipaddress
|
|
except ImportError:
|
|
import ipaddr as ipaddress
|
|
ipaddress.ip_address = ipaddress.IPAddress
|
|
ipaddress.ip_network = ipaddress.IPNetwork
|
|
|
|
|
|
__all__ = [
|
|
"ipaddress", "uses_pycache", "console_to_str", "native_str",
|
|
"get_path_uid", "stdlib_pkgs", "WINDOWS", "samefile",
|
|
]
|
|
|
|
|
|
if sys.version_info >= (3, 4):
|
|
uses_pycache = True
|
|
from importlib.util import cache_from_source
|
|
else:
|
|
import imp
|
|
uses_pycache = hasattr(imp, 'cache_from_source')
|
|
if uses_pycache:
|
|
cache_from_source = imp.cache_from_source
|
|
else:
|
|
cache_from_source = None
|
|
|
|
|
|
if sys.version_info >= (3,):
|
|
def console_to_str(s):
|
|
try:
|
|
return s.decode(sys.__stdout__.encoding)
|
|
except UnicodeDecodeError:
|
|
return s.decode('utf_8')
|
|
|
|
def native_str(s, replace=False):
|
|
if isinstance(s, bytes):
|
|
return s.decode('utf-8', 'replace' if replace else 'strict')
|
|
return s
|
|
|
|
else:
|
|
def console_to_str(s):
|
|
return s
|
|
|
|
def native_str(s, replace=False):
|
|
# Replace is ignored -- unicode to UTF-8 can't fail
|
|
if isinstance(s, text_type):
|
|
return s.encode('utf-8')
|
|
return s
|
|
|
|
|
|
def get_path_uid(path):
|
|
"""
|
|
Return path's uid.
|
|
|
|
Does not follow symlinks:
|
|
https://github.com/pypa/pip/pull/935#discussion_r5307003
|
|
|
|
Placed this function in compat due to differences on AIX and
|
|
Jython, that should eventually go away.
|
|
|
|
:raises OSError: When path is a symlink or can't be read.
|
|
"""
|
|
if hasattr(os, 'O_NOFOLLOW'):
|
|
fd = os.open(path, os.O_RDONLY | os.O_NOFOLLOW)
|
|
file_uid = os.fstat(fd).st_uid
|
|
os.close(fd)
|
|
else: # AIX and Jython
|
|
# WARNING: time of check vulnerability, but best we can do w/o NOFOLLOW
|
|
if not os.path.islink(path):
|
|
# older versions of Jython don't have `os.fstat`
|
|
file_uid = os.stat(path).st_uid
|
|
else:
|
|
# raise OSError for parity with os.O_NOFOLLOW above
|
|
raise OSError(
|
|
"%s is a symlink; Will not return uid for symlinks" % path
|
|
)
|
|
return file_uid
|
|
|
|
|
|
def expanduser(path):
|
|
"""
|
|
Expand ~ and ~user constructions.
|
|
|
|
Includes a workaround for http://bugs.python.org/issue14768
|
|
"""
|
|
expanded = os.path.expanduser(path)
|
|
if path.startswith('~/') and expanded.startswith('//'):
|
|
expanded = expanded[1:]
|
|
return expanded
|
|
|
|
|
|
# packages in the stdlib that may have installation metadata, but should not be
|
|
# considered 'installed'. this theoretically could be determined based on
|
|
# dist.location (py27:`sysconfig.get_paths()['stdlib']`,
|
|
# py26:sysconfig.get_config_vars('LIBDEST')), but fear platform variation may
|
|
# make this ineffective, so hard-coding
|
|
stdlib_pkgs = {"python", "wsgiref", "argparse"}
|
|
|
|
|
|
# windows detection, covers cpython and ironpython
|
|
WINDOWS = (sys.platform.startswith("win") or
|
|
(sys.platform == 'cli' and os.name == 'nt'))
|
|
|
|
|
|
def samefile(file1, file2):
|
|
"""Provide an alternative for os.path.samefile on Windows/Python2"""
|
|
if hasattr(os.path, 'samefile'):
|
|
return os.path.samefile(file1, file2)
|
|
else:
|
|
path1 = os.path.normcase(os.path.abspath(file1))
|
|
path2 = os.path.normcase(os.path.abspath(file2))
|
|
return path1 == path2
|