Liderahenk/Ahenk
1
0
Fork 0
mirror of https://github.com/Pardus-LiderAhenk/ahenk synced 2024-11-22 15:32:19 +03:00
Code Issues Projects Releases Packages Wiki Activity

implemented execution manager for lider events

Browse source
This commit is contained in:
Volkan Şahin 2016-03-11 17:36:11 +02:00
parent 0644a12548
commit 8eac2d6615
1 changed files with 51 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

81
opt/ahenk/base/execution/ExecutionManager.py
View file

@ -1,43 +1,64 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com> # Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
import sys
import logging,subprocess import subprocess
import logging.config
from base.Scope import Scope from base.Scope import Scope
from base.messaging.MessageSender import MessageSender
import hashlib,json,os,stat,shutil
class ExecutionManager(object): class ExecutionManager(object):
"""docstring for FileTransferManager""" """docstring for FileTransferManager"""
def __init__(self): def __init__(self):
super(ExecutionManager, self).__init__() super(ExecutionManager, self).__init__()
scope = Scope.getInstance() scope = Scope.getInstance()
self.config_manager = scope.getConfigurationManager() self.config_manager = scope.getConfigurationManager()
self.event_manager = scope.getEventManager() self.event_manager = scope.getEventManager()
self.event_manager.register_event('EXECUTE_TASK',self.execute_task) self.event_manager.register_event('EXECUTE_TASK',self.execute_task)
self.event_manager.register_event('EXECUTE_SCRIPT',self.execute_script) self.event_manager.register_event('EXECUTE_SCRIPT',self.execute_script)
self.event_manager.register_event('SEND_FILE',self.send_file) self.event_manager.register_event('REQUEST_FILE',self.request_file)
self.event_manager.register_event('MOVE_FILE',self.move_file)
def execute_task(self,arg): def execute_task(self,arg):
print("execute_task") #TODO
self.logger.debug('[ExecutionManager] Executing task...')
def execute_script(self,arg): def move_file(self,arg):
print("execute_script") default_file_path=self.config_manager.get('CONNECTION', 'receiveFileParam')
j = json.loads(arg) j = json.loads(arg)
msg_id =str(j['id']).lower() #msg_id =str(j['id']).lower()
file_name =str(j['filePath']).lower() target_file_path =str(j['filepath']).lower()
time_stamp=str(j['timestamp']).lower() file_name =str(j['filename']).lower()
subprocess.call("/bin/sh "+self.conf_manager.get('CONNECTION', 'receivefileparam')+file_name, shell=True) self.logger.debug('[ExecutionManager] %s will be moved to %s' % file_name,target_file_path)
shutil.move(default_file_path+file_name,target_file_path+file_name)
#need to move somewhere else def execute_script(self,arg):
def send_file(self,arg): j = json.loads(arg)
print("send_file") #msg_id =str(j['id']).lower()
j = json.loads(arg) file_path =str(j['filepath']).lower()
msg_id =str(j['id']).lower() time_stamp=str(j['timestamp']).lower()
file_path =str(j['filePath']).lower() self.logger.debug('[ExecutionManager] Making executable file (%s) for execution' % file_path)
time_stamp=str(j['timestamp']).lower() st = os.stat(file_path)
os.chmod(file_path, st.st_mode | stat.S_IEXEC)
subprocess.call("/bin/sh "+file_path, shell=True)
message_sender=MessageSender(None,file_path) #need to move somewhere else
message_sender.connect_to_server() def request_file(self,arg):
j = json.loads(arg)
#msg_id =str(j['id']).lower()
file_path =str(j['filepath']).lower()
time_stamp=str(j['timestamp']).lower()
self.logger.debug('[ExecutionManager] Requested file is '+file_path)
message_sender=MessageSender(None,file_path)
message_sender.connect_to_server()
def get_md5_file(self,fname):
self.logger.debug('[ExecutionManager] md5 hashing')
hash_md5 = hashlib.md5()
with open(fname, "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
hash_md5.update(chunk)
return str(hash_md5.hexdigest())