python-pq 0.1.7__tar.gz → 0.2.0__tar.gz

{python_pq-0.1.7 → python_pq-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: python-pq
-Version: 0.1.7
+Version: 0.2.0
 Summary: Postgres-backed job queue for Python
 Author: ricwo
 Author-email: ricwo <r@cogram.com>

{python_pq-0.1.7 → python_pq-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-pq"
-version = "0.1.7"
+version = "0.2.0"
 description = "Postgres-backed job queue for Python"
 readme = "README.md"
 authors = [

{python_pq-0.1.7 → python_pq-0.2.0}/src/pq/client.py

@@ -3,7 +3,8 @@
 from collections.abc import Callable, Set
 from contextlib import contextmanager
 from datetime import UTC, datetime, timedelta
-from typing import Any
+from types import TracebackType
+from typing import Any, Self
 
 from croniter import croniter
 from croniter.croniter import CroniterBadCronError
@@ -43,6 +44,27 @@ class PQ:
         finally:
             session.close()
 
+    def close(self) -> None:
+        """Close all database connections and dispose the engine.
+
+        This closes all connections in the connection pool. After calling
+        this method, the PQ instance should not be used.
+        """
+        self._engine.dispose()
+
+    def __enter__(self) -> Self:
+        """Enter context manager."""
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        """Exit context manager and close connections."""
+        self.close()
+
     def create_tables(self) -> None:
         """Create all tables (for testing)."""
         Base.metadata.create_all(self._engine)