Ahenk/opt/ahenk/base/scheduler/custom/custom_scheduler.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>

from base.scheduler.base_scheduler import BaseScheduler
from base.Scope import Scope
from base.scheduler.custom.scheduledb import ScheduleTaskDB
from base.scheduler.custom.schedule_job import ScheduleTaskJob
from datetime import datetime, timedelta
import time


class CustomScheduler(BaseScheduler):

    def __init__(self):
        self.events = []
        self.keep_run = True
        self.logger = Scope.getInstance().getLogger()
        self.scheduledb = ScheduleTaskDB()

    def initialize(self):
        self.scheduledb.initialize()
        tasks = self.scheduledb.load()
        if tasks:
            for task in tasks:
                self.add_job(ScheduleTaskJob(task))

    def add_job(self, job):
        self.events.append(job)

    def save_and_add_job(self, task):
        try:
            self.logger.debug('[CustomScheduler] Saving scheduled task to database...')
            self.scheduledb.save(task)
            self.logger.debug('[CustomScheduler] Adding scheduled task to events...')
            self.events.append(ScheduleTaskJob(task))
        except Exception as e:
            self.logger.error('[CustomScheduler] A problem occurred while saving and adding job. Error Message: {}'.format(str(e)))

    def remove_job(self, task_id):
        for event in self.events:
            if event.task.get_id() == task_id:
                self.scheduledb.delete(task_id)
                self.logger.debug('[CustomScheduler] Task was deleted from scheduled tasks table')
                self.events.remove(event)
                self.logger.debug('[CustomScheduler] Task was removed from events')

    def remove_job_via_task_id(self,task_id):
        for event in self.events:
            if event.task.get_id() == task_id:
                self.scheduledb.delete(event.task)
                self.events.remove(event)

    def stop(self):
        self.keep_run = False

    def list_schedule_tasks(self):
        return self.scheduledb.load()

    def run(self):
        t = datetime(*datetime.now().timetuple()[:5])
        while 1 and self.keep_run:
            for e in self.events:
                e.check(t)

            t += timedelta(minutes=1)
            while datetime.now() < t:
                time.sleep((t - datetime.now()).seconds)