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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.125.21
+Version: 0.125.22
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

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

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=w16go3vHbb763oQxxUHlPrGNReaPnuMsxxD4YRz2JeE,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
@@ -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.21.dist-info/METADATA,sha256=Jq0hDSrXzEM1SXuG6bYbXjJb3dVNQaACnhjTHT_4yhA,12852
-dycw_utilities-0.125.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.125.21.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.125.21.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.21"
+__version__ = "0.125.22"

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"]