pgdevkit 0.7.0__py3-none-any.whl → 0.7.2__py3-none-any.whl

pgdevkit/testdb/__init__.py

@@ -7,6 +7,7 @@ from .api import (
     run_sql,
     shell_argv,
     status,
+    workspace_db_names,
 )
 
 __all__ = [
@@ -18,4 +19,5 @@ __all__ = [
     "run_sql",
     "shell_argv",
     "status",
+    "workspace_db_names",
 ]

pgdevkit/testdb/api.py

@@ -189,6 +189,20 @@ def find_orphaned_dbs(project_root: Path | None = None) -> list[str]:
     return asyncio.run(_find_orphaned_dbs(config))
 
 
+def workspace_db_names(project_root: Path | None = None) -> frozenset[str]:
+    """Every DB name this exact workspace (the branch currently checked out
+    at `project_root`) owns: its main workspace DB plus one
+    `<main>{suffix}` sibling per configured `extra_db_suffixes` entry. The
+    single-workspace analog of what `find_orphaned_dbs` computes across
+    every *live* worktree -- for a caller that wants "which DBs belong to
+    this one worktree right now" (e.g. to drop them before removing the
+    worktree itself), as opposed to a whole-project orphan sweep. Engine
+    (postgres/mssql) doesn't affect naming, so this doesn't dispatch on it."""
+    config = load_config(project_root)
+    branch = current_branch(config.root)
+    return frozenset(expected_db_names(config, [branch]))
+
+
 def clean_testdb(project_root: Path | None = None, all: bool = False, orphaned: bool = False) -> None:
     """Drop this workspace's database. With all=True, drop every database
     belonging to this project (matched by its name-slug prefix), across

pgdevkit/testdb/naming.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import hashlib
+import os
 import re
 import subprocess
 from pathlib import Path
@@ -25,7 +26,15 @@ def slugify(value: str) -> str:
 
 
 def current_branch(cwd: Path | None = None) -> str:
-    """Return the branch checked out in the git worktree rooted at cwd."""
+    """Return the branch checked out in the git worktree rooted at cwd.
+
+    PGDEVKIT_TESTDB_BRANCH overrides this when set -- lets concurrent CI
+    checkouts (which all otherwise resolve to detached-HEAD "HEAD") get
+    distinct database names on a shared test-DB container.
+    """
+    override = os.environ.get("PGDEVKIT_TESTDB_BRANCH")
+    if override:
+        return override
     result = subprocess.run(
         ["git", "rev-parse", "--abbrev-ref", "HEAD"],
         cwd=cwd,

{pgdevkit-0.7.0.dist-info → pgdevkit-0.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: pgdevkit
-Version: 0.7.0
+Version: 0.7.2
 Summary: A helper for developing with Postgres
 Requires-Python: >=3.14
 Requires-Dist: docker>=7.1.0
@@ -252,6 +252,13 @@ belongs to a live worktree too:
 extra_db_suffixes = ["_myservice"]
 ```
 
+`pgdevkit.testdb.workspace_db_names(project_root=None)` is the
+single-workspace analog of `find_orphaned_dbs()`: it returns the exact set
+of DB names (main + any `extra_db_suffixes`) owned by the branch currently
+checked out at `project_root`, without touching Postgres or diffing against
+other worktrees. Useful for a caller that's about to remove one specific
+worktree and wants to know exactly which DB(s) go with it.
+
 Container connection defaults (`localhost:54322`, `postgres`/`testpwd`) can
 be overridden with `PGDEVKIT_TESTDB_HOST`, `PGDEVKIT_TESTDB_PORT`,
 `PGDEVKIT_TESTDB_USER`, `PGDEVKIT_TESTDB_PASSWORD`. Before touching the

{pgdevkit-0.7.0.dist-info → pgdevkit-0.7.2.dist-info}/RECORD

@@ -26,13 +26,13 @@ pgdevkit/db/loader.py,sha256=0Swmp0h0s7I4acG0Dv3cem2YD4lFvq8aFEbFQPPQrxg,616
 pgdevkit/db/model.py,sha256=ntuXonq-SCGkbNw-lvdMS5VuzLjxYRT8O6xaVqiw56I,980
 pgdevkit/db/mssql_crud.py,sha256=yN4izZ0fsFwAs14008JECHH4dqjNbdHUHGEG8kidyNs,11374
 pgdevkit/db/mssql_sql.py,sha256=DO3UMsTFN9pg0tNdaPGb0wf6lWlHGcqwqzkrGVhJF2g,2292
-pgdevkit/testdb/__init__.py,sha256=wxXvB00ZZVCIVYddo6zaJvH1xTgIiWRKAs_JGJ9BR8g,316
+pgdevkit/testdb/__init__.py,sha256=tphuORo4t0QHQp_gFi6ZHXXgd32w6qUoKlaQCab6izA,366
 pgdevkit/testdb/_docker.py,sha256=kj_0xrGO_mxHlK83ilsB-yE0ELX1kpH22ruYrS-N3iU,1467
-pgdevkit/testdb/api.py,sha256=6Vo0V-G232hPL8GdCPV88aH23a8WWmiyMmeDpQdQM9U,8912
+pgdevkit/testdb/api.py,sha256=13YgTz6hPcYWfsUlQJhJ7k20-W_4hzepedVaT8gWhxI,9709
 pgdevkit/testdb/config.py,sha256=NyeskDeThUgwOuxiMgHVnUOiHW-xVO9s2lJQoJcQwW4,2839
 pgdevkit/testdb/constants.py,sha256=oykc7xoX2R0ykhjXpqIdwkZoeHcScme8S2UPYvi_d_w,1718
 pgdevkit/testdb/container.py,sha256=PD2xUksIlAWXXus2VtAil_8UnwlG5DUcyUvbZx0eZTg,3123
-pgdevkit/testdb/naming.py,sha256=R7zEZaqFqwqlXS8rkfh1zGnuD3jnzdt1A7aCibSnSpM,4060
+pgdevkit/testdb/naming.py,sha256=iHZVYdBQjHOD5sihgmZFe2qOC1djgBb6iP6R8wLUfL0,4377
 pgdevkit/testdb/query.py,sha256=iYpy_EoYIq5USfwr6pNktmDjcGi_yQx0vHvjHZNGRr0,3854
 pgdevkit/testdb/schema.py,sha256=XYBK3xyay98a5gzFW6SAQ336Jge-NhOQhildSQDOaf4,15101
 pgdevkit/testdb/mssql/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
@@ -43,7 +43,7 @@ pgdevkit/docs/database-layout.md,sha256=j9N8GuT63D2oVf9rIOEv8GTWs4yTtjZTGH-zM5Wr
 pgdevkit/skills/pgdevkit/SKILL.md,sha256=ADqEaQtMJPUUAiieqokzUUmIf27bWWJNB1ZvsfgiKAE,12205
 pgdevkit/skills/pgdevkit/references/dynamic-sql.md,sha256=aTR1sx81moiTwaMaiOecaqk-9rT8y5Zxz7W1BoC3vK0,2774
 pgdevkit/skills/pgdevkit/references/temporal-tables.md,sha256=LwZnItvQWI7bB0oRHxIfUc0ynnq7N0V5Mnv-QHvFN3Q,1364
-pgdevkit-0.7.0.dist-info/METADATA,sha256=rDdxAZz8tcUdv_WsexR8My5Tj28P80_qs7bYDI8hOTQ,19945
-pgdevkit-0.7.0.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-pgdevkit-0.7.0.dist-info/entry_points.txt,sha256=FK2P55lf0WaGlOZJIINOGgCt8WB7ep0OnPsgnrp80bg,42
-pgdevkit-0.7.0.dist-info/RECORD,,
+pgdevkit-0.7.2.dist-info/METADATA,sha256=6LQs_7tl-vQ-I8WnTSpl7uRIesuDropRBTL8fcE22GE,20367
+pgdevkit-0.7.2.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+pgdevkit-0.7.2.dist-info/entry_points.txt,sha256=FK2P55lf0WaGlOZJIINOGgCt8WB7ep0OnPsgnrp80bg,42
+pgdevkit-0.7.2.dist-info/RECORD,,

{pgdevkit-0.7.0.dist-info → pgdevkit-0.7.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.32.0
+Generator: hatchling 1.32.4
 Root-Is-Purelib: true
 Tag: py3-none-any