81 lines
3.5 KiB
Python
81 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"))
|