threadmill 0.6.0__tar.gz → 0.6.2__tar.gz

{threadmill-0.6.0 → threadmill-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: threadmill
-Version: 0.6.0
+Version: 0.6.2
 Summary: The most reliable backend for Django's task framework.
 Keywords: Django,tasks,worker
 Author-email: Johannes Maron <johannes@maron.family>

{threadmill-0.6.0 → threadmill-0.6.2}/threadmill/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.6.0'
-__version_tuple__ = version_tuple = (0, 6, 0)
+__version__ = version = '0.6.2'
+__version_tuple__ = version_tuple = (0, 6, 2)
 
-__commit_id__ = commit_id = 'g2842de1d6'
+__commit_id__ = commit_id = 'g9ded4014b'

threadmill-0.6.2/threadmill/apps.py

@@ -0,0 +1,8 @@
+from django.apps import AppConfig
+
+
+class ThreadmillConfig(AppConfig):
+    name = "threadmill"
+
+    def ready(self) -> None:
+        from . import signals  # noqa: F401

{threadmill-0.6.0 → threadmill-0.6.2}/threadmill/executor.py

@@ -15,6 +15,7 @@ from inspect import iscoroutinefunction
 from queue import Empty
 from traceback import format_exception
 
+import django
 from django.tasks import TaskResult, task_backends
 from django.tasks.base import TaskContext, TaskError, TaskResultStatus
 from django.tasks.signals import task_finished, task_started
@@ -146,6 +147,7 @@ class WorkerProcess(multiprocessing.Process):
 
     def run(self) -> None:
         """Start consumer execution inside this process."""
+        django.setup()
         logger.info("Starting worker process %s", self.name)
         self.lock = threading.Lock()
         self.expired = threading.Event()

threadmill-0.6.2/threadmill/signals.py

@@ -0,0 +1,10 @@
+from django.db import close_old_connections
+from django.dispatch import receiver
+from django.tasks.signals import task_finished, task_started
+
+from .executor import TaskExecutor
+
+
+@receiver([task_started, task_finished], sender=TaskExecutor)
+def close_task_database_connection(sender, task_result, **kwargs):
+    close_old_connections()

{threadmill-0.6.0 → threadmill-0.6.2}/pyproject.toml

@@ -98,5 +98,5 @@ ini_options.DJANGO_SETTINGS_MODULE = "tests.testapp.settings"
 
 [tool.coverage]
 run.source = [ "threadmill" ]
-report.show_missing = true
 report.skip_covered = true
+report.show_missing = true