➖ Deprecate pkg_resources (#165)

* ➖ Deprecate pkg_resources

* 🔥 Remove unnecessary wheels

* Fix lint

Author: ddelange

.pre-commit-config.yaml

@@ -36,7 +36,7 @@ repos:
             'flake8-cognitive-complexity~=0.1.0',
             'flake8-comprehensions~=3.14.0',
             # 'flake8-docstrings~=1.7.0',
-            'flake8-logging-format~=0.9.0',
+            # 'flake8-logging-format~=0.9.0',
             'flake8-mutable~=1.2.0',
             'flake8-print~=4.0.0',
             'flake8-printf-formatting~=1.1.2',

README.md

@@ -129,12 +129,11 @@ Exhaustive dependency trees without the need to install any packages ([at most b
 ```
 $ pipgrip --tree pipgrip
 
-pipgrip (0.12.0)
+pipgrip (0.13.0)
 ├── anytree>=2.4.1 (2.13.0)
 ├── click>=7 (8.3.1)
 ├── packaging>=17 (26.0)
-├── pip>=22.2 (26.0)
-├── setuptools<81,>=38.3 (80.10.2)
+├── pip>=22.2 (26.0.1)
 └── wheel (0.46.3)
     └── packaging>=24.0 (26.0)
 ```

requirements/prod.txt

@@ -1,7 +1,6 @@
 anytree>=2.4.1
 click>=7
 packaging>=17
-setuptools>=38.3,<81  # for pkg_resources ref https://github.com/pypa/setuptools/commit/9c28cdffd423f83e43dbfd39fc793c251da48585
 wheel
 pip>=22.2; python_version>'3.6'  # >=22.2 for --dry-run --report ref ddelange/pipgrip#114
 pip>=7.1.0; python_version<='3.6'  # >=7.1.0 for --constraint

src/pipgrip/__init__.py

@@ -31,12 +31,4 @@
 #
 # SPDX-License-Identifier: BSD-3-Clause
 # version based on .git/refs/tags - make a tag/release locally, or on GitHub (and pull)
-import warnings
-
 from pipgrip._repo_version import version as __version__  # noqa:F401
-
-warnings.filterwarnings(
-    action="ignore",
-    message=".*pkg_resources",
-    category=UserWarning,
-)

src/pipgrip/cli.py

@@ -46,7 +46,6 @@
 from anytree.exporter import DictExporter
 from packaging.markers import default_environment
 from packaging.requirements import InvalidRequirement
-from pkg_resources import RequirementParseError
 
 from pipgrip import __version__
 from pipgrip.compat import PIP_VERSION
@@ -624,7 +623,7 @@ def main(
         for root_dependency in dependencies:
             try:
                 source.root_dep(root_dependency)
-            except (InvalidRequirement, RequirementParseError) as e:
+            except InvalidRequirement as e:
                 if skip_invalid_input:
                     logger.warning(
                         "Skipping invalid requirement '%s': %s", root_dependency, str(e)

src/pipgrip/compat.py

@@ -31,12 +31,12 @@
 #
 # SPDX-License-Identifier: BSD-3-Clause
 import pip
-from pkg_resources import parse_version
+from packaging.version import Version
 
 try:
     from urllib.parse import urlparse
 except ImportError:
     # Python 2
     from urlparse import urlparse  # noqa:F401
 
-PIP_VERSION = list(parse_version(pip.__version__)._version.release)
+PIP_VERSION = list(Version(pip.__version__).release)

src/pipgrip/libs/mixology/package.py

@@ -30,9 +30,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #
 # SPDX-License-Identifier: BSD-3-Clause
-import pkg_resources
-
-from pipgrip.pipper import parse_req
+from pipgrip.pipper import ParsedRequirement, parse_req
 
 
 class Package(object):
@@ -52,7 +50,7 @@ def name(self):  # type: () -> str
         return self._name
 
     @property
-    def req(self):  # type: () -> pkg_resources.Requirement
+    def req(self):  # type: () -> ParsedRequirement
         return self._req
 
     def __eq__(self, other):  # type: () -> bool

src/pipgrip/pipper.py

@@ -40,9 +40,9 @@
 import sys
 from tempfile import NamedTemporaryFile, mkdtemp
 
-import pkg_resources
 from click import echo as _echo
 from packaging.markers import default_environment
+from packaging.requirements import Requirement as PkgRequirement
 from packaging.utils import canonicalize_name
 
 from pipgrip.compat import PIP_VERSION, urlparse
@@ -65,6 +65,34 @@ def read_requirements(path):
         raise RuntimeError("{} is broken".format(path))
 
 
+class ParsedRequirement(object):
+    """Parsed requirement with attributes compatible with the old pkg_resources interface."""
+
+    __slots__ = (
+        "key",
+        "name",
+        "extras",
+        "extras_name",
+        "specs",
+        "url",
+        "marker",
+        "_str",
+    )
+
+    def __init__(self, key, name, extras, extras_name, specs, url, marker, full_str):
+        self.key = key
+        self.name = name
+        self.extras = extras
+        self.extras_name = extras_name
+        self.specs = specs
+        self.url = url
+        self.marker = marker
+        self._str = full_str
+
+    def __str__(self):
+        return self._str
+
+
 _parse_req_cache = {}
 
 
@@ -74,32 +102,37 @@ def parse_req(requirement, extras=None):
     if cache_key in _parse_req_cache:
         return _parse_req_cache[cache_key]
     if requirement == "_root_" or requirement == "." or requirement.startswith(".["):
-        req = pkg_resources.Requirement.parse(
+        parsed = PkgRequirement(
             requirement.replace(".", "rubbish", 1)
             if requirement.startswith(".[")
             else "rubbish"
         )
         if extras:
-            req.extras = extras
-        req.key = "." if requirement.startswith(".[") else requirement
-        full_str = req.__str__().replace(req.name, req.key)
-        req.name = req.key
+            parsed.extras = extras
+        key = "." if requirement.startswith(".[") else requirement
+        full_str = str(parsed).replace(parsed.name, key)
     else:
-        req = pkg_resources.Requirement.parse(requirement)
+        parsed = PkgRequirement(requirement)
         if extras:
-            req.extras = extras
-        req.key = canonicalize_name(req.key)
-        req.name = req.key
-        full_str = req.__str__()  # .replace(req.name, req.key)
+            parsed.extras = extras
+        key = canonicalize_name(parsed.name)
+        full_str = str(parsed).replace(parsed.name, key, 1)
 
-    def __str__():
-        return full_str
-
-    req.__str__ = __str__
-    req.extras_name = (
-        req.name + "[" + ",".join(sorted(req.extras)) + "]" if req.extras else req.name
+    extras_frozen = frozenset(parsed.extras)
+    extras_name = (
+        key + "[" + ",".join(sorted(extras_frozen)) + "]" if extras_frozen else key
+    )
+    specs = [(s.operator, s.version) for s in parsed.specifier]
+    req = ParsedRequirement(
+        key=key,
+        name=key,
+        extras=extras_frozen,
+        extras_name=extras_name,
+        specs=specs,
+        url=parsed.url,
+        marker=parsed.marker,
+        full_str=full_str,
     )
-    req.extras = frozenset(req.extras)
     _parse_req_cache[cache_key] = req
     return req