dycw-utilities 0.125.20__py3-none-any.whl → 0.125.22__py3-none-any.whl

{dycw_utilities-0.125.20.dist-info → dycw_utilities-0.125.22.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.125.20
+Version: 0.125.22
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
 Requires-Dist: typing-extensions<4.14,>=4.13.1
 Provides-Extra: test
-Requires-Dist: hypothesis<6.132,>=6.131.30; extra == 'test'
+Requires-Dist: hypothesis<6.132,>=6.131.32; extra == 'test'
 Requires-Dist: pytest-asyncio<1.1,>=1.0.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'
 Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
@@ -78,7 +78,7 @@ Provides-Extra: zzz-test-hypothesis
 Requires-Dist: aiosqlite<0.22,>=0.21.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: asyncpg<0.31,>=0.30.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: greenlet<3.3,>=3.2.0; extra == 'zzz-test-hypothesis'
-Requires-Dist: hypothesis<6.132,>=6.131.30; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.132,>=6.131.32; extra == 'zzz-test-hypothesis'
 Requires-Dist: luigi<3.7,>=3.6.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: numpy<2.3,>=2.2.6; extra == 'zzz-test-hypothesis'
 Requires-Dist: pathvalidate<3.3,>=3.2.3; extra == 'zzz-test-hypothesis'

{dycw_utilities-0.125.20.dist-info → dycw_utilities-0.125.22.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=7jjcydn2HpkHk2ldvyVCqxPJFAZwBb7SiotXzuwh50k,61
+utilities/__init__.py,sha256=mOuoQC76dPEs9bAglMk52el5SSD3hguWO9Zm_CqFpl4,61
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/asyncio.py,sha256=pQ5GRcNyBqmDyeAeSarYugrjSNxmX3E0G9pye-kkPt4,51085
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
@@ -6,14 +6,14 @@ utilities/atools.py,sha256=IYMuFSFGSKyuQmqD6v5IUtDlz8PPw0Sr87Cub_gRU3M,1168
 utilities/cachetools.py,sha256=C1zqOg7BYz0IfQFK8e3qaDDgEZxDpo47F15RTfJM37Q,2910
 utilities/click.py,sha256=SRVkUkWxyO_7AaYvvDl0hNCzdpneX04lf1O97i9MfPw,14311
 utilities/concurrent.py,sha256=s2scTEd2AhXVTW4hpASU2qxV_DiVLALfms55cCQzCvM,2886
-utilities/contextlib.py,sha256=OOIIEa5lXKGzFAnauaul40nlQnQko6Na4ryiMJcHkIg,478
+utilities/contextlib.py,sha256=lpaLJBy3X0UGLWjM98jkQZZq8so4fRmoK-Bheq0uOW4,1027
 utilities/contextvars.py,sha256=RsSGGrbQqqZ67rOydnM7WWIsM2lIE31UHJLejnHJPWY,505
 utilities/cryptography.py,sha256=_CiK_K6c_-uQuUhsUNjNjTL-nqxAh4_1zTfS11Xe120,972
 utilities/cvxpy.py,sha256=Rv1-fD-XYerosCavRF8Pohop2DBkU3AlFaGTfD8AEAA,13776
 utilities/dataclasses.py,sha256=iiC1wpGXWhaocIikzwBt8bbLWyImoUlOlcDZJGejaIg,33011
 utilities/datetime.py,sha256=uYoaOi_C1YtNXGfTN9xlTrW62Re2b1_4Skuv14_MeYQ,38985
 utilities/enum.py,sha256=HoRwVCWzsnH0vpO9ZEcAAIZLMv0Sn2vJxxA4sYMQgDs,5793
-utilities/errors.py,sha256=C89Xm-DHUjfqJKROo1pW-RGat8zDvTzZuy6zBNMkj_s,837
+utilities/errors.py,sha256=wg6Jrgz5T_KbmG_PXF6pbDFSFso9CG93RGz1Om6Q0Lk,848
 utilities/eventkit.py,sha256=6M5Xu1SzN-juk9PqBHwy5dS-ta7T0qA6SMpDsakOJ0E,13039
 utilities/fastapi.py,sha256=LG1-Q8RDi7wsyVN6v74qptPYX8WGXPkFOQFniMvtzjc,2439
 utilities/fpdf2.py,sha256=y1NGXR5chWqLXWpewGV3hlRGMr_5yV1lVRkPBhPEgJI,1843
@@ -50,6 +50,7 @@ utilities/platform.py,sha256=48IOKx1IC6ZJXWG-b56ZQptITcNFhWRjELW72o2dGTA,2398
 utilities/polars.py,sha256=QlmUpYTqHNkcLnWOQh1TW22W2QyLzvifCvBcbsqhpdE,63272
 utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
+utilities/psutil.py,sha256=0dXHKW7t9J3uzY3YtHGiRiZVRWr0uozccioZRCVFQfk,3761
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=f6qtR5mO2YMuyvNmbaEj5YeD9eGA4YYfb7Bjzh9jUs0,1845
 utilities/pyinstrument.py,sha256=OJFDh4o1CWIa4aYPYURdQjgap_nvP45KUsCEe94rQHY,829
@@ -89,7 +90,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=jS31ZAY5OMxFxLja_Yo5Fidi87Pd-GoVZ7Vi_teqVDA,16743
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-5j7IQ9nb7gR43rdgA7ms05im-XuqhAk9EJnQBXxCoQ,1874
-dycw_utilities-0.125.20.dist-info/METADATA,sha256=xMxMTQI22apLwASoFNF6uBV9K2oGR4cBmxYYzv6wzKs,12852
-dycw_utilities-0.125.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.125.20.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.125.20.dist-info/RECORD,,
+dycw_utilities-0.125.22.dist-info/METADATA,sha256=1cHkVeipeZn1O9uboTuGQ1v8cOEjkg3GIBqUnmmaIow,12852
+dycw_utilities-0.125.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.125.22.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.125.22.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.125.20"
+__version__ = "0.125.22"

utilities/contextlib.py

@@ -1,12 +1,17 @@
 from __future__ import annotations
 
+import re
+from contextlib import contextmanager
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator
     from types import TracebackType
 
 
 class NoOpContextManager:
+    """Context-manager for no-op."""
+
     def __enter__(self) -> None:
         return None
 
@@ -20,4 +25,20 @@ class NoOpContextManager:
         return False
 
 
-__all__ = ["NoOpContextManager"]
+##
+
+
+_SUPER_OBJECT_HAS_NO_ATTRIBUTE = re.compile(r"'super' object has no attribute '\w+'")
+
+
+@contextmanager
+def suppress_super_object_attribute_error() -> Iterator[None]:
+    """Suppress the super() attribute error, for mix-ins."""
+    try:
+        yield
+    except AttributeError as error:
+        if not _SUPER_OBJECT_HAS_NO_ATTRIBUTE.search(error.args[0]):
+            raise
+
+
+__all__ = ["NoOpContextManager", "suppress_super_object_attribute_error"]

utilities/errors.py

@@ -15,6 +15,7 @@ class ImpossibleCaseError(Exception):
     def __str__(self) -> str:
         desc = ", ".join(self.case)
         return f"Case must be possible: {desc}."
+        ##
 
 
 ##

utilities/psutil.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from json import dumps
+from logging import getLogger
+from math import isclose, nan
+from pathlib import Path
+from typing import TYPE_CHECKING, Self, override
+
+from psutil import swap_memory, virtual_memory
+
+from utilities.asyncio import Looper
+from utilities.contextlib import suppress_super_object_attribute_error
+from utilities.datetime import SECOND, get_now
+
+if TYPE_CHECKING:
+    import datetime as dt
+    from logging import Logger
+
+    from utilities.types import Duration, PathLike
+
+
+@dataclass(kw_only=True)
+class MemoryMonitorService(Looper[None]):
+    """Service to monitor memory usage."""
+
+    # base
+    freq: Duration = field(default=10 * SECOND, repr=False)
+    backoff: Duration = field(default=10 * SECOND, repr=False)
+    # self
+    console: str | None = field(default=None, repr=False)
+    path: PathLike = "memory.txt"
+    _console: Logger | None = field(init=False, repr=False)
+    _max_age: int | None = field(default=None, init=False, repr=False)
+    _path: Path = field(init=False, repr=False)
+
+    @override
+    def __post_init__(self) -> None:
+        super().__post_init__()
+        if self.console is not None:
+            self._console = getLogger(self.console)
+        self._path = Path(self.path)
+
+    @override
+    async def core(self) -> None:
+        await super().core()
+        memory = MemoryUsage.new()
+        mapping = {
+            "datetime": memory.datetime.strftime("%Y-%m-%d %H:%M:%S"),
+            "virtual used (mb)": memory.virtual_used_mb,
+            "virtual total (mb)": memory.virtual_total_mb,
+            "virtual (%)": memory.virtual_pct,
+            "swap used (mb)": memory.swap_used_mb,
+            "swap total (mb)": memory.swap_total_mb,
+            "swap (%)": memory.swap_pct,
+        }
+        ser = dumps(mapping)
+        with self._path.open(mode="a") as fh:
+            _ = fh.write(f"{ser}\n")
+        if self._console is not None:
+            self._console.info("%s", mapping)
+
+
+##
+
+
+@dataclass(kw_only=True)
+class MemoryUsage:
+    """A memory usage."""
+
+    datetime: dt.datetime = field(default_factory=get_now)
+    virtual_used: int = field(repr=False)
+    virtual_used_mb: int = field(init=False)
+    virtual_total: int = field(repr=False)
+    virtual_total_mb: int = field(init=False)
+    virtual_pct: float = field(init=False)
+    swap_used: int = field(repr=False)
+    swap_used_mb: int = field(init=False)
+    swap_total: int = field(repr=False)
+    swap_total_mb: int = field(init=False)
+    swap_pct: float = field(init=False)
+
+    def __post_init__(self) -> None:
+        with suppress_super_object_attribute_error():
+            super().__post_init__()  # pyright: ignore[reportAttributeAccessIssue]
+        self.virtual_used_mb = self._to_mb(self.virtual_used)
+        self.virtual_total_mb = self._to_mb(self.virtual_total)
+        self.virtual_pct = (
+            nan
+            if isclose(self.virtual_total, 0.0)
+            else self.virtual_used / self.virtual_total
+        )
+        self.swap_used_mb = self._to_mb(self.swap_used)
+        self.swap_total_mb = self._to_mb(self.swap_total)
+        self.swap_pct = (
+            nan if isclose(self.swap_total, 0.0) else self.swap_used / self.swap_total
+        )
+
+    @classmethod
+    def new(cls) -> Self:
+        virtual = virtual_memory()
+        virtual_total = virtual.total
+        swap = swap_memory()
+        return cls(
+            virtual_used=virtual_total - virtual.available,
+            virtual_total=virtual_total,
+            swap_used=swap.used,
+            swap_total=swap.total,
+        )
+
+    def _to_mb(self, bytes_: int) -> int:
+        return round(bytes_ / (1024**2))
+
+
+__all__ = ["MemoryMonitorService", "MemoryUsage"]