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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.150.3
+Version: 0.150.5
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -12,7 +12,7 @@ Provides-Extra: logging
 Requires-Dist: coloredlogs<15.1,>=15.0.1; extra == 'logging'
 Provides-Extra: test
 Requires-Dist: dycw-pytest-only<2.2,>=2.1.1; extra == 'test'
-Requires-Dist: hypothesis<6.136,>=6.135.33; extra == 'test'
+Requires-Dist: hypothesis<6.137,>=6.136.0; extra == 'test'
 Requires-Dist: pudb<2025.2,>=2025.1; extra == 'test'
 Requires-Dist: pytest-asyncio<1.2,>=1.1.0; extra == 'test'
 Requires-Dist: pytest-cov<6.3,>=6.2.1; extra == 'test'

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

@@ -41,7 +41,7 @@ fastapi-test = ["httpx", "uvicorn"]
 fpdf2 = ["fpdf2 >= 2.8.3, < 2.9"]
 hashlib-test = ["orjson"]
 http-test = ["orjson"]
-hypothesis = ["hypothesis >= 6.135.33, < 6.136"]
+hypothesis = ["hypothesis >= 6.136.0, < 6.137"]
 hypothesis-test = ["libcst", "pathvalidate", "numpy", "pytest-rerunfailures"]
 inflect = ["inflect >= 7.5.0, < 7.6"]
 jupyter-test = ["pandas", "polars"]
@@ -103,7 +103,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.150.3"
+version = "0.150.5"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -115,7 +115,7 @@ logging = [
 ]
 test = [
   "dycw-pytest-only >= 2.1.1, < 2.2",
-  "hypothesis >= 6.135.33, < 6.136",
+  "hypothesis >= 6.136.0, < 6.137",
   "pudb >= 2025.1, < 2025.2",
   "pytest >= 8.4.1, < 8.5",
   "pytest-asyncio >= 1.1.0, < 1.2",
@@ -136,7 +136,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.150.3"
+current_version = "0.150.5"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.150.3 → dycw_utilities-0.150.5}/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.5}/src/utilities/__init__.py

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

{dycw_utilities-0.150.3 → dycw_utilities-0.150.5}/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:
@@ -190,6 +194,7 @@ async def restore(
     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,
@@ -207,6 +212,7 @@ async def restore(
         database=database,
         data_only=data_only,
         clean=clean,
+        create=create,
         jobs=jobs,
         schema=schema,
         schema_exc=schema_exc,
@@ -255,6 +261,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 +278,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 +296,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 +314,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 +322,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: