Ahenk/usr/share/ahenk/base/task/task_manager.py
2019-05-17 09:53:14 +03:00

80 lines
3.5 KiB
Python

#!/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"))