Liderahenk/Ahenk
1
0
Fork 0
mirror of https://github.com/Pardus-LiderAhenk/ahenk synced 2024-11-29 06:12:21 +03:00
Code Issues Projects Releases Packages Wiki Activity

minor fixes

Browse source
This commit is contained in:
Volkan Şahin 2016-04-25 17:12:45 +03:00
parent ffa0c32b61
commit 965ba8da2c
1 changed files with 23 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
opt/ahenk/base/plugin/Plugin.py
View file

@ -5,13 +5,25 @@ import subprocess
import threading import threading
from base.Scope import Scope from base.Scope import Scope
from base.model.MessageType import MessageType
from base.model.Response import Response from base.model.Response import Response
from base.model.enum.MessageType import MessageType
class Context(object): class Context(object):
def __init__(self): def __init__(self):
self.data = {} self.data = {}
self.scope = Scope().getInstance()
self.logger = self.scope.getLogger()
self.config_manager = self.scope.getConfigurationManager()
def debug(self, message):
self.logger.debug('[PLUGIN]' + message)
def info(self, message):
self.logger.info('[PLUGIN]' + message)
def error(self, message):
self.logger.error('[PLUGIN]' + message)
def put(self, var_name, data): def put(self, var_name, data):
self.data[var_name] = data self.data[var_name] = data
@ -25,7 +37,11 @@ class Context(object):
def execute(self, command): def execute(self, command):
return subprocess.Popen(command, shell=True) return subprocess.Popen(command, shell=True)
#TODO send file,... def get_path(self):
return self.config_manager.get('PLUGIN', 'pluginfolderpath')
# TODO send file,...
class Plugin(threading.Thread): class Plugin(threading.Thread):
""" """
@ -86,28 +102,27 @@ class Plugin(threading.Thread):
self.keep_run = False self.keep_run = False
self.logger.debug('[Plugin] Killing queue ! Plugin Name : ' + str(self.name)) self.logger.debug('[Plugin] Killing queue ! Plugin Name : ' + str(self.name))
elif obj_name == "SAFE_MODE": elif obj_name == "SAFE_MODE":
username = item_obj.username username = item_obj.username
safe_mode_module = Scope.getInstance().getPluginManager().find_safe_mode_module(self.name) safe_mode_module = Scope.getInstance().getPluginManager().find_safe_mode_module(self.name)
safe_mode_module.handle_safe_mode(username,self.context) safe_mode_module.handle_safe_mode(username, self.context)
self.context.empty_data() self.context.empty_data()
else: else:
self.logger.warning("[Plugin] Not supported object type " + obj_name) self.logger.warning("[Plugin] Not supported object type " + obj_name)
except Exception as e: except Exception as e:
# TODO error log here self.logger.error("[Plugin] Plugin running exception. Exception Message: {} ".format(str(e)))
self.logger.error("[Plugin] Plugin running exception " + str(e))
def get_execution_id(self, profile_id): def get_execution_id(self, profile_id):
try: try:
return self.db_service.select_one_result('policy', 'execution_id', ' id={}'.format(profile_id)) return self.db_service.select_one_result('policy', 'execution_id', ' id={}'.format(profile_id))
except Exception as e: except Exception as e:
print(str(e)) self.logger.error("[Plugin] A problem occurred while getting execution id. Exception Message: {} ".format(str(e)))
return None return None
def get_policy_version(self, profile_id): def get_policy_version(self, profile_id):
try: try:
return self.db_service.select_one_result('policy', 'version', ' id={}'.format(profile_id)) return self.db_service.select_one_result('policy', 'version', ' id={}'.format(profile_id))
except Exception as e: except Exception as e:
print(str(e)) self.logger.error("[Plugin] A problem occurred while getting policy version . Exception Message: {} ".format(str(e)))
return None return None
def getName(self): def getName(self):