mirror of https://github.com/evilhero/mylar
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from datetime import datetime, timedelta
|
||
|
|
from math import ceil
|
||
|
|
|
||
|
|
from lib.apscheduler.util import convert_to_datetime, timedelta_seconds
|
||
|
|
|
||
|
|
|
||
|
|
class IntervalTrigger(object):
|
||
|
|
def __init__(self, interval, start_date=None):
|
||
|
|
if not isinstance(interval, timedelta):
|
||
|
|
raise TypeError('interval must be a timedelta')
|
||
|
|
if start_date:
|
||
|
|
start_date = convert_to_datetime(start_date)
|
||
|
|
|
||
|
|
self.interval = interval
|
||
|
|
self.interval_length = timedelta_seconds(self.interval)
|
||
|
|
if self.interval_length == 0:
|
||
|
|
self.interval = timedelta(seconds=1)
|
||
|
|
self.interval_length = 1
|
||
|
|
|
||
|
|
if start_date is None:
|
||
|
|
self.start_date = datetime.now() + self.interval
|
||
|
|
else:
|
||
|
|
self.start_date = convert_to_datetime(start_date)
|
||
|
|
|
||
|
|
def get_next_fire_time(self, start_date):
|
||
|
|
if start_date < self.start_date:
|
||
|
|
return self.start_date
|
||
|
|
|
||
|
|
timediff_seconds = timedelta_seconds(start_date - self.start_date)
|
||
|
|
next_interval_num = int(ceil(timediff_seconds / self.interval_length))
|
||
|
|
return self.start_date + self.interval * next_interval_num
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return 'interval[%s]' % str(self.interval)
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return "<%s (interval=%s, start_date=%s)>" % (
|
||
|
|
self.__class__.__name__, repr(self.interval),
|
||
|
|
repr(self.start_date))
|