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

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

from base.scope import Scope
from base.model.message_factory import MessageFactory
from base.model.enum.message_type import MessageType
import json


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

    def __init__(self):
        # super(TaskManager, self).__init__()
        scope = Scope.get_instance()
        self.pluginManager = scope.get_plugin_manager()
        self.logger = scope.get_logger()
        self.db_service = scope.get_db_service()
        self.scheduler = scope.get_scheduler()

    def addTask(self, task):
        try:
            self.saveTask(task)
            if task.get_cron_str() is None or task.get_cron_str() == '':
                self.logger.debug('Adding task ... ')
                self.pluginManager.process_task(task)
            else:
                self.scheduler.save_and_add_job(task)

        except Exception as e:
            self.logger.debug('Exception occurred when adding task. Error Message: {0}'.format(str(e)))

    def addPolicy(self, policy):
        try:
            self.pluginManager.process_policy(policy)
        except Exception as e:
            self.logger.error("Exception occurred when adding policy. Error Message: {0}".format(str(e)))

    def saveTask(self, task):
        try:
            self.logger.debug('task save')
            task_cols = ['id', 'create_date', 'modify_date', 'command_cls_id', 'parameter_map', 'deleted', 'plugin',
                         'cron_expr', 'file_server']
            plu_cols = ['active', 'create_date', 'deleted', 'description', 'machine_oriented', 'modify_date', 'name',
                        'policy_plugin', 'user_oriented', 'version', 'task_plugin', 'x_based']
            plugin_args = [str(task.get_plugin().get_active()), str(task.get_plugin().get_create_date()),
                           str(task.get_plugin().get_deleted()), str(task.get_plugin().get_description()),
                           str(task.get_plugin().get_machine_oriented()), str(task.get_plugin().get_modify_date()),
                           str(task.get_plugin().get_name()), str(task.get_plugin().get_policy_plugin()),
                           str(task.get_plugin().get_user_oriented()), str(task.get_plugin().get_version()),
                           str(task.get_plugin().get_task_plugin()), str(task.get_plugin().get_x_based())]
            plugin_id = self.db_service.update('plugin', plu_cols, plugin_args)

            params = json.dumps(task.get_parameter_map())

            values = [str(task.get_id()), str(task.get_create_date()), str(task.get_modify_date()),
                      str(task.get_command_cls_id()),  str(params), str(task.get_deleted()),
                      str(plugin_id), str(task.get_cron_str()), str(task.get_file_server())]
            self.db_service.update('task', task_cols, values)
        except Exception as e:
            self.logger.error("Exception occurred while saving task. Error Message: {0}".format(str(e)))

    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"))