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())
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()