Exemple #1
0
 def save_model(self, request, obj, form, change):
     obj.user = request.user
     if not obj.run_at and obj.id:
         periodic_task_remove('delay_%s' % obj.id)
     if obj.run_at:
         obj.save()
         delay_task_generator(obj.id, obj.run_at)
     functions = {
         1: generate_cron_never,
         2: generate_cron_daily,
         3: generate_cron_weekly,
         4: generate_cron_monthly,
     }
     cron_string = functions.get(obj.repeat_each, generate_cron_never)(
         day=obj.repeat_each_day,
         weekday=obj.repeat_each_weekday,
         hour_min=obj.repeat_at,
     )
     if obj.cron != cron_string:  # cron changed
         obj.save()
         obj.cron = cron_string
         if obj.cron:
             periodic_task_generator(obj.id, obj.cron)
         else:
             periodic_task_remove(obj.id)
     obj.save()
    def test_task_remove(self):
        task, created = PeriodicTask.objects.get_or_create(
                        name='test_task',
                        task="w3af_webui.tasks.test_task",
                        )
        task.save()
        init_count = PeriodicTask.objects.count()
        periodic_task_remove('test_task') # remove exist task
        self.assertEqual(init_count - 1, PeriodicTask.objects.count())

        periodic_task_remove('not_exist') # remove not exist task
        self.assertEqual(init_count - 1, PeriodicTask.objects.count())
Exemple #3
0
    def test_task_remove(self):
        task, created = PeriodicTask.objects.get_or_create(
            name='test_task',
            task="w3af_webui.tasks.test_task",
        )
        task.save()
        init_count = PeriodicTask.objects.count()
        periodic_task_remove('test_task')  # remove exist task
        self.assertEqual(init_count - 1, PeriodicTask.objects.count())

        periodic_task_remove('not_exist')  # remove not exist task
        self.assertEqual(init_count - 1, PeriodicTask.objects.count())
Exemple #4
0
 def save_model(self, request, obj, form, change):
     obj.user = request.user
     if not obj.run_at and obj.id:
         periodic_task_remove('delay_%s' % obj.id)
     if obj.run_at:
         obj.save()
         delay_task_generator(obj.id, obj.run_at)
     functions = {1: generate_cron_never,
                  2: generate_cron_daily,
                  3: generate_cron_weekly,
                  4: generate_cron_monthly,}
     cron_string = functions.get(obj.repeat_each, generate_cron_never)(
                                         day=obj.repeat_each_day,
                                         weekday=obj.repeat_each_weekday,
                                         hour_min=obj.repeat_at,
                                         )
     if obj.cron != cron_string: # cron changed
         obj.save()
         obj.cron = cron_string
         if obj.cron:
             periodic_task_generator(obj.id, obj.cron)
         else:
             periodic_task_remove(obj.id)
     obj.save()