p1-taskqueue 0.1.23__tar.gz → 0.1.24__tar.gz

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.23
+Version: 0.1.24
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "p1-taskqueue"
 # DO NOT CHANGE THIS VERSION - it gets automatically replaced by CI/CD with the git tag version
-version = "0.1.23"
+version = "0.1.24"
 description = "A Task Queue Wrapper for Dekoruma Backend"
 authors = [
     {name = "Chalvin", email = "engineering@dekoruma.com"}

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/src/p1_taskqueue.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.23
+Version: 0.1.24
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/src/taskqueue/cmanager.py

@@ -317,11 +317,12 @@ class CManager:
                             queue_name: str, job_timeout: int, retry_policy: dict, countdown: int, eta: Any) -> str:
         task_id = str(uuid.uuid4())
 
+        task_result_obj = None
         try:
             from django_celery_results.models import TaskResult
             from django.utils import timezone
 
-            TaskResult.objects.create(
+            task_result_obj = TaskResult.objects.create(
                 task_id=task_id,
                 task_name=task_name,
                 status='PENDING',
@@ -339,7 +340,12 @@ class CManager:
             pickled_task_args = self._pickle_data(task_args)
             pickled_task_kwargs = self._pickle_data(task_kwargs)
 
+            if task_result_obj is None:
+                raise ValueError(
+                    f"Failed to create TaskResult for task id: {task_id}")
+
             TaskReconstruction.objects.create(
+                task_result=task_result_obj,
                 task_id=task_id,
                 task_name=task_name,
                 task_args=pickled_task_args,

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/src/taskqueue/migrations/0001_initial.py

@@ -12,8 +12,14 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             name='TaskReconstruction',
             fields=[
+                ('id', models.AutoField(
+                    auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('task_result', models.OneToOneField(
+                    on_delete=models.deletion.CASCADE,
+                    to='django_celery_results.taskresult',
+                )),
                 ('task_id', models.CharField(
-                    max_length=255, primary_key=True, serialize=False)),
+                    db_index=True, max_length=255, unique=True)),
                 ('task_name', models.CharField(max_length=255)),
                 ('task_args', models.TextField()),
                 ('task_kwargs', models.TextField()),

{p1_taskqueue-0.1.23 → p1_taskqueue-0.1.24}/src/taskqueue/models.py

@@ -2,7 +2,12 @@ from django.db import models
 
 
 class TaskReconstruction(models.Model):
-    task_id = models.CharField(max_length=255, primary_key=True)
+    task_result = models.OneToOneField(
+        'django_celery_results.TaskResult',
+        on_delete=models.CASCADE,
+    )
+    # @note(chalvin): same task task_result.task_id. Saved for performance.
+    task_id = models.CharField(max_length=255, unique=True, db_index=True)
     task_name = models.CharField(max_length=255)
     task_args = models.TextField()
     task_kwargs = models.TextField()