| | """ |
| | 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()) |
| |
|