django-qstash 0.0.7__tar.gz → 0.0.9__tar.gz

{django_qstash-0.0.7 → django_qstash-0.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-qstash
-Version: 0.0.7
+Version: 0.0.9
 Summary: A drop-in replacement for Celery's shared_task with Upstash QStash.
 Author-email: Justin Mitchel <justin@codingforentrepreneurs.com>
 Project-URL: Changelog, https://github.com/jmitchel3/django-qstash

{django_qstash-0.0.7 → django_qstash-0.0.9}/pyproject.toml

@@ -6,7 +6,7 @@ requires = [
 
 [project]
 name = "django-qstash"
-version = "0.0.7"
+version = "0.0.9"
 description = "A drop-in replacement for Celery's shared_task with Upstash QStash."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -70,7 +70,6 @@ branch = true
 parallel = true
 source = [
   "django_qstash",
-  "tests",
 ]
 omit = [
   "*/migrations/*",
@@ -80,8 +79,8 @@ omit = [
 
 [tool.coverage.paths]
 source = [
-  "src",
-  ".tox/**/site-packages",
+  "src/django_qstash",
+  ".tox/*/lib/python*/site-packages/django_qstash",
 ]
 
 [tool.coverage.report]

{django_qstash-0.0.7 → django_qstash-0.0.9}/src/django_qstash/__init__.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-__version__ = "0.0.7"
+__version__ = "0.0.9"
 
-from .tasks import shared_task
+from django_qstash.app import shared_task
 
 __all__ = ["shared_task"]

django_qstash-0.0.9/src/django_qstash/app/__init__.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from .base import AsyncResult
+from .base import QStashTask
+from .decorators import shared_task
+
+__all__ = ["AsyncResult", "QStashTask", "shared_task"]

django_qstash-0.0.7/src/django_qstash/tasks.py → django_qstash-0.0.9/src/django_qstash/app/base.py

@@ -118,26 +118,3 @@ class AsyncResult:
     @property
     def id(self) -> str:
         return self.task_id
-
-
-def shared_task(
-    func: Callable | None = None,
-    name: str | None = None,
-    deduplicated: bool = False,
-    **options: dict[str, Any],
-) -> QStashTask:
-    """
-    Decorator that mimics Celery's shared_task
-
-    Can be used as:
-        @shared_task
-        def my_task():
-            pass
-
-        @shared_task(name="custom_name", deduplicated=True)
-        def my_task():
-            pass
-    """
-    if func is not None:
-        return QStashTask(func, name=name, deduplicated=deduplicated, **options)
-    return lambda f: QStashTask(f, name=name, deduplicated=deduplicated, **options)

django_qstash-0.0.9/src/django_qstash/app/decorators.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from typing import Any
+from typing import Callable
+
+from django_qstash.app.base import QStashTask
+
+
+def shared_task(
+    func: Callable | None = None,
+    name: str | None = None,
+    deduplicated: bool = False,
+    **options: dict[str, Any],
+) -> QStashTask:
+    """
+    Decorator that mimics Celery's shared_task
+
+    Can be used as:
+        @shared_task
+        def my_task():
+            pass
+
+        @shared_task(name="custom_name", deduplicated=True)
+        def my_task():
+            pass
+    """
+    if func is not None:
+        return QStashTask(func, name=name, deduplicated=deduplicated, **options)
+    return lambda f: QStashTask(f, name=name, deduplicated=deduplicated, **options)

{django_qstash-0.0.7 → django_qstash-0.0.9}/src/django_qstash/discovery/utils.py

@@ -32,7 +32,7 @@ def discover_tasks() -> list[tuple[str, str]]:
             ('other_app.tasks.custom_task', 'special_name')
         ]
     """
-    from django_qstash.tasks import QStashTask
+    from django_qstash.app import QStashTask
 
     discovered_tasks = []
     packages = []
@@ -71,7 +71,7 @@ def discover_tasks() -> list[tuple[str, str]]:
                     if attr.name == attr_name:
                         label = value
                     else:
-                        label = f"{attr.name} ({value})"
+                        label = f"{attr.name} ({package}.tasks)"
                     discovered_tasks.append((value, label))
         except Exception as e:
             warnings.warn(

django_qstash-0.0.9/src/django_qstash/management/commands/clear_stale_results.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+
+from django.conf import settings
+from django.core.management.base import BaseCommand
+
+from django_qstash.results.tasks import clear_stale_results_task
+
+DJANGO_QSTASH_RESULT_TTL = getattr(settings, "DJANGO_QSTASH_RESULT_TTL", 604800)
+
+
+class Command(BaseCommand):
+    help = f"""Clears stale task results older than\n
+    {DJANGO_QSTASH_RESULT_TTL} seconds (settings.DJANGO_QSTASH_RESULT_TTL)"""
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--no-input",
+            action="store_true",
+            help="Do not ask for confirmation",
+        )
+        parser.add_argument(
+            "--since",
+            type=int,
+            help="The number of seconds ago to clear results for",
+        )
+        parser.add_argument(
+            "--delay", action="store_true", help="Offload request using django_qstash"
+        )
+
+    def handle(self, *args, **options):
+        delay = options["delay"]
+        no_input = options["no_input"]
+        since = options.get("since") or DJANGO_QSTASH_RESULT_TTL
+        user_confirm = not no_input
+        if not delay:
+            clear_stale_results_task(
+                since=since, user_confirm=user_confirm, stdout=self.stdout
+            )
+        else:
+            clear_stale_results_task.delay(
+                since=since, user_confirm=user_confirm, stdout=self.stdout
+            )

django_qstash-0.0.9/src/django_qstash/results/tasks.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+import logging
+from datetime import timedelta
+
+from django.apps import apps
+from django.conf import settings
+from django.utils import timezone
+
+from django_qstash import shared_task
+
+DJANGO_QSTASH_RESULT_TTL = getattr(settings, "DJANGO_QSTASH_RESULT_TTL", 604800)
+
+logger = logging.getLogger(__name__)
+
+
+@shared_task(name="Cleanup Task Results")
+def clear_stale_results_task(
+    since=None, stdout=None, user_confirm=False, *args, **options
+):
+    delta_seconds = since or DJANGO_QSTASH_RESULT_TTL
+    cutoff_date = timezone.now() - timedelta(seconds=delta_seconds)
+    TaskResult = None
+    try:
+        TaskResult = apps.get_model("django_qstash_results", "TaskResult")
+    except LookupError as e:
+        msg = "Django QStash Results not installed.\nAdd `django_qstash.results` to INSTALLED_APPS and run migrations."
+        if stdout is not None:
+            stdout.write(msg)
+        logger.exception(msg)
+        raise e
+    qs_to_delete = TaskResult.objects.filter(date_done__lt=cutoff_date)
+
+    if user_confirm:
+        user_input = input("Are you sure? (Y/n): ")
+        if f"{user_input}".lower() != "y":
+            msg = "Skipping deletion"
+            if stdout is not None:
+                stdout.write(msg)
+            logger.info(msg)
+            return
+
+    if not qs_to_delete.exists():
+        msg = "No stale Django QStash task results found"
+        if stdout is not None:
+            stdout.write(msg)
+        else:
+            logger.info(msg)
+        return
+    delete_msg = f"Deleting {qs_to_delete.count()} task results older than {cutoff_date} ({DJANGO_QSTASH_RESULT_TTL} seconds)"
+    if stdout is not None:
+        stdout.write(delete_msg)
+    else:
+        logger.info(delete_msg)
+    try:
+        deleted_count, _ = qs_to_delete.delete()
+        msg = f"Successfully deleted {deleted_count} stale results."
+        if stdout is not None:
+            stdout.write(msg)
+        else:
+            logger.info(msg)
+    except Exception as e:
+        msg = f"Error deleting stale results: {e}"
+        if stdout is not None:
+            stdout.write(msg)
+        logger.exception(msg)
+        raise e

{django_qstash-0.0.7 → django_qstash-0.0.9}/src/django_qstash/schedules/services.py

@@ -20,7 +20,9 @@ def sync_task_schedule_instance_to_qstash(instance: TaskSchedule) -> TaskSchedul
     and updates existing schedules when active.
     """
     data = format_task_schedule_for_qstash(instance)
-    qstash_client.schedule.create(**data)
+    schedule_id = qstash_client.schedule.create(**data)
+    if not instance.schedule_id:
+        TaskSchedule.objects.filter(id=instance.id).update(schedule_id=schedule_id)
 
     sync_state_changes(instance)
 

{django_qstash-0.0.7 → django_qstash-0.0.9}/src/django_qstash.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-qstash
-Version: 0.0.7
+Version: 0.0.9
 Summary: A drop-in replacement for Celery's shared_task with Upstash QStash.
 Author-email: Justin Mitchel <justin@codingforentrepreneurs.com>
 Project-URL: Changelog, https://github.com/jmitchel3/django-qstash

{django_qstash-0.0.7 → django_qstash-0.0.9}/src/django_qstash.egg-info/SOURCES.txt

@@ -6,7 +6,6 @@ src/django_qstash/client.py
 src/django_qstash/exceptions.py
 src/django_qstash/handlers.py
 src/django_qstash/settings.py
-src/django_qstash/tasks.py
 src/django_qstash/utils.py
 src/django_qstash/views.py
 src/django_qstash.egg-info/PKG-INFO
@@ -14,6 +13,9 @@ src/django_qstash.egg-info/SOURCES.txt
 src/django_qstash.egg-info/dependency_links.txt
 src/django_qstash.egg-info/requires.txt
 src/django_qstash.egg-info/top_level.txt
+src/django_qstash/app/__init__.py
+src/django_qstash/app/base.py
+src/django_qstash/app/decorators.py
 src/django_qstash/discovery/__init__.py
 src/django_qstash/discovery/fields.py
 src/django_qstash/discovery/models.py
@@ -28,6 +30,7 @@ src/django_qstash/results/admin.py
 src/django_qstash/results/apps.py
 src/django_qstash/results/models.py
 src/django_qstash/results/services.py
+src/django_qstash/results/tasks.py
 src/django_qstash/results/migrations/0001_initial.py
 src/django_qstash/results/migrations/__init__.py
 src/django_qstash/schedules/__init__.py
@@ -48,6 +51,5 @@ tests/test_exceptions.py
 tests/test_handlers.py
 tests/test_results_models.py
 tests/test_settings.py
-tests/test_tasks.py
 tests/test_utils.py
 tests/test_views.py

django_qstash-0.0.7/src/django_qstash/management/commands/clear_stale_results.py

@@ -1,61 +0,0 @@
-from __future__ import annotations
-
-from datetime import timedelta
-
-from django.apps import apps
-from django.conf import settings
-from django.core.management.base import BaseCommand
-from django.utils import timezone
-
-DJANGO_QSTASH_RESULT_TTL = getattr(settings, "DJANGO_QSTASH_RESULT_TTL", 604800)
-
-
-class Command(BaseCommand):
-    help = f"""Clears stale task results older than\n
-    {DJANGO_QSTASH_RESULT_TTL} seconds (settings.DJANGO_QSTASH_RESULT_TTL)"""
-
-    def add_arguments(self, parser):
-        parser.add_argument(
-            "--no-input",
-            action="store_true",
-            help="Do not ask for confirmation",
-        )
-        parser.add_argument(
-            "--since",
-            type=int,
-            help="The number of seconds ago to clear results for",
-        )
-
-    def handle(self, *args, **options):
-        no_input = options["no_input"]
-        since = options.get("since") or DJANGO_QSTASH_RESULT_TTL
-        cutoff_date = timezone.now() - timedelta(seconds=since)
-        try:
-            TaskResult = apps.get_model("django_qstash_results", "TaskResult")
-        except LookupError:
-            self.stdout.write(
-                self.style.ERROR(
-                    "Django QStash Results not installed.\nAdd `django_qstash.results` to INSTALLED_APPS and run migrations."
-                )
-            )
-            return
-        to_delete = TaskResult.objects.filter(date_done__lt=cutoff_date)
-
-        if not to_delete.exists():
-            self.stdout.write("No stale Django QStash task results found")
-            return
-
-        # use input to confirm  deletion
-        self.stdout.write(
-            f"Deleting {to_delete.count()} task results older than {cutoff_date} ({DJANGO_QSTASH_RESULT_TTL} seconds)"
-        )
-        if not no_input:
-            if input("Are you sure? (y/n): ") != "y":
-                self.stdout.write("Skipping deletion")
-                return
-
-        deleted_count, _ = to_delete.delete()
-
-        self.stdout.write(
-            self.style.SUCCESS(f"Successfully deleted {deleted_count} stale results.")
-        )

django_qstash-0.0.7/tests/test_tasks.py

@@ -1,53 +0,0 @@
-from unittest.mock import Mock, patch
-
-import pytest
-
-from django_qstash.tasks import shared_task
-
-
-@shared_task
-def sample_task(x, y):
-    return x + y
-
-
-@shared_task(name="custom_task", deduplicated=True)
-def sample_task_with_options(x, y):
-    return x * y
-
-
-@pytest.mark.django_db
-class TestQStashTasks:
-    def test_basic_task_execution(self):
-        """Test that tasks can be executed directly"""
-        result = sample_task(2, 3)
-        assert result == 5
-
-    def test_task_with_options(self):
-        """Test that tasks with custom options work"""
-        result = sample_task_with_options(4, 5)
-        assert result == 20
-
-    @patch("django_qstash.tasks.qstash_client")
-    def test_task_delay(self, mock_client):
-        """Test that delay() sends task to QStash"""
-        mock_response = Mock()
-        mock_response.message_id = "test-id-123"
-        mock_client.message.publish_json.return_value = mock_response
-
-        result = sample_task.delay(2, 3)
-
-        assert result.task_id == "test-id-123"
-        mock_client.message.publish_json.assert_called_once()
-
-    @patch("django_qstash.tasks.qstash_client")
-    def test_task_apply_async(self, mock_client):
-        """Test that apply_async() works with countdown"""
-        mock_response = Mock()
-        mock_response.message_id = "test-id-456"
-        mock_client.message.publish_json.return_value = mock_response
-
-        result = sample_task.apply_async(args=(2, 3), countdown=60)
-
-        assert result.task_id == "test-id-456"
-        call_kwargs = mock_client.message.publish_json.call_args[1]
-        assert call_kwargs["delay"] == "60s"