dycw-utilities 0.147.1__py3-none-any.whl → 0.147.3__py3-none-any.whl

{dycw_utilities-0.147.1.dist-info → dycw_utilities-0.147.3.dist-info}/METADATA

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

{dycw_utilities-0.147.1.dist-info → dycw_utilities-0.147.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=J75bM8p_57A3_ySjPtqrNeaMUduKbBl5noITru0C5Ig,60
+utilities/__init__.py,sha256=CYx_3txVREoGWejJRJBItmHBOZ94QymP1DKxtNU8fIo,60
 utilities/altair.py,sha256=92E2lCdyHY4Zb-vCw6rEJIsWdKipuu-Tu2ab1ufUfAk,9079
 utilities/asyncio.py,sha256=aB0EtUbUJ5ZKQ5ET-Xfyx6wfUJG2G4vihEX0blK4TGE,14964
 utilities/atomicwrites.py,sha256=xcOWenTBRS0oat3kg7Sqe51AohNThMQ2ixPL7QCG8hw,5795
@@ -48,8 +48,8 @@ utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
 utilities/platform.py,sha256=Ue9LSxYvg9yUXGKuz5aZoy_qkUEXde-v6B09exgSctU,2813
 utilities/polars.py,sha256=BgiDryAVOapi41ddfJqN0wYh_sDj8BNEYtPB36LaHdo,71824
 utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
-utilities/postgres.py,sha256=jAxd0ouhpt5J42JuegcTsR95nFFEYpQw3rYgQQKGFpU,7607
-utilities/pottery.py,sha256=doqA59CMGvErUTK3vubTUe1sYqY90VvdnIjws4W5Ta0,6953
+utilities/postgres.py,sha256=dbQrUFOoV6huNeuYTnSpNhSNq-QhXkBfrJPMtoAGJtY,7877
+utilities/pottery.py,sha256=JluZG7SUSw-JnN73_7QV-vJyPrTsct3q86p8eqVUasc,7019
 utilities/pqdm.py,sha256=BTsYPtbKQWwX-iXF4qCkfPG7DPxIB54J989n83bXrIo,3092
 utilities/psutil.py,sha256=KUlu4lrUw9Zg1V7ZGetpWpGb9DB8l_SSDWGbANFNCPU,2104
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -88,8 +88,8 @@ utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
 utilities/pytest_plugins/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
 utilities/pytest_plugins/pytest_randomly.py,sha256=NXzCcGKbpgYouz5yehKb4jmxmi2SexKKpgF4M65bi10,414
 utilities/pytest_plugins/pytest_regressions.py,sha256=Iwhfv_OJH7UCPZCfoh7ugZ2Xjqjil-BBBsOb8sDwiGI,1471
-dycw_utilities-0.147.1.dist-info/METADATA,sha256=eMrWFM8cz1gqxHYD9FBko7fpl62DYiU4auzbbUWXXIA,1697
-dycw_utilities-0.147.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.147.1.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.147.1.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.147.1.dist-info/RECORD,,
+dycw_utilities-0.147.3.dist-info/METADATA,sha256=0s33DB1w4KCYCicIAaLfq_aP_xxUUqvwFV9rpelm0R0,1697
+dycw_utilities-0.147.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.147.3.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.147.3.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.147.3.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.147.1"
+__version__ = "0.147.3"

utilities/postgres.py

@@ -27,6 +27,7 @@ async def pg_dump(
     path: PathLike,
     /,
     *,
+    docker: str | None = None,
     format_: _PGDumpFormat = "plain",
     jobs: int | None = None,
     schemas: MaybeListStr | None = None,
@@ -43,7 +44,10 @@ async def pg_dump(
         raise _PGDumpHostError(url=url)
     if url.port is None:
         raise _PGDumpPortError(url=url)
-    parts: list[str] = [
+    parts: list[str] = []
+    if docker is not None:
+        parts.extend(["docker", "exec", "-it", docker])
+    parts.extend([
         "pg_dump",
         # general options
         f"--dbname={url.database}",
@@ -60,7 +64,7 @@ async def pg_dump(
         f"--host={url.host}",
         f"--port={url.port}",
         "--no-password",
-    ]
+    ])
     if (format_ == "directory") and (jobs is not None):
         parts.append(f"--jobs={jobs}")
     if schemas is not None:
@@ -136,6 +140,7 @@ async def pg_restore(
     /,
     *,
     database: str | None = None,
+    docker: str | None = None,
     data_only: bool = False,
     jobs: int | None = None,
     schemas: MaybeListStr | None = None,
@@ -157,7 +162,10 @@ async def pg_restore(
         raise _PGRestoreHostError(url=url)
     if url.port is None:
         raise _PGRestorePortError(url=url)
-    parts: list[str] = [
+    parts: list[str] = []
+    if docker is not None:
+        parts.extend(["docker", "exec", "-it", docker])
+    parts.extend([
         "pg_restore",
         # general options
         f"--dbname={database_use}",
@@ -171,7 +179,7 @@ async def pg_restore(
         f"--host={url.host}",
         f"--port={url.port}",
         "--no-password",
-    ]
+    ])
     if data_only:
         parts.append("--data-only")
     else:

utilities/pottery.py

@@ -175,7 +175,9 @@ async def _get_first_available_lock(
     )
     async with timeout_td(timeout, error=error):  # skipif-ci-and-not-linux
         while True:
-            if (result := await _get_first_available_lock_if_any(locks)) is not None:
+            if (
+                (result := await _get_first_available_lock_if_any(locks)) is not None
+            ) and (await result.locked() > 0.0):
                 return result
             await sleep_td(sleep)