| """
|
| Package resource API
|
| --------------------
|
|
|
| A resource is a logical file contained within a package, or a logical
|
| subdirectory thereof. The package resource API expects resource names
|
| to have their path parts separated with ``/``, *not* whatever the local
|
| path separator is. Do not use os.path operations to manipulate resource
|
| names being passed into the API.
|
|
|
| The package resource API is designed to work with normal filesystem packages,
|
| .egg files, and unpacked .egg files. It can also work in a limited way with
|
| .zip files and with custom PEP 302 loaders that support the ``get_data()``
|
| method.
|
| """
|
|
|
| import sys
|
| import os
|
| import io
|
| import time
|
| import re
|
| import types
|
| import zipfile
|
| import zipimport
|
| import warnings
|
| import stat
|
| import functools
|
| import pkgutil
|
| import operator
|
| import platform
|
| import collections
|
| import plistlib
|
| import email.parser
|
| import errno
|
| import tempfile
|
| import textwrap
|
| import itertools
|
| import inspect
|
| import ntpath
|
| import posixpath
|
| import importlib
|
| from pkgutil import get_importer
|
|
|
| try:
|
| import _imp
|
| except ImportError:
|
|
|
| import imp as _imp
|
|
|
| try:
|
| FileExistsError
|
| except NameError:
|
| FileExistsError = OSError
|
|
|
|
|
| from os import utime
|
| try:
|
| from os import mkdir, rename, unlink
|
| WRITE_SUPPORT = True
|
| except ImportError:
|
|
|
| WRITE_SUPPORT = False
|
|
|
| from os import open as os_open
|
| from os.path import isdir, split
|
|
|
| try:
|
| import importlib.machinery as importlib_machinery
|
|
|
| importlib_machinery.__name__
|
| except ImportError:
|
| importlib_machinery = None
|
|
|
| from pkg_resources.extern.jaraco.text import (
|
| yield_lines,
|
| drop_comment,
|
| join_continuation,
|
| )
|
|
|
| from pkg_resources.extern import appdirs
|
| from pkg_resources.extern import packaging
|
| __import__('pkg_resources.extern.packaging.version')
|
| __import__('pkg_resources.extern.packaging.specifiers')
|
| __import__('pkg_resources.extern.packaging.requirements')
|
| __import__('pkg_resources.extern.packaging.markers')
|
| __import__('pkg_resources.extern.packaging.utils')
|
|
|
| if sys.version_info < (3, 5):
|
| raise RuntimeError("Python 3.5 or later is required")
|
|
|
|
|
|
|
| require = None
|
| working_set = None
|
| add_activation_listener = None
|
| resources_stream = None
|
| cleanup_resources = None
|
| resource_dir = None
|
| resource_stream = None
|
| set_extraction_path = None
|
| resource_isdir = None
|
| resource_string = None
|
| iter_entry_points = None
|
| resource_listdir = None
|
| resource_filename = None
|
| resource_exists = None
|
| _distribution_finders = None
|
| _namespace_handlers = None
|
| _namespace_packages = None
|
|
|
|
|
| class PEP440Warning(RuntimeWarning):
|
| """
|
| Used when there is an issue with a version or specifier not complying with
|
| PEP 440.
|
| """
|
|
|
|
|
| def parse_version(v):
|
| try:
|
| return packaging.version.Version(v)
|
| except packaging.version.InvalidVersion:
|
| warnings.warn(
|
| f"{v} is an invalid version and will not be supported in "
|
| "a future release",
|
| PkgResourcesDeprecationWarning,
|
| )
|
| return packaging.version.LegacyVersion(v)
|
|
|
|
|
| _state_vars = {}
|
|
|
|
|
| def _declare_state(vartype, **kw):
|
| globals().update(kw)
|
| _state_vars.update(dict.fromkeys(kw, vartype))
|
|
|
|
|
| def __getstate__():
|
| state = {}
|
| g = globals()
|
| for k, v in _state_vars.items():
|
| state[k] = g['_sget_' + v](g[k])
|
| return state
|
|
|
|
|
| def __setstate__(state):
|
| g = globals()
|
| for k, v in state.items():
|
| g['_sset_' + _state_vars[k]](k, g[k], v)
|
| return state
|
|
|
|
|
| def _sget_dict(val):
|
| return val.copy()
|
|
|
|
|
| def _sset_dict(key, ob, state):
|
| ob.clear()
|
| ob.update(state)
|
|
|
|
|
| def _sget_object(val):
|
| return val.__getstate__()
|
|
|
|
|
| def _sset_object(key, ob, state):
|
| ob.__setstate__(state)
|
|
|
|
|
| _sget_none = _sset_none = lambda *args: None
|
|
|
|
|
| def get_supported_platform():
|
| """Return this platform's maximum compatible version.
|
|
|
| distutils.util.get_platform() normally reports the minimum version
|
| of macOS that would be required to *use* extensions produced by
|
| distutils. But what we want when checking compatibility is to know the
|
| version of macOS that we are *running*. To allow usage of packages that
|
| explicitly require a newer version of macOS, we must also know the
|
| current version of the OS.
|
|
|
| If this condition occurs for any other platform with a version in its
|
| platform strings, this function should be extended accordingly.
|
| """
|
| plat = get_build_platform()
|
| m = macosVersionString.match(plat)
|
| if m is not None and sys.platform == "darwin":
|
| try:
|
| plat = 'macosx-%s-%s' % ('.'.join(_macos_vers()[:2]), m.group(3))
|
| except ValueError:
|
|
|
| pass
|
| return plat
|
|
|
|
|
| __all__ = [
|
|
|
| 'require', 'run_script', 'get_provider', 'get_distribution',
|
| 'load_entry_point', 'get_entry_map', 'get_entry_info',
|
| 'iter_entry_points',
|
| 'resource_string', 'resource_stream', 'resource_filename',
|
| 'resource_listdir', 'resource_exists', 'resource_isdir',
|
|
|
|
|
| 'declare_namespace', 'working_set', 'add_activation_listener',
|
| 'find_distributions', 'set_extraction_path', 'cleanup_resources',
|
| 'get_default_cache',
|
|
|
|
|
| 'Environment', 'WorkingSet', 'ResourceManager',
|
| 'Distribution', 'Requirement', 'EntryPoint',
|
|
|
|
|
| 'ResolutionError', 'VersionConflict', 'DistributionNotFound',
|
| 'UnknownExtra', 'ExtractionError',
|
|
|
|
|
| 'PEP440Warning',
|
|
|
|
|
| 'parse_requirements', 'parse_version', 'safe_name', 'safe_version',
|
| 'get_platform', 'compatible_platforms', 'yield_lines', 'split_sections',
|
| 'safe_extra', 'to_filename', 'invalid_marker', 'evaluate_marker',
|
|
|
|
|
| 'ensure_directory', 'normalize_path',
|
|
|
|
|
| 'EGG_DIST', 'BINARY_DIST', 'SOURCE_DIST', 'CHECKOUT_DIST', 'DEVELOP_DIST',
|
|
|
|
|
| 'IMetadataProvider', 'IResourceProvider', 'FileMetadata',
|
| 'PathMetadata', 'EggMetadata', 'EmptyProvider', 'empty_provider',
|
| 'NullProvider', 'EggProvider', 'DefaultProvider', 'ZipProvider',
|
| 'register_finder', 'register_namespace_handler', 'register_loader_type',
|
| 'fixup_namespace_packages', 'get_importer',
|
|
|
|
|
| 'PkgResourcesDeprecationWarning',
|
|
|
|
|
| 'run_main', 'AvailableDistributions',
|
| ]
|
|
|
|
|
| class ResolutionError(Exception):
|
| """Abstract base for dependency resolution errors"""
|
|
|
| def __repr__(self):
|
| return self.__class__.__name__ + repr(self.args)
|
|
|
|
|
| class VersionConflict(ResolutionError):
|
| """
|
| An already-installed version conflicts with the requested version.
|
|
|
| Should be initialized with the installed Distribution and the requested
|
| Requirement.
|
| """
|
|
|
| _template = "{self.dist} is installed but {self.req} is required"
|
|
|
| @property
|
| def dist(self):
|
| return self.args[0]
|
|
|
| @property
|
| def req(self):
|
| return self.args[1]
|
|
|
| def report(self):
|
| return self._template.format(**locals())
|
|
|
| def with_context(self, required_by):
|
| """
|
| If required_by is non-empty, return a version of self that is a
|
| ContextualVersionConflict.
|
| """
|
| if not required_by:
|
| return self
|
| args = self.args + (required_by,)
|
| return ContextualVersionConflict(*args)
|
|
|
|
|
| class ContextualVersionConflict(VersionConflict):
|
| """
|
| A VersionConflict that accepts a third parameter, the set of the
|
| requirements that required the installed Distribution.
|
| """
|
|
|
| _template = VersionConflict._template + ' by {self.required_by}'
|
|
|
| @property
|
| def required_by(self):
|
| return self.args[2]
|
|
|
|
|
| class DistributionNotFound(ResolutionError):
|
| """A requested distribution was not found"""
|
|
|
| _template = ("The '{self.req}' distribution was not found "
|
| "and is required by {self.requirers_str}")
|
|
|
| @property
|
| def req(self):
|
| return self.args[0]
|
|
|
| @property
|
| def requirers(self):
|
| return self.args[1]
|
|
|
| @property
|
| def requirers_str(self):
|
| if not self.requirers:
|
| return 'the application'
|
| return ', '.join(self.requirers)
|
|
|
| def report(self):
|
| return self._template.format(**locals())
|
|
|
| def __str__(self):
|
| return self.report()
|
|
|
|
|
| class UnknownExtra(ResolutionError):
|
| """Distribution doesn't have an "extra feature" of the given name"""
|
|
|
|
|
| _provider_factories = {}
|
|
|
| PY_MAJOR = '{}.{}'.format(*sys.version_info)
|
| EGG_DIST = 3
|
| BINARY_DIST = 2
|
| SOURCE_DIST = 1
|
| CHECKOUT_DIST = 0
|
| DEVELOP_DIST = -1
|
|
|
|
|
| def register_loader_type(loader_type, provider_factory):
|
| """Register `provider_factory` to make providers for `loader_type`
|
|
|
| `loader_type` is the type or class of a PEP 302 ``module.__loader__``,
|
| and `provider_factory` is a function that, passed a *module* object,
|
| returns an ``IResourceProvider`` for that module.
|
| """
|
| _provider_factories[loader_type] = provider_factory
|
|
|
|
|
| def get_provider(moduleOrReq):
|
| """Return an IResourceProvider for the named module or requirement"""
|
| if isinstance(moduleOrReq, Requirement):
|
| return working_set.find(moduleOrReq) or require(str(moduleOrReq))[0]
|
| try:
|
| module = sys.modules[moduleOrReq]
|
| except KeyError:
|
| __import__(moduleOrReq)
|
| module = sys.modules[moduleOrReq]
|
| loader = getattr(module, '__loader__', None)
|
| return _find_adapter(_provider_factories, loader)(module)
|
|
|
|
|
| def _macos_vers(_cache=[]):
|
| if not _cache:
|
| version = platform.mac_ver()[0]
|
|
|
| if version == '':
|
| plist = '/System/Library/CoreServices/SystemVersion.plist'
|
| if os.path.exists(plist):
|
| if hasattr(plistlib, 'readPlist'):
|
| plist_content = plistlib.readPlist(plist)
|
| if 'ProductVersion' in plist_content:
|
| version = plist_content['ProductVersion']
|
|
|
| _cache.append(version.split('.'))
|
| return _cache[0]
|
|
|
|
|
| def _macos_arch(machine):
|
| return {'PowerPC': 'ppc', 'Power_Macintosh': 'ppc'}.get(machine, machine)
|
|
|
|
|
| def get_build_platform():
|
| """Return this platform's string for platform-specific distributions
|
|
|
| XXX Currently this is the same as ``distutils.util.get_platform()``, but it
|
| needs some hacks for Linux and macOS.
|
| """
|
| from sysconfig import get_platform
|
|
|
| plat = get_platform()
|
| if sys.platform == "darwin" and not plat.startswith('macosx-'):
|
| try:
|
| version = _macos_vers()
|
| machine = os.uname()[4].replace(" ", "_")
|
| return "macosx-%d.%d-%s" % (
|
| int(version[0]), int(version[1]),
|
| _macos_arch(machine),
|
| )
|
| except ValueError:
|
|
|
|
|
| pass
|
| return plat
|
|
|
|
|
| macosVersionString = re.compile(r"macosx-(\d+)\.(\d+)-(.*)")
|
| darwinVersionString = re.compile(r"darwin-(\d+)\.(\d+)\.(\d+)-(.*)")
|
|
|
| get_platform = get_build_platform
|
|
|
|
|
| def compatible_platforms(provided, required):
|
| """Can code for the `provided` platform run on the `required` platform?
|
|
|
| Returns true if either platform is ``None``, or the platforms are equal.
|
|
|
| XXX Needs compatibility checks for Linux and other unixy OSes.
|
| """
|
| if provided is None or required is None or provided == required:
|
|
|
| return True
|
|
|
|
|
| reqMac = macosVersionString.match(required)
|
| if reqMac:
|
| provMac = macosVersionString.match(provided)
|
|
|
|
|
| if not provMac:
|
|
|
|
|
|
|
| provDarwin = darwinVersionString.match(provided)
|
| if provDarwin:
|
| dversion = int(provDarwin.group(1))
|
| macosversion = "%s.%s" % (reqMac.group(1), reqMac.group(2))
|
| if dversion == 7 and macosversion >= "10.3" or \
|
| dversion == 8 and macosversion >= "10.4":
|
| return True
|
|
|
| return False
|
|
|
|
|
| if provMac.group(1) != reqMac.group(1) or \
|
| provMac.group(3) != reqMac.group(3):
|
| return False
|
|
|
|
|
| if int(provMac.group(2)) > int(reqMac.group(2)):
|
| return False
|
|
|
| return True
|
|
|
|
|
| return False
|
|
|
|
|
| def run_script(dist_spec, script_name):
|
| """Locate distribution `dist_spec` and run its `script_name` script"""
|
| ns = sys._getframe(1).f_globals
|
| name = ns['__name__']
|
| ns.clear()
|
| ns['__name__'] = name
|
| require(dist_spec)[0].run_script(script_name, ns)
|
|
|
|
|
|
|
| run_main = run_script
|
|
|
|
|
| def get_distribution(dist):
|
| """Return a current distribution object for a Requirement or string"""
|
| if isinstance(dist, str):
|
| dist = Requirement.parse(dist)
|
| if isinstance(dist, Requirement):
|
| dist = get_provider(dist)
|
| if not isinstance(dist, Distribution):
|
| raise TypeError("Expected string, Requirement, or Distribution", dist)
|
| return dist
|
|
|
|
|
| def load_entry_point(dist, group, name):
|
| """Return `name` entry point of `group` for `dist` or raise ImportError"""
|
| return get_distribution(dist).load_entry_point(group, name)
|
|
|
|
|
| def get_entry_map(dist, group=None):
|
| """Return the entry point map for `group`, or the full entry map"""
|
| return get_distribution(dist).get_entry_map(group)
|
|
|
|
|
| def get_entry_info(dist, group, name):
|
| """Return the EntryPoint object for `group`+`name`, or ``None``"""
|
| return get_distribution(dist).get_entry_info(group, name)
|
|
|
|
|
| class IMetadataProvider:
|
| def has_metadata(name):
|
| """Does the package's distribution contain the named metadata?"""
|
|
|
| def get_metadata(name):
|
| """The named metadata resource as a string"""
|
|
|
| def get_metadata_lines(name):
|
| """Yield named metadata resource as list of non-blank non-comment lines
|
|
|
| Leading and trailing whitespace is stripped from each line, and lines
|
| with ``#`` as the first non-blank character are omitted."""
|
|
|
| def metadata_isdir(name):
|
| """Is the named metadata a directory? (like ``os.path.isdir()``)"""
|
|
|
| def metadata_listdir(name):
|
| """List of metadata names in the directory (like ``os.listdir()``)"""
|
|
|
| def run_script(script_name, namespace):
|
| """Execute the named script in the supplied namespace dictionary"""
|
|
|
|
|
| class IResourceProvider(IMetadataProvider):
|
| """An object that provides access to package resources"""
|
|
|
| def get_resource_filename(manager, resource_name):
|
| """Return a true filesystem path for `resource_name`
|
|
|
| `manager` must be an ``IResourceManager``"""
|
|
|
| def get_resource_stream(manager, resource_name):
|
| """Return a readable file-like object for `resource_name`
|
|
|
| `manager` must be an ``IResourceManager``"""
|
|
|
| def get_resource_string(manager, resource_name):
|
| """Return a string containing the contents of `resource_name`
|
|
|
| `manager` must be an ``IResourceManager``"""
|
|
|
| def has_resource(resource_name):
|
| """Does the package contain the named resource?"""
|
|
|
| def resource_isdir(resource_name):
|
| """Is the named resource a directory? (like ``os.path.isdir()``)"""
|
|
|
| def resource_listdir(resource_name):
|
| """List of resource names in the directory (like ``os.listdir()``)"""
|
|
|
|
|
| class WorkingSet:
|
| """A collection of active distributions on sys.path (or a similar list)"""
|
|
|
| def __init__(self, entries=None):
|
| """Create working set from list of path entries (default=sys.path)"""
|
| self.entries = []
|
| self.entry_keys = {}
|
| self.by_key = {}
|
| self.normalized_to_canonical_keys = {}
|
| self.callbacks = []
|
|
|
| if entries is None:
|
| entries = sys.path
|
|
|
| for entry in entries:
|
| self.add_entry(entry)
|
|
|
| @classmethod
|
| def _build_master(cls):
|
| """
|
| Prepare the master working set.
|
| """
|
| ws = cls()
|
| try:
|
| from __main__ import __requires__
|
| except ImportError:
|
|
|
| return ws
|
|
|
|
|
| try:
|
| ws.require(__requires__)
|
| except VersionConflict:
|
| return cls._build_from_requirements(__requires__)
|
|
|
| return ws
|
|
|
| @classmethod
|
| def _build_from_requirements(cls, req_spec):
|
| """
|
| Build a working set from a requirement spec. Rewrites sys.path.
|
| """
|
|
|
|
|
| ws = cls([])
|
| reqs = parse_requirements(req_spec)
|
| dists = ws.resolve(reqs, Environment())
|
| for dist in dists:
|
| ws.add(dist)
|
|
|
|
|
| for entry in sys.path:
|
| if entry not in ws.entries:
|
| ws.add_entry(entry)
|
|
|
|
|
| sys.path[:] = ws.entries
|
| return ws
|
|
|
| def add_entry(self, entry):
|
| """Add a path item to ``.entries``, finding any distributions on it
|
|
|
| ``find_distributions(entry, True)`` is used to find distributions
|
| corresponding to the path entry, and they are added. `entry` is
|
| always appended to ``.entries``, even if it is already present.
|
| (This is because ``sys.path`` can contain the same value more than
|
| once, and the ``.entries`` of the ``sys.path`` WorkingSet should always
|
| equal ``sys.path``.)
|
| """
|
| self.entry_keys.setdefault(entry, [])
|
| self.entries.append(entry)
|
| for dist in find_distributions(entry, True):
|
| self.add(dist, entry, False)
|
|
|
| def __contains__(self, dist):
|
| """True if `dist` is the active distribution for its project"""
|
| return self.by_key.get(dist.key) == dist
|
|
|
| def find(self, req):
|
| """Find a distribution matching requirement `req`
|
|
|
| If there is an active distribution for the requested project, this
|
| returns it as long as it meets the version requirement specified by
|
| `req`. But, if there is an active distribution for the project and it
|
| does *not* meet the `req` requirement, ``VersionConflict`` is raised.
|
| If there is no active distribution for the requested project, ``None``
|
| is returned.
|
| """
|
| dist = self.by_key.get(req.key)
|
|
|
| if dist is None:
|
| canonical_key = self.normalized_to_canonical_keys.get(req.key)
|
|
|
| if canonical_key is not None:
|
| req.key = canonical_key
|
| dist = self.by_key.get(canonical_key)
|
|
|
| if dist is not None and dist not in req:
|
|
|
| raise VersionConflict(dist, req)
|
| return dist
|
|
|
| def iter_entry_points(self, group, name=None):
|
| """Yield entry point objects from `group` matching `name`
|
|
|
| If `name` is None, yields all entry points in `group` from all
|
| distributions in the working set, otherwise only ones matching
|
| both `group` and `name` are yielded (in distribution order).
|
| """
|
| return (
|
| entry
|
| for dist in self
|
| for entry in dist.get_entry_map(group).values()
|
| if name is None or name == entry.name
|
| )
|
|
|
| def run_script(self, requires, script_name):
|
| """Locate distribution for `requires` and run `script_name` script"""
|
| ns = sys._getframe(1).f_globals
|
| name = ns['__name__']
|
| ns.clear()
|
| ns['__name__'] = name
|
| self.require(requires)[0].run_script(script_name, ns)
|
|
|
| def __iter__(self):
|
| """Yield distributions for non-duplicate projects in the working set
|
|
|
| The yield order is the order in which the items' path entries were
|
| added to the working set.
|
| """
|
| seen = {}
|
| for item in self.entries:
|
| if item not in self.entry_keys:
|
|
|
| continue
|
|
|
| for key in self.entry_keys[item]:
|
| if key not in seen:
|
| seen[key] = 1
|
| yield self.by_key[key]
|
|
|
| def add(self, dist, entry=None, insert=True, replace=False):
|
| """Add `dist` to working set, associated with `entry`
|
|
|
| If `entry` is unspecified, it defaults to the ``.location`` of `dist`.
|
| On exit from this routine, `entry` is added to the end of the working
|
| set's ``.entries`` (if it wasn't already present).
|
|
|
| `dist` is only added to the working set if it's for a project that
|
| doesn't already have a distribution in the set, unless `replace=True`.
|
| If it's added, any callbacks registered with the ``subscribe()`` method
|
| will be called.
|
| """
|
| if insert:
|
| dist.insert_on(self.entries, entry, replace=replace)
|
|
|
| if entry is None:
|
| entry = dist.location
|
| keys = self.entry_keys.setdefault(entry, [])
|
| keys2 = self.entry_keys.setdefault(dist.location, [])
|
| if not replace and dist.key in self.by_key:
|
|
|
| return
|
|
|
| self.by_key[dist.key] = dist
|
| normalized_name = packaging.utils.canonicalize_name(dist.key)
|
| self.normalized_to_canonical_keys[normalized_name] = dist.key
|
| if dist.key not in keys:
|
| keys.append(dist.key)
|
| if dist.key not in keys2:
|
| keys2.append(dist.key)
|
| self._added_new(dist)
|
|
|
|
|
| def resolve(self, requirements, env=None, installer=None,
|
| replace_conflicting=False, extras=None):
|
| """List all distributions needed to (recursively) meet `requirements`
|
|
|
| `requirements` must be a sequence of ``Requirement`` objects. `env`,
|
| if supplied, should be an ``Environment`` instance. If
|
| not supplied, it defaults to all distributions available within any
|
| entry or distribution in the working set. `installer`, if supplied,
|
| will be invoked with each requirement that cannot be met by an
|
| already-installed distribution; it should return a ``Distribution`` or
|
| ``None``.
|
|
|
| Unless `replace_conflicting=True`, raises a VersionConflict exception
|
| if
|
| any requirements are found on the path that have the correct name but
|
| the wrong version. Otherwise, if an `installer` is supplied it will be
|
| invoked to obtain the correct version of the requirement and activate
|
| it.
|
|
|
| `extras` is a list of the extras to be used with these requirements.
|
| This is important because extra requirements may look like `my_req;
|
| extra = "my_extra"`, which would otherwise be interpreted as a purely
|
| optional requirement. Instead, we want to be able to assert that these
|
| requirements are truly required.
|
| """
|
|
|
|
|
| requirements = list(requirements)[::-1]
|
|
|
| processed = {}
|
|
|
| best = {}
|
| to_activate = []
|
|
|
| req_extras = _ReqExtras()
|
|
|
|
|
|
|
| required_by = collections.defaultdict(set)
|
|
|
| while requirements:
|
|
|
| req = requirements.pop(0)
|
| if req in processed:
|
|
|
| continue
|
|
|
| if not req_extras.markers_pass(req, extras):
|
| continue
|
|
|
| dist = best.get(req.key)
|
| if dist is None:
|
|
|
| dist = self.by_key.get(req.key)
|
| if dist is None or (dist not in req and replace_conflicting):
|
| ws = self
|
| if env is None:
|
| if dist is None:
|
| env = Environment(self.entries)
|
| else:
|
|
|
|
|
|
|
| env = Environment([])
|
| ws = WorkingSet([])
|
| dist = best[req.key] = env.best_match(
|
| req, ws, installer,
|
| replace_conflicting=replace_conflicting
|
| )
|
| if dist is None:
|
| requirers = required_by.get(req, None)
|
| raise DistributionNotFound(req, requirers)
|
| to_activate.append(dist)
|
| if dist not in req:
|
|
|
| dependent_req = required_by[req]
|
| raise VersionConflict(dist, req).with_context(dependent_req)
|
|
|
|
|
| new_requirements = dist.requires(req.extras)[::-1]
|
| requirements.extend(new_requirements)
|
|
|
|
|
| for new_requirement in new_requirements:
|
| required_by[new_requirement].add(req.project_name)
|
| req_extras[new_requirement] = req.extras
|
|
|
| processed[req] = True
|
|
|
|
|
| return to_activate
|
|
|
| def find_plugins(
|
| self, plugin_env, full_env=None, installer=None, fallback=True):
|
| """Find all activatable distributions in `plugin_env`
|
|
|
| Example usage::
|
|
|
| distributions, errors = working_set.find_plugins(
|
| Environment(plugin_dirlist)
|
| )
|
| # add plugins+libs to sys.path
|
| map(working_set.add, distributions)
|
| # display errors
|
| print('Could not load', errors)
|
|
|
| The `plugin_env` should be an ``Environment`` instance that contains
|
| only distributions that are in the project's "plugin directory" or
|
| directories. The `full_env`, if supplied, should be an ``Environment``
|
| contains all currently-available distributions. If `full_env` is not
|
| supplied, one is created automatically from the ``WorkingSet`` this
|
| method is called on, which will typically mean that every directory on
|
| ``sys.path`` will be scanned for distributions.
|
|
|
| `installer` is a standard installer callback as used by the
|
| ``resolve()`` method. The `fallback` flag indicates whether we should
|
| attempt to resolve older versions of a plugin if the newest version
|
| cannot be resolved.
|
|
|
| This method returns a 2-tuple: (`distributions`, `error_info`), where
|
| `distributions` is a list of the distributions found in `plugin_env`
|
| that were loadable, along with any other distributions that are needed
|
| to resolve their dependencies. `error_info` is a dictionary mapping
|
| unloadable plugin distributions to an exception instance describing the
|
| error that occurred. Usually this will be a ``DistributionNotFound`` or
|
| ``VersionConflict`` instance.
|
| """
|
|
|
| plugin_projects = list(plugin_env)
|
|
|
| plugin_projects.sort()
|
|
|
| error_info = {}
|
| distributions = {}
|
|
|
| if full_env is None:
|
| env = Environment(self.entries)
|
| env += plugin_env
|
| else:
|
| env = full_env + plugin_env
|
|
|
| shadow_set = self.__class__([])
|
|
|
| list(map(shadow_set.add, self))
|
|
|
| for project_name in plugin_projects:
|
|
|
| for dist in plugin_env[project_name]:
|
|
|
| req = [dist.as_requirement()]
|
|
|
| try:
|
| resolvees = shadow_set.resolve(req, env, installer)
|
|
|
| except ResolutionError as v:
|
|
|
| error_info[dist] = v
|
| if fallback:
|
|
|
| continue
|
| else:
|
|
|
| break
|
|
|
| else:
|
| list(map(shadow_set.add, resolvees))
|
| distributions.update(dict.fromkeys(resolvees))
|
|
|
|
|
| break
|
|
|
| distributions = list(distributions)
|
| distributions.sort()
|
|
|
| return distributions, error_info
|
|
|
| def require(self, *requirements):
|
| """Ensure that distributions matching `requirements` are activated
|
|
|
| `requirements` must be a string or a (possibly-nested) sequence
|
| thereof, specifying the distributions and versions required. The
|
| return value is a sequence of the distributions that needed to be
|
| activated to fulfill the requirements; all relevant distributions are
|
| included, even if they were already activated in this working set.
|
| """
|
| needed = self.resolve(parse_requirements(requirements))
|
|
|
| for dist in needed:
|
| self.add(dist)
|
|
|
| return needed
|
|
|
| def subscribe(self, callback, existing=True):
|
| """Invoke `callback` for all distributions
|
|
|
| If `existing=True` (default),
|
| call on all existing ones, as well.
|
| """
|
| if callback in self.callbacks:
|
| return
|
| self.callbacks.append(callback)
|
| if not existing:
|
| return
|
| for dist in self:
|
| callback(dist)
|
|
|
| def _added_new(self, dist):
|
| for callback in self.callbacks:
|
| callback(dist)
|
|
|
| def __getstate__(self):
|
| return (
|
| self.entries[:], self.entry_keys.copy(), self.by_key.copy(),
|
| self.normalized_to_canonical_keys.copy(), self.callbacks[:]
|
| )
|
|
|
| def __setstate__(self, e_k_b_n_c):
|
| entries, keys, by_key, normalized_to_canonical_keys, callbacks = e_k_b_n_c
|
| self.entries = entries[:]
|
| self.entry_keys = keys.copy()
|
| self.by_key = by_key.copy()
|
| self.normalized_to_canonical_keys = normalized_to_canonical_keys.copy()
|
| self.callbacks = callbacks[:]
|
|
|
|
|
| class _ReqExtras(dict):
|
| """
|
| Map each requirement to the extras that demanded it.
|
| """
|
|
|
| def markers_pass(self, req, extras=None):
|
| """
|
| Evaluate markers for req against each extra that
|
| demanded it.
|
|
|
| Return False if the req has a marker and fails
|
| evaluation. Otherwise, return True.
|
| """
|
| extra_evals = (
|
| req.marker.evaluate({'extra': extra})
|
| for extra in self.get(req, ()) + (extras or (None,))
|
| )
|
| return not req.marker or any(extra_evals)
|
|
|
|
|
| class Environment:
|
| """Searchable snapshot of distributions on a search path"""
|
|
|
| def __init__(
|
| self, search_path=None, platform=get_supported_platform(),
|
| python=PY_MAJOR):
|
| """Snapshot distributions available on a search path
|
|
|
| Any distributions found on `search_path` are added to the environment.
|
| `search_path` should be a sequence of ``sys.path`` items. If not
|
| supplied, ``sys.path`` is used.
|
|
|
| `platform` is an optional string specifying the name of the platform
|
| that platform-specific distributions must be compatible with. If
|
| unspecified, it defaults to the current platform. `python` is an
|
| optional string naming the desired version of Python (e.g. ``'3.6'``);
|
| it defaults to the current version.
|
|
|
| You may explicitly set `platform` (and/or `python`) to ``None`` if you
|
| wish to map *all* distributions, not just those compatible with the
|
| running platform or Python version.
|
| """
|
| self._distmap = {}
|
| self.platform = platform
|
| self.python = python
|
| self.scan(search_path)
|
|
|
| def can_add(self, dist):
|
| """Is distribution `dist` acceptable for this environment?
|
|
|
| The distribution must match the platform and python version
|
| requirements specified when this environment was created, or False
|
| is returned.
|
| """
|
| py_compat = (
|
| self.python is None
|
| or dist.py_version is None
|
| or dist.py_version == self.python
|
| )
|
| return py_compat and compatible_platforms(dist.platform, self.platform)
|
|
|
| def remove(self, dist):
|
| """Remove `dist` from the environment"""
|
| self._distmap[dist.key].remove(dist)
|
|
|
| def scan(self, search_path=None):
|
| """Scan `search_path` for distributions usable in this environment
|
|
|
| Any distributions found are added to the environment.
|
| `search_path` should be a sequence of ``sys.path`` items. If not
|
| supplied, ``sys.path`` is used. Only distributions conforming to
|
| the platform/python version defined at initialization are added.
|
| """
|
| if search_path is None:
|
| search_path = sys.path
|
|
|
| for item in search_path:
|
| for dist in find_distributions(item):
|
| self.add(dist)
|
|
|
| def __getitem__(self, project_name):
|
| """Return a newest-to-oldest list of distributions for `project_name`
|
|
|
| Uses case-insensitive `project_name` comparison, assuming all the
|
| project's distributions use their project's name converted to all
|
| lowercase as their key.
|
|
|
| """
|
| distribution_key = project_name.lower()
|
| return self._distmap.get(distribution_key, [])
|
|
|
| def add(self, dist):
|
| """Add `dist` if we ``can_add()`` it and it has not already been added
|
| """
|
| if self.can_add(dist) and dist.has_version():
|
| dists = self._distmap.setdefault(dist.key, [])
|
| if dist not in dists:
|
| dists.append(dist)
|
| dists.sort(key=operator.attrgetter('hashcmp'), reverse=True)
|
|
|
| def best_match(
|
| self, req, working_set, installer=None, replace_conflicting=False):
|
| """Find distribution best matching `req` and usable on `working_set`
|
|
|
| This calls the ``find(req)`` method of the `working_set` to see if a
|
| suitable distribution is already active. (This may raise
|
| ``VersionConflict`` if an unsuitable version of the project is already
|
| active in the specified `working_set`.) If a suitable distribution
|
| isn't active, this method returns the newest distribution in the
|
| environment that meets the ``Requirement`` in `req`. If no suitable
|
| distribution is found, and `installer` is supplied, then the result of
|
| calling the environment's ``obtain(req, installer)`` method will be
|
| returned.
|
| """
|
| try:
|
| dist = working_set.find(req)
|
| except VersionConflict:
|
| if not replace_conflicting:
|
| raise
|
| dist = None
|
| if dist is not None:
|
| return dist
|
| for dist in self[req.key]:
|
| if dist in req:
|
| return dist
|
|
|
| return self.obtain(req, installer)
|
|
|
| def obtain(self, requirement, installer=None):
|
| """Obtain a distribution matching `requirement` (e.g. via download)
|
|
|
| Obtain a distro that matches requirement (e.g. via download). In the
|
| base ``Environment`` class, this routine just returns
|
| ``installer(requirement)``, unless `installer` is None, in which case
|
| None is returned instead. This method is a hook that allows subclasses
|
| to attempt other ways of obtaining a distribution before falling back
|
| to the `installer` argument."""
|
| if installer is not None:
|
| return installer(requirement)
|
|
|
| def __iter__(self):
|
| """Yield the unique project names of the available distributions"""
|
| for key in self._distmap.keys():
|
| if self[key]:
|
| yield key
|
|
|
| def __iadd__(self, other):
|
| """In-place addition of a distribution or environment"""
|
| if isinstance(other, Distribution):
|
| self.add(other)
|
| elif isinstance(other, Environment):
|
| for project in other:
|
| for dist in other[project]:
|
| self.add(dist)
|
| else:
|
| raise TypeError("Can't add %r to environment" % (other,))
|
| return self
|
|
|
| def __add__(self, other):
|
| """Add an environment or distribution to an environment"""
|
| new = self.__class__([], platform=None, python=None)
|
| for env in self, other:
|
| new += env
|
| return new
|
|
|
|
|
|
|
| AvailableDistributions = Environment
|
|
|
|
|
| class ExtractionError(RuntimeError):
|
| """An error occurred extracting a resource
|
|
|
| The following attributes are available from instances of this exception:
|
|
|
| manager
|
| The resource manager that raised this exception
|
|
|
| cache_path
|
| The base directory for resource extraction
|
|
|
| original_error
|
| The exception instance that caused extraction to fail
|
| """
|
|
|
|
|
| class ResourceManager:
|
| """Manage resource extraction and packages"""
|
| extraction_path = None
|
|
|
| def __init__(self):
|
| self.cached_files = {}
|
|
|
| def resource_exists(self, package_or_requirement, resource_name):
|
| """Does the named resource exist?"""
|
| return get_provider(package_or_requirement).has_resource(resource_name)
|
|
|
| def resource_isdir(self, package_or_requirement, resource_name):
|
| """Is the named resource an existing directory?"""
|
| return get_provider(package_or_requirement).resource_isdir(
|
| resource_name
|
| )
|
|
|
| def resource_filename(self, package_or_requirement, resource_name):
|
| """Return a true filesystem path for specified resource"""
|
| return get_provider(package_or_requirement).get_resource_filename(
|
| self, resource_name
|
| )
|
|
|
| def resource_stream(self, package_or_requirement, resource_name):
|
| """Return a readable file-like object for specified resource"""
|
| return get_provider(package_or_requirement).get_resource_stream(
|
| self, resource_name
|
| )
|
|
|
| def resource_string(self, package_or_requirement, resource_name):
|
| """Return specified resource as a string"""
|
| return get_provider(package_or_requirement).get_resource_string(
|
| self, resource_name
|
| )
|
|
|
| def resource_listdir(self, package_or_requirement, resource_name):
|
| """List the contents of the named resource directory"""
|
| return get_provider(package_or_requirement).resource_listdir(
|
| resource_name
|
| )
|
|
|
| def extraction_error(self):
|
| """Give an error message for problems extracting file(s)"""
|
|
|
| old_exc = sys.exc_info()[1]
|
| cache_path = self.extraction_path or get_default_cache()
|
|
|
| tmpl = textwrap.dedent("""
|
| Can't extract file(s) to egg cache
|
|
|
| The following error occurred while trying to extract file(s)
|
| to the Python egg cache:
|
|
|
| {old_exc}
|
|
|
| The Python egg cache directory is currently set to:
|
|
|
| {cache_path}
|
|
|
| Perhaps your account does not have write access to this directory?
|
| You can change the cache directory by setting the PYTHON_EGG_CACHE
|
| environment variable to point to an accessible directory.
|
| """).lstrip()
|
| err = ExtractionError(tmpl.format(**locals()))
|
| err.manager = self
|
| err.cache_path = cache_path
|
| err.original_error = old_exc
|
| raise err
|
|
|
| def get_cache_path(self, archive_name, names=()):
|
| """Return absolute location in cache for `archive_name` and `names`
|
|
|
| The parent directory of the resulting path will be created if it does
|
| not already exist. `archive_name` should be the base filename of the
|
| enclosing egg (which may not be the name of the enclosing zipfile!),
|
| including its ".egg" extension. `names`, if provided, should be a
|
| sequence of path name parts "under" the egg's extraction location.
|
|
|
| This method should only be called by resource providers that need to
|
| obtain an extraction location, and only for names they intend to
|
| extract, as it tracks the generated names for possible cleanup later.
|
| """
|
| extract_path = self.extraction_path or get_default_cache()
|
| target_path = os.path.join(extract_path, archive_name + '-tmp', *names)
|
| try:
|
| _bypass_ensure_directory(target_path)
|
| except Exception:
|
| self.extraction_error()
|
|
|
| self._warn_unsafe_extraction_path(extract_path)
|
|
|
| self.cached_files[target_path] = 1
|
| return target_path
|
|
|
| @staticmethod
|
| def _warn_unsafe_extraction_path(path):
|
| """
|
| If the default extraction path is overridden and set to an insecure
|
| location, such as /tmp, it opens up an opportunity for an attacker to
|
| replace an extracted file with an unauthorized payload. Warn the user
|
| if a known insecure location is used.
|
|
|
| See Distribute #375 for more details.
|
| """
|
| if os.name == 'nt' and not path.startswith(os.environ['windir']):
|
|
|
|
|
|
|
| return
|
| mode = os.stat(path).st_mode
|
| if mode & stat.S_IWOTH or mode & stat.S_IWGRP:
|
| msg = (
|
| "Extraction path is writable by group/others "
|
| "and vulnerable to attack when "
|
| "used with get_resource_filename ({path}). "
|
| "Consider a more secure "
|
| "location (set with .set_extraction_path or the "
|
| "PYTHON_EGG_CACHE environment variable)."
|
| ).format(**locals())
|
| warnings.warn(msg, UserWarning)
|
|
|
| def postprocess(self, tempname, filename):
|
| """Perform any platform-specific postprocessing of `tempname`
|
|
|
| This is where Mac header rewrites should be done; other platforms don't
|
| have anything special they should do.
|
|
|
| Resource providers should call this method ONLY after successfully
|
| extracting a compressed resource. They must NOT call it on resources
|
| that are already in the filesystem.
|
|
|
| `tempname` is the current (temporary) name of the file, and `filename`
|
| is the name it will be renamed to by the caller after this routine
|
| returns.
|
| """
|
|
|
| if os.name == 'posix':
|
|
|
| mode = ((os.stat(tempname).st_mode) | 0o555) & 0o7777
|
| os.chmod(tempname, mode)
|
|
|
| def set_extraction_path(self, path):
|
| """Set the base path where resources will be extracted to, if needed.
|
|
|
| If you do not call this routine before any extractions take place, the
|
| path defaults to the return value of ``get_default_cache()``. (Which
|
| is based on the ``PYTHON_EGG_CACHE`` environment variable, with various
|
| platform-specific fallbacks. See that routine's documentation for more
|
| details.)
|
|
|
| Resources are extracted to subdirectories of this path based upon
|
| information given by the ``IResourceProvider``. You may set this to a
|
| temporary directory, but then you must call ``cleanup_resources()`` to
|
| delete the extracted files when done. There is no guarantee that
|
| ``cleanup_resources()`` will be able to remove all extracted files.
|
|
|
| (Note: you may not change the extraction path for a given resource
|
| manager once resources have been extracted, unless you first call
|
| ``cleanup_resources()``.)
|
| """
|
| if self.cached_files:
|
| raise ValueError(
|
| "Can't change extraction path, files already extracted"
|
| )
|
|
|
| self.extraction_path = path
|
|
|
| def cleanup_resources(self, force=False):
|
| """
|
| Delete all extracted resource files and directories, returning a list
|
| of the file and directory names that could not be successfully removed.
|
| This function does not have any concurrency protection, so it should
|
| generally only be called when the extraction path is a temporary
|
| directory exclusive to a single process. This method is not
|
| automatically called; you must call it explicitly or register it as an
|
| ``atexit`` function if you wish to ensure cleanup of a temporary
|
| directory used for extractions.
|
| """
|
|
|
|
|
|
|
| def get_default_cache():
|
| """
|
| Return the ``PYTHON_EGG_CACHE`` environment variable
|
| or a platform-relevant user cache dir for an app
|
| named "Python-Eggs".
|
| """
|
| return (
|
| os.environ.get('PYTHON_EGG_CACHE')
|
| or appdirs.user_cache_dir(appname='Python-Eggs')
|
| )
|
|
|
|
|
| def safe_name(name):
|
| """Convert an arbitrary string to a standard distribution name
|
|
|
| Any runs of non-alphanumeric/. characters are replaced with a single '-'.
|
| """
|
| return re.sub('[^A-Za-z0-9.]+', '-', name)
|
|
|
|
|
| def safe_version(version):
|
| """
|
| Convert an arbitrary string to a standard version string
|
| """
|
| try:
|
|
|
| return str(packaging.version.Version(version))
|
| except packaging.version.InvalidVersion:
|
| version = version.replace(' ', '.')
|
| return re.sub('[^A-Za-z0-9.]+', '-', version)
|
|
|
|
|
| def safe_extra(extra):
|
| """Convert an arbitrary string to a standard 'extra' name
|
|
|
| Any runs of non-alphanumeric characters are replaced with a single '_',
|
| and the result is always lowercased.
|
| """
|
| return re.sub('[^A-Za-z0-9.-]+', '_', extra).lower()
|
|
|
|
|
| def to_filename(name):
|
| """Convert a project or version name to its filename-escaped form
|
|
|
| Any '-' characters are currently replaced with '_'.
|
| """
|
| return name.replace('-', '_')
|
|
|
|
|
| def invalid_marker(text):
|
| """
|
| Validate text as a PEP 508 environment marker; return an exception
|
| if invalid or False otherwise.
|
| """
|
| try:
|
| evaluate_marker(text)
|
| except SyntaxError as e:
|
| e.filename = None
|
| e.lineno = None
|
| return e
|
| return False
|
|
|
|
|
| def evaluate_marker(text, extra=None):
|
| """
|
| Evaluate a PEP 508 environment marker.
|
| Return a boolean indicating the marker result in this environment.
|
| Raise SyntaxError if marker is invalid.
|
|
|
| This implementation uses the 'pyparsing' module.
|
| """
|
| try:
|
| marker = packaging.markers.Marker(text)
|
| return marker.evaluate()
|
| except packaging.markers.InvalidMarker as e:
|
| raise SyntaxError(e) from e
|
|
|
|
|
| class NullProvider:
|
| """Try to implement resources and metadata for arbitrary PEP 302 loaders"""
|
|
|
| egg_name = None
|
| egg_info = None
|
| loader = None
|
|
|
| def __init__(self, module):
|
| self.loader = getattr(module, '__loader__', None)
|
| self.module_path = os.path.dirname(getattr(module, '__file__', ''))
|
|
|
| def get_resource_filename(self, manager, resource_name):
|
| return self._fn(self.module_path, resource_name)
|
|
|
| def get_resource_stream(self, manager, resource_name):
|
| return io.BytesIO(self.get_resource_string(manager, resource_name))
|
|
|
| def get_resource_string(self, manager, resource_name):
|
| return self._get(self._fn(self.module_path, resource_name))
|
|
|
| def has_resource(self, resource_name):
|
| return self._has(self._fn(self.module_path, resource_name))
|
|
|
| def _get_metadata_path(self, name):
|
| return self._fn(self.egg_info, name)
|
|
|
| def has_metadata(self, name):
|
| if not self.egg_info:
|
| return self.egg_info
|
|
|
| path = self._get_metadata_path(name)
|
| return self._has(path)
|
|
|
| def get_metadata(self, name):
|
| if not self.egg_info:
|
| return ""
|
| path = self._get_metadata_path(name)
|
| value = self._get(path)
|
| try:
|
| return value.decode('utf-8')
|
| except UnicodeDecodeError as exc:
|
|
|
|
|
| exc.reason += ' in {} file at path: {}'.format(name, path)
|
| raise
|
|
|
| def get_metadata_lines(self, name):
|
| return yield_lines(self.get_metadata(name))
|
|
|
| def resource_isdir(self, resource_name):
|
| return self._isdir(self._fn(self.module_path, resource_name))
|
|
|
| def metadata_isdir(self, name):
|
| return self.egg_info and self._isdir(self._fn(self.egg_info, name))
|
|
|
| def resource_listdir(self, resource_name):
|
| return self._listdir(self._fn(self.module_path, resource_name))
|
|
|
| def metadata_listdir(self, name):
|
| if self.egg_info:
|
| return self._listdir(self._fn(self.egg_info, name))
|
| return []
|
|
|
| def run_script(self, script_name, namespace):
|
| script = 'scripts/' + script_name
|
| if not self.has_metadata(script):
|
| raise ResolutionError(
|
| "Script {script!r} not found in metadata at {self.egg_info!r}"
|
| .format(**locals()),
|
| )
|
| script_text = self.get_metadata(script).replace('\r\n', '\n')
|
| script_text = script_text.replace('\r', '\n')
|
| script_filename = self._fn(self.egg_info, script)
|
| namespace['__file__'] = script_filename
|
| if os.path.exists(script_filename):
|
| with open(script_filename) as fid:
|
| source = fid.read()
|
| code = compile(source, script_filename, 'exec')
|
| exec(code, namespace, namespace)
|
| else:
|
| from linecache import cache
|
| cache[script_filename] = (
|
| len(script_text), 0, script_text.split('\n'), script_filename
|
| )
|
| script_code = compile(script_text, script_filename, 'exec')
|
| exec(script_code, namespace, namespace)
|
|
|
| def _has(self, path):
|
| raise NotImplementedError(
|
| "Can't perform this operation for unregistered loader type"
|
| )
|
|
|
| def _isdir(self, path):
|
| raise NotImplementedError(
|
| "Can't perform this operation for unregistered loader type"
|
| )
|
|
|
| def _listdir(self, path):
|
| raise NotImplementedError(
|
| "Can't perform this operation for unregistered loader type"
|
| )
|
|
|
| def _fn(self, base, resource_name):
|
| self._validate_resource_path(resource_name)
|
| if resource_name:
|
| return os.path.join(base, *resource_name.split('/'))
|
| return base
|
|
|
| @staticmethod
|
| def _validate_resource_path(path):
|
| """
|
| Validate the resource paths according to the docs.
|
| https://setuptools.pypa.io/en/latest/pkg_resources.html#basic-resource-access
|
|
|
| >>> warned = getfixture('recwarn')
|
| >>> warnings.simplefilter('always')
|
| >>> vrp = NullProvider._validate_resource_path
|
| >>> vrp('foo/bar.txt')
|
| >>> bool(warned)
|
| False
|
| >>> vrp('../foo/bar.txt')
|
| >>> bool(warned)
|
| True
|
| >>> warned.clear()
|
| >>> vrp('/foo/bar.txt')
|
| >>> bool(warned)
|
| True
|
| >>> vrp('foo/../../bar.txt')
|
| >>> bool(warned)
|
| True
|
| >>> warned.clear()
|
| >>> vrp('foo/f../bar.txt')
|
| >>> bool(warned)
|
| False
|
|
|
| Windows path separators are straight-up disallowed.
|
| >>> vrp(r'\\foo/bar.txt')
|
| Traceback (most recent call last):
|
| ...
|
| ValueError: Use of .. or absolute path in a resource path \
|
| is not allowed.
|
|
|
| >>> vrp(r'C:\\foo/bar.txt')
|
| Traceback (most recent call last):
|
| ...
|
| ValueError: Use of .. or absolute path in a resource path \
|
| is not allowed.
|
|
|
| Blank values are allowed
|
|
|
| >>> vrp('')
|
| >>> bool(warned)
|
| False
|
|
|
| Non-string values are not.
|
|
|
| >>> vrp(None)
|
| Traceback (most recent call last):
|
| ...
|
| AttributeError: ...
|
| """
|
| invalid = (
|
| os.path.pardir in path.split(posixpath.sep) or
|
| posixpath.isabs(path) or
|
| ntpath.isabs(path)
|
| )
|
| if not invalid:
|
| return
|
|
|
| msg = "Use of .. or absolute path in a resource path is not allowed."
|
|
|
|
|
| if ntpath.isabs(path) and not posixpath.isabs(path):
|
| raise ValueError(msg)
|
|
|
|
|
|
|
| warnings.warn(
|
| msg[:-1] + " and will raise exceptions in a future release.",
|
| DeprecationWarning,
|
| stacklevel=4,
|
| )
|
|
|
| def _get(self, path):
|
| if hasattr(self.loader, 'get_data'):
|
| return self.loader.get_data(path)
|
| raise NotImplementedError(
|
| "Can't perform this operation for loaders without 'get_data()'"
|
| )
|
|
|
|
|
| register_loader_type(object, NullProvider)
|
|
|
|
|
| def _parents(path):
|
| """
|
| yield all parents of path including path
|
| """
|
| last = None
|
| while path != last:
|
| yield path
|
| last = path
|
| path, _ = os.path.split(path)
|
|
|
|
|
| class EggProvider(NullProvider):
|
| """Provider based on a virtual filesystem"""
|
|
|
| def __init__(self, module):
|
| super().__init__(module)
|
| self._setup_prefix()
|
|
|
| def _setup_prefix(self):
|
|
|
|
|
| eggs = filter(_is_egg_path, _parents(self.module_path))
|
| egg = next(eggs, None)
|
| egg and self._set_egg(egg)
|
|
|
| def _set_egg(self, path):
|
| self.egg_name = os.path.basename(path)
|
| self.egg_info = os.path.join(path, 'EGG-INFO')
|
| self.egg_root = path
|
|
|
|
|
| class DefaultProvider(EggProvider):
|
| """Provides access to package resources in the filesystem"""
|
|
|
| def _has(self, path):
|
| return os.path.exists(path)
|
|
|
| def _isdir(self, path):
|
| return os.path.isdir(path)
|
|
|
| def _listdir(self, path):
|
| return os.listdir(path)
|
|
|
| def get_resource_stream(self, manager, resource_name):
|
| return open(self._fn(self.module_path, resource_name), 'rb')
|
|
|
| def _get(self, path):
|
| with open(path, 'rb') as stream:
|
| return stream.read()
|
|
|
| @classmethod
|
| def _register(cls):
|
| loader_names = 'SourceFileLoader', 'SourcelessFileLoader',
|
| for name in loader_names:
|
| loader_cls = getattr(importlib_machinery, name, type(None))
|
| register_loader_type(loader_cls, cls)
|
|
|
|
|
| DefaultProvider._register()
|
|
|
|
|
| class EmptyProvider(NullProvider):
|
| """Provider that returns nothing for all requests"""
|
|
|
| module_path = None
|
|
|
| _isdir = _has = lambda self, path: False
|
|
|
| def _get(self, path):
|
| return ''
|
|
|
| def _listdir(self, path):
|
| return []
|
|
|
| def __init__(self):
|
| pass
|
|
|
|
|
| empty_provider = EmptyProvider()
|
|
|
|
|
| class ZipManifests(dict):
|
| """
|
| zip manifest builder
|
| """
|
|
|
| @classmethod
|
| def build(cls, path):
|
| """
|
| Build a dictionary similar to the zipimport directory
|
| caches, except instead of tuples, store ZipInfo objects.
|
|
|
| Use a platform-specific path separator (os.sep) for the path keys
|
| for compatibility with pypy on Windows.
|
| """
|
| with zipfile.ZipFile(path) as zfile:
|
| items = (
|
| (
|
| name.replace('/', os.sep),
|
| zfile.getinfo(name),
|
| )
|
| for name in zfile.namelist()
|
| )
|
| return dict(items)
|
|
|
| load = build
|
|
|
|
|
| class MemoizedZipManifests(ZipManifests):
|
| """
|
| Memoized zipfile manifests.
|
| """
|
| manifest_mod = collections.namedtuple('manifest_mod', 'manifest mtime')
|
|
|
| def load(self, path):
|
| """
|
| Load a manifest at path or return a suitable manifest already loaded.
|
| """
|
| path = os.path.normpath(path)
|
| mtime = os.stat(path).st_mtime
|
|
|
| if path not in self or self[path].mtime != mtime:
|
| manifest = self.build(path)
|
| self[path] = self.manifest_mod(manifest, mtime)
|
|
|
| return self[path].manifest
|
|
|
|
|
| class ZipProvider(EggProvider):
|
| """Resource support for zips and eggs"""
|
|
|
| eagers = None
|
| _zip_manifests = MemoizedZipManifests()
|
|
|
| def __init__(self, module):
|
| super().__init__(module)
|
| self.zip_pre = self.loader.archive + os.sep
|
|
|
| def _zipinfo_name(self, fspath):
|
|
|
|
|
| fspath = fspath.rstrip(os.sep)
|
| if fspath == self.loader.archive:
|
| return ''
|
| if fspath.startswith(self.zip_pre):
|
| return fspath[len(self.zip_pre):]
|
| raise AssertionError(
|
| "%s is not a subpath of %s" % (fspath, self.zip_pre)
|
| )
|
|
|
| def _parts(self, zip_path):
|
|
|
|
|
| fspath = self.zip_pre + zip_path
|
| if fspath.startswith(self.egg_root + os.sep):
|
| return fspath[len(self.egg_root) + 1:].split(os.sep)
|
| raise AssertionError(
|
| "%s is not a subpath of %s" % (fspath, self.egg_root)
|
| )
|
|
|
| @property
|
| def zipinfo(self):
|
| return self._zip_manifests.load(self.loader.archive)
|
|
|
| def get_resource_filename(self, manager, resource_name):
|
| if not self.egg_name:
|
| raise NotImplementedError(
|
| "resource_filename() only supported for .egg, not .zip"
|
| )
|
|
|
| zip_path = self._resource_to_zip(resource_name)
|
| eagers = self._get_eager_resources()
|
| if '/'.join(self._parts(zip_path)) in eagers:
|
| for name in eagers:
|
| self._extract_resource(manager, self._eager_to_zip(name))
|
| return self._extract_resource(manager, zip_path)
|
|
|
| @staticmethod
|
| def _get_date_and_size(zip_stat):
|
| size = zip_stat.file_size
|
|
|
| date_time = zip_stat.date_time + (0, 0, -1)
|
|
|
| timestamp = time.mktime(date_time)
|
| return timestamp, size
|
|
|
|
|
| def _extract_resource(self, manager, zip_path):
|
|
|
| if zip_path in self._index():
|
| for name in self._index()[zip_path]:
|
| last = self._extract_resource(
|
| manager, os.path.join(zip_path, name)
|
| )
|
|
|
| return os.path.dirname(last)
|
|
|
| timestamp, size = self._get_date_and_size(self.zipinfo[zip_path])
|
|
|
| if not WRITE_SUPPORT:
|
| raise IOError('"os.rename" and "os.unlink" are not supported '
|
| 'on this platform')
|
| try:
|
|
|
| real_path = manager.get_cache_path(
|
| self.egg_name, self._parts(zip_path)
|
| )
|
|
|
| if self._is_current(real_path, zip_path):
|
| return real_path
|
|
|
| outf, tmpnam = _mkstemp(
|
| ".$extract",
|
| dir=os.path.dirname(real_path),
|
| )
|
| os.write(outf, self.loader.get_data(zip_path))
|
| os.close(outf)
|
| utime(tmpnam, (timestamp, timestamp))
|
| manager.postprocess(tmpnam, real_path)
|
|
|
| try:
|
| rename(tmpnam, real_path)
|
|
|
| except os.error:
|
| if os.path.isfile(real_path):
|
| if self._is_current(real_path, zip_path):
|
|
|
|
|
| return real_path
|
|
|
| elif os.name == 'nt':
|
| unlink(real_path)
|
| rename(tmpnam, real_path)
|
| return real_path
|
| raise
|
|
|
| except os.error:
|
|
|
| manager.extraction_error()
|
|
|
| return real_path
|
|
|
| def _is_current(self, file_path, zip_path):
|
| """
|
| Return True if the file_path is current for this zip_path
|
| """
|
| timestamp, size = self._get_date_and_size(self.zipinfo[zip_path])
|
| if not os.path.isfile(file_path):
|
| return False
|
| stat = os.stat(file_path)
|
| if stat.st_size != size or stat.st_mtime != timestamp:
|
| return False
|
|
|
| zip_contents = self.loader.get_data(zip_path)
|
| with open(file_path, 'rb') as f:
|
| file_contents = f.read()
|
| return zip_contents == file_contents
|
|
|
| def _get_eager_resources(self):
|
| if self.eagers is None:
|
| eagers = []
|
| for name in ('native_libs.txt', 'eager_resources.txt'):
|
| if self.has_metadata(name):
|
| eagers.extend(self.get_metadata_lines(name))
|
| self.eagers = eagers
|
| return self.eagers
|
|
|
| def _index(self):
|
| try:
|
| return self._dirindex
|
| except AttributeError:
|
| ind = {}
|
| for path in self.zipinfo:
|
| parts = path.split(os.sep)
|
| while parts:
|
| parent = os.sep.join(parts[:-1])
|
| if parent in ind:
|
| ind[parent].append(parts[-1])
|
| break
|
| else:
|
| ind[parent] = [parts.pop()]
|
| self._dirindex = ind
|
| return ind
|
|
|
| def _has(self, fspath):
|
| zip_path = self._zipinfo_name(fspath)
|
| return zip_path in self.zipinfo or zip_path in self._index()
|
|
|
| def _isdir(self, fspath):
|
| return self._zipinfo_name(fspath) in self._index()
|
|
|
| def _listdir(self, fspath):
|
| return list(self._index().get(self._zipinfo_name(fspath), ()))
|
|
|
| def _eager_to_zip(self, resource_name):
|
| return self._zipinfo_name(self._fn(self.egg_root, resource_name))
|
|
|
| def _resource_to_zip(self, resource_name):
|
| return self._zipinfo_name(self._fn(self.module_path, resource_name))
|
|
|
|
|
| register_loader_type(zipimport.zipimporter, ZipProvider)
|
|
|
|
|
| class FileMetadata(EmptyProvider):
|
| """Metadata handler for standalone PKG-INFO files
|
|
|
| Usage::
|
|
|
| metadata = FileMetadata("/path/to/PKG-INFO")
|
|
|
| This provider rejects all data and metadata requests except for PKG-INFO,
|
| which is treated as existing, and will be the contents of the file at
|
| the provided location.
|
| """
|
|
|
| def __init__(self, path):
|
| self.path = path
|
|
|
| def _get_metadata_path(self, name):
|
| return self.path
|
|
|
| def has_metadata(self, name):
|
| return name == 'PKG-INFO' and os.path.isfile(self.path)
|
|
|
| def get_metadata(self, name):
|
| if name != 'PKG-INFO':
|
| raise KeyError("No metadata except PKG-INFO is available")
|
|
|
| with io.open(self.path, encoding='utf-8', errors="replace") as f:
|
| metadata = f.read()
|
| self._warn_on_replacement(metadata)
|
| return metadata
|
|
|
| def _warn_on_replacement(self, metadata):
|
| replacement_char = '�'
|
| if replacement_char in metadata:
|
| tmpl = "{self.path} could not be properly decoded in UTF-8"
|
| msg = tmpl.format(**locals())
|
| warnings.warn(msg)
|
|
|
| def get_metadata_lines(self, name):
|
| return yield_lines(self.get_metadata(name))
|
|
|
|
|
| class PathMetadata(DefaultProvider):
|
| """Metadata provider for egg directories
|
|
|
| Usage::
|
|
|
| # Development eggs:
|
|
|
| egg_info = "/path/to/PackageName.egg-info"
|
| base_dir = os.path.dirname(egg_info)
|
| metadata = PathMetadata(base_dir, egg_info)
|
| dist_name = os.path.splitext(os.path.basename(egg_info))[0]
|
| dist = Distribution(basedir, project_name=dist_name, metadata=metadata)
|
|
|
| # Unpacked egg directories:
|
|
|
| egg_path = "/path/to/PackageName-ver-pyver-etc.egg"
|
| metadata = PathMetadata(egg_path, os.path.join(egg_path,'EGG-INFO'))
|
| dist = Distribution.from_filename(egg_path, metadata=metadata)
|
| """
|
|
|
| def __init__(self, path, egg_info):
|
| self.module_path = path
|
| self.egg_info = egg_info
|
|
|
|
|
| class EggMetadata(ZipProvider):
|
| """Metadata provider for .egg files"""
|
|
|
| def __init__(self, importer):
|
| """Create a metadata provider from a zipimporter"""
|
|
|
| self.zip_pre = importer.archive + os.sep
|
| self.loader = importer
|
| if importer.prefix:
|
| self.module_path = os.path.join(importer.archive, importer.prefix)
|
| else:
|
| self.module_path = importer.archive
|
| self._setup_prefix()
|
|
|
|
|
| _declare_state('dict', _distribution_finders={})
|
|
|
|
|
| def register_finder(importer_type, distribution_finder):
|
| """Register `distribution_finder` to find distributions in sys.path items
|
|
|
| `importer_type` is the type or class of a PEP 302 "Importer" (sys.path item
|
| handler), and `distribution_finder` is a callable that, passed a path
|
| item and the importer instance, yields ``Distribution`` instances found on
|
| that path item. See ``pkg_resources.find_on_path`` for an example."""
|
| _distribution_finders[importer_type] = distribution_finder
|
|
|
|
|
| def find_distributions(path_item, only=False):
|
| """Yield distributions accessible via `path_item`"""
|
| importer = get_importer(path_item)
|
| finder = _find_adapter(_distribution_finders, importer)
|
| return finder(importer, path_item, only)
|
|
|
|
|
| def find_eggs_in_zip(importer, path_item, only=False):
|
| """
|
| Find eggs in zip files; possibly multiple nested eggs.
|
| """
|
| if importer.archive.endswith('.whl'):
|
|
|
|
|
| return
|
| metadata = EggMetadata(importer)
|
| if metadata.has_metadata('PKG-INFO'):
|
| yield Distribution.from_filename(path_item, metadata=metadata)
|
| if only:
|
|
|
| return
|
| for subitem in metadata.resource_listdir(''):
|
| if _is_egg_path(subitem):
|
| subpath = os.path.join(path_item, subitem)
|
| dists = find_eggs_in_zip(zipimport.zipimporter(subpath), subpath)
|
| for dist in dists:
|
| yield dist
|
| elif subitem.lower().endswith(('.dist-info', '.egg-info')):
|
| subpath = os.path.join(path_item, subitem)
|
| submeta = EggMetadata(zipimport.zipimporter(subpath))
|
| submeta.egg_info = subpath
|
| yield Distribution.from_location(path_item, subitem, submeta)
|
|
|
|
|
| register_finder(zipimport.zipimporter, find_eggs_in_zip)
|
|
|
|
|
| def find_nothing(importer, path_item, only=False):
|
| return ()
|
|
|
|
|
| register_finder(object, find_nothing)
|
|
|
|
|
| def _by_version_descending(names):
|
| """
|
| Given a list of filenames, return them in descending order
|
| by version number.
|
|
|
| >>> names = 'bar', 'foo', 'Python-2.7.10.egg', 'Python-2.7.2.egg'
|
| >>> _by_version_descending(names)
|
| ['Python-2.7.10.egg', 'Python-2.7.2.egg', 'bar', 'foo']
|
| >>> names = 'Setuptools-1.2.3b1.egg', 'Setuptools-1.2.3.egg'
|
| >>> _by_version_descending(names)
|
| ['Setuptools-1.2.3.egg', 'Setuptools-1.2.3b1.egg']
|
| >>> names = 'Setuptools-1.2.3b1.egg', 'Setuptools-1.2.3.post1.egg'
|
| >>> _by_version_descending(names)
|
| ['Setuptools-1.2.3.post1.egg', 'Setuptools-1.2.3b1.egg']
|
| """
|
| def try_parse(name):
|
| """
|
| Attempt to parse as a version or return a null version.
|
| """
|
| try:
|
| return packaging.version.Version(name)
|
| except Exception:
|
| return packaging.version.Version('0')
|
|
|
| def _by_version(name):
|
| """
|
| Parse each component of the filename
|
| """
|
| name, ext = os.path.splitext(name)
|
| parts = itertools.chain(name.split('-'), [ext])
|
| return [try_parse(part) for part in parts]
|
|
|
| return sorted(names, key=_by_version, reverse=True)
|
|
|
|
|
| def find_on_path(importer, path_item, only=False):
|
| """Yield distributions accessible on a sys.path directory"""
|
| path_item = _normalize_cached(path_item)
|
|
|
| if _is_unpacked_egg(path_item):
|
| yield Distribution.from_filename(
|
| path_item, metadata=PathMetadata(
|
| path_item, os.path.join(path_item, 'EGG-INFO')
|
| )
|
| )
|
| return
|
|
|
| entries = (
|
| os.path.join(path_item, child)
|
| for child in safe_listdir(path_item)
|
| )
|
|
|
|
|
|
|
|
|
| filtered = (
|
| entry
|
| for entry in entries
|
| if dist_factory(path_item, entry, only)
|
| )
|
|
|
|
|
| path_item_entries = _by_version_descending(filtered)
|
| for entry in path_item_entries:
|
| fullpath = os.path.join(path_item, entry)
|
| factory = dist_factory(path_item, entry, only)
|
| for dist in factory(fullpath):
|
| yield dist
|
|
|
|
|
| def dist_factory(path_item, entry, only):
|
| """Return a dist_factory for the given entry."""
|
| lower = entry.lower()
|
| is_egg_info = lower.endswith('.egg-info')
|
| is_dist_info = (
|
| lower.endswith('.dist-info') and
|
| os.path.isdir(os.path.join(path_item, entry))
|
| )
|
| is_meta = is_egg_info or is_dist_info
|
| return (
|
| distributions_from_metadata
|
| if is_meta else
|
| find_distributions
|
| if not only and _is_egg_path(entry) else
|
| resolve_egg_link
|
| if not only and lower.endswith('.egg-link') else
|
| NoDists()
|
| )
|
|
|
|
|
| class NoDists:
|
| """
|
| >>> bool(NoDists())
|
| False
|
|
|
| >>> list(NoDists()('anything'))
|
| []
|
| """
|
| def __bool__(self):
|
| return False
|
|
|
| def __call__(self, fullpath):
|
| return iter(())
|
|
|
|
|
| def safe_listdir(path):
|
| """
|
| Attempt to list contents of path, but suppress some exceptions.
|
| """
|
| try:
|
| return os.listdir(path)
|
| except (PermissionError, NotADirectoryError):
|
| pass
|
| except OSError as e:
|
|
|
|
|
| if e.errno not in (errno.ENOTDIR, errno.EACCES, errno.ENOENT):
|
| raise
|
| return ()
|
|
|
|
|
| def distributions_from_metadata(path):
|
| root = os.path.dirname(path)
|
| if os.path.isdir(path):
|
| if len(os.listdir(path)) == 0:
|
|
|
| return
|
| metadata = PathMetadata(root, path)
|
| else:
|
| metadata = FileMetadata(path)
|
| entry = os.path.basename(path)
|
| yield Distribution.from_location(
|
| root, entry, metadata, precedence=DEVELOP_DIST,
|
| )
|
|
|
|
|
| def non_empty_lines(path):
|
| """
|
| Yield non-empty lines from file at path
|
| """
|
| with open(path) as f:
|
| for line in f:
|
| line = line.strip()
|
| if line:
|
| yield line
|
|
|
|
|
| def resolve_egg_link(path):
|
| """
|
| Given a path to an .egg-link, resolve distributions
|
| present in the referenced path.
|
| """
|
| referenced_paths = non_empty_lines(path)
|
| resolved_paths = (
|
| os.path.join(os.path.dirname(path), ref)
|
| for ref in referenced_paths
|
| )
|
| dist_groups = map(find_distributions, resolved_paths)
|
| return next(dist_groups, ())
|
|
|
|
|
| register_finder(pkgutil.ImpImporter, find_on_path)
|
|
|
| if hasattr(importlib_machinery, 'FileFinder'):
|
| register_finder(importlib_machinery.FileFinder, find_on_path)
|
|
|
| _declare_state('dict', _namespace_handlers={})
|
| _declare_state('dict', _namespace_packages={})
|
|
|
|
|
| def register_namespace_handler(importer_type, namespace_handler):
|
| """Register `namespace_handler` to declare namespace packages
|
|
|
| `importer_type` is the type or class of a PEP 302 "Importer" (sys.path item
|
| handler), and `namespace_handler` is a callable like this::
|
|
|
| def namespace_handler(importer, path_entry, moduleName, module):
|
| # return a path_entry to use for child packages
|
|
|
| Namespace handlers are only called if the importer object has already
|
| agreed that it can handle the relevant path item, and they should only
|
| return a subpath if the module __path__ does not already contain an
|
| equivalent subpath. For an example namespace handler, see
|
| ``pkg_resources.file_ns_handler``.
|
| """
|
| _namespace_handlers[importer_type] = namespace_handler
|
|
|
|
|
| def _handle_ns(packageName, path_item):
|
| """Ensure that named package includes a subpath of path_item (if needed)"""
|
|
|
| importer = get_importer(path_item)
|
| if importer is None:
|
| return None
|
|
|
|
|
| try:
|
| spec = importer.find_spec(packageName)
|
| except AttributeError:
|
|
|
| with warnings.catch_warnings():
|
| warnings.simplefilter("ignore")
|
| loader = importer.find_module(packageName)
|
| else:
|
| loader = spec.loader if spec else None
|
|
|
| if loader is None:
|
| return None
|
| module = sys.modules.get(packageName)
|
| if module is None:
|
| module = sys.modules[packageName] = types.ModuleType(packageName)
|
| module.__path__ = []
|
| _set_parent_ns(packageName)
|
| elif not hasattr(module, '__path__'):
|
| raise TypeError("Not a package:", packageName)
|
| handler = _find_adapter(_namespace_handlers, importer)
|
| subpath = handler(importer, path_item, packageName, module)
|
| if subpath is not None:
|
| path = module.__path__
|
| path.append(subpath)
|
| importlib.import_module(packageName)
|
| _rebuild_mod_path(path, packageName, module)
|
| return subpath
|
|
|
|
|
| def _rebuild_mod_path(orig_path, package_name, module):
|
| """
|
| Rebuild module.__path__ ensuring that all entries are ordered
|
| corresponding to their sys.path order
|
| """
|
| sys_path = [_normalize_cached(p) for p in sys.path]
|
|
|
| def safe_sys_path_index(entry):
|
| """
|
| Workaround for #520 and #513.
|
| """
|
| try:
|
| return sys_path.index(entry)
|
| except ValueError:
|
| return float('inf')
|
|
|
| def position_in_sys_path(path):
|
| """
|
| Return the ordinal of the path based on its position in sys.path
|
| """
|
| path_parts = path.split(os.sep)
|
| module_parts = package_name.count('.') + 1
|
| parts = path_parts[:-module_parts]
|
| return safe_sys_path_index(_normalize_cached(os.sep.join(parts)))
|
|
|
| new_path = sorted(orig_path, key=position_in_sys_path)
|
| new_path = [_normalize_cached(p) for p in new_path]
|
|
|
| if isinstance(module.__path__, list):
|
| module.__path__[:] = new_path
|
| else:
|
| module.__path__ = new_path
|
|
|
|
|
| def declare_namespace(packageName):
|
| """Declare that package 'packageName' is a namespace package"""
|
|
|
| _imp.acquire_lock()
|
| try:
|
| if packageName in _namespace_packages:
|
| return
|
|
|
| path = sys.path
|
| parent, _, _ = packageName.rpartition('.')
|
|
|
| if parent:
|
| declare_namespace(parent)
|
| if parent not in _namespace_packages:
|
| __import__(parent)
|
| try:
|
| path = sys.modules[parent].__path__
|
| except AttributeError as e:
|
| raise TypeError("Not a package:", parent) from e
|
|
|
|
|
|
|
| _namespace_packages.setdefault(parent or None, []).append(packageName)
|
| _namespace_packages.setdefault(packageName, [])
|
|
|
| for path_item in path:
|
|
|
|
|
| _handle_ns(packageName, path_item)
|
|
|
| finally:
|
| _imp.release_lock()
|
|
|
|
|
| def fixup_namespace_packages(path_item, parent=None):
|
| """Ensure that previously-declared namespace packages include path_item"""
|
| _imp.acquire_lock()
|
| try:
|
| for package in _namespace_packages.get(parent, ()):
|
| subpath = _handle_ns(package, path_item)
|
| if subpath:
|
| fixup_namespace_packages(subpath, package)
|
| finally:
|
| _imp.release_lock()
|
|
|
|
|
| def file_ns_handler(importer, path_item, packageName, module):
|
| """Compute an ns-package subpath for a filesystem or zipfile importer"""
|
|
|
| subpath = os.path.join(path_item, packageName.split('.')[-1])
|
| normalized = _normalize_cached(subpath)
|
| for item in module.__path__:
|
| if _normalize_cached(item) == normalized:
|
| break
|
| else:
|
|
|
| return subpath
|
|
|
|
|
| register_namespace_handler(pkgutil.ImpImporter, file_ns_handler)
|
| register_namespace_handler(zipimport.zipimporter, file_ns_handler)
|
|
|
| if hasattr(importlib_machinery, 'FileFinder'):
|
| register_namespace_handler(importlib_machinery.FileFinder, file_ns_handler)
|
|
|
|
|
| def null_ns_handler(importer, path_item, packageName, module):
|
| return None
|
|
|
|
|
| register_namespace_handler(object, null_ns_handler)
|
|
|
|
|
| def normalize_path(filename):
|
| """Normalize a file/dir name for comparison purposes"""
|
| return os.path.normcase(os.path.realpath(os.path.normpath(
|
| _cygwin_patch(filename))))
|
|
|
|
|
| def _cygwin_patch(filename):
|
| """
|
| Contrary to POSIX 2008, on Cygwin, getcwd (3) contains
|
| symlink components. Using
|
| os.path.abspath() works around this limitation. A fix in os.getcwd()
|
| would probably better, in Cygwin even more so, except
|
| that this seems to be by design...
|
| """
|
| return os.path.abspath(filename) if sys.platform == 'cygwin' else filename
|
|
|
|
|
| def _normalize_cached(filename, _cache={}):
|
| try:
|
| return _cache[filename]
|
| except KeyError:
|
| _cache[filename] = result = normalize_path(filename)
|
| return result
|
|
|
|
|
| def _is_egg_path(path):
|
| """
|
| Determine if given path appears to be an egg.
|
| """
|
| return _is_zip_egg(path) or _is_unpacked_egg(path)
|
|
|
|
|
| def _is_zip_egg(path):
|
| return (
|
| path.lower().endswith('.egg') and
|
| os.path.isfile(path) and
|
| zipfile.is_zipfile(path)
|
| )
|
|
|
|
|
| def _is_unpacked_egg(path):
|
| """
|
| Determine if given path appears to be an unpacked egg.
|
| """
|
| return (
|
| path.lower().endswith('.egg') and
|
| os.path.isfile(os.path.join(path, 'EGG-INFO', 'PKG-INFO'))
|
| )
|
|
|
|
|
| def _set_parent_ns(packageName):
|
| parts = packageName.split('.')
|
| name = parts.pop()
|
| if parts:
|
| parent = '.'.join(parts)
|
| setattr(sys.modules[parent], name, sys.modules[packageName])
|
|
|
|
|
| MODULE = re.compile(r"\w+(\.\w+)*$").match
|
| EGG_NAME = re.compile(
|
| r"""
|
| (?P<name>[^-]+) (
|
| -(?P<ver>[^-]+) (
|
| -py(?P<pyver>[^-]+) (
|
| -(?P<plat>.+)
|
| )?
|
| )?
|
| )?
|
| """,
|
| re.VERBOSE | re.IGNORECASE,
|
| ).match
|
|
|
|
|
| class EntryPoint:
|
| """Object representing an advertised importable object"""
|
|
|
| def __init__(self, name, module_name, attrs=(), extras=(), dist=None):
|
| if not MODULE(module_name):
|
| raise ValueError("Invalid module name", module_name)
|
| self.name = name
|
| self.module_name = module_name
|
| self.attrs = tuple(attrs)
|
| self.extras = tuple(extras)
|
| self.dist = dist
|
|
|
| def __str__(self):
|
| s = "%s = %s" % (self.name, self.module_name)
|
| if self.attrs:
|
| s += ':' + '.'.join(self.attrs)
|
| if self.extras:
|
| s += ' [%s]' % ','.join(self.extras)
|
| return s
|
|
|
| def __repr__(self):
|
| return "EntryPoint.parse(%r)" % str(self)
|
|
|
| def load(self, require=True, *args, **kwargs):
|
| """
|
| Require packages for this EntryPoint, then resolve it.
|
| """
|
| if not require or args or kwargs:
|
| warnings.warn(
|
| "Parameters to load are deprecated. Call .resolve and "
|
| ".require separately.",
|
| PkgResourcesDeprecationWarning,
|
| stacklevel=2,
|
| )
|
| if require:
|
| self.require(*args, **kwargs)
|
| return self.resolve()
|
|
|
| def resolve(self):
|
| """
|
| Resolve the entry point from its module and attrs.
|
| """
|
| module = __import__(self.module_name, fromlist=['__name__'], level=0)
|
| try:
|
| return functools.reduce(getattr, self.attrs, module)
|
| except AttributeError as exc:
|
| raise ImportError(str(exc)) from exc
|
|
|
| def require(self, env=None, installer=None):
|
| if self.extras and not self.dist:
|
| raise UnknownExtra("Can't require() without a distribution", self)
|
|
|
|
|
|
|
|
|
|
|
|
|
| reqs = self.dist.requires(self.extras)
|
| items = working_set.resolve(reqs, env, installer, extras=self.extras)
|
| list(map(working_set.add, items))
|
|
|
| pattern = re.compile(
|
| r'\s*'
|
| r'(?P<name>.+?)\s*'
|
| r'=\s*'
|
| r'(?P<module>[\w.]+)\s*'
|
| r'(:\s*(?P<attr>[\w.]+))?\s*'
|
| r'(?P<extras>\[.*\])?\s*$'
|
| )
|
|
|
| @classmethod
|
| def parse(cls, src, dist=None):
|
| """Parse a single entry point from string `src`
|
|
|
| Entry point syntax follows the form::
|
|
|
| name = some.module:some.attr [extra1, extra2]
|
|
|
| The entry name and module name are required, but the ``:attrs`` and
|
| ``[extras]`` parts are optional
|
| """
|
| m = cls.pattern.match(src)
|
| if not m:
|
| msg = "EntryPoint must be in 'name=module:attrs [extras]' format"
|
| raise ValueError(msg, src)
|
| res = m.groupdict()
|
| extras = cls._parse_extras(res['extras'])
|
| attrs = res['attr'].split('.') if res['attr'] else ()
|
| return cls(res['name'], res['module'], attrs, extras, dist)
|
|
|
| @classmethod
|
| def _parse_extras(cls, extras_spec):
|
| if not extras_spec:
|
| return ()
|
| req = Requirement.parse('x' + extras_spec)
|
| if req.specs:
|
| raise ValueError()
|
| return req.extras
|
|
|
| @classmethod
|
| def parse_group(cls, group, lines, dist=None):
|
| """Parse an entry point group"""
|
| if not MODULE(group):
|
| raise ValueError("Invalid group name", group)
|
| this = {}
|
| for line in yield_lines(lines):
|
| ep = cls.parse(line, dist)
|
| if ep.name in this:
|
| raise ValueError("Duplicate entry point", group, ep.name)
|
| this[ep.name] = ep
|
| return this
|
|
|
| @classmethod
|
| def parse_map(cls, data, dist=None):
|
| """Parse a map of entry point groups"""
|
| if isinstance(data, dict):
|
| data = data.items()
|
| else:
|
| data = split_sections(data)
|
| maps = {}
|
| for group, lines in data:
|
| if group is None:
|
| if not lines:
|
| continue
|
| raise ValueError("Entry points must be listed in groups")
|
| group = group.strip()
|
| if group in maps:
|
| raise ValueError("Duplicate group name", group)
|
| maps[group] = cls.parse_group(group, lines, dist)
|
| return maps
|
|
|
|
|
| def _version_from_file(lines):
|
| """
|
| Given an iterable of lines from a Metadata file, return
|
| the value of the Version field, if present, or None otherwise.
|
| """
|
| def is_version_line(line):
|
| return line.lower().startswith('version:')
|
| version_lines = filter(is_version_line, lines)
|
| line = next(iter(version_lines), '')
|
| _, _, value = line.partition(':')
|
| return safe_version(value.strip()) or None
|
|
|
|
|
| class Distribution:
|
| """Wrap an actual or potential sys.path entry w/metadata"""
|
| PKG_INFO = 'PKG-INFO'
|
|
|
| def __init__(
|
| self, location=None, metadata=None, project_name=None,
|
| version=None, py_version=PY_MAJOR, platform=None,
|
| precedence=EGG_DIST):
|
| self.project_name = safe_name(project_name or 'Unknown')
|
| if version is not None:
|
| self._version = safe_version(version)
|
| self.py_version = py_version
|
| self.platform = platform
|
| self.location = location
|
| self.precedence = precedence
|
| self._provider = metadata or empty_provider
|
|
|
| @classmethod
|
| def from_location(cls, location, basename, metadata=None, **kw):
|
| project_name, version, py_version, platform = [None] * 4
|
| basename, ext = os.path.splitext(basename)
|
| if ext.lower() in _distributionImpl:
|
| cls = _distributionImpl[ext.lower()]
|
|
|
| match = EGG_NAME(basename)
|
| if match:
|
| project_name, version, py_version, platform = match.group(
|
| 'name', 'ver', 'pyver', 'plat'
|
| )
|
| return cls(
|
| location, metadata, project_name=project_name, version=version,
|
| py_version=py_version, platform=platform, **kw
|
| )._reload_version()
|
|
|
| def _reload_version(self):
|
| return self
|
|
|
| @property
|
| def hashcmp(self):
|
| return (
|
| self.parsed_version,
|
| self.precedence,
|
| self.key,
|
| self.location,
|
| self.py_version or '',
|
| self.platform or '',
|
| )
|
|
|
| def __hash__(self):
|
| return hash(self.hashcmp)
|
|
|
| def __lt__(self, other):
|
| return self.hashcmp < other.hashcmp
|
|
|
| def __le__(self, other):
|
| return self.hashcmp <= other.hashcmp
|
|
|
| def __gt__(self, other):
|
| return self.hashcmp > other.hashcmp
|
|
|
| def __ge__(self, other):
|
| return self.hashcmp >= other.hashcmp
|
|
|
| def __eq__(self, other):
|
| if not isinstance(other, self.__class__):
|
|
|
| return False
|
| return self.hashcmp == other.hashcmp
|
|
|
| def __ne__(self, other):
|
| return not self == other
|
|
|
|
|
|
|
|
|
|
|
| @property
|
| def key(self):
|
| try:
|
| return self._key
|
| except AttributeError:
|
| self._key = key = self.project_name.lower()
|
| return key
|
|
|
| @property
|
| def parsed_version(self):
|
| if not hasattr(self, "_parsed_version"):
|
| self._parsed_version = parse_version(self.version)
|
|
|
| return self._parsed_version
|
|
|
| def _warn_legacy_version(self):
|
| LV = packaging.version.LegacyVersion
|
| is_legacy = isinstance(self._parsed_version, LV)
|
| if not is_legacy:
|
| return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| if not self.version:
|
| return
|
|
|
| tmpl = textwrap.dedent("""
|
| '{project_name} ({version})' is being parsed as a legacy,
|
| non PEP 440,
|
| version. You may find odd behavior and sort order.
|
| In particular it will be sorted as less than 0.0. It
|
| is recommended to migrate to PEP 440 compatible
|
| versions.
|
| """).strip().replace('\n', ' ')
|
|
|
| warnings.warn(tmpl.format(**vars(self)), PEP440Warning)
|
|
|
| @property
|
| def version(self):
|
| try:
|
| return self._version
|
| except AttributeError as e:
|
| version = self._get_version()
|
| if version is None:
|
| path = self._get_metadata_path_for_display(self.PKG_INFO)
|
| msg = (
|
| "Missing 'Version:' header and/or {} file at path: {}"
|
| ).format(self.PKG_INFO, path)
|
| raise ValueError(msg, self) from e
|
|
|
| return version
|
|
|
| @property
|
| def _dep_map(self):
|
| """
|
| A map of extra to its list of (direct) requirements
|
| for this distribution, including the null extra.
|
| """
|
| try:
|
| return self.__dep_map
|
| except AttributeError:
|
| self.__dep_map = self._filter_extras(self._build_dep_map())
|
| return self.__dep_map
|
|
|
| @staticmethod
|
| def _filter_extras(dm):
|
| """
|
| Given a mapping of extras to dependencies, strip off
|
| environment markers and filter out any dependencies
|
| not matching the markers.
|
| """
|
| for extra in list(filter(None, dm)):
|
| new_extra = extra
|
| reqs = dm.pop(extra)
|
| new_extra, _, marker = extra.partition(':')
|
| fails_marker = marker and (
|
| invalid_marker(marker)
|
| or not evaluate_marker(marker)
|
| )
|
| if fails_marker:
|
| reqs = []
|
| new_extra = safe_extra(new_extra) or None
|
|
|
| dm.setdefault(new_extra, []).extend(reqs)
|
| return dm
|
|
|
| def _build_dep_map(self):
|
| dm = {}
|
| for name in 'requires.txt', 'depends.txt':
|
| for extra, reqs in split_sections(self._get_metadata(name)):
|
| dm.setdefault(extra, []).extend(parse_requirements(reqs))
|
| return dm
|
|
|
| def requires(self, extras=()):
|
| """List of Requirements needed for this distro if `extras` are used"""
|
| dm = self._dep_map
|
| deps = []
|
| deps.extend(dm.get(None, ()))
|
| for ext in extras:
|
| try:
|
| deps.extend(dm[safe_extra(ext)])
|
| except KeyError as e:
|
| raise UnknownExtra(
|
| "%s has no such extra feature %r" % (self, ext)
|
| ) from e
|
| return deps
|
|
|
| def _get_metadata_path_for_display(self, name):
|
| """
|
| Return the path to the given metadata file, if available.
|
| """
|
| try:
|
|
|
|
|
|
|
| path = self._provider._get_metadata_path(name)
|
|
|
|
|
|
|
| except Exception:
|
| return '[could not detect]'
|
|
|
| return path
|
|
|
| def _get_metadata(self, name):
|
| if self.has_metadata(name):
|
| for line in self.get_metadata_lines(name):
|
| yield line
|
|
|
| def _get_version(self):
|
| lines = self._get_metadata(self.PKG_INFO)
|
| version = _version_from_file(lines)
|
|
|
| return version
|
|
|
| def activate(self, path=None, replace=False):
|
| """Ensure distribution is importable on `path` (default=sys.path)"""
|
| if path is None:
|
| path = sys.path
|
| self.insert_on(path, replace=replace)
|
| if path is sys.path:
|
| fixup_namespace_packages(self.location)
|
| for pkg in self._get_metadata('namespace_packages.txt'):
|
| if pkg in sys.modules:
|
| declare_namespace(pkg)
|
|
|
| def egg_name(self):
|
| """Return what this distribution's standard .egg filename should be"""
|
| filename = "%s-%s-py%s" % (
|
| to_filename(self.project_name), to_filename(self.version),
|
| self.py_version or PY_MAJOR
|
| )
|
|
|
| if self.platform:
|
| filename += '-' + self.platform
|
| return filename
|
|
|
| def __repr__(self):
|
| if self.location:
|
| return "%s (%s)" % (self, self.location)
|
| else:
|
| return str(self)
|
|
|
| def __str__(self):
|
| try:
|
| version = getattr(self, 'version', None)
|
| except ValueError:
|
| version = None
|
| version = version or "[unknown version]"
|
| return "%s %s" % (self.project_name, version)
|
|
|
| def __getattr__(self, attr):
|
| """Delegate all unrecognized public attributes to .metadata provider"""
|
| if attr.startswith('_'):
|
| raise AttributeError(attr)
|
| return getattr(self._provider, attr)
|
|
|
| def __dir__(self):
|
| return list(
|
| set(super(Distribution, self).__dir__())
|
| | set(
|
| attr for attr in self._provider.__dir__()
|
| if not attr.startswith('_')
|
| )
|
| )
|
|
|
| @classmethod
|
| def from_filename(cls, filename, metadata=None, **kw):
|
| return cls.from_location(
|
| _normalize_cached(filename), os.path.basename(filename), metadata,
|
| **kw
|
| )
|
|
|
| def as_requirement(self):
|
| """Return a ``Requirement`` that matches this distribution exactly"""
|
| if isinstance(self.parsed_version, packaging.version.Version):
|
| spec = "%s==%s" % (self.project_name, self.parsed_version)
|
| else:
|
| spec = "%s===%s" % (self.project_name, self.parsed_version)
|
|
|
| return Requirement.parse(spec)
|
|
|
| def load_entry_point(self, group, name):
|
| """Return the `name` entry point of `group` or raise ImportError"""
|
| ep = self.get_entry_info(group, name)
|
| if ep is None:
|
| raise ImportError("Entry point %r not found" % ((group, name),))
|
| return ep.load()
|
|
|
| def get_entry_map(self, group=None):
|
| """Return the entry point map for `group`, or the full entry map"""
|
| try:
|
| ep_map = self._ep_map
|
| except AttributeError:
|
| ep_map = self._ep_map = EntryPoint.parse_map(
|
| self._get_metadata('entry_points.txt'), self
|
| )
|
| if group is not None:
|
| return ep_map.get(group, {})
|
| return ep_map
|
|
|
| def get_entry_info(self, group, name):
|
| """Return the EntryPoint object for `group`+`name`, or ``None``"""
|
| return self.get_entry_map(group).get(name)
|
|
|
|
|
| def insert_on(self, path, loc=None, replace=False):
|
| """Ensure self.location is on path
|
|
|
| If replace=False (default):
|
| - If location is already in path anywhere, do nothing.
|
| - Else:
|
| - If it's an egg and its parent directory is on path,
|
| insert just ahead of the parent.
|
| - Else: add to the end of path.
|
| If replace=True:
|
| - If location is already on path anywhere (not eggs)
|
| or higher priority than its parent (eggs)
|
| do nothing.
|
| - Else:
|
| - If it's an egg and its parent directory is on path,
|
| insert just ahead of the parent,
|
| removing any lower-priority entries.
|
| - Else: add it to the front of path.
|
| """
|
|
|
| loc = loc or self.location
|
| if not loc:
|
| return
|
|
|
| nloc = _normalize_cached(loc)
|
| bdir = os.path.dirname(nloc)
|
| npath = [(p and _normalize_cached(p) or p) for p in path]
|
|
|
| for p, item in enumerate(npath):
|
| if item == nloc:
|
| if replace:
|
| break
|
| else:
|
|
|
|
|
| return
|
| elif item == bdir and self.precedence == EGG_DIST:
|
|
|
|
|
| if (not replace) and nloc in npath[p:]:
|
| return
|
| if path is sys.path:
|
| self.check_version_conflict()
|
| path.insert(p, loc)
|
| npath.insert(p, nloc)
|
| break
|
| else:
|
| if path is sys.path:
|
| self.check_version_conflict()
|
| if replace:
|
| path.insert(0, loc)
|
| else:
|
| path.append(loc)
|
| return
|
|
|
|
|
| while True:
|
| try:
|
| np = npath.index(nloc, p + 1)
|
| except ValueError:
|
| break
|
| else:
|
| del npath[np], path[np]
|
|
|
| p = np
|
|
|
| return
|
|
|
| def check_version_conflict(self):
|
| if self.key == 'setuptools':
|
|
|
| return
|
|
|
| nsp = dict.fromkeys(self._get_metadata('namespace_packages.txt'))
|
| loc = normalize_path(self.location)
|
| for modname in self._get_metadata('top_level.txt'):
|
| if (modname not in sys.modules or modname in nsp
|
| or modname in _namespace_packages):
|
| continue
|
| if modname in ('pkg_resources', 'setuptools', 'site'):
|
| continue
|
| fn = getattr(sys.modules[modname], '__file__', None)
|
| if fn and (normalize_path(fn).startswith(loc) or
|
| fn.startswith(self.location)):
|
| continue
|
| issue_warning(
|
| "Module %s was already imported from %s, but %s is being added"
|
| " to sys.path" % (modname, fn, self.location),
|
| )
|
|
|
| def has_version(self):
|
| try:
|
| self.version
|
| except ValueError:
|
| issue_warning("Unbuilt egg for " + repr(self))
|
| return False
|
| return True
|
|
|
| def clone(self, **kw):
|
| """Copy this distribution, substituting in any changed keyword args"""
|
| names = 'project_name version py_version platform location precedence'
|
| for attr in names.split():
|
| kw.setdefault(attr, getattr(self, attr, None))
|
| kw.setdefault('metadata', self._provider)
|
| return self.__class__(**kw)
|
|
|
| @property
|
| def extras(self):
|
| return [dep for dep in self._dep_map if dep]
|
|
|
|
|
| class EggInfoDistribution(Distribution):
|
| def _reload_version(self):
|
| """
|
| Packages installed by distutils (e.g. numpy or scipy),
|
| which uses an old safe_version, and so
|
| their version numbers can get mangled when
|
| converted to filenames (e.g., 1.11.0.dev0+2329eae to
|
| 1.11.0.dev0_2329eae). These distributions will not be
|
| parsed properly
|
| downstream by Distribution and safe_version, so
|
| take an extra step and try to get the version number from
|
| the metadata file itself instead of the filename.
|
| """
|
| md_version = self._get_version()
|
| if md_version:
|
| self._version = md_version
|
| return self
|
|
|
|
|
| class DistInfoDistribution(Distribution):
|
| """
|
| Wrap an actual or potential sys.path entry
|
| w/metadata, .dist-info style.
|
| """
|
| PKG_INFO = 'METADATA'
|
| EQEQ = re.compile(r"([\(,])\s*(\d.*?)\s*([,\)])")
|
|
|
| @property
|
| def _parsed_pkg_info(self):
|
| """Parse and cache metadata"""
|
| try:
|
| return self._pkg_info
|
| except AttributeError:
|
| metadata = self.get_metadata(self.PKG_INFO)
|
| self._pkg_info = email.parser.Parser().parsestr(metadata)
|
| return self._pkg_info
|
|
|
| @property
|
| def _dep_map(self):
|
| try:
|
| return self.__dep_map
|
| except AttributeError:
|
| self.__dep_map = self._compute_dependencies()
|
| return self.__dep_map
|
|
|
| def _compute_dependencies(self):
|
| """Recompute this distribution's dependencies."""
|
| dm = self.__dep_map = {None: []}
|
|
|
| reqs = []
|
|
|
| for req in self._parsed_pkg_info.get_all('Requires-Dist') or []:
|
| reqs.extend(parse_requirements(req))
|
|
|
| def reqs_for_extra(extra):
|
| for req in reqs:
|
| if not req.marker or req.marker.evaluate({'extra': extra}):
|
| yield req
|
|
|
| common = types.MappingProxyType(dict.fromkeys(reqs_for_extra(None)))
|
| dm[None].extend(common)
|
|
|
| for extra in self._parsed_pkg_info.get_all('Provides-Extra') or []:
|
| s_extra = safe_extra(extra.strip())
|
| dm[s_extra] = [r for r in reqs_for_extra(extra) if r not in common]
|
|
|
| return dm
|
|
|
|
|
| _distributionImpl = {
|
| '.egg': Distribution,
|
| '.egg-info': EggInfoDistribution,
|
| '.dist-info': DistInfoDistribution,
|
| }
|
|
|
|
|
| def issue_warning(*args, **kw):
|
| level = 1
|
| g = globals()
|
| try:
|
|
|
|
|
| while sys._getframe(level).f_globals is g:
|
| level += 1
|
| except ValueError:
|
| pass
|
| warnings.warn(stacklevel=level + 1, *args, **kw)
|
|
|
|
|
| def parse_requirements(strs):
|
| """
|
| Yield ``Requirement`` objects for each specification in `strs`.
|
|
|
| `strs` must be a string, or a (possibly-nested) iterable thereof.
|
| """
|
| return map(Requirement, join_continuation(map(drop_comment, yield_lines(strs))))
|
|
|
|
|
| class RequirementParseError(packaging.requirements.InvalidRequirement):
|
| "Compatibility wrapper for InvalidRequirement"
|
|
|
|
|
| class Requirement(packaging.requirements.Requirement):
|
| def __init__(self, requirement_string):
|
| """DO NOT CALL THIS UNDOCUMENTED METHOD; use Requirement.parse()!"""
|
| super(Requirement, self).__init__(requirement_string)
|
| self.unsafe_name = self.name
|
| project_name = safe_name(self.name)
|
| self.project_name, self.key = project_name, project_name.lower()
|
| self.specs = [
|
| (spec.operator, spec.version) for spec in self.specifier]
|
| self.extras = tuple(map(safe_extra, self.extras))
|
| self.hashCmp = (
|
| self.key,
|
| self.url,
|
| self.specifier,
|
| frozenset(self.extras),
|
| str(self.marker) if self.marker else None,
|
| )
|
| self.__hash = hash(self.hashCmp)
|
|
|
| def __eq__(self, other):
|
| return (
|
| isinstance(other, Requirement) and
|
| self.hashCmp == other.hashCmp
|
| )
|
|
|
| def __ne__(self, other):
|
| return not self == other
|
|
|
| def __contains__(self, item):
|
| if isinstance(item, Distribution):
|
| if item.key != self.key:
|
| return False
|
|
|
| item = item.version
|
|
|
|
|
|
|
|
|
| return self.specifier.contains(item, prereleases=True)
|
|
|
| def __hash__(self):
|
| return self.__hash
|
|
|
| def __repr__(self):
|
| return "Requirement.parse(%r)" % str(self)
|
|
|
| @staticmethod
|
| def parse(s):
|
| req, = parse_requirements(s)
|
| return req
|
|
|
|
|
| def _always_object(classes):
|
| """
|
| Ensure object appears in the mro even
|
| for old-style classes.
|
| """
|
| if object not in classes:
|
| return classes + (object,)
|
| return classes
|
|
|
|
|
| def _find_adapter(registry, ob):
|
| """Return an adapter factory for `ob` from `registry`"""
|
| types = _always_object(inspect.getmro(getattr(ob, '__class__', type(ob))))
|
| for t in types:
|
| if t in registry:
|
| return registry[t]
|
|
|
|
|
| def ensure_directory(path):
|
| """Ensure that the parent directory of `path` exists"""
|
| dirname = os.path.dirname(path)
|
| os.makedirs(dirname, exist_ok=True)
|
|
|
|
|
| def _bypass_ensure_directory(path):
|
| """Sandbox-bypassing version of ensure_directory()"""
|
| if not WRITE_SUPPORT:
|
| raise IOError('"os.mkdir" not supported on this platform.')
|
| dirname, filename = split(path)
|
| if dirname and filename and not isdir(dirname):
|
| _bypass_ensure_directory(dirname)
|
| try:
|
| mkdir(dirname, 0o755)
|
| except FileExistsError:
|
| pass
|
|
|
|
|
| def split_sections(s):
|
| """Split a string or iterable thereof into (section, content) pairs
|
|
|
| Each ``section`` is a stripped version of the section header ("[section]")
|
| and each ``content`` is a list of stripped lines excluding blank lines and
|
| comment-only lines. If there are any such lines before the first section
|
| header, they're returned in a first ``section`` of ``None``.
|
| """
|
| section = None
|
| content = []
|
| for line in yield_lines(s):
|
| if line.startswith("["):
|
| if line.endswith("]"):
|
| if section or content:
|
| yield section, content
|
| section = line[1:-1].strip()
|
| content = []
|
| else:
|
| raise ValueError("Invalid section heading", line)
|
| else:
|
| content.append(line)
|
|
|
|
|
| yield section, content
|
|
|
|
|
| def _mkstemp(*args, **kw):
|
| old_open = os.open
|
| try:
|
|
|
| os.open = os_open
|
| return tempfile.mkstemp(*args, **kw)
|
| finally:
|
|
|
| os.open = old_open
|
|
|
|
|
|
|
|
|
|
|
|
|
| warnings.filterwarnings("ignore", category=PEP440Warning, append=True)
|
|
|
|
|
|
|
| def _call_aside(f, *args, **kwargs):
|
| f(*args, **kwargs)
|
| return f
|
|
|
|
|
| @_call_aside
|
| def _initialize(g=globals()):
|
| "Set up global resource manager (deliberately not state-saved)"
|
| manager = ResourceManager()
|
| g['_manager'] = manager
|
| g.update(
|
| (name, getattr(manager, name))
|
| for name in dir(manager)
|
| if not name.startswith('_')
|
| )
|
|
|
|
|
| class PkgResourcesDeprecationWarning(Warning):
|
| """
|
| Base class for warning about deprecations in ``pkg_resources``
|
|
|
| This class is not derived from ``DeprecationWarning``, and as such is
|
| visible by default.
|
| """
|
|
|
|
|
| @_call_aside
|
| def _initialize_master_working_set():
|
| """
|
| Prepare the master working set and make the ``require()``
|
| API available.
|
|
|
| This function has explicit effects on the global state
|
| of pkg_resources. It is intended to be invoked once at
|
| the initialization of this module.
|
|
|
| Invocation by other packages is unsupported and done
|
| at their own risk.
|
| """
|
| working_set = WorkingSet._build_master()
|
| _declare_state('object', working_set=working_set)
|
|
|
| require = working_set.require
|
| iter_entry_points = working_set.iter_entry_points
|
| add_activation_listener = working_set.subscribe
|
| run_script = working_set.run_script
|
|
|
| run_main = run_script
|
|
|
|
|
|
|
|
|
| tuple(
|
| dist.activate(replace=False)
|
| for dist in working_set
|
| )
|
| add_activation_listener(
|
| lambda dist: dist.activate(replace=True),
|
| existing=False,
|
| )
|
| working_set.entries = []
|
|
|
| list(map(working_set.add_entry, sys.path))
|
| globals().update(locals())
|
|
|