dycw-utilities 0.175.12__py3-none-any.whl → 0.175.14__py3-none-any.whl

{dycw_utilities-0.175.12.dist-info → dycw_utilities-0.175.14.dist-info}/METADATA

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

{dycw_utilities-0.175.12.dist-info → dycw_utilities-0.175.14.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=VXQVfBDPDHfSTokbF78T_FrlVXj_bZnVPxu5nh8fDqA,61
+utilities/__init__.py,sha256=BPQLImIyoL3MqLo2IQpQFaliGFVZmxQJLSwMiVrwkhA,61
 utilities/altair.py,sha256=TLfRFbG9HwG7SLXoJ-v0r-t49ZaGgTQZD82cpjVi4vs,9085
 utilities/asyncio.py,sha256=aJySVxBY0gqsIYnoNmH7-1r8djKuf4vSsU69VCD08t8,16772
 utilities/atomicwrites.py,sha256=tPo6r-Rypd9u99u66B9z86YBPpnLrlHtwox_8Z7T34Y,5790
@@ -80,7 +80,7 @@ utilities/sqlalchemy.py,sha256=HQYpd7LFxdTF5WYVWYtCJeEBI71EJm7ytvCGyAH9B-U,37163
 utilities/sqlalchemy_polars.py,sha256=JCGhB37raSR7fqeWV5dTsciRTMVzIdVT9YSqKT0piT0,13370
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/string.py,sha256=shmBK87zZwzGyixuNuXCiUbqzfeZ9xlrFwz6JTaRvDk,582
-utilities/subprocess.py,sha256=NmPV4ffjNmZjOKPTrmKqYu-9CrENgQC424iyedMmUes,41275
+utilities/subprocess.py,sha256=pagoVFbp_xJyHk4hei3vwZnaSZuMKgodKY6KF1ToqX8,42379
 utilities/tempfile.py,sha256=QyvIdfV4r4YZ0NeNYsg0tCijThLKa7Z32u5Kxy6ZsGo,3619
 utilities/testbook.py,sha256=j1KmaVbrX9VrbeMgtPh5gk55myAsn3dyRUn7jGbPbRk,1294
 utilities/text.py,sha256=7SvwcSR2l_5cOrm1samGnR4C-ZI6qyFLHLzSpO1zeHQ,13958
@@ -97,7 +97,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.175.12.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
-dycw_utilities-0.175.12.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
-dycw_utilities-0.175.12.dist-info/METADATA,sha256=eGXw6LhVNB5hf61GEagQHjxDqu74NI5RTBMeKleAkaw,1443
-dycw_utilities-0.175.12.dist-info/RECORD,,
+dycw_utilities-0.175.14.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
+dycw_utilities-0.175.14.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
+dycw_utilities-0.175.14.dist-info/METADATA,sha256=2aXs3olCrTO08_DVJn8u6uvXpl6IQpm2yiNY6oxUcVA,1443
+dycw_utilities-0.175.14.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.175.12"
+__version__ = "0.175.14"

utilities/subprocess.py

@@ -18,16 +18,18 @@ from typing import IO, TYPE_CHECKING, Literal, assert_never, overload, override
 from utilities.errors import ImpossibleCaseError
 from utilities.iterables import always_iterable
 from utilities.logging import to_logger
+from utilities.pathlib import PWD
 from utilities.permissions import Permissions, ensure_perms
 from utilities.tempfile import TemporaryDirectory
 from utilities.text import strip_and_dedent
-from utilities.whenever import to_seconds
+from utilities.whenever import SECOND, to_seconds
 
 if TYPE_CHECKING:
     from collections.abc import Callable, Iterator
 
     from utilities.permissions import PermissionsLike
     from utilities.types import (
+        Delta,
         LoggerLike,
         MaybeIterable,
         PathLike,
@@ -349,6 +351,24 @@ def mv_cmd(src: PathLike, dest: PathLike, /) -> list[str]:
 ##
 
 
+def ripgrep(*args: str, path: PathLike = PWD) -> str | None:
+    """Search for lines."""
+    try:  # skipif-ci
+        return run(*ripgrep_cmd(*args, path=path), return_=True)
+    except CalledProcessError as error:  # skipif-ci
+        if error.returncode == 1:
+            return None
+        raise
+
+
+def ripgrep_cmd(*args: str, path: PathLike = PWD) -> list[str]:
+    """Command to use 'ripgrep' to search for lines."""
+    return ["rg", *args, str(path)]
+
+
+##
+
+
 def rm(path: PathLike, /, *paths: PathLike, sudo: bool = False) -> None:
     """Remove a file/directory."""
     if sudo:  # pragma: no cover
@@ -1128,6 +1148,30 @@ def ssh_opts_cmd(
 ##
 
 
+def ssh_await(
+    user: str,
+    hostname: str,
+    /,
+    *,
+    logger: LoggerLike | None = None,
+    delta: Delta = SECOND,
+) -> None:
+    while True:  # skipif-ci
+        if logger is not None:
+            to_logger(logger).info("Waiting for '%s'...", hostname)
+        try:
+            ssh(user, hostname, "true")
+        except CalledProcessError:
+            sleep(to_seconds(delta))
+        else:
+            if logger is not None:
+                to_logger(logger).info("'%s' is up", hostname)
+            return
+
+
+##
+
+
 def ssh_keyscan(
     hostname: str, /, *, path: PathLike = KNOWN_HOSTS, port: int | None = None
 ) -> None:
@@ -1445,6 +1489,8 @@ __all__ = [
     "mkdir_cmd",
     "mv",
     "mv_cmd",
+    "ripgrep",
+    "ripgrep_cmd",
     "rm",
     "rm_cmd",
     "rsync",
@@ -1453,6 +1499,7 @@ __all__ = [
     "run",
     "set_hostname_cmd",
     "ssh",
+    "ssh_await",
     "ssh_cmd",
     "ssh_keygen_remove",
     "ssh_keygen_remove_cmd",