dycw-utilities 0.176.4__py3-none-any.whl → 0.176.6__py3-none-any.whl

{dycw_utilities-0.176.4.dist-info → dycw_utilities-0.176.6.dist-info}/METADATA

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

{dycw_utilities-0.176.4.dist-info → dycw_utilities-0.176.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=SJhOje1TStT2FY-F-Dfs6lWe6hk-1K8sp3Q6dwJm87Y,60
+utilities/__init__.py,sha256=30qJasMvZPS_VQwLBhylCmlhjcGr2IwhMF8Ehsdi7-M,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,6 +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=UGD2O2kW8fRYCPpwrBS2qq1j-3f1WsYDkyvaP2bxg3s,3096
 utilities/parse.py,sha256=g7Qm9eBOIeDId2tGA021CIaeF6jp1TI8rx4srdvlyoo,17937
 utilities/pathlib.py,sha256=N4Ip8R9eCM-6GfvxUJ3T9oQIle2C2P52F-13BCFRdTg,9345
 utilities/permissions.py,sha256=vLXlWztSVYffbrxptne7ksj6dU1HLekm4fEvS4ny_4Q,8944
@@ -97,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.4.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
-dycw_utilities-0.176.4.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
-dycw_utilities-0.176.4.dist-info/METADATA,sha256=pJbc4nUEtf3VzoMYkgMGiVlgf1LAV6Yh9eVG-sG9ZVY,1398
-dycw_utilities-0.176.4.dist-info/RECORD,,
+dycw_utilities-0.176.6.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
+dycw_utilities-0.176.6.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
+dycw_utilities-0.176.6.dist-info/METADATA,sha256=2x7y41YJEnsO9-W1RdfA0Ndxo7GSBDyS_Ip3uz4NazY,1398
+dycw_utilities-0.176.6.dist-info/RECORD,,

{dycw_utilities-0.176.4.dist-info → dycw_utilities-0.176.6.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.21
+Generator: uv 0.9.22
 Root-Is-Purelib: true
 Tag: py3-none-any

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.176.4"
+__version__ = "0.176.6"

utilities/packaging.py

@@ -0,0 +1,104 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Self, overload, override
+
+import packaging._parser
+import packaging.requirements
+from packaging.requirements import _parse_requirement
+from packaging.specifiers import Specifier, SpecifierSet
+
+from utilities.iterables import OneEmptyError, one
+
+if TYPE_CHECKING:
+    from packaging._parser import MarkerList
+
+
+@dataclass(order=True, unsafe_hash=True, kw_only=True, slots=True)
+class Requirement:
+    requirement: str
+    _parsed_req: packaging._parser.ParsedRequirement
+    _custom_req: _CustomRequirement
+
+    def __getitem__(self, operator: str, /) -> str:
+        return self.specifier_set[operator]
+
+    @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
+
+    @property
+    def name(self) -> str:
+        return self._parsed_req.name
+
+    @property
+    def specifier(self) -> str:
+        return self._parsed_req.specifier
+
+    @property
+    def specifier_set(self) -> _CustomSpecifierSet:
+        return _CustomSpecifierSet(_parse_requirement(self.requirement).specifier)
+
+    @property
+    def url(self) -> str:
+        return self._parsed_req.url
+
+
+class _CustomRequirement(packaging.requirements.Requirement):
+    @override
+    def __init__(self, requirement_string: str) -> None:
+        super().__init__(requirement_string)
+        parsed = _parse_requirement(requirement_string)
+        self.specifier = _CustomSpecifierSet(parsed.specifier)
+
+
+class _CustomSpecifierSet(SpecifierSet):
+    def __getitem__(self, operator: str, /) -> str:
+        try:
+            return one(s.version for s in self if s.operator == operator)
+        except OneEmptyError:
+            raise KeyError(operator) from None
+
+    @override
+    def __str__(self) -> str:
+        specs = sorted(self._specs, key=self._sort_key)
+        return ", ".join(map(str, specs))
+
+    @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 _sort_key(self, spec: Specifier, /) -> int:
+        return [">=", "<"].index(spec.operator)
+
+
+__all__ = ["Requirement"]