127 lines
4.1 KiB
Python
127 lines
4.1 KiB
Python
|
import zipfile
|
||
|
|
from typing import Iterator, List, Optional
|
||
|
|
|
||
|
|
from pip._vendor import pkg_resources
|
||
|
|
from pip._vendor.packaging.utils import canonicalize_name
|
||
|
|
from pip._vendor.packaging.version import parse as parse_version
|
||
|
|
|
||
|
|
from pip._internal.utils import misc # TODO: Move definition here.
|
||
|
|
from pip._internal.utils.packaging import get_installer
|
||
|
|
from pip._internal.utils.wheel import pkg_resources_distribution_for_wheel
|
||
|
|
|
||
|
|
from .base import BaseDistribution, BaseEnvironment, DistributionVersion
|
||
|
|
|
||
|
|
|
||
|
|
class Distribution(BaseDistribution):
|
||
|
|
def __init__(self, dist):
|
||
|
|
# type: (pkg_resources.Distribution) -> None
|
||
|
|
self._dist = dist
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_wheel(cls, path, name):
|
||
|
|
# type: (str, str) -> Distribution
|
||
|
|
with zipfile.ZipFile(path, allowZip64=True) as zf:
|
||
|
|
dist = pkg_resources_distribution_for_wheel(zf, name, path)
|
||
|
|
return cls(dist)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def location(self):
|
||
|
|
# type: () -> Optional[str]
|
||
|
|
return self._dist.location
|
||
|
|
|
||
|
|
@property
|
||
|
|
def metadata_version(self):
|
||
|
|
# type: () -> Optional[str]
|
||
|
|
for line in self._dist.get_metadata_lines(self._dist.PKG_INFO):
|
||
|
|
if line.lower().startswith("metadata-version:"):
|
||
|
|
return line.split(":", 1)[-1].strip()
|
||
|
|
return None
|
||
|
|
|
||
|
|
@property
|
||
|
|
def canonical_name(self):
|
||
|
|
# type: () -> str
|
||
|
|
return canonicalize_name(self._dist.project_name)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def version(self):
|
||
|
|
# type: () -> DistributionVersion
|
||
|
|
return parse_version(self._dist.version)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def installer(self):
|
||
|
|
# type: () -> str
|
||
|
|
return get_installer(self._dist)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def editable(self):
|
||
|
|
# type: () -> bool
|
||
|
|
return misc.dist_is_editable(self._dist)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def local(self):
|
||
|
|
# type: () -> bool
|
||
|
|
return misc.dist_is_local(self._dist)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def in_usersite(self):
|
||
|
|
# type: () -> bool
|
||
|
|
return misc.dist_in_usersite(self._dist)
|
||
|
|
|
||
|
|
|
||
|
|
class Environment(BaseEnvironment):
|
||
|
|
def __init__(self, ws):
|
||
|
|
# type: (pkg_resources.WorkingSet) -> None
|
||
|
|
self._ws = ws
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def default(cls):
|
||
|
|
# type: () -> BaseEnvironment
|
||
|
|
return cls(pkg_resources.working_set)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_paths(cls, paths):
|
||
|
|
# type: (Optional[List[str]]) -> BaseEnvironment
|
||
|
|
return cls(pkg_resources.WorkingSet(paths))
|
||
|
|
|
||
|
|
def _search_distribution(self, name):
|
||
|
|
# type: (str) -> Optional[BaseDistribution]
|
||
|
|
"""Find a distribution matching the ``name`` in the environment.
|
||
|
|
|
||
|
|
This searches from *all* distributions available in the environment, to
|
||
|
|
match the behavior of ``pkg_resources.get_distribution()``.
|
||
|
|
"""
|
||
|
|
canonical_name = canonicalize_name(name)
|
||
|
|
for dist in self.iter_distributions():
|
||
|
|
if dist.canonical_name == canonical_name:
|
||
|
|
return dist
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_distribution(self, name):
|
||
|
|
# type: (str) -> Optional[BaseDistribution]
|
||
|
|
|
||
|
|
# Search the distribution by looking through the working set.
|
||
|
|
dist = self._search_distribution(name)
|
||
|
|
if dist:
|
||
|
|
return dist
|
||
|
|
|
||
|
|
# If distribution could not be found, call working_set.require to
|
||
|
|
# update the working set, and try to find the distribution again.
|
||
|
|
# This might happen for e.g. when you install a package twice, once
|
||
|
|
# using setup.py develop and again using setup.py install. Now when
|
||
|
|
# running pip uninstall twice, the package gets removed from the
|
||
|
|
# working set in the first uninstall, so we have to populate the
|
||
|
|
# working set again so that pip knows about it and the packages gets
|
||
|
|
# picked up and is successfully uninstalled the second time too.
|
||
|
|
try:
|
||
|
|
# We didn't pass in any version specifiers, so this can never
|
||
|
|
# raise pkg_resources.VersionConflict.
|
||
|
|
self._ws.require(name)
|
||
|
|
except pkg_resources.DistributionNotFound:
|
||
|
|
return None
|
||
|
|
return self._search_distribution(name)
|
||
|
|
|
||
|
|
def _iter_distributions(self):
|
||
|
|
# type: () -> Iterator[BaseDistribution]
|
||
|
|
for dist in self._ws:
|
||
|
|
yield Distribution(dist)
|