perf: add __slots__ to Specifiers (#1002) (#1004)

Author: henryiii

src/packaging/specifiers.py

@@ -65,6 +65,12 @@ class InvalidSpecifier(ValueError):
 
 class BaseSpecifier(metaclass=abc.ABCMeta):
     __slots__ = ()
+    __match_args__ = ("_str",)
+
+    @property
+    def _str(self) -> str:
+        """Internal property for match_args"""
+        return str(self)
 
     @abc.abstractmethod
     def __str__(self) -> str:

src/packaging/version.py

@@ -268,6 +268,9 @@ class Version(_BaseVersion):
     """
 
     __slots__ = ("_dev", "_epoch", "_key_cache", "_local", "_post", "_pre", "_release")
+    __match_args__ = ("_str",)
+
+    _regex = re.compile(r"\s*" + VERSION_PATTERN + r"\s*", re.VERBOSE | re.IGNORECASE)
 
     _epoch: int
     _release: tuple[int, ...]
@@ -278,8 +281,6 @@ class Version(_BaseVersion):
 
     _key_cache: CmpKey | None
 
-    _regex = re.compile(r"\s*" + VERSION_PATTERN + r"\s*", re.VERBOSE | re.IGNORECASE)
-
     def __init__(self, version: str) -> None:
         """Initialize a Version object.
 
@@ -386,6 +387,11 @@ def __str__(self) -> str:
 
         return "".join(parts)
 
+    @property
+    def _str(self) -> str:
+        """Internal property for match_args"""
+        return str(self)
+
     @property
     def epoch(self) -> int:
         """The epoch of the version.

tests/test_specifiers.py

@@ -46,6 +46,10 @@ class TestSpecifier:
     def test_specifiers_valid(self, specifier: str) -> None:
         Specifier(specifier)
 
+    def test_match_args(self) -> None:
+        assert Specifier.__match_args__ == ("_str",)
+        assert Specifier(">=1.0")._str == ">=1.0"
+
     @pytest.mark.parametrize(
         "specifier",
         [
@@ -829,6 +833,10 @@ def test_create_from_specifiers(self) -> None:
         spec = SpecifierSet(iter(specs))
         assert set(spec) == set(specs)
 
+    def test_match_args(self) -> None:
+        assert SpecifierSet.__match_args__ == ("_str",)
+        assert SpecifierSet(">=1.0,<2")._str == str(SpecifierSet(">=1.0,<2"))
+
     @pytest.mark.parametrize(
         (
             "initial_prereleases",

tests/test_version.py

@@ -108,6 +108,10 @@ class TestVersion:
     def test_valid_versions(self, version: str) -> None:
         Version(version)
 
+    def test_match_args(self) -> None:
+        assert Version.__match_args__ == ("_str",)
+        assert Version("1.2")._str == "1.2"
+
     @pytest.mark.parametrize(
         "version",
         [