summaryrefslogtreecommitdiffstats
path: root/ogcp/views.py
diff options
context:
space:
mode:
Diffstat (limited to 'ogcp/views.py')
-rw-r--r--ogcp/views.py71
1 files changed, 70 insertions, 1 deletions
diff --git a/ogcp/views.py b/ogcp/views.py
index 92dbf32..109957a 100644
--- a/ogcp/views.py
+++ b/ogcp/views.py
@@ -14,7 +14,7 @@ from ogcp.forms.action_forms import (
RoomForm, DeleteRoomForm, CenterForm, DeleteCenterForm, OgliveForm,
GenericForm, SelectClientForm, ImageUpdateForm, ImportClientsForm,
ServerForm, DeleteRepositoryForm, RepoForm, FolderForm, CacheForm,
- ClientMoveForm
+ ClientMoveForm, RunScriptForm
)
from flask_login import (
current_user, LoginManager,
@@ -1812,6 +1812,75 @@ def action_client_delete():
else:
return redirect(url_for('scopes'))
+@app.route('/action/script/run', methods=['GET', 'POST'])
+@login_required
+def action_run_script():
+ form = RunScriptForm(request.form)
+ if request.method == 'POST':
+ ips = form.ips.data.split(' ')
+ if not validate_elements(ips):
+ return redirect(url_for('commands'))
+
+ arguments = form.arguments.data.split(' ')
+ cmd_elems = [form.script.data] + arguments
+
+ payload = {
+ 'clients': ips,
+ 'run': ';|\n\r'.join(cmd_elems),
+ 'echo': True
+ }
+ server = get_server_from_clients(ips)
+ r = server.post('/shell/run', payload)
+
+ if not r:
+ return ogserver_down('commands')
+ if r.status_code != requests.codes.ok:
+ return ogserver_error('commands')
+
+ flash(_('Script run sent successfully'), category='info')
+ return redirect(url_for('commands'))
+ else:
+ ips = parse_elements(request.args.to_dict())
+ form.ips.data = " ".join(ips)
+ if not validate_elements(ips):
+ return redirect(url_for('commands'))
+
+ server = get_server_from_clients(ips)
+ different_setups = False
+ reference_patitioning = None
+ for ip in ips:
+ r = server.get('/client/setup', payload={'client': [ip]})
+ if not r:
+ return ogserver_down('commands')
+ if r.status_code != requests.codes.ok:
+ return ogserver_error('commands')
+
+ partitions = r.json()['partitions'][1:]
+ if not reference_patitioning:
+ reference_patitioning = partitions
+ elif reference_patitioning != partitions:
+ different_setups = True
+
+ if different_setups:
+ flash(_('Some clients don\'t have same configuration'), category='info')
+
+ r = server.get('/shell/list')
+ if not r:
+ return ogserver_down('commands')
+ if r.status_code != requests.codes.ok:
+ return ogserver_error('commands')
+
+ scripts = r.json()['scripts']
+
+ for script in scripts:
+ form.script.choices.append((script, script))
+
+ scopes, clients = get_scopes(set(ips))
+ selected_clients = list(get_selected_clients(scopes['scope']).items())
+ return render_template('actions/script_run.html', form=form,
+ selected_clients=selected_clients,
+ scopes=scopes)
+
def get_clients_modes(ips, server):
modes = {}
for ip in ips: