wip

Author: cunla

scheduler/management/commands/import.py

@@ -18,6 +18,7 @@ def job_model_str(model_str: str) -> str:
         return model_str[:-3] + "Task"
     return model_str
 
+
 def get_task_type(model_str: str) -> TaskType:
     model_str = job_model_str(model_str)
     if model_str not in MODEL_NAMES:
@@ -29,6 +30,7 @@ def get_task_type(model_str: str) -> TaskType:
     elif model_str == "ScheduledTask":
         return TaskType.ONCE
 
+
 def create_task_from_dict(task_dict: Dict[str, Any], update):
     existing_job = Task.objects.filter(name=task_dict["name"]).first()
     task_type = get_task_type(task_dict["model"])

scheduler/models/scheduled_task.py

@@ -406,6 +406,7 @@ class Meta:
 
 class RepeatableTask(RepeatableMixin, ScheduledTimeMixin, BaseTask):
     task_type = "RepeatableTask"
+
     class TimeUnits(models.TextChoices):
         SECONDS = "seconds", _("seconds")
         MINUTES = "minutes", _("minutes")

scheduler/models/task.py

@@ -69,6 +69,7 @@ class TaskType(models.TextChoices):
     REPEATABLE = "RepeatableTask", _("Repeatable Task")
     ONCE = "OnceTask", _("Run once")
 
+
 class Task(models.Model):
     class TimeUnits(models.TextChoices):
         SECONDS = "seconds", _("seconds")