mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-10 18:12:35 +03:00
ahenk main py format, log and style fixed
This commit is contained in:
parent
bc582b7616
commit
a77718e48d
1 changed files with 264 additions and 207 deletions
|
@ -33,29 +33,37 @@ from base.timer.setup_timer import SetupTimer
|
||||||
from base.timer.timer import Timer
|
from base.timer.timer import Timer
|
||||||
from base.util.util import Util
|
from base.util.util import Util
|
||||||
|
|
||||||
ahenkdaemon = None
|
ahenk_daemon = None
|
||||||
|
|
||||||
|
|
||||||
class AhenkDeamon(BaseDaemon):
|
class AhenkDaemon(BaseDaemon):
|
||||||
"""docstring for AhenkDeamon"""
|
"""Ahenk service base class which initializes services and maintains events/commands"""
|
||||||
|
|
||||||
def reload(self):
|
@staticmethod
|
||||||
|
def reload():
|
||||||
|
""" docstring"""
|
||||||
# reload service here
|
# reload service here
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def init_logger(self):
|
@staticmethod
|
||||||
|
def init_logger():
|
||||||
|
""" docstring"""
|
||||||
logger = Logger()
|
logger = Logger()
|
||||||
logger.info('[AhenkDeamon] Log was set')
|
logger.info('[AhenkDaemon] Log was set')
|
||||||
Scope.getInstance().setLogger(logger)
|
Scope.getInstance().setLogger(logger)
|
||||||
return logger
|
return logger
|
||||||
|
|
||||||
def init_config_manager(self, configFilePath, configfileFolderPath):
|
@staticmethod
|
||||||
configManager = ConfigManager(configFilePath, configfileFolderPath)
|
def init_config_manager(config_file_path, configfile_folder_path):
|
||||||
config = configManager.read()
|
""" docstring"""
|
||||||
|
config_manager = ConfigManager(config_file_path, configfile_folder_path)
|
||||||
|
config = config_manager.read()
|
||||||
Scope.getInstance().setConfigurationManager(config)
|
Scope.getInstance().setConfigurationManager(config)
|
||||||
return config
|
return config
|
||||||
|
|
||||||
def init_scheduler(self):
|
@staticmethod
|
||||||
|
def init_scheduler():
|
||||||
|
""" docstring"""
|
||||||
scheduler_ins = SchedulerFactory.get_intstance()
|
scheduler_ins = SchedulerFactory.get_intstance()
|
||||||
scheduler_ins.initialize()
|
scheduler_ins.initialize()
|
||||||
Scope.getInstance().set_scheduler(scheduler_ins)
|
Scope.getInstance().set_scheduler(scheduler_ins)
|
||||||
|
@ -64,87 +72,113 @@ class AhenkDeamon(BaseDaemon):
|
||||||
sc_thread.start()
|
sc_thread.start()
|
||||||
return scheduler_ins
|
return scheduler_ins
|
||||||
|
|
||||||
def init_event_manager(self):
|
@staticmethod
|
||||||
eventManager = EventManager()
|
def init_event_manager():
|
||||||
Scope.getInstance().setEventManager(eventManager)
|
""" docstring"""
|
||||||
return eventManager
|
event_manager = EventManager()
|
||||||
|
Scope.getInstance().setEventManager(event_manager)
|
||||||
|
return event_manager
|
||||||
|
|
||||||
def init_ahenk_db(self):
|
@staticmethod
|
||||||
|
def init_ahenk_db():
|
||||||
|
""" docstring"""
|
||||||
db_service = AhenkDbService()
|
db_service = AhenkDbService()
|
||||||
db_service.connect()
|
db_service.connect()
|
||||||
db_service.initialize_table()
|
db_service.initialize_table()
|
||||||
Scope.getInstance().setDbService(db_service)
|
Scope.getInstance().setDbService(db_service)
|
||||||
return db_service
|
return db_service
|
||||||
|
|
||||||
def init_messaging(self):
|
@staticmethod
|
||||||
messageManager = Messaging()
|
def init_messaging():
|
||||||
Scope.getInstance().setMessageManager(messageManager)
|
""" docstring"""
|
||||||
return messageManager
|
message_manager = Messaging()
|
||||||
|
Scope.getInstance().setMessageManager(message_manager)
|
||||||
|
return message_manager
|
||||||
|
|
||||||
def init_plugin_manager(self):
|
@staticmethod
|
||||||
pluginManager = PluginManagerFactory.get_instance()
|
def init_plugin_manager():
|
||||||
Scope.getInstance().setPluginManager(pluginManager)
|
""" docstring"""
|
||||||
|
plugin_manager = PluginManagerFactory.get_instance()
|
||||||
|
Scope.getInstance().setPluginManager(plugin_manager)
|
||||||
# order changed, problem?
|
# order changed, problem?
|
||||||
pluginManager.load_plugins()
|
plugin_manager.load_plugins()
|
||||||
return pluginManager
|
return plugin_manager
|
||||||
|
|
||||||
def init_task_manager(self):
|
@staticmethod
|
||||||
taskManager = TaskManager()
|
def init_task_manager():
|
||||||
Scope.getInstance().setTaskManager(taskManager)
|
""" docstring"""
|
||||||
return taskManager
|
task_manager = TaskManager()
|
||||||
|
Scope.getInstance().setTaskManager(task_manager)
|
||||||
|
return task_manager
|
||||||
|
|
||||||
def init_registration(self):
|
@staticmethod
|
||||||
|
def init_registration():
|
||||||
|
""" docstring"""
|
||||||
registration = Registration()
|
registration = Registration()
|
||||||
Scope.getInstance().setRegistration(registration)
|
Scope.getInstance().setRegistration(registration)
|
||||||
return registration
|
return registration
|
||||||
|
|
||||||
def init_execution_manager(self):
|
@staticmethod
|
||||||
|
def init_execution_manager():
|
||||||
|
""" docstring"""
|
||||||
execution_manager = ExecutionManager()
|
execution_manager = ExecutionManager()
|
||||||
Scope.getInstance().setExecutionManager(execution_manager)
|
Scope.getInstance().setExecutionManager(execution_manager)
|
||||||
return execution_manager
|
return execution_manager
|
||||||
|
|
||||||
def init_messenger(self):
|
@staticmethod
|
||||||
messenger = Messenger()
|
def init_messenger():
|
||||||
messenger.connect_to_server()
|
""" docstring"""
|
||||||
Scope.getInstance().setMessenger(messenger)
|
messenger_ = Messenger()
|
||||||
return messenger
|
messenger_.connect_to_server()
|
||||||
|
Scope.getInstance().setMessenger(messenger_)
|
||||||
|
return messenger_
|
||||||
|
|
||||||
def init_message_response_queue(self):
|
@staticmethod
|
||||||
responseQueue = queue.Queue()
|
def init_message_response_queue():
|
||||||
messageResponseQueue = MessageResponseQueue(responseQueue)
|
""" docstring"""
|
||||||
messageResponseQueue.setDaemon(True)
|
response_queue = queue.Queue()
|
||||||
messageResponseQueue.start()
|
message_response_queue = MessageResponseQueue(response_queue)
|
||||||
Scope.getInstance().setResponseQueue(responseQueue)
|
message_response_queue.setDaemon(True)
|
||||||
return responseQueue
|
message_response_queue.start()
|
||||||
|
Scope.getInstance().setResponseQueue(response_queue)
|
||||||
|
return response_queue
|
||||||
|
|
||||||
def check_registration(self):
|
def check_registration(self):
|
||||||
max_attemp_number = int(System.Hardware.Network.interface_size()) * 3
|
""" docstring"""
|
||||||
logger = Scope.getInstance().getLogger()
|
max_attempt_number = int(System.Hardware.Network.interface_size()) * 3
|
||||||
|
# self.logger.debug()
|
||||||
|
# logger = Scope.getInstance().getLogger()
|
||||||
registration = Scope.getInstance().getRegistration()
|
registration = Scope.getInstance().getRegistration()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
while registration.is_registered() is False:
|
while registration.is_registered() is False:
|
||||||
max_attemp_number -= 1
|
max_attempt_number -= 1
|
||||||
logger.debug('[AhenkDeamon] Ahenk is not registered. Attempting for registration')
|
self.logger.debug('[AhenkDaemon] Ahenk is not registered. Attempting for registration')
|
||||||
registration.registration_request()
|
registration.registration_request()
|
||||||
if max_attemp_number < 0:
|
if max_attempt_number < 0:
|
||||||
logger.warning('[AhenkDeamon] Number of Attempting for registration is over')
|
self.logger.warning('[AhenkDaemon] Number of Attempting for registration is over')
|
||||||
self.registration_failed()
|
self.registration_failed()
|
||||||
break
|
break
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error('[AhenkDeamon] Registration failed. Error message: {}'.format(str(e)))
|
self.logger.error('[AhenkDaemon] Registration failed. Error message: {0}'.format(str(e)))
|
||||||
|
|
||||||
def shutdown_mode(self):
|
@staticmethod
|
||||||
|
def shutdown_mode():
|
||||||
|
""" docstring"""
|
||||||
scope = Scope().getInstance()
|
scope = Scope().getInstance()
|
||||||
plugin_manager = scope.getPluginManager()
|
plugin_manager = scope.getPluginManager()
|
||||||
plugin_manager.process_mode('shutdown')
|
plugin_manager.process_mode('shutdown')
|
||||||
|
|
||||||
def registration_failed(self):
|
def registration_failed(self):
|
||||||
self.logger.error('[AhenkDeamon] Registration failed. All registration attemps were failed. Ahenk is stopping...')
|
""" docstring"""
|
||||||
|
self.logger.error(
|
||||||
|
'[AhenkDaemon] Registration failed. All registration attempts were failed. Ahenk is stopping...')
|
||||||
print('Registration failed. Ahenk is stopping..')
|
print('Registration failed. Ahenk is stopping..')
|
||||||
ahenkdaemon.stop()
|
ahenk_daemon.stop()
|
||||||
|
|
||||||
def reload_plugins(self):
|
@staticmethod
|
||||||
|
def reload_plugins():
|
||||||
|
""" docstring"""
|
||||||
Scope.getInstance().getPluginManager().reloadPlugins()
|
Scope.getInstance().getPluginManager().reloadPlugins()
|
||||||
|
|
||||||
def reload_configuration(self):
|
def reload_configuration(self):
|
||||||
|
@ -160,90 +194,95 @@ class AhenkDeamon(BaseDaemon):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def update_plugin_manager(self):
|
def update_plugin_manager(self):
|
||||||
|
""" docstring"""
|
||||||
# TODO destroy plugin manager here
|
# TODO destroy plugin manager here
|
||||||
self.init_plugin_manager()
|
self.init_plugin_manager()
|
||||||
|
|
||||||
def init_signal_listener(self):
|
def init_signal_listener(self):
|
||||||
|
""" docstring"""
|
||||||
try:
|
try:
|
||||||
signal.signal(signal.SIGALRM, self.run_command_from_fifo)
|
signal.signal(signal.SIGALRM, self.run_command_from_fifo)
|
||||||
self.logger.info('[AhenkDeamon] Signal handler is set up')
|
self.logger.info('[AhenkDaemon] Signal handler is set up')
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.logger.error('[AhenkDeamon] Signal handler could not set up. Error Message: {} '.format(str(e)))
|
self.logger.error('[AhenkDaemon] Signal handler could not set up. Error Message: {0} '.format(str(e)))
|
||||||
|
|
||||||
def init_pid_file(self):
|
@staticmethod
|
||||||
|
def init_pid_file():
|
||||||
|
""" docstring"""
|
||||||
with open(System.Ahenk.pid_path(), 'w+') as f:
|
with open(System.Ahenk.pid_path(), 'w+') as f:
|
||||||
f.write(str(os.getpid()))
|
f.write(str(os.getpid()))
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
|
""" docstring"""
|
||||||
print('Ahenk running...')
|
print('Ahenk running...')
|
||||||
|
|
||||||
globalscope = Scope()
|
global_scope = Scope()
|
||||||
globalscope.setInstance(globalscope)
|
global_scope.setInstance(global_scope)
|
||||||
|
|
||||||
configfileFolderPath = '/etc/ahenk/config.d/'
|
config_file_folder_path = '/etc/ahenk/config.d/'
|
||||||
|
|
||||||
# configuration manager must be first load
|
# configuration manager must be first load
|
||||||
self.init_config_manager(System.Ahenk.config_path(), configfileFolderPath)
|
self.init_config_manager(System.Ahenk.config_path(), config_file_folder_path)
|
||||||
|
|
||||||
# Logger must be second
|
# Logger must be second
|
||||||
self.logger = self.init_logger()
|
self.logger = self.init_logger()
|
||||||
|
|
||||||
self.init_pid_file()
|
self.init_pid_file()
|
||||||
self.logger.info('[AhenkDeamon] Pid file was created')
|
self.logger.info('[AhenkDaemon] Pid file was created')
|
||||||
|
|
||||||
self.init_event_manager()
|
self.init_event_manager()
|
||||||
self.logger.info('[AhenkDeamon] Event Manager was set')
|
self.logger.info('[AhenkDaemon] Event Manager was set')
|
||||||
|
|
||||||
self.init_ahenk_db()
|
self.init_ahenk_db()
|
||||||
self.logger.info('[AhenkDeamon] DataBase Service was set')
|
self.logger.info('[AhenkDaemon] DataBase Service was set')
|
||||||
|
|
||||||
self.init_messaging()
|
self.init_messaging()
|
||||||
self.logger.info('[AhenkDeamon] Message Manager was set')
|
self.logger.info('[AhenkDaemon] Message Manager was set')
|
||||||
|
|
||||||
self.init_plugin_manager()
|
self.init_plugin_manager()
|
||||||
self.logger.info('[AhenkDeamon] Plugin Manager was set')
|
self.logger.info('[AhenkDaemon] Plugin Manager was set')
|
||||||
|
|
||||||
self.init_scheduler()
|
self.init_scheduler()
|
||||||
self.logger.info('[AhenkDeamon] Scheduler was set')
|
self.logger.info('[AhenkDaemon] Scheduler was set')
|
||||||
|
|
||||||
self.init_task_manager()
|
self.init_task_manager()
|
||||||
self.logger.info('[AhenkDeamon] Task Manager was set')
|
self.logger.info('[AhenkDaemon] Task Manager was set')
|
||||||
|
|
||||||
self.init_registration()
|
self.init_registration()
|
||||||
self.logger.info('[AhenkDeamon] Registration was set')
|
self.logger.info('[AhenkDaemon] Registration was set')
|
||||||
|
|
||||||
self.init_execution_manager()
|
self.init_execution_manager()
|
||||||
self.logger.info('[AhenkDeamon] Execution Manager was set')
|
self.logger.info('[AhenkDaemon] Execution Manager was set')
|
||||||
|
|
||||||
self.check_registration()
|
self.check_registration()
|
||||||
self.logger.info('[AhenkDeamon] Ahenk was registered')
|
self.logger.info('[AhenkDaemon] Ahenk was registered')
|
||||||
|
|
||||||
self.messenger = self.init_messenger()
|
self.messenger = self.init_messenger()
|
||||||
self.logger.info('[AhenkDeamon] Messenger was set')
|
self.logger.info('[AhenkDaemon] Messenger was set')
|
||||||
|
|
||||||
self.init_signal_listener()
|
self.init_signal_listener()
|
||||||
self.logger.info('[AhenkDeamon] Signals listeners was set')
|
self.logger.info('[AhenkDaemon] Signals listeners was set')
|
||||||
|
|
||||||
Agreement().agreement_contract_update()
|
Agreement().agreement_contract_update()
|
||||||
|
|
||||||
self.init_message_response_queue()
|
self.init_message_response_queue()
|
||||||
|
|
||||||
# if registration.is_ldap_registered() is False:
|
# if registration.is_ldap_registered() is False:
|
||||||
# logger.debug('[AhenkDeamon] Attempting to registering ldap')
|
# logger.debug('[AhenkDaemon] Attempting to registering ldap')
|
||||||
# registration.ldap_registration_request() #TODO work on message
|
# registration.ldap_registration_request() #TODO work on message
|
||||||
|
|
||||||
self.logger.info('[AhenkDeamon] LDAP registration of Ahenk is completed')
|
self.logger.info('[AhenkDaemon] LDAP registration of Ahenk is completed')
|
||||||
|
|
||||||
self.messenger.send_direct_message('test')
|
self.messenger.send_direct_message('test')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
# if messager.is_connected() is False:
|
|
||||||
# self.logger.debug('reconnecting')
|
|
||||||
# Scope.getInstance().getLogger().warning('[AhenkDeamon] Connection is lost. Ahenk is trying for reconnection')
|
|
||||||
# messager = self.init_messager()
|
|
||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
def running_plugin(self):
|
@staticmethod
|
||||||
|
def running_plugin():
|
||||||
|
""" docstring"""
|
||||||
scope = Scope().getInstance()
|
scope = Scope().getInstance()
|
||||||
plugin_manager = scope.getPluginManager()
|
plugin_manager = scope.getPluginManager()
|
||||||
|
|
||||||
|
@ -253,7 +292,7 @@ class AhenkDeamon(BaseDaemon):
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def run_command_from_fifo(self, num, stack):
|
def run_command_from_fifo(self, num, stack):
|
||||||
|
""" docstring"""
|
||||||
scope = Scope().getInstance()
|
scope = Scope().getInstance()
|
||||||
plugin_manager = scope.getPluginManager()
|
plugin_manager = scope.getPluginManager()
|
||||||
message_manager = scope.getMessageManager()
|
message_manager = scope.getMessageManager()
|
||||||
|
@ -262,24 +301,28 @@ class AhenkDeamon(BaseDaemon):
|
||||||
db_service = scope.getDbService()
|
db_service = scope.getDbService()
|
||||||
execute_manager = scope.getExecutionManager()
|
execute_manager = scope.getExecutionManager()
|
||||||
|
|
||||||
event=''
|
while True:
|
||||||
try:
|
try:
|
||||||
event = Commander().get_event()
|
event = Commander().get_event()
|
||||||
|
if event is None:
|
||||||
|
break
|
||||||
json_data = json.loads(event)
|
json_data = json.loads(event)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.logger.error('[AhenkDeamon] A problem occurred while loading json. Check json format! Error Message: {0}.Event={1} '.format(str(e),str(event)))
|
self.logger.error(
|
||||||
|
'[AhenkDaemon] A problem occurred while loading json. Check json format! Error Message: {0}.'
|
||||||
|
' Event = {1}'.format(str(e), str(event)))
|
||||||
return
|
return
|
||||||
|
|
||||||
if json_data is not None:
|
if json_data is not None:
|
||||||
|
|
||||||
self.logger.debug('[AhenkDeamon] Signal handled')
|
self.logger.debug('[AhenkDaemon] Signal handled')
|
||||||
self.logger.debug('[AhenkDeamon] Signal is :{}'.format(str(json_data['event'])))
|
self.logger.debug('[AhenkDaemon] Signal is :{0}'.format(str(json_data['event'])))
|
||||||
|
|
||||||
if 'login' == str(json_data['event']):
|
if str(json_data['event']) == 'login':
|
||||||
username = json_data['username']
|
username = json_data['username']
|
||||||
display = json_data['display']
|
display = json_data['display']
|
||||||
desktop = json_data['desktop']
|
desktop = json_data['desktop']
|
||||||
self.logger.info('[AhenkDeamon] login event is handled for user: {}'.format(username))
|
self.logger.info('[AhenkDaemon] login event is handled for user: {0}'.format(username))
|
||||||
login_message = message_manager.login_msg(username)
|
login_message = message_manager.login_msg(username)
|
||||||
messenger.send_direct_message(login_message)
|
messenger.send_direct_message(login_message)
|
||||||
|
|
||||||
|
@ -287,7 +330,7 @@ class AhenkDeamon(BaseDaemon):
|
||||||
agreement_choice = None
|
agreement_choice = None
|
||||||
|
|
||||||
if agreement.check_agreement(username) is not True:
|
if agreement.check_agreement(username) is not True:
|
||||||
self.logger.debug('[AhenkDeamon] User {} has not accepted agreement.'.format(username))
|
self.logger.debug('[AhenkDaemon] User {0} has not accepted agreement.'.format(username))
|
||||||
thread_ask = Process(target=agreement.ask, args=(username, display,))
|
thread_ask = Process(target=agreement.ask, args=(username, display,))
|
||||||
thread_ask.start()
|
thread_ask.start()
|
||||||
|
|
||||||
|
@ -297,22 +340,24 @@ class AhenkDeamon(BaseDaemon):
|
||||||
timer = time.time()
|
timer = time.time()
|
||||||
while 1:
|
while 1:
|
||||||
if thread_ask.is_alive() is False:
|
if thread_ask.is_alive() is False:
|
||||||
self.logger.warning('[AhenkDeamon] {} was answered the question '.format(username))
|
self.logger.warning('[AhenkDaemon] {0} was answered the question '.format(username))
|
||||||
if Agreement().check_agreement(username) is True:
|
if Agreement().check_agreement(username) is True:
|
||||||
self.logger.warning('[AhenkDeamon] Choice of {} is YES'.format(username))
|
self.logger.warning('[AhenkDaemon] Choice of {0} is YES'.format(username))
|
||||||
agreement_choice = True
|
agreement_choice = True
|
||||||
break
|
break
|
||||||
elif Agreement().check_agreement(username) is False:
|
elif Agreement().check_agreement(username) is False:
|
||||||
self.logger.warning('[AhenkDeamon] Choice of {} is NO'.format(username))
|
self.logger.warning('[AhenkDaemon] Choice of {0} is NO'.format(username))
|
||||||
agreement_choice = False
|
agreement_choice = False
|
||||||
Util.execute('pkill -9 -u {}'.format(username))
|
Util.close_session(username)
|
||||||
break
|
break
|
||||||
|
|
||||||
if (time.time() - timer) > timeout:
|
if (time.time() - timer) > timeout:
|
||||||
if thread_ask.is_alive():
|
if thread_ask.is_alive():
|
||||||
thread_ask.terminate()
|
thread_ask.terminate()
|
||||||
Util.execute('pkill -9 -u {}'.format(username))
|
Util.close_session(username)
|
||||||
self.logger.warning('[AhenkDeamon] Session of {} was ended because of timeout of contract agreement'.format(username))
|
self.logger.warning(
|
||||||
|
'[AhenkDaemon] Session of {0} was ended because of timeout of contract agreement'.format(
|
||||||
|
username))
|
||||||
break
|
break
|
||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
|
@ -324,102 +369,114 @@ class AhenkDeamon(BaseDaemon):
|
||||||
if agreement_choice is True:
|
if agreement_choice is True:
|
||||||
db_service.delete('session', 'username=\'{0}\''.format(username))
|
db_service.delete('session', 'username=\'{0}\''.format(username))
|
||||||
|
|
||||||
self.logger.info('[AhenkDeamon] Display is {0}, desktop env is {1} for {2}'.format(display, desktop, username))
|
self.logger.info(
|
||||||
|
'[AhenkDaemon] Display is {0}, desktop env is {1} for {2}'.format(display, desktop, username))
|
||||||
|
|
||||||
db_service.update('session', scope.getDbService().get_cols('session'), [username, display, desktop, Util.timestamp()])
|
db_service.update('session', scope.getDbService().get_cols('session'),
|
||||||
|
[username, display, desktop, Util.timestamp()])
|
||||||
get_policy_message = message_manager.policy_request_msg(username)
|
get_policy_message = message_manager.policy_request_msg(username)
|
||||||
|
|
||||||
plugin_manager.process_mode('safe', username)
|
plugin_manager.process_mode('safe', username)
|
||||||
plugin_manager.process_mode('login', username)
|
plugin_manager.process_mode('login', username)
|
||||||
|
|
||||||
kward = {}
|
kward = dict()
|
||||||
kward['timeout_args'] = username
|
kward['timeout_args'] = username
|
||||||
kward['checker_args'] = username
|
kward['checker_args'] = username
|
||||||
|
|
||||||
SetupTimer.start(Timer(timeout=System.Ahenk.get_policy_timeout(), timeout_function=execute_manager.execute_default_policy, checker_func=execute_manager.is_policy_executed, kwargs=kward))
|
SetupTimer.start(Timer(timeout=System.Ahenk.get_policy_timeout(),
|
||||||
|
timeout_function=execute_manager.execute_default_policy,
|
||||||
|
checker_func=execute_manager.is_policy_executed, kwargs=kward))
|
||||||
|
|
||||||
self.logger.info('[AhenkDeamon] Requesting updated policies from Lider. If Ahenk could not reach updated policies in {0} sec, booked policies will be executed'.format(System.Ahenk.get_policy_timeout()))
|
self.logger.info(
|
||||||
|
'[AhenkDaemon] Requesting updated policies from Lider. If Ahenk could not reach updated '
|
||||||
|
'policies in {0} sec, booked policies will be executed'.format(
|
||||||
|
System.Ahenk.get_policy_timeout()))
|
||||||
messenger.send_direct_message(get_policy_message)
|
messenger.send_direct_message(get_policy_message)
|
||||||
|
|
||||||
elif 'logout' == str(json_data['event']):
|
elif str(json_data['event']) == 'logout':
|
||||||
username = json_data['username']
|
username = json_data['username']
|
||||||
db_service.delete('session', 'username=\'{}\''.format(username))
|
db_service.delete('session', 'username=\'{0}\''.format(username))
|
||||||
execute_manager.remove_user_executed_policy_dict(username)
|
execute_manager.remove_user_executed_policy_dict(username)
|
||||||
# TODO delete all user records while initializing
|
# TODO delete all user records while initializing
|
||||||
self.logger.info('[AhenkDeamon] logout event is handled for user: {}'.format(username))
|
self.logger.info('[AhenkDaemon] logout event is handled for user: {0}'.format(username))
|
||||||
logout_message = message_manager.logout_msg(username)
|
logout_message = message_manager.logout_msg(username)
|
||||||
messenger.send_direct_message(logout_message)
|
messenger.send_direct_message(logout_message)
|
||||||
|
|
||||||
plugin_manager.process_mode('logout', username)
|
plugin_manager.process_mode('logout', username)
|
||||||
plugin_manager.process_mode('safe', username)
|
plugin_manager.process_mode('safe', username)
|
||||||
|
|
||||||
elif 'send' == str(json_data['event']):
|
elif str(json_data['event']) == 'send':
|
||||||
self.logger.info('[AhenkDeamon] Sending message over ahenkd command. Response Message: {}'.format(json.dumps(json_data['message'])))
|
self.logger.info('[AhenkDaemon] Sending message over ahenkd command. Response Message: {0}'.format(
|
||||||
|
json.dumps(json_data['message'])))
|
||||||
message = json.dumps(json_data['message'])
|
message = json.dumps(json_data['message'])
|
||||||
messenger.send_direct_message(message)
|
messenger.send_direct_message(message)
|
||||||
|
|
||||||
elif 'load' == str(json_data['event']):
|
elif str(json_data['event']) == 'load':
|
||||||
plugin_name = str(json_data['plugins'])
|
plugin_name = str(json_data['plugins'])
|
||||||
|
|
||||||
if plugin_name == 'all':
|
if plugin_name == 'all':
|
||||||
self.logger.debug('[AhenkDeamon] All plugins are loading to ahenk')
|
self.logger.debug('[AhenkDaemon] All plugins are loading to ahenk')
|
||||||
plugin_manager.load_plugins()
|
plugin_manager.load_plugins()
|
||||||
else:
|
else:
|
||||||
for p_name in plugin_name.split(','):
|
for p_name in plugin_name.split(','):
|
||||||
self.logger.debug('[AhenkDeamon] {} plugin is loading to ahenk'.format(p_name))
|
self.logger.debug('[AhenkDaemon] {0} plugin is loading to ahenk'.format(p_name))
|
||||||
plugin_manager.load_single_plugin(p_name)
|
plugin_manager.load_single_plugin(p_name)
|
||||||
|
|
||||||
elif 'reload' == str(json_data['event']):
|
elif str(json_data['event']) == 'reload':
|
||||||
plugin_name = str(json_data['plugins'])
|
plugin_name = str(json_data['plugins'])
|
||||||
|
|
||||||
if plugin_name == 'all':
|
if plugin_name == 'all':
|
||||||
self.logger.debug('[AhenkDeamon] All plugins are reloading to ahenk')
|
self.logger.debug('[AhenkDaemon] All plugins are reloading to ahenk')
|
||||||
plugin_manager.reload_plugins()
|
plugin_manager.reload_plugins()
|
||||||
else:
|
else:
|
||||||
for p_name in plugin_name.split(','):
|
for p_name in plugin_name.split(','):
|
||||||
self.logger.debug('[AhenkDeamon] {} plugin is reloading to ahenk'.format(p_name))
|
self.logger.debug('[AhenkDaemon] {0} plugin is reloading to ahenk'.format(p_name))
|
||||||
plugin_manager.reload_single_plugin(p_name)
|
plugin_manager.reload_single_plugin(p_name)
|
||||||
|
|
||||||
elif 'remove' == str(json_data['event']):
|
elif str(json_data['event']) == 'remove':
|
||||||
plugin_name = str(json_data['plugins'])
|
plugin_name = str(json_data['plugins'])
|
||||||
|
|
||||||
if plugin_name == 'all':
|
if plugin_name == 'all':
|
||||||
self.logger.debug('[AhenkDeamon] All plugins are removing from ahenk')
|
self.logger.debug('[AhenkDaemon] All plugins are removing from ahenk')
|
||||||
plugin_manager.remove_plugins()
|
plugin_manager.remove_plugins()
|
||||||
else:
|
else:
|
||||||
for p_name in plugin_name.split(','):
|
for p_name in plugin_name.split(','):
|
||||||
self.logger.debug('[AhenkDeamon] {} plugin is removing from ahenk'.format(p_name))
|
self.logger.debug('[AhenkDaemon] {0} plugin is removing from ahenk'.format(p_name))
|
||||||
plugin_manager.remove_single_plugin(p_name)
|
plugin_manager.remove_single_plugin(p_name)
|
||||||
|
|
||||||
elif 'stop' == str(json_data['event']):
|
elif str(json_data['event']) == 'stop':
|
||||||
self.shutdown_mode()
|
self.shutdown_mode()
|
||||||
self.logger.info('[AhenkDeamon] Shutdown mode activated.')
|
self.logger.info('[AhenkDaemon] Shutdown mode activated.')
|
||||||
|
|
||||||
# TODO timeout
|
# TODO timeout
|
||||||
while self.running_plugin() is False:
|
while self.running_plugin() is False:
|
||||||
self.logger.debug('[AhenkDeamon] Waiting for progress of plugins...')
|
self.logger.debug('[AhenkDaemon] Waiting for progress of plugins...')
|
||||||
time.sleep(0.5)
|
time.sleep(0.5)
|
||||||
ahenkdaemon.stop()
|
|
||||||
|
if Util.is_exist('/tmp/liderahenk.fifo'):
|
||||||
|
Util.delete_file('/tmp/liderahenk.fifo')
|
||||||
|
ahenk_daemon.stop()
|
||||||
else:
|
else:
|
||||||
self.logger.error('[AhenkDeamon] Unknown command error. Command:' + json_data['event'])
|
self.logger.error('[AhenkDaemon] Unknown command error. Command:' + json_data['event'])
|
||||||
self.logger.debug('[AhenkDeamon] Processing of handled event is completed')
|
self.logger.debug('[AhenkDaemon] Processing of handled event is completed')
|
||||||
return True
|
# return True
|
||||||
else:
|
# else:
|
||||||
return False
|
# return False
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
||||||
ahenkdaemon = AhenkDeamon(System.Ahenk.pid_path())
|
ahenk_daemon = AhenkDaemon(System.Ahenk.pid_path())
|
||||||
try:
|
try:
|
||||||
if len(sys.argv) == 2 and (sys.argv[1] in ('start', 'stop', 'restart', 'status')):
|
if len(sys.argv) == 2 and (sys.argv[1] in ('start', 'stop', 'restart', 'status')):
|
||||||
if sys.argv[1] == 'start':
|
if sys.argv[1] == 'start':
|
||||||
if System.Ahenk.is_running() is True:
|
if System.Ahenk.is_running() is True:
|
||||||
print('There is already running Ahenk service. It will be killed.[{}]'.format(str(System.Ahenk.get_pid_number())))
|
print('There is already running Ahenk service. It will be killed.[{0}]'.format(
|
||||||
|
str(System.Ahenk.get_pid_number())))
|
||||||
System.Process.kill_by_pid(int(System.Ahenk.get_pid_number()))
|
System.Process.kill_by_pid(int(System.Ahenk.get_pid_number()))
|
||||||
else:
|
else:
|
||||||
print('Ahenk starting...')
|
print('Ahenk starting...')
|
||||||
ahenkdaemon.run()
|
ahenk_daemon.run()
|
||||||
elif sys.argv[1] == 'stop':
|
elif sys.argv[1] == 'stop':
|
||||||
if System.Ahenk.is_running() is True:
|
if System.Ahenk.is_running() is True:
|
||||||
raise SystemExit
|
raise SystemExit
|
||||||
|
@ -428,10 +485,10 @@ if __name__ == '__main__':
|
||||||
elif sys.argv[1] == 'restart':
|
elif sys.argv[1] == 'restart':
|
||||||
if System.Ahenk.is_running() is True:
|
if System.Ahenk.is_running() is True:
|
||||||
print('Ahenk restarting...')
|
print('Ahenk restarting...')
|
||||||
ahenkdaemon.restart()
|
ahenk_daemon.restart()
|
||||||
else:
|
else:
|
||||||
print('Ahenk starting...')
|
print('Ahenk starting...')
|
||||||
ahenkdaemon.run()
|
ahenk_daemon.run()
|
||||||
elif sys.argv[1] == 'status':
|
elif sys.argv[1] == 'status':
|
||||||
print(Commander().status())
|
print(Commander().status())
|
||||||
else:
|
else:
|
||||||
|
@ -453,4 +510,4 @@ if __name__ == '__main__':
|
||||||
if System.Ahenk.is_running() is True:
|
if System.Ahenk.is_running() is True:
|
||||||
os.kill(int(System.Ahenk.get_pid_number()), signal.SIGALRM)
|
os.kill(int(System.Ahenk.get_pid_number()), signal.SIGALRM)
|
||||||
else:
|
else:
|
||||||
ahenkdaemon.stop()
|
ahenk_daemon.stop()
|
||||||
|
|
Loading…
Reference in a new issue