Ahenk/opt/ahenk/base/task/TaskManager.py

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

from base.Scope import Scope
from base.model.enum.MessageType import MessageType


class TaskManager(object):
    """docstring for TaskManager"""

    def __init__(self):
        # super(TaskManager, self).__init__()
        scope = Scope.getInstance()
        self.pluginManager = scope.getPluginManager()
        self.logger = scope.getLogger()
        self.db_service = scope.getDbService()
        self.scheduler = scope.get_scheduler()

    def addTask(self, task):
        try:
            if task.get_cron_str() == None or task.get_cron_str() == '':
                self.logger.debug('Adding task ... ')
                #self.saveTask(task)
                self.logger.info('Task saved ')
                # TODO send task received message
                self.pluginManager.processTask(task)
            else:
                self.scheduler.save_and_add_job(task)

        except Exception as e:
            # TODO error log here
            self.logger.debug('Exception occured when adding task ' + str(e))
            pass

    def addPolicy(self, policy):
        try:
            self.pluginManager.processPolicy(policy)
        except Exception as e:
            self.logger.error("Exception occured when adding policy ")
            pass

    def saveTask(self, task):

        cols = ['id', 'create_date', 'modify_date', 'command_cls_id', 'parameter_map', 'deleted', 'plugin']
        values = [str(task.get_id()), str(task.get_create_date()), str(task.get_modify_date()), str(task.get_command_cls_id()), str(task.get_parameter_map()), str(task.get_deleted()), task.plugin.to_string()]
        self.db_service.update('task', cols, values, None)
        self.logger.debug('[TaskManager] Task has been saved to database (Task id:' + task.id + ')')

    def updateTask(self, task):
        # TODO not implemented yet
        # This is updates task status processing - processed ...
        pass

    def deleteTask(self, task):
        # TODO not implemented yet
        # remove task if it is processed
        pass

    def sendMessage(self, type, message):
        # TODO not implemented yet
        pass


if __name__ == '__main__':
    print(MessageFactory.createMessage(MessageType.TASK_PROCESSING, "my message"))