dycw-utilities 0.175.6__py3-none-any.whl → 0.175.7__py3-none-any.whl

{dycw_utilities-0.175.6.dist-info → dycw_utilities-0.175.7.dist-info}/METADATA

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

{dycw_utilities-0.175.6.dist-info → dycw_utilities-0.175.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=7y6XPRHYRyZjE6Vfy185IzZfDUJT8hoQcrfKUT3_4RU,60
+utilities/__init__.py,sha256=B5bIogCgY2363uBTrXOQ-P5Y0xUozj2Nfbz2ZiV4uhs,60
 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=C9H8GiV4gsNRRJY5mo8_2DcNqfeonxcfT5o6icQeSkg,40599
+utilities/subprocess.py,sha256=pTmRcfsIraSfsM182R0gzqvDrBS7-_dRmwOABU1ZB14,41018
 utilities/tempfile.py,sha256=Lx6qa16lL1XVH6WdmD_G9vlN6gLI8nrIurxmsFkPKvg,3022
 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.6.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
-dycw_utilities-0.175.6.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
-dycw_utilities-0.175.6.dist-info/METADATA,sha256=761YNYgBWP5Xs0xN_xnVhdet11rc3wj-eIQfF_XoqNk,1442
-dycw_utilities-0.175.6.dist-info/RECORD,,
+dycw_utilities-0.175.7.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
+dycw_utilities-0.175.7.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
+dycw_utilities-0.175.7.dist-info/METADATA,sha256=inDD1FvpQAWndmJ70WVWYhpWPrGG1MIhPIOR8Yfz98o,1442
+dycw_utilities-0.175.7.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.175.6"
+__version__ = "0.175.7"

utilities/subprocess.py

@@ -217,6 +217,13 @@ def echo_cmd(text: str, /) -> list[str]:
 ##
 
 
+def env_cmds(env: StrStrMapping, /) -> list[str]:
+    return [f"{key}={value}" for key, value in env.items()]
+
+
+##
+
+
 def expand_path(
     path: PathLike, /, *, subs: StrMapping | None = None, sudo: bool = False
 ) -> Path:
@@ -893,11 +900,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,
     print: bool = False,
     print_stdout: bool = False,
@@ -913,11 +921,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,
     print: bool = False,
     print_stdout: bool = False,
@@ -933,11 +942,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,
     print: bool = False,
     print_stdout: bool = False,
@@ -953,11 +963,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,
     print: bool = False,
     print_stdout: bool = False,
@@ -973,11 +984,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,
     print: bool = False,
     print_stdout: bool = False,
@@ -992,11 +1004,12 @@ def ssh(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
     input: str | None = None,  # noqa: A002
     print: bool = False,  # noqa: A002
     print_stdout: bool = False,
@@ -1008,18 +1021,19 @@ def ssh(
     logger: LoggerLike | None = None,
 ) -> str | None:
     """Execute a command on a remote machine."""
-    cmd_and_args = ssh_cmd(  # skipif-ci
+    run_cmd_and_args = ssh_cmd(  # skipif-ci
         user,
         hostname,
-        *cmd_and_cmds_or_args,
+        *cmd_and_args,
         batch_mode=batch_mode,
         host_key_algorithms=host_key_algorithms,
         strict_host_key_checking=strict_host_key_checking,
         port=port,
+        env=env,
     )
     try:  # skipif-ci
         return run(
-            *cmd_and_args,
+            *run_cmd_and_args,
             input=input,
             print=print,
             print_stdout=print_stdout,
@@ -1038,7 +1052,7 @@ def ssh(
         return ssh(
             user,
             hostname,
-            *cmd_and_cmds_or_args,
+            *cmd_and_args,
             batch_mode=batch_mode,
             host_key_algorithms=host_key_algorithms,
             strict_host_key_checking=strict_host_key_checking,
@@ -1072,11 +1086,12 @@ def ssh_cmd(
     user: str,
     hostname: str,
     /,
-    *cmd_and_cmds_or_args: str,
+    *cmd_and_args: str,
     batch_mode: bool = True,
     host_key_algorithms: list[str] = _HOST_KEY_ALGORITHMS,
     strict_host_key_checking: bool = True,
     port: int | None = None,
+    env: StrStrMapping | None = None,
 ) -> list[str]:
     """Command to use 'ssh' to execute a command on a remote machine."""
     args: list[str] = ssh_opts_cmd(
@@ -1085,7 +1100,10 @@ def ssh_cmd(
         strict_host_key_checking=strict_host_key_checking,
         port=port,
     )
-    return [*args, f"{user}@{hostname}", *cmd_and_cmds_or_args]
+    args.append(f"{user}@{hostname}")
+    if env is not None:
+        args.extend(env_cmds(env))
+    return [*args, *cmd_and_args]
 
 
 def ssh_opts_cmd(
@@ -1407,6 +1425,7 @@ __all__ = [
     "cp",
     "cp_cmd",
     "echo_cmd",
+    "env_cmds",
     "expand_path",
     "git_branch_current",
     "git_checkout",