dycw-utilities 0.148.3__tar.gz → 0.148.5__tar.gz

{dycw_utilities-0.148.3 → dycw_utilities-0.148.5}/PKG-INFO

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

{dycw_utilities-0.148.3 → dycw_utilities-0.148.5}/pyproject.toml

@@ -102,7 +102,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.148.3"
+version = "0.148.5"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -135,7 +135,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.148.3"
+current_version = "0.148.5"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.148.3 → dycw_utilities-0.148.5}/src/tests/test_postgres.py

@@ -64,6 +64,7 @@ class TestPGDump:
         tables=tables() | none(),
         tables_exc=tables() | none(),
         inserts=booleans(),
+        on_conflict_do_nothing=booleans(),
         logger=text_ascii(min_size=1) | none(),
     )
     async def test_main(
@@ -79,6 +80,7 @@ class TestPGDump:
         tables: list[Table | str] | None,
         tables_exc: list[Table | str] | None,
         inserts: bool,
+        on_conflict_do_nothing: bool,
         logger: str | None,
     ) -> None:
         _ = await pg_dump(
@@ -92,6 +94,7 @@ class TestPGDump:
             tables=tables,
             tables_exc=tables_exc,
             inserts=inserts,
+            on_conflict_do_nothing=on_conflict_do_nothing,
             logger=logger,
             dry_run=True,
         )

{dycw_utilities-0.148.3 → dycw_utilities-0.148.5}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.148.3"
+__version__ = "0.148.5"

{dycw_utilities-0.148.3 → dycw_utilities-0.148.5}/src/utilities/postgres.py

@@ -39,6 +39,7 @@ async def pg_dump(
     tables: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
     tables_exc: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
     inserts: bool = False,
+    on_conflict_do_nothing: bool = False,
     logger: LoggerOrName | None = None,
     dry_run: bool = False,
 ) -> None:
@@ -86,12 +87,14 @@ async def pg_dump(
         ])
     if inserts:
         parts.append("--inserts")
+    if on_conflict_do_nothing:
+        parts.append("--on-conflict-do-nothing")
     if url.username is not None:
         parts.append(f"--username={url.username}")
     cmd = " ".join(parts)
     if dry_run:
         if logger is not None:
-            get_logger(logger=logger).info("Would run %r", str(path))
+            get_logger(logger=logger).info("Would run %r", str(cmd))
         return
     with temp_environ(PGPASSWORD=url.password), Timer() as timer:  # pragma: no cover
         try:
@@ -213,7 +216,7 @@ async def pg_restore(
     cmd = " ".join(parts)
     if dry_run:
         if logger is not None:
-            get_logger(logger=logger).info("Would run %r", str(path))
+            get_logger(logger=logger).info("Would run %r", str(cmd))
         return
     with temp_environ(PGPASSWORD=url.password), Timer() as timer:  # pragma: no cover
         try: