forked from platypush/platypush
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
|
import datetime
|
||
|
|
import logging
|
||
|
|
import re
|
||
|
|
import time
|
||
|
|
|
||
|
|
from threading import Thread
|
||
|
|
|
||
|
|
from platypush.event.hook import EventAction
|
||
|
|
|
||
|
|
class Cronjob(Thread):
|
||
|
|
def __init__(self, name, cron_expression, actions, *args, **kwargs):
|
||
|
|
super().__init__()
|
||
|
|
self.cron_expression = cron_expression
|
||
|
|
self.actions = []
|
||
|
|
|
||
|
|
for action in actions:
|
||
|
|
self.actions.append(EventAction.build(action))
|
||
|
|
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
logging.info('Running cronjob {}'.format(self.name))
|
||
|
|
response = None
|
||
|
|
context = {}
|
||
|
|
|
||
|
|
for action in self.actions:
|
||
|
|
response = action.execute(async=False, **context)
|
||
|
|
logging.info('Response from cronjob {}: {}'.format(self.name, response))
|
||
|
|
|
||
|
|
|
||
|
|
def should_run(self):
|
||
|
|
units = ('minute', 'hour', 'day', 'month', 'year')
|
||
|
|
now = datetime.datetime.fromtimestamp(time.time())
|
||
|
|
cron_units = re.split('\s+', self.cron_expression)
|
||
|
|
|
||
|
|
for i in range(0, len(units)):
|
||
|
|
unit = units[i]
|
||
|
|
now_unit = getattr(now, unit)
|
||
|
|
cron_unit = cron_units[i].replace('*', str(now_unit))
|
||
|
|
m = re.match('(\d+)(/(\d+))?', cron_unit)
|
||
|
|
|
||
|
|
if m.group(3):
|
||
|
|
if int(m.group(1)) % int(m.group(3)):
|
||
|
|
return False
|
||
|
|
elif m:
|
||
|
|
if int(m.group(1)) != now_unit:
|
||
|
|
return False
|
||
|
|
else:
|
||
|
|
raise RuntimeError('Invalid cron expression for job {}: {}'.
|
||
|
|
format(self.name, self.cron_expression))
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class CronScheduler(Thread):
|
||
|
|
def __init__(self, jobs, *args, **kwargs):
|
||
|
|
super().__init__()
|
||
|
|
self.jobs_config = jobs
|
||
|
|
logging.info('Cron scheduler initialized with {} jobs'
|
||
|
|
.format(len(self.jobs_config)))
|
||
|
|
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _build_job(cls, job_config):
|
||
|
|
if isinstance(job_config, dict):
|
||
|
|
job = Cronjob(cron_expression=job_config['cron_expression'],
|
||
|
|
name=job_config['name'],
|
||
|
|
actions=job_config['actions'])
|
||
|
|
|
||
|
|
assert isinstance(job, Cronjob)
|
||
|
|
return job
|
||
|
|
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
logging.info('Running cron scheduler')
|
||
|
|
|
||
|
|
while True:
|
||
|
|
for job_config in self.jobs_config:
|
||
|
|
job = self._build_job(job_config)
|
||
|
|
if job.should_run():
|
||
|
|
job.start()
|
||
|
|
|
||
|
|
time.sleep(60)
|
||
|
|
|
||
|
|
|
||
|
|
# vim:sw=4:ts=4:et:
|
||
|
|
|