django-lambda-tasks 0.7.0__tar.gz → 0.8.1__tar.gz

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/.github/workflows/ci.yml

@@ -30,12 +30,26 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
+    services:
+      postgres:
+        image: postgres:17
+        env:
+          POSTGRES_DB: lambda_tasks_test
+          POSTGRES_USER: postgres
+          POSTGRES_PASSWORD: postgres
+        ports:
+          - 5432:5432
+        options: >-
+          --health-cmd "pg_isready -U postgres"
+          --health-interval 10s
+          --health-timeout 5s
+          --health-retries 5
     steps:
       - uses: actions/checkout@v4
       - name: Install uv and set the python version
         uses: astral-sh/setup-uv@v6
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Run uv
+      - name: Run tests
         run: |
           uv run pytest --cov-branch --cov-report term-missing --cov=tests/ --cov=lambda_tasks/

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: django-lambda-tasks
-Version: 0.7.0
+Version: 0.8.1
 Summary: Run async tasks in a lambda function
 Requires-Python: >=3.10
 Requires-Dist: awslambdaric
@@ -16,6 +16,8 @@ A Django library for offloading work to AWS Lambda or AWS Batch outside of the r
 
 > **Platform note:** Unix-based systems only (Linux, macOS). Timeout enforcement relies on `SIGALRM`.
 
+> **Database note:** PostgreSQL is required. The `TaskRecord` model uses PostgreSQL-specific features.
+
 ---
 
 ## Installation
@@ -90,6 +92,7 @@ A runnable Django project is included in the [`example/`](example/) directory. I
 
 ```bash
 cd example
+docker compose up -d
 uv run python manage.py migrate
 uv run python manage.py createsuperuser
 uv run python manage.py runserver

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/README.md

@@ -4,6 +4,8 @@ A Django library for offloading work to AWS Lambda or AWS Batch outside of the r
 
 > **Platform note:** Unix-based systems only (Linux, macOS). Timeout enforcement relies on `SIGALRM`.
 
+> **Database note:** PostgreSQL is required. The `TaskRecord` model uses PostgreSQL-specific features.
+
 ---
 
 ## Installation
@@ -78,6 +80,7 @@ A runnable Django project is included in the [`example/`](example/) directory. I
 
 ```bash
 cd example
+docker compose up -d
 uv run python manage.py migrate
 uv run python manage.py createsuperuser
 uv run python manage.py runserver

django_lambda_tasks-0.8.1/docker-compose.yml

@@ -0,0 +1,11 @@
+services:
+  postgres:
+    image: postgres:17
+    environment:
+      POSTGRES_DB: lambda_tasks_test
+      POSTGRES_USER: postgres
+      POSTGRES_PASSWORD: postgres
+    ports:
+      - "5432:5432"
+    tmpfs:
+      - /var/lib/postgresql/data

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/example/README.md

@@ -1,27 +1,34 @@
 # Example App
 
-A minimal Django app demonstrating `django-lambda-tasks` with `LAMBDA_TASKS_EAGER = True`. In eager mode, tasks execute synchronously in-process — no AWS infrastructure required.
+A minimal Django app demonstrating `django-lambda-tasks` with `LAMBDA_TASKS_LOCAL_WORKERS = 2`. Tasks execute in a background process pool — no AWS infrastructure required.
 
 ## Prerequisites
 
 - `uv` installed
+- Docker installed (for PostgreSQL)
 - All commands run from the `example/` directory
 
 ## Setup
 
-**1. Apply migrations**
+**1. Start PostgreSQL**
+
+```bash
+docker compose up -d
+```
+
+**2. Apply migrations**
 
 ```bash
 uv run python manage.py migrate
 ```
 
-**2. Create a superuser** (to access the admin interface)
+**3. Create a superuser** (to access the admin interface)
 
 ```bash
 uv run python manage.py createsuperuser
 ```
 
-**3. Start the development server**
+**4. Start the development server**
 
 ```bash
 uv run python manage.py runserver
@@ -33,7 +40,7 @@ uv run python manage.py runserver
 
 Visit: `http://127.0.0.1:8000/trigger/?name=Alice`
 
-The task runs synchronously and the response confirms it was triggered.
+The task runs in a background worker process and the response confirms it was triggered.
 
 **View task records in the admin**
 

django_lambda_tasks-0.8.1/example/docker-compose.yml

@@ -0,0 +1,14 @@
+services:
+  postgres:
+    image: postgres:17
+    environment:
+      POSTGRES_DB: example
+      POSTGRES_USER: postgres
+      POSTGRES_PASSWORD: postgres
+    ports:
+      - "5432:5432"
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+
+volumes:
+  postgres_data:

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/example/example_project/settings.py

@@ -17,6 +17,7 @@ INSTALLED_APPS = [
     "django.contrib.sessions",
     "django.contrib.messages",
     "django.contrib.staticfiles",
+    "django.contrib.postgres",
     "lambda_tasks",
     "example_app",
 ]
@@ -53,8 +54,12 @@ WSGI_APPLICATION = "example_project.wsgi.application"
 
 DATABASES = {
     "default": {
-        "ENGINE": "django.db.backends.sqlite3",
-        "NAME": BASE_DIR / "db.sqlite3",
+        "ENGINE": "django.db.backends.postgresql",
+        "NAME": "example",
+        "USER": "postgres",
+        "PASSWORD": "postgres",
+        "HOST": "localhost",
+        "PORT": "5432",
     }
 }
 

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/lambda_tasks/checks.py

@@ -1,7 +1,27 @@
 """Django deployment checks for lambda_tasks."""
 
 from django.conf import settings as django_settings
-from django.core.checks import Warning, register
+from django.core.checks import Error, Warning, register
+
+
+@register("lambda_tasks")
+def check_contrib_postgres_installed(
+    *, app_configs: object, **kwargs: object
+) -> list[Error]:
+    """Error if django.contrib.postgres is not in INSTALLED_APPS."""
+    errors: list[Error] = []
+
+    if "django.contrib.postgres" not in django_settings.INSTALLED_APPS:
+        errors.append(
+            Error(
+                "django.contrib.postgres is not in INSTALLED_APPS.",
+                hint="Add 'django.contrib.postgres' to INSTALLED_APPS. "
+                "It is required for PostgreSQL-specific indexes used by lambda_tasks.",
+                id="lambda_tasks.E001",
+            )
+        )
+
+    return errors
 
 
 @register("lambda_tasks", deploy=True)

django_lambda_tasks-0.8.1/lambda_tasks/migrations/0003_taskrecord_taskrecord_kwargs_trgm.py

@@ -0,0 +1,29 @@
+# Generated by Django 6.0.3 on 2026-08-17 12:45
+
+import django.contrib.postgres.indexes
+import django.db.models.functions.comparison
+from django.contrib.postgres.operations import TrigramExtension
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("lambda_tasks", "0002_alter_taskrecord_status"),
+    ]
+
+    operations = [
+        TrigramExtension(),
+        migrations.AddIndex(
+            model_name="taskrecord",
+            index=django.contrib.postgres.indexes.GinIndex(
+                django.contrib.postgres.indexes.OpClass(
+                    django.db.models.functions.comparison.Cast(
+                        "kwargs", output_field=models.TextField()
+                    ),
+                    name="gin_trgm_ops",
+                ),
+                name="taskrecord_kwargs_trgm",
+            ),
+        ),
+    ]

django_lambda_tasks-0.8.1/lambda_tasks/migrations/0004_remove_taskrecord_taskrecord_kwargs_trgm_and_more.py

@@ -0,0 +1,48 @@
+# Generated by Django 6.0.3 on 2026-08-18 09:47
+
+import django.contrib.postgres.indexes
+import django.db.models.functions.comparison
+import django.db.models.functions.text
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("lambda_tasks", "0003_taskrecord_taskrecord_kwargs_trgm"),
+    ]
+
+    operations = [
+        migrations.RemoveIndex(
+            model_name="taskrecord",
+            name="taskrecord_kwargs_trgm",
+        ),
+        migrations.AddIndex(
+            model_name="taskrecord",
+            index=django.contrib.postgres.indexes.GinIndex(
+                django.contrib.postgres.indexes.OpClass(
+                    django.db.models.functions.text.Upper(
+                        django.db.models.functions.comparison.Cast(
+                            "id", output_field=models.TextField()
+                        )
+                    ),
+                    name="gin_trgm_ops",
+                ),
+                name="taskrecord_id_trgm",
+            ),
+        ),
+        migrations.AddIndex(
+            model_name="taskrecord",
+            index=django.contrib.postgres.indexes.GinIndex(
+                django.contrib.postgres.indexes.OpClass(
+                    django.db.models.functions.text.Upper(
+                        django.db.models.functions.comparison.Cast(
+                            "kwargs", output_field=models.TextField()
+                        )
+                    ),
+                    name="gin_trgm_ops",
+                ),
+                name="taskrecord_kwargs_trgm",
+            ),
+        ),
+    ]

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/lambda_tasks/models.py

@@ -8,10 +8,12 @@ import traceback
 import uuid
 
 import boto3
+from django.contrib.postgres.indexes import GinIndex, OpClass
 from django.core.cache import caches
 from django.core.exceptions import ImproperlyConfigured
 from django.db import models, transaction
 from django.db.models import Q
+from django.db.models.functions import Cast, Upper
 from django.utils.module_loading import import_string
 from django.utils.timezone import now
 from pydantic import BaseModel, ConfigDict, Field
@@ -74,6 +76,20 @@ class TaskRecord(models.Model):
             models.Index(fields=["task_name"]),
             models.Index(fields=["status"]),
             models.Index(fields=["-start_time"]),
+            GinIndex(
+                OpClass(
+                    Upper(Cast("id", output_field=models.TextField())),
+                    name="gin_trgm_ops",
+                ),
+                name="taskrecord_id_trgm",
+            ),
+            GinIndex(
+                OpClass(
+                    Upper(Cast("kwargs", output_field=models.TextField())),
+                    name="gin_trgm_ops",
+                ),
+                name="taskrecord_kwargs_trgm",
+            ),
         ]
         constraints = [
             models.CheckConstraint(

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["lambda_tasks"]
 
 [project]
 name = "django-lambda-tasks"
-version = "0.7.0"
+version = "0.8.1"
 description = "Run async tasks in a lambda function"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -30,6 +30,7 @@ dev = [
     "pytest-xdist",
     "pytest-cov",
     "ty",
+    "psycopg",
 ]
 
 [tool.ty.rules]

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/tests/settings.py

@@ -1,7 +1,11 @@
 DATABASES = {
     "default": {
-        "ENGINE": "django.db.backends.sqlite3",
-        "NAME": ":memory:",
+        "ENGINE": "django.db.backends.postgresql",
+        "NAME": "lambda_tasks_test",
+        "USER": "postgres",
+        "PASSWORD": "postgres",
+        "HOST": "localhost",
+        "PORT": "5432",
     }
 }
 
@@ -9,6 +13,7 @@ INSTALLED_APPS = [
     "django.contrib.admin",
     "django.contrib.contenttypes",
     "django.contrib.auth",
+    "django.contrib.postgres",
     "lambda_tasks",
 ]
 

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/tests/test_admin.py

@@ -281,3 +281,30 @@ class TestReplayAction:
 
         action_names = list(model_admin.get_actions(request).keys())
         assert "replay_tasks" in action_names
+
+
+@pytest.mark.django_db()
+class TestKwargsSearchUsesIndex:
+    def test_admin_search_uses_gin_trigram_indexes(self) -> None:
+        """Verify the GIN trigram indexes are used for admin search queries."""
+        from django.db import connection
+        from django.test import RequestFactory
+
+        model_admin = admin.site._registry[TaskRecord]
+        request = RequestFactory().get("/", data={"q": "29992d75"})
+        request.user = User(is_staff=True, is_superuser=True)
+
+        changelist = model_admin.get_changelist_instance(request)
+        queryset = changelist.get_queryset(request)
+
+        sql, params = queryset.query.sql_with_params()
+        with connection.cursor() as cursor:
+            cursor.execute("SET enable_seqscan = off")
+            cursor.execute("SET enable_indexscan = off")
+            cursor.execute(f"EXPLAIN {sql}", params)
+            plan = "\n".join(row[0] for row in cursor.fetchall())
+            cursor.execute("SET enable_indexscan = on")
+            cursor.execute("SET enable_seqscan = on")
+
+        assert "taskrecord_id_trgm" in plan
+        assert "taskrecord_kwargs_trgm" in plan

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/tests/test_batch_worker.py

@@ -1,6 +1,7 @@
 """Tests for lambda_tasks.batch_worker entry point."""
 
 import uuid
+from unittest.mock import patch
 
 import pytest
 
@@ -8,6 +9,14 @@ from lambda_tasks.decorators import lambda_task
 from lambda_tasks.handler import main
 from lambda_tasks.models import SQSLambdaTaskMessage, TaskRecord
 
+
+@pytest.fixture(autouse=True)
+def _patch_close_old_connections():
+    """Prevent close_old_connections from closing the test DB connection."""
+    with patch("lambda_tasks.handler.db.close_old_connections"):
+        yield
+
+
 QUEUES = {
     "default": {"queue_url": "https://sqs.us-east-1.amazonaws.com/000000000000/default"}
 }

django_lambda_tasks-0.8.1/tests/test_checks.py

@@ -0,0 +1,21 @@
+"""Tests for the django.contrib.postgres system check."""
+
+from lambda_tasks.checks import check_contrib_postgres_installed
+
+
+def test_check_passes_when_contrib_postgres_installed(settings):
+    settings.INSTALLED_APPS = [
+        "django.contrib.postgres",
+        "lambda_tasks",
+    ]
+    errors = check_contrib_postgres_installed(app_configs=None)
+    assert errors == []
+
+
+def test_check_errors_when_contrib_postgres_missing(settings):
+    settings.INSTALLED_APPS = [
+        "lambda_tasks",
+    ]
+    errors = check_contrib_postgres_installed(app_configs=None)
+    assert len(errors) == 1
+    assert errors[0].id == "lambda_tasks.E001"

{django_lambda_tasks-0.7.0 → django_lambda_tasks-0.8.1}/tests/test_models.py

@@ -1096,7 +1096,14 @@ def test_property_4_retry_increments_n_retries(n_retries, exc_type_name):
 @pytest.mark.django_db(transaction=True)
 @given(
     x=st.integers(),
-    label=st.text(min_size=1, max_size=20),
+    label=st.text(
+        min_size=1,
+        max_size=20,
+        alphabet=st.characters(
+            blacklist_characters="\x00",
+            blacklist_categories=("Cs",),
+        ),
+    ),
 )
 @h_settings(max_examples=50, suppress_health_check=[HealthCheck.too_slow])
 def test_property_5_matching_exc_enqueues_retry_same_kwargs(x, label):