dycw-utilities 0.176.5__py3-none-any.whl → 0.177.0__py3-none-any.whl

{dycw_utilities-0.176.5.dist-info → dycw_utilities-0.177.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dycw-utilities
-Version: 0.176.5
+Version: 0.177.0
 Summary: Miscellaneous Python utilities
 Author: Derek Wan
 Author-email: Derek Wan <d.wan@icloud.com>

{dycw_utilities-0.176.5.dist-info → dycw_utilities-0.177.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=chOgc1NV48pH7zK_NN0RHPlWQSfdIFGjwBbOXK_Es2A,60
+utilities/__init__.py,sha256=RsM8wAOFee7SDF_pWygxynYpc3vDkccPqwghqyUS4wU,60
 utilities/altair.py,sha256=TLfRFbG9HwG7SLXoJ-v0r-t49ZaGgTQZD82cpjVi4vs,9085
 utilities/asyncio.py,sha256=aJySVxBY0gqsIYnoNmH7-1r8djKuf4vSsU69VCD08t8,16772
 utilities/atomicwrites.py,sha256=tPo6r-Rypd9u99u66B9z86YBPpnLrlHtwox_8Z7T34Y,5790
@@ -44,7 +44,7 @@ utilities/operator.py,sha256=C3NylZWGTVWRpwYHOPVhaLgRhw0DfpS4_XQ8KfPhBLQ,3613
 utilities/optuna.py,sha256=C-fhWYiXHVPo1l8QctYkFJ4DyhbSrGorzP1dJb_qvd8,1933
 utilities/orjson.py,sha256=T_0SlK811ysg46d3orvIPY3JpBa4FRMpP2wlPQo7-gU,41854
 utilities/os.py,sha256=kjKKSQfnRqFTTZ315iavaaGd3gGuYNoSWlxVLCJjyQs,4852
-utilities/packaging.py,sha256=kuGenKbyAwxZWL_mbkMzkNXKjhyi9a9tfZQwitrYYlM,2401
+utilities/packaging.py,sha256=z_3v2ofEkqIMLhiEvP8zvQUaUFIqY6d-9XK_ZfsZTb0,3723
 utilities/parse.py,sha256=g7Qm9eBOIeDId2tGA021CIaeF6jp1TI8rx4srdvlyoo,17937
 utilities/pathlib.py,sha256=N4Ip8R9eCM-6GfvxUJ3T9oQIle2C2P52F-13BCFRdTg,9345
 utilities/permissions.py,sha256=vLXlWztSVYffbrxptne7ksj6dU1HLekm4fEvS4ny_4Q,8944
@@ -98,7 +98,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=F4ek0-OBWxHYrZdmoZt76N2RnNyKY5KrEHt7rqO4AQE,60183
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=tdIScrTB2-B-LH0ukb1HUXKooLknOfJNwHk10MuMYvA,3619
-dycw_utilities-0.176.5.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
-dycw_utilities-0.176.5.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
-dycw_utilities-0.176.5.dist-info/METADATA,sha256=g6FzLC0r4g8uwbM_zouJhCaaHM0IsQ1q33xQyS5tZp0,1398
-dycw_utilities-0.176.5.dist-info/RECORD,,
+dycw_utilities-0.177.0.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
+dycw_utilities-0.177.0.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
+dycw_utilities-0.177.0.dist-info/METADATA,sha256=s_2LJzGzSZmuxFdy-JTQqxGi-FJ0iiXla3SpYnoZdrI,1398
+dycw_utilities-0.177.0.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.176.5"
+__version__ = "0.177.0"

utilities/packaging.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
-from typing import TYPE_CHECKING, Self, override
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING, Self, overload, override
 
 import packaging._parser
 import packaging.requirements
@@ -14,31 +14,34 @@ if TYPE_CHECKING:
     from packaging._parser import MarkerList
 
 
-@dataclass(order=True, unsafe_hash=True, kw_only=True, slots=True)
+@dataclass(order=True, unsafe_hash=True, slots=True)
 class Requirement:
     requirement: str
-    _parsed_req: packaging._parser.ParsedRequirement
-    _custom_req: _CustomRequirement
+    _parsed_req: packaging._parser.ParsedRequirement = field(init=False, repr=False)
+    _custom_req: _CustomRequirement = field(init=False, repr=False)
 
-    def __getitem__(self, key: str, /) -> str:
-        return self.specifier_set[key]
+    def __getitem__(self, operator: str, /) -> str:
+        return self.specifier_set[operator]
+
+    def __post_init__(self) -> None:
+        self._parsed_req = _parse_requirement(self.requirement)
+        self._custom_req = _CustomRequirement(self.requirement)
 
     @override
     def __str__(self) -> str:
         return str(self._custom_req)
 
-    @classmethod
-    def new(cls, requirement: str, /) -> Self:
-        return cls(
-            requirement=requirement,
-            _parsed_req=_parse_requirement(requirement),
-            _custom_req=_CustomRequirement(requirement),
-        )
-
     @property
     def extras(self) -> list[str]:
         return self._parsed_req.extras
 
+    @overload
+    def get(self, operator: str, default: str, /) -> str: ...
+    @overload
+    def get(self, operator: str, default: None = None, /) -> str | None: ...
+    def get(self, operator: str, default: str | None = None, /) -> str | None:
+        return self.specifier_set.get(operator, default)
+
     @property
     def marker(self) -> MarkerList | None:
         return self._parsed_req.marker
@@ -47,6 +50,9 @@ class Requirement:
     def name(self) -> str:
         return self._parsed_req.name
 
+    def replace(self, operator: str, version: str, /) -> Self:
+        return type(self)(str(self._custom_req.replace(operator, version)))
+
     @property
     def specifier(self) -> str:
         return self._parsed_req.specifier
@@ -61,26 +67,48 @@ class Requirement:
 
 
 class _CustomRequirement(packaging.requirements.Requirement):
+    specifier: _CustomSpecifierSet
+
     @override
     def __init__(self, requirement_string: str) -> None:
         super().__init__(requirement_string)
         parsed = _parse_requirement(requirement_string)
-        self.specifier = _CustomSpecifierSet(parsed.specifier)
+        self.specifier = _CustomSpecifierSet(parsed.specifier)  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    def replace(self, operator: str, version: str, /) -> Self:
+        new = type(self)(super().__str__())
+        new.specifier = self.specifier.replace(operator, version)
+        return new
 
 
 class _CustomSpecifierSet(SpecifierSet):
-    def __getitem__(self, key: str, /) -> str:
+    def __getitem__(self, operator: str, /) -> str:
         try:
-            return one(s.version for s in self if s.operator == key)
+            return one(s.version for s in self if s.operator == operator)
         except OneEmptyError:
-            raise KeyError(key) from None
+            raise KeyError(operator) from None
 
     @override
     def __str__(self) -> str:
-        specs = sorted(self._specs, key=self._key)
+        specs = sorted(self._specs, key=self._sort_key)
         return ", ".join(map(str, specs))
 
-    def _key(self, spec: Specifier, /) -> int:
+    @overload
+    def get(self, operator: str, default: str, /) -> str: ...
+    @overload
+    def get(self, operator: str, default: None = None, /) -> str | None: ...
+    def get(self, operator: str, default: str | None = None, /) -> str | None:
+        try:
+            return self[operator]
+        except KeyError:
+            return default
+
+    def replace(self, operator: str, version: str, /) -> Self:
+        new = Specifier(spec=f"{operator}{version}")
+        remainder = (s for s in self if s.operator != operator)
+        return type(self)([new, *remainder])
+
+    def _sort_key(self, spec: Specifier, /) -> int:
         return [">=", "<"].index(spec.operator)