django-lambda-tasks 0.4.5__tar.gz → 0.4.7__tar.gz

{django_lambda_tasks-0.4.5 → django_lambda_tasks-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-lambda-tasks
-Version: 0.4.5
+Version: 0.4.7
 Summary: Run async tasks in a lambda function
 Requires-Python: >=3.10
 Requires-Dist: awslambdaric

{django_lambda_tasks-0.4.5 → django_lambda_tasks-0.4.7}/lambda_tasks/local_executor.py

@@ -1,5 +1,6 @@
 """Process pool executor for async local task execution."""
 
+import atexit
 import logging
 import uuid
 from concurrent.futures import Future, ProcessPoolExecutor
@@ -18,6 +19,14 @@ def _pool_initializer() -> None:
     django.setup()
 
 
+def _shutdown_pool() -> None:
+    """Shut down the pool at interpreter exit to release semaphores."""
+    global _pool
+    if _pool is not None:
+        _pool.shutdown(wait=True, cancel_futures=True)
+        _pool = None
+
+
 def get_pool() -> ProcessPoolExecutor:
     """Return the shared ProcessPoolExecutor, creating it on first call."""
     global _pool
@@ -27,6 +36,7 @@ def get_pool() -> ProcessPoolExecutor:
             max_workers=conf.LOCAL_WORKERS,
             initializer=_pool_initializer,
         )
+        atexit.register(_shutdown_pool)
     return _pool
 
 

{django_lambda_tasks-0.4.5 → django_lambda_tasks-0.4.7}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["lambda_tasks"]
 
 [project]
 name = "django-lambda-tasks"
-version = "0.4.5"
+version = "0.4.7"
 description = "Run async tasks in a lambda function"
 readme = "README.md"
 requires-python = ">=3.10"

{django_lambda_tasks-0.4.5 → django_lambda_tasks-0.4.7}/tests/test_local_executor.py

@@ -126,12 +126,11 @@ def test_mutual_exclusion_of_eager_and_local_workers(n):
 def _reset_pool():
     """Reset the module-level _pool after each test to avoid cross-test contamination."""
     yield
-    try:
-        import lambda_tasks.local_executor
+    import lambda_tasks.local_executor
 
+    if lambda_tasks.local_executor._pool is not None:
+        lambda_tasks.local_executor._pool.shutdown(wait=True, cancel_futures=True)
         lambda_tasks.local_executor._pool = None
-    except ModuleNotFoundError:
-        pass
 
 
 class TestGetPool:
@@ -216,7 +215,11 @@ def test_pool_created_with_correct_worker_count(n):
             pool = lambda_tasks.local_executor.get_pool()
             assert pool._max_workers == n
         finally:
-            # Always reset to avoid pool reuse across examples
+            # Shut down properly to avoid leaked semaphores
+            if lambda_tasks.local_executor._pool is not None:
+                lambda_tasks.local_executor._pool.shutdown(
+                    wait=True, cancel_futures=True
+                )
             lambda_tasks.local_executor._pool = None