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_delay_task_generator(self):
run_at = datetime.now() + dt.timedelta(days=1)
delay_task = delay_task_generator(self.scan_task.id, run_at)
delay_task, created = PeriodicTask.objects.get_or_create(
name='delay_%s' % self.scan_task.id, )
self.assertFalse(created)
self.assertTrue(delay_task.enabled)
self.assertNotEqual(delay_task.interval, None)
def test_delay_task_generator(self):
run_at = datetime.now() + dt.timedelta(days=1)
delay_task = delay_task_generator(self.scan_task.id, run_at)
delay_task, created = PeriodicTask.objects.get_or_create(
name='delay_%s' % self.scan_task.id,
)
self.assertFalse(created)
self.assertTrue(delay_task.enabled)
self.assertNotEqual(delay_task.interval, None)
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()
