opengnsys
/
ogagent
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

#750: Adapting route {{GET /command}}} parameters.

oglive^2^2^2
Ramón M. Gómez 2018-07-04 12:15:50 +02:00
parent 3806f120c6
commit 91bdf9040d
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/opengnsys/modules/server/OpenGnSys/__init__.py
View File

@ -85,6 +85,7 @@ def check_locked_partition(sync=False):
class OpenGnSysWorker(ServerWorker):
name = 'opengnsys'
interface = None # Binded interface for OpenGnsys
REST = None # REST object
loggedin = False # User session flag
locked = {} # Locked partitions
commands = [] # Running commands
@ -106,7 +107,6 @@ class OpenGnSysWorker(ServerWorker):
"""
Sends OGAgent activation notification to OpenGnsys server
"""
self.cmd = None
# Ensure cfg has required configuration variables or an exception will be thrown
url = self.service.config.get('opengnsys', 'remote')
if operations.os_type == 'ogLive' and 'oglive' in os.environ:
@ -337,10 +337,11 @@ class OpenGnSysWorker(ServerWorker):
(stat, out, err) = operations.exec_command(code)
# Removing command from the list
for c in self.commands:
if c.has_key('id') and c['id'] == op_id:
if c.getName() == op_id:
self.commands.remove(c)
# Sending results
self.REST.sendMessage(route, {'id': op_id, 'status': stat, 'output': out, 'error': err})
self.REST.sendMessage(route, {'client': self.interface.ip, 'trace': op_id, 'status': stat, 'output': out,
'error': err})
def process_command(self, path, get_params, post_params, server):
"""
@ -349,8 +350,8 @@ class OpenGnSysWorker(ServerWorker):
:param get_params: ignored
:param post_params: object with format:
id: operation id.
code: command code
route: callback URL
script: command code
redirect_url: callback REST route
:param server: headers data
:rtype: object with launching status
"""
@ -358,17 +359,17 @@ class OpenGnSysWorker(ServerWorker):
self.checkSecret(server)
# Processing data
try:
code = post_params.get('code')
script = post_params.get('script')
op_id = post_params.get('id')
route = post_params.get('route')
route = post_params.get('redirect_url')
# Checking if the thread id. exists
for c in self.commands:
if c.has_key('id') and c['id'] == op_id:
if c.getName() == op_id:
raise Exception('Task id. already exists: {}'.format(op_id))
# Launching a new thread
thr = threading.Thread(name=op_id, target=self.task_command, args=(code, route, op_id))
thr = threading.Thread(name=op_id, target=self.task_command, args=(script, route, op_id))
thr.start()
self.commands.append({'id': op_id, 'code': code})
self.commands.append(thr)
except Exception as e:
logger.error('Got exception {}'.format(e))
return {'error': e}
@ -383,12 +384,14 @@ class OpenGnSysWorker(ServerWorker):
:param server:
:return: object
"""
#data = []
#for c in self.commands:
# if c.is_alive():
# data.append({'name': c.getName(), 'code': c.__dict__['_Thread__args']})
#return data
return self.commands
data = []
logger.debug('Received execinfo operation')
self.checkSecret(server)
# Returning the arguments of all running threads
for c in self.commands:
if c.is_alive():
data.append(c.__dict__['_Thread__args'])
return data
def process_hardware(self, path, get_params, post_params, server):
"""
@ -399,7 +402,7 @@ class OpenGnSysWorker(ServerWorker):
:param server:
"""
data = []
logger.debug('Recieved hardware operation')
logger.debug('Received hardware operation')
self.checkSecret(server)
# Processing data
try:
@ -419,5 +422,5 @@ class OpenGnSysWorker(ServerWorker):
:param server:
:return:
"""
logger.debug('Recieved software operation with params: {}'.format(post_params))
logger.debug('Received software operation with params: {}'.format(post_params))
return operations.get_software(post_params.get('disk'), post_params.get('part'))