django-lambda-tasks 0.4.6__tar.gz → 0.4.8__tar.gz

{django_lambda_tasks-0.4.6 → django_lambda_tasks-0.4.8}/PKG-INFO

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

{django_lambda_tasks-0.4.6 → django_lambda_tasks-0.4.8}/lambda_tasks/local_executor.py

@@ -2,6 +2,7 @@
 
 import atexit
 import logging
+import signal
 import uuid
 from concurrent.futures import Future, ProcessPoolExecutor
 
@@ -13,7 +14,16 @@ _pool: ProcessPoolExecutor | None = None
 
 
 def _pool_initializer() -> None:
-    """Run once per worker process. Sets up Django."""
+    """Run once per worker process.
+
+    Ignores SIGINT so that Ctrl+C is handled exclusively by the parent
+    process, which shuts down the pool cleanly via atexit. This prevents
+    workers from being killed mid-operation and leaking semaphores.
+
+    Then sets up Django for task execution.
+    """
+    signal.signal(signal.SIGINT, signal.SIG_IGN)
+
     import django
 
     django.setup()
@@ -23,7 +33,7 @@ def _shutdown_pool() -> None:
     """Shut down the pool at interpreter exit to release semaphores."""
     global _pool
     if _pool is not None:
-        _pool.shutdown(wait=False)
+        _pool.shutdown(wait=True, cancel_futures=True)
         _pool = None
 
 

{django_lambda_tasks-0.4.6 → django_lambda_tasks-0.4.8}/pyproject.toml

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

{django_lambda_tasks-0.4.6 → django_lambda_tasks-0.4.8}/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
 
 
@@ -567,6 +570,21 @@ class TestPoolInitializer:
 
         mock_django_setup.assert_called_once()
 
+    def test_pool_initializer_ignores_sigint(self):
+        """_pool_initializer() sets SIGINT to SIG_IGN so workers don't receive Ctrl+C."""
+        import signal
+
+        from lambda_tasks.local_executor import _pool_initializer
+
+        original_handler = signal.getsignal(signal.SIGINT)
+        try:
+            with patch("django.setup"):
+                _pool_initializer()
+
+            assert signal.getsignal(signal.SIGINT) == signal.SIG_IGN
+        finally:
+            signal.signal(signal.SIGINT, original_handler)
+
 
 @pytest.mark.django_db(transaction=True)
 class TestTransactionCommitIntegration: