summaryrefslogtreecommitdiffstats
path: root/ogcp
diff options
context:
space:
mode:
Diffstat (limited to 'ogcp')
-rw-r--r--ogcp/forms/action_forms.py5
-rw-r--r--ogcp/templates/actions/mode.html13
-rw-r--r--ogcp/templates/scopes.html2
-rw-r--r--ogcp/views.py30
4 files changed, 49 insertions, 1 deletions
diff --git a/ogcp/forms/action_forms.py b/ogcp/forms/action_forms.py
index a73b4d1..44e2d58 100644
--- a/ogcp/forms/action_forms.py
+++ b/ogcp/forms/action_forms.py
@@ -76,6 +76,11 @@ class ClientDetailsForm(FlaskForm):
boot = SelectField(label=_('Boot Mode'))
create = SubmitField(label=_('Create'))
+class BootModeForm(FlaskForm):
+ ips = HiddenField()
+ boot = SelectField(label=_('Boot mode'))
+ ok = SubmitField(label=_('Ok'))
+
class ImageCreateForm(FlaskForm):
ip = HiddenField()
os = SelectField(label=_('OS'), choices=[])
diff --git a/ogcp/templates/actions/mode.html b/ogcp/templates/actions/mode.html
new file mode 100644
index 0000000..d93e8a7
--- /dev/null
+++ b/ogcp/templates/actions/mode.html
@@ -0,0 +1,13 @@
+{% extends 'base.html' %}
+{% import "bootstrap/wtf.html" as wtf %}
+
+{% block content %}
+
+{{ wtf.quick_form(form,
+ action=url_for('action_mode'),
+ method='post',
+ button_map={'ok': 'primary'},
+ extra_classes="m-5") }}
+
+{% endblock %}
+
diff --git a/ogcp/templates/scopes.html b/ogcp/templates/scopes.html
index 8eba78c..879f5f0 100644
--- a/ogcp/templates/scopes.html
+++ b/ogcp/templates/scopes.html
@@ -56,6 +56,8 @@
formaction="{{ url_for('action_client_add') }}" formmethod="get">
<input class="dropdown-item" type="submit" value="{{ _('Create image') }}"
formaction="{{ url_for('action_image_create') }}" formmethod="get">
+ <input class="dropdown-item" type="submit" value="{{ _('Set boot mode') }}"
+ formaction="{{ url_for('action_mode') }}" formmethod="get">
</div>
</div>
</form>
diff --git a/ogcp/views.py b/ogcp/views.py
index 2635195..37c59db 100644
--- a/ogcp/views.py
+++ b/ogcp/views.py
@@ -3,7 +3,7 @@ from flask import (
)
from ogcp.forms.action_forms import (
WOLForm, PartitionForm, ClientDetailsForm, HardwareForm, SessionForm,
- ImageRestoreForm, ImageCreateForm, SoftwareForm
+ ImageRestoreForm, ImageCreateForm, SoftwareForm, BootModeForm
)
from ogcp.og_server import OGServer
from flask_babel import _
@@ -439,6 +439,34 @@ def action_client_add():
form.create.render_kw = {"formaction": url_for('action_client_add')}
return render_template('actions/client_details.html', form=form)
+@app.route('/action/mode', methods=['GET', 'POST'])
+def action_mode():
+ form = BootModeForm(request.form)
+ if request.method == 'POST':
+ ips = form.ips.data.split(' ')
+ payload = { 'clients': ips, 'mode': form.boot.data }
+ print(payload)
+ r = g.server.post('/mode', payload)
+ if r.status_code == requests.codes.ok:
+ flash(_('Client set boot mode request sent successfully'), category='info')
+ else:
+ flash(_('Ogserver replied with status code not ok'), category='error')
+ return redirect(url_for("scopes"))
+
+ else:
+ r = g.server.get('/mode')
+ available_modes = [(mode, mode) for mode in r.json()['modes']]
+ form.boot.choices = list(available_modes)
+
+ ips = parse_ips(request.args.to_dict())
+ form.ips.data = " ".join(ips)
+ if not validate_ips(ips):
+ return redirect(url_for("scopes"))
+
+ form.ok.render_kw = { 'formaction': url_for('action_mode') }
+ return render_template('actions/mode.html', form=form)
+
+
@app.route('/action/image/create', methods=['GET', 'POST'])
def action_image_create():
form = ImageCreateForm(request.form)