dycw-utilities 0.175.34__py3-none-any.whl → 0.175.36__py3-none-any.whl

{dycw_utilities-0.175.34.dist-info → dycw_utilities-0.175.36.dist-info}/METADATA

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

{dycw_utilities-0.175.34.dist-info → dycw_utilities-0.175.36.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=mL4P-Ukiy8UcfSj8x_ckAI3_hnXE8m5ck5YhALwBabM,61
+utilities/__init__.py,sha256=7QvekuGqm3-JdDS498qZrn1_4tah8Zape8NgpwbTKUc,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,8 +80,8 @@ 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=mdEnkFjCbMz9VVwjw3AfXnWDi5AIfz3M_Vy85IjCNZY,53523
-utilities/tempfile.py,sha256=a3_M1QyxGZql_VcGkBOQBeWbbkItjgkfIpVyzU1UAic,3843
+utilities/subprocess.py,sha256=4IAEpK0z5OUQ3hsCJFZgkF2TW-osdrBJODaEhMAuQ_Q,53751
+utilities/tempfile.py,sha256=4kRGd4hyINDX4hpcYMtwzDcd-4IKjpTWT0MnjnKD4hE,4221
 utilities/testbook.py,sha256=j1KmaVbrX9VrbeMgtPh5gk55myAsn3dyRUn7jGbPbRk,1294
 utilities/text.py,sha256=7SvwcSR2l_5cOrm1samGnR4C-ZI6qyFLHLzSpO1zeHQ,13958
 utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
@@ -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.34.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
-dycw_utilities-0.175.34.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
-dycw_utilities-0.175.34.dist-info/METADATA,sha256=-obfKB86Pl-Bbb8dtSMilaGoUNqGNACDimr7OecLJGc,1443
-dycw_utilities-0.175.34.dist-info/RECORD,,
+dycw_utilities-0.175.36.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
+dycw_utilities-0.175.36.dist-info/entry_points.txt,sha256=cOGtKeJI0KXLSV7MJ8Dhc2G8jPgDcBDm53MVNJU4ycI,136
+dycw_utilities-0.175.36.dist-info/METADATA,sha256=baQ9cqXdr7o8gURScaioDS4t0Cg3ODkxCO79ptarTL8,1443
+dycw_utilities-0.175.36.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.175.34"
+__version__ = "0.175.36"

utilities/subprocess.py

@@ -192,36 +192,46 @@ def chown(
     /,
     *,
     sudo: bool = False,
+    recursive: bool = False,
     user: str | int | None = None,
     group: str | int | None = None,
 ) -> None:
     """Change file owner and/or group."""
     if sudo:  # pragma: no cover
-        match user, group:
-            case None, None:
-                ...
-            case str() | int() | None, str() | int() | None:
-                run(*sudo_cmd(*chown_cmd(path, user=user, group=group)))
-            case never:
-                assert_never(never)
+        if (user is not None) or (group is not None):
+            args = sudo_cmd(
+                *chown_cmd(path, recursive=recursive, user=user, group=group)
+            )
+            run(*args)
     else:
-        match user, group:
-            case None, None:
-                ...
-            case str() | int(), None:
-                shutil.chown(path, user, group)
-            case None, str() | int():
-                shutil.chown(path, user, group)
-            case str() | int(), str() | int():
-                shutil.chown(path, user, group)
-            case never:
-                assert_never(never)
+        path = Path(path)
+        paths = list(path.rglob("**/*")) if recursive else [path]
+        for p in paths:
+            match user, group:
+                case None, None:
+                    ...
+                case str() | int(), None:
+                    shutil.chown(p, user, group)
+                case None, str() | int():
+                    shutil.chown(p, user, group)
+                case str() | int(), str() | int():
+                    shutil.chown(p, user, group)
+                case never:
+                    assert_never(never)
 
 
 def chown_cmd(
-    path: PathLike, /, *, user: str | int | None = None, group: str | int | None = None
+    path: PathLike,
+    /,
+    *,
+    recursive: bool = False,
+    user: str | int | None = None,
+    group: str | int | None = None,
 ) -> list[str]:
     """Command to use 'chown' to change file owner and/or group."""
+    args: list[str] = ["chown"]
+    if recursive:
+        args.append("-R")
     match user, group:
         case None, None:
             raise ChownCmdError
@@ -233,7 +243,7 @@ def chown_cmd(
             ownership = f"{user}:{group}"
         case never:
             assert_never(never)
-    return ["chown", ownership, str(path)]
+    return [*args, ownership, str(path)]
 
 
 @dataclass(kw_only=True, slots=True)

utilities/tempfile.py

@@ -92,12 +92,19 @@ def TemporaryFile(  # noqa: N802
                 delete=delete,
             ) as temp_dir,
             _temporary_file_outer(
-                temp_dir, delete=delete, name=name, text=text
+                temp_dir,
+                suffix=suffix,
+                prefix=prefix,
+                delete=delete,
+                name=name,
+                text=text,
             ) as temp,
         ):
             yield temp
     else:
-        with _temporary_file_outer(dir, delete=delete, name=name, text=text) as temp:
+        with _temporary_file_outer(
+            dir, suffix=suffix, prefix=prefix, delete=delete, name=name, text=text
+        ) as temp:
             yield temp
 
 
@@ -106,11 +113,15 @@ def _temporary_file_outer(
     path: PathLike,
     /,
     *,
+    suffix: str | None = None,
+    prefix: str | None = None,
     delete: bool = True,
     name: str | None = None,
     text: str | None = None,
 ) -> Iterator[Path]:
-    with _temporary_file_inner(path, delete=delete, name=name) as temp:
+    with _temporary_file_inner(
+        path, suffix=suffix, prefix=prefix, delete=delete, name=name
+    ) as temp:
         if text is not None:
             _ = temp.write_text(text)
         yield temp
@@ -118,11 +129,17 @@ def _temporary_file_outer(
 
 @contextmanager
 def _temporary_file_inner(
-    path: PathLike, /, *, delete: bool = True, name: str | None = None
+    path: PathLike,
+    /,
+    *,
+    suffix: str | None = None,
+    prefix: str | None = None,
+    delete: bool = True,
+    name: str | None = None,
 ) -> Iterator[Path]:
     path = Path(path)
     temp = _NamedTemporaryFile(  # noqa: SIM115
-        dir=path, delete=delete, delete_on_close=False
+        suffix=suffix, prefix=prefix, dir=path, delete=delete, delete_on_close=False
     )
     if name is None:
         yield path / temp.name