dycw-utilities 0.150.5__tar.gz → 0.150.7__tar.gz

{dycw_utilities-0.150.5 → dycw_utilities-0.150.7}/PKG-INFO

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

{dycw_utilities-0.150.5 → dycw_utilities-0.150.7}/pyproject.toml

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

{dycw_utilities-0.150.5 → dycw_utilities-0.150.7}/src/tests/test_postgres.py

@@ -27,6 +27,8 @@ from utilities.typing import get_literal_elements
 if TYPE_CHECKING:
     from pathlib import Path
 
+    from utilities.types import MaybeListStr
+
 
 @composite
 def tables(draw: DrawFn, /) -> list[Table | str]:
@@ -66,6 +68,8 @@ class TestPGDump:
         format_=sampled_from(get_literal_elements(_PGDumpFormat)),
         jobs=integers(min_value=0) | none(),
         create=booleans(),
+        extension=lists(text_ascii(min_size=1)) | none(),
+        extension_exc=lists(text_ascii(min_size=1)) | none(),
         schema=lists(text_ascii(min_size=1)) | none(),
         schema_exc=lists(text_ascii(min_size=1)) | none(),
         table=tables() | none(),
@@ -83,6 +87,8 @@ class TestPGDump:
         format_: _PGDumpFormat,
         jobs: int | None,
         create: bool,
+        extension: MaybeListStr | None,
+        extension_exc: MaybeListStr | None,
         schema: list[str] | None,
         schema_exc: list[str] | None,
         table: list[Table | str] | None,
@@ -98,6 +104,8 @@ class TestPGDump:
             format_=format_,
             jobs=jobs,
             create=create,
+            extension=extension,
+            extension_exc=extension_exc,
             schema=schema,
             schema_exc=schema_exc,
             table=table,

{dycw_utilities-0.150.5 → dycw_utilities-0.150.7}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.150.5"
+__version__ = "0.150.7"

{dycw_utilities-0.150.5 → dycw_utilities-0.150.7}/src/utilities/postgres.py

@@ -37,6 +37,8 @@ async def pg_dump(
     data_only: bool = False,
     clean: bool = False,
     create: bool = False,
+    extension: MaybeListStr | None = None,
+    extension_exc: MaybeListStr | None = None,
     schema: MaybeListStr | None = None,
     schema_exc: MaybeListStr | None = None,
     table: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
@@ -59,6 +61,8 @@ async def pg_dump(
         data_only=data_only,
         clean=clean,
         create=create,
+        extension=extension,
+        extension_exc=extension_exc,
         schema=schema,
         schema_exc=schema_exc,
         table=table,
@@ -70,7 +74,7 @@ async def pg_dump(
     )
     if dry_run:
         if logger is not None:
-            get_logger(logger=logger).info("Would run %r", str(cmd))
+            get_logger(logger=logger).info("Would run:\n\t%r", str(cmd))
         return
     with temp_environ(PGPASSWORD=url.password), Timer() as timer:  # pragma: no cover
         try:
@@ -109,6 +113,8 @@ def _build_pg_dump(
     data_only: bool = False,
     clean: bool = False,
     create: bool = False,
+    extension: MaybeListStr | None = None,
+    extension_exc: MaybeListStr | None = None,
     schema: MaybeListStr | None = None,
     schema_exc: MaybeListStr | None = None,
     table: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
@@ -141,6 +147,12 @@ def _build_pg_dump(
         parts.append(f"--jobs={jobs}")
     if create:
         parts.append("--create")
+    if extension is not None:
+        parts.extend([f"--extension={e}" for e in always_iterable(extension)])
+    if extension_exc is not None:
+        parts.extend([
+            f"--exclude-extension={e}" for e in always_iterable(extension_exc)
+        ])
     if schema is not None:
         parts.extend([f"--schema={s}" for s in always_iterable(schema)])
     if schema_exc is not None:
@@ -222,7 +234,7 @@ async def restore(
     )
     if dry_run:
         if logger is not None:
-            get_logger(logger=logger).info("Would run %r", str(cmd))
+            get_logger(logger=logger).info("Would run:\n\t%r", str(cmd))
         return
     with temp_environ(PGPASSWORD=url.password), Timer() as timer:  # pragma: no cover
         try: