dycw-utilities 0.150.3__tar.gz → 0.150.4__tar.gz

{dycw_utilities-0.150.3 → dycw_utilities-0.150.4}/PKG-INFO

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

{dycw_utilities-0.150.3 → dycw_utilities-0.150.4}/pyproject.toml

@@ -103,7 +103,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.150.3"
+version = "0.150.4"
 
 [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.3"
+current_version = "0.150.4"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.150.3 → dycw_utilities-0.150.4}/src/tests/test_postgres.py

@@ -65,6 +65,7 @@ class TestPGDump:
         path=temp_paths(),
         format_=sampled_from(get_literal_elements(_PGDumpFormat)),
         jobs=integers(min_value=0) | none(),
+        create=booleans(),
         schema=lists(text_ascii(min_size=1)) | none(),
         schema_exc=lists(text_ascii(min_size=1)) | none(),
         table=tables() | none(),
@@ -81,6 +82,7 @@ class TestPGDump:
         path: Path,
         format_: _PGDumpFormat,
         jobs: int | None,
+        create: bool,
         schema: list[str] | None,
         schema_exc: list[str] | None,
         table: list[Table | str] | None,
@@ -95,6 +97,7 @@ class TestPGDump:
             path,
             format_=format_,
             jobs=jobs,
+            create=create,
             schema=schema,
             schema_exc=schema_exc,
             table=table,
@@ -135,6 +138,7 @@ class TestRestore:
         path=temp_paths(),
         psql=booleans(),
         database=text_ascii(min_size=1) | none(),
+        create=booleans(),
         jobs=integers(min_value=0) | none(),
         schema=lists(text_ascii(min_size=1)) | none(),
         schema_exc=lists(text_ascii(min_size=1)) | none(),
@@ -149,6 +153,7 @@ class TestRestore:
         path: Path,
         psql: bool,
         database: str | None,
+        create: bool,
         jobs: int | None,
         schema: list[str] | None,
         schema_exc: list[str] | None,
@@ -161,6 +166,7 @@ class TestRestore:
             path,
             psql=psql,
             database=database,
+            create=create,
             jobs=jobs,
             schema=schema,
             schema_exc=schema_exc,

{dycw_utilities-0.150.3 → dycw_utilities-0.150.4}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.150.3"
+__version__ = "0.150.4"

{dycw_utilities-0.150.3 → dycw_utilities-0.150.4}/src/utilities/postgres.py

@@ -36,6 +36,7 @@ async def pg_dump(
     jobs: int | None = None,
     data_only: bool = False,
     clean: bool = False,
+    create: bool = False,
     schema: MaybeListStr | None = None,
     schema_exc: MaybeListStr | None = None,
     table: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
@@ -57,6 +58,7 @@ async def pg_dump(
         jobs=jobs,
         data_only=data_only,
         clean=clean,
+        create=create,
         schema=schema,
         schema_exc=schema_exc,
         table=table,
@@ -106,6 +108,7 @@ def _build_pg_dump(
     jobs: int | None = None,
     data_only: bool = False,
     clean: bool = False,
+    create: bool = False,
     schema: MaybeListStr | None = None,
     schema_exc: MaybeListStr | None = None,
     table: MaybeSequence[TableOrORMInstOrClass | str] | None = None,
@@ -127,7 +130,6 @@ def _build_pg_dump(
         # output options
         *_resolve_data_only_and_clean(data_only=data_only, clean=clean),
         "--large-objects",
-        "--create",
         "--no-owner",
         "--no-privileges",
         # connection options
@@ -137,6 +139,8 @@ def _build_pg_dump(
     ]
     if (format_ == "directory") and (jobs is not None):
         parts.append(f"--jobs={jobs}")
+    if create:
+        parts.append("--create")
     if schema is not None:
         parts.extend([f"--schema={s}" for s in always_iterable(schema)])
     if schema_exc is not None:
@@ -255,6 +259,7 @@ def _build_pg_restore_or_psql(
     database: str | None = None,
     data_only: bool = False,
     clean: bool = False,
+    create: bool = False,
     jobs: int | None = None,
     schema: MaybeListStr | None = None,
     schema_exc: MaybeListStr | None = None,
@@ -271,6 +276,7 @@ def _build_pg_restore_or_psql(
         database=database,
         data_only=data_only,
         clean=clean,
+        create=create,
         jobs=jobs,
         schemas=schema,
         schemas_exc=schema_exc,
@@ -288,6 +294,7 @@ def _build_pg_restore(
     database: str | None = None,
     data_only: bool = False,
     clean: bool = False,
+    create: bool = False,
     jobs: int | None = None,
     schemas: MaybeListStr | None = None,
     schemas_exc: MaybeListStr | None = None,
@@ -305,7 +312,6 @@ def _build_pg_restore(
         "--verbose",
         # restore options
         *_resolve_data_only_and_clean(data_only=data_only, clean=clean),
-        "--create",
         "--exit-on-error",
         "--no-owner",
         "--no-privileges",
@@ -314,6 +320,8 @@ def _build_pg_restore(
         f"--port={port}",
         "--no-password",
     ]
+    if create:
+        parts.append("--create")
     if jobs is not None:
         parts.append(f"--jobs={jobs}")
     if schemas is not None: