wip

Author: cunla

scheduler/admin/old_task_models.py

@@ -134,6 +134,7 @@ class TaskAdmin(admin.ModelAdmin):
         "function_string",
         "is_scheduled",
         "queue",
+        "new_task_id",
     )
     list_display_links = ("name",)
     readonly_fields = ("job_id",)

scheduler/management/commands/import.py

@@ -52,7 +52,8 @@ def create_task_from_dict(task_dict: Dict[str, Any], update):
         if not settings.USE_TZ and not timezone.is_naive(target):
             target = timezone.make_naive(target)
         kwargs["scheduled_time"] = target
-    model_fields = set(map(lambda field: field.attname, Task._meta.get_fields()))
+    model_fields = filter(lambda field:hasattr(field,'attname'),Task._meta.get_fields())
+    model_fields = set(map(lambda field: field.attname, model_fields))
     keys_to_ignore = list(filter(lambda _k: _k not in model_fields, kwargs.keys()))
     for k in keys_to_ignore:
         del kwargs[k]

scheduler/migrations/0019_task.py renamed to scheduler/migrations/0019_task_crontask_new_task_id_repeatabletask_new_task_id_and_more.py

@@ -1,5 +1,6 @@
-# Generated by Django 5.1.3 on 2024-11-19 23:36
+# Generated by Django 5.1.3 on 2024-11-20 20:32
 
+import django.db.models.deletion
 import scheduler.models.task
 from django.db import migrations, models
 
@@ -161,4 +162,25 @@ class Migration(migrations.Migration):
                 ),
             ],
         ),
+        migrations.AddField(
+            model_name="crontask",
+            name="new_task_id",
+            field=models.ForeignKey(
+                blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to="scheduler.task"
+            ),
+        ),
+        migrations.AddField(
+            model_name="repeatabletask",
+            name="new_task_id",
+            field=models.ForeignKey(
+                blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to="scheduler.task"
+            ),
+        ),
+        migrations.AddField(
+            model_name="scheduledtask",
+            name="new_task_id",
+            field=models.ForeignKey(
+                blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to="scheduler.task"
+            ),
+        ),
     ]

scheduler/models/__init__.py

@@ -1,4 +1,10 @@
-from .args import TaskKwarg, TaskArg, BaseTaskArg  # noqa: F401
-from .queue import Queue  # noqa: F401
-from .scheduled_task import BaseTask, ScheduledTask, RepeatableTask, CronTask  # noqa: F401
-from .task import Task  # noqa: F401
+from .args import TaskKwarg, TaskArg
+from .scheduled_task import BaseTask, ScheduledTask, RepeatableTask, CronTask
+from .queue import Queue
+from .task import Task
+
+__all__ = [
+    "TaskKwarg", "TaskArg",
+    "ScheduledTask", "RepeatableTask", "CronTask",
+    "Queue", "Task",
+]

scheduler/models/scheduled_task.py

@@ -124,6 +124,7 @@ class BaseTask(models.Model):
                 >0: Result expires after n seconds."""
         ),
     )
+    new_task_id = models.ForeignKey('Task', on_delete=models.CASCADE, blank=True, null=True)
 
     def callable_func(self):
         """Translate callable string to callable"""