dycw-utilities 0.167.0__tar.gz → 0.167.1__tar.gz

{dycw_utilities-0.167.0 → dycw_utilities-0.167.1}/PKG-INFO

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

{dycw_utilities-0.167.0 → dycw_utilities-0.167.1}/pyproject.toml

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

{dycw_utilities-0.167.0 → dycw_utilities-0.167.1}/src/tests/test_sqlalchemy.py

@@ -69,6 +69,7 @@ from utilities.sqlalchemy import (
     create_engine,
     ensure_database_created,
     ensure_database_dropped,
+    ensure_database_users_disconnected,
     ensure_tables_created,
     ensure_tables_dropped,
     enum_name,
@@ -249,6 +250,21 @@ class TestEnsureDatabaseCreatedAndDropped:
             _ = (await conn.execute(sel)).all()
 
 
+@SKIPIF_CI
+class TestEnsureDatabaseUsersDisconnected:
+    async def test_main(self) -> None:
+        url = URL.create(
+            "postgresql+asyncpg",
+            username=getuser(),
+            password="password",  # noqa: S106
+            host="localhost",
+            port=5432,
+            database="postgres",
+        )
+        database = f"testing_{_table_names()}"
+        await ensure_database_users_disconnected(url, database)
+
+
 class TestEnsureTablesCreated:
     async def test_table(self, *, test_async_engine: AsyncEngine) -> None:
         table = Table(

{dycw_utilities-0.167.0 → dycw_utilities-0.167.1}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.167.0"
+__version__ = "0.167.1"

{dycw_utilities-0.167.0 → dycw_utilities-0.167.1}/src/utilities/sqlalchemy.py

@@ -331,6 +331,20 @@ async def ensure_database_dropped(super_: URL, database: str, /) -> None:
         _ = await conn.execute(text(f"DROP DATABASE IF EXISTS {database}"))
 
 
+async def ensure_database_users_disconnected(super_: URL, database: str, /) -> None:
+    """Ensure a databases' users are disconnected."""
+    engine = create_async_engine(super_, isolation_level="AUTOCOMMIT")
+    match dialect := _get_dialect(engine):
+        case "postgresql":  # skipif-ci-and-not-linux
+            query = f"SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = {database!r} AND pid <> pg_backend_pid()"  # noqa: S608
+        case "mssql" | "mysql" | "oracle" | "sqlite":  # pragma: no cover
+            raise NotImplementedError(dialect)
+        case never:
+            assert_never(never)
+    async with engine.begin() as conn:
+        _ = await conn.execute(text(query))
+
+
 ##
 
 
@@ -1166,6 +1180,7 @@ __all__ = [
     "create_engine",
     "ensure_database_created",
     "ensure_database_dropped",
+    "ensure_database_users_disconnected",
     "ensure_tables_created",
     "ensure_tables_dropped",
     "enum_name",