def start(self): try: schedule = EventSchedule.objects.get(description=SCHEDULE_DESC) except EventSchedule.DoesNotExist: schedule = EventSchedule(description=SCHEDULE_DESC) schedule.callback = CALLBACK schedule.minutes = '*' schedule.save() self.info('started')
def test_one_shot(self): """ Test scheduler in real time""" global callback_counter self.router.start() schedule = EventSchedule(callback="scheduler.tests.slow.callback_func", minutes=ALL, callback_args=([3])) schedule.save() time.sleep(180.0) self.assertEquals(callback_counter, 9) self.router.stop()
def test_one_shot(self): """ Test scheduler in real time""" global callback_counter self.router.start() schedule = EventSchedule(callback="scheduler.tests.slow.callback_func", minutes=ALL, callback_args=([3])) schedule.save() time.sleep(180.0) self.assertEquals(callback_counter,9) self.router.stop()
def test_recurring(self): """ Test regular recurring schedules """ global callback_counter self.router.start() self.router.get_app('scheduler').schedule_thread._debug_speedup(minutes=1) schedule = EventSchedule(callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3])) schedule.save() time.sleep(2.9) self.assertEquals(callback_counter,9) self.router.stop()
def test_one_shot(self): """ Basically test 'count'""" global callback_counter self.router.start() # speedup the scheduler so that 1 second == 1 minute self.router.get_app('scheduler').schedule_thread._debug_speedup(minutes=1) schedule = EventSchedule(callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3]), count=1) schedule.save() time.sleep(3.0) self.assertEquals(callback_counter,3) self.router.stop()
def schedule_notifications (reg, reg_time): """schedule all future notificaitons to be sent for this patient""" for day in config.notification_days: send_at = calc_send_time(reg_time, reg.contact_time, day) #scheduler works in local server time instead of UTC local_send_at = send_at.astimezone(pytz.timezone(config.server_tz)).replace(tzinfo=None) schedule = EventSchedule(callback="circumcision.apps.circumcision.schedule.send_notification", \ start_time=local_send_at, count=1, minutes='*', callback_args=[reg.pk, day], description='patient %s; day %d' % (reg.patient_id, day)) schedule.save()
def start(self): # create a scheduled event if not present # due to bug in fixture handling in scheduler event_desc = 'group_messaging_main' if EventSchedule.objects.filter(description=event_desc).count() > 0: return schedule = EventSchedule(description=event_desc, \ callback="group_messaging.utils.process_queue_callback", \ minutes=([00,58]), \ callback_args=('self.router')) schedule.save() self.debug(u"Created Event Schedule %s" % event_desc)
def test_timestart_timestop(self): """ Test timebound schedules """ global callback_counter self.router.start() self.router.get_app('scheduler').schedule_thread._debug_speedup(minutes=1) start_time = datetime.now() + timedelta(minutes=2) end_time = datetime.now() + timedelta(minutes=4) schedule = EventSchedule(callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3]), \ start_time = start_time, end_time = end_time) schedule.save() time.sleep(7.0) self.assertEquals(callback_counter,6) self.router.stop()
def test_recurring(self): """ Test regular recurring schedules """ global callback_counter self.router.start() self.router.get_app('scheduler').schedule_thread._debug_speedup( minutes=1) schedule = EventSchedule( callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3])) schedule.save() time.sleep(2.9) self.assertEquals(callback_counter, 9) self.router.stop()
def schedule_response_reminders(self, d): self.debug('in App.schedulecallback(): self.router: %s', self.router) cb = d.pop('callback') m = d.pop('minutes') reps = d.pop('repetitions') self.debug('callback:%s; minutes:%s; repetitions:%s; kwargs:%s',cb,m,reps,d) t = datetime.now() s = timedelta(minutes=m) # for n in [(t + 1*s), (t + 2*s), ... (t + r+s)], where r goes from [1, reps+1) for st in [t + r*s for r in range(1,reps+1)]: self.debug('scheduling a reminder to fire after %s', st) schedule = EventSchedule(callback=cb, minutes=ALL, callback_kwargs=d, start_time=st, count=1) schedule.save()
def test_timestart_timestop(self): """ Test timebound schedules """ global callback_counter self.router.start() self.router.get_app('scheduler').schedule_thread._debug_speedup( minutes=1) start_time = datetime.now() + timedelta(minutes=2) end_time = datetime.now() + timedelta(minutes=4) schedule = EventSchedule(callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3]), \ start_time = start_time, end_time = end_time) schedule.save() time.sleep(7.0) self.assertEquals(callback_counter, 6) self.router.stop()
def test_one_shot(self): """ Basically test 'count'""" global callback_counter self.router.start() # speedup the scheduler so that 1 second == 1 minute self.router.get_app('scheduler').schedule_thread._debug_speedup( minutes=1) schedule = EventSchedule( callback="scheduler.tests.speedup.callback_func", minutes=ALL, callback_args=([3]), count=1) schedule.save() time.sleep(3.0) self.assertEquals(callback_counter, 3) self.router.stop()
def schedule_response_reminder(self, d): self.debug("in App.schedulecallback(): self.router: %s", self.router) cb = d.pop("callback") m = d.pop("minutes") reps = d.pop("repetitions") self.debug("callback:%s; minutes:%s; repetitions:%s; kwargs:%s", cb, m, reps, d) t = datetime.now() s = timedelta(minutes=m) # for n in [(t + 1*s), (t + 2*s), ... (t + r+s)], where r goes from [1, reps+1) # for st in [t + r*s for r in range(1,reps+1)]: # MLL: Changed to do one at a time, so resend will schedule the next one schedule = EventSchedule(callback=cb, minutes=ALL, callback_kwargs=d, start_time=t + s, count=1) schedule.save() self.debug("scheduling a reminder to fire after %s at %s, id=%d", s, s + t, schedule.id)
def LoadSchedules(): count = 0 callbacks = ['ilsgateway.callbacks.run_reminders'] # 'ilsgateway.callbacks.district_randr_reminder', # 'ilsgateway.callbacks.facility_delivery_reminder', # 'ilsgateway.callbacks.district_delivery_reminder', # 'ilsgateway.callbacks.district_delinquent_deliveries_summary', # 'ilsgateway.callbacks.facility_soh_reminder',] for callback in callbacks: if not EventSchedule.objects.filter(callback=callback): e = EventSchedule(callback=callback, description=callback, days_of_month='*', hours='*', minutes=[0,15,30,45]) e.save() count = count + 1 print "Loaded %d EventSchedules" % count