summaryrefslogtreecommitdiffstats
path: root/ogcp/views.py
diff options
context:
space:
mode:
authorJavier Sánchez Parra <jsanchez@soleta.eu>2022-04-06 17:29:44 +0200
committerJavier Sánchez Parra <jsanchez@soleta.eu>2022-04-08 08:03:32 +0200
commitfcbaf7d66d5e344b12b57d8d8c4c182c7497e764 (patch)
treef3a30c47d0ae91838495485e3a1c45c39e6c54d9 /ogcp/views.py
parentd2fef8e85dcd348201631911711cbb9b5db30ec3 (diff)
Add Update image to commands
Allows updating the image file for an existing image. This action is related to /image/create in ogServer API. If ogServer receives an POST /image/create without the parameter "description", it does not create a new image and only updates.
Diffstat (limited to 'ogcp/views.py')
-rw-r--r--ogcp/views.py69
1 files changed, 68 insertions, 1 deletions
diff --git a/ogcp/views.py b/ogcp/views.py
index ef85f05..ffd5ebf 100644
--- a/ogcp/views.py
+++ b/ogcp/views.py
@@ -12,7 +12,7 @@ from ogcp.forms.action_forms import (
WOLForm, SetupForm, ClientDetailsForm, ImageDetailsForm, HardwareForm,
SessionForm, ImageRestoreForm, ImageCreateForm, SoftwareForm, BootModeForm,
RoomForm, DeleteRoomForm, CenterForm, DeleteCenterForm, OgliveForm,
- GenericForm, SelectClientForm
+ GenericForm, SelectClientForm, ImageUpdateForm
)
from flask_login import (
current_user, LoginManager,
@@ -906,6 +906,73 @@ def action_image_create():
return render_template('actions/image_create.html', form=form,
scopes=scopes)
+
+@app.route('/action/image/update', methods=['GET', 'POST'])
+@login_required
+def action_image_update():
+ form = ImageUpdateForm(request.form)
+ if request.method == 'POST':
+ ip = form.ip.data
+ disk, partition, code = form.os.data.split(' ')
+ image_id = form.image.data
+ r = g.server.get('/images')
+ images_list = r.json()['images']
+ image = search_image(images_list, int(image_id))
+ if not image:
+ flash(_('Image to restore was not found'), category='error')
+ return redirect(url_for('commands'))
+ payload = {'clients': [ip],
+ 'disk': disk,
+ 'partition': partition,
+ 'code': code,
+ 'name': image['name'],
+ 'repository': g.server.ip,
+ 'id': str(image['id']),
+ # Dummy parameters, not used by ogServer on image update.
+ 'group_id': 0,
+ 'center_id': 0}
+ r = g.server.post('/image/create', payload)
+ if r.status_code == requests.codes.ok:
+ flash(_('Image update command sent sucessfully'), category='info')
+ else:
+ flash(_('There was a problem sending the image update command'),
+ category='error')
+ return redirect(url_for('commands'))
+
+ ips = parse_elements(request.args.to_dict())
+ if not validate_elements(ips, max_len=1):
+ return redirect(url_for('commands'))
+ form.ip.data = ' '.join(ips)
+
+ r = g.server.get('/images')
+ for image in r.json()['images']:
+ form.image.choices.append((image['id'], image['name']))
+
+ r = g.server.get('/client/setup', payload={'client': list(ips)})
+ for partition in r.json()['partitions']:
+ disk_id = partition['disk']
+ part_id = partition['partition']
+ fs_id = partition['filesystem']
+ code = partition['code']
+
+ if part_id == 0:
+ # This is the disk data, not a partition.
+ continue
+
+ choice_value = f'{disk_id} {part_id} {code}'
+ choice_name = (f"{_('Disk')} {disk_id} | "
+ f"{_('Partition')} {part_id} | "
+ f"{_('FS')} {FS_CODES[fs_id]}")
+ form.os.choices.append((choice_value, choice_name))
+
+ scopes, _clients = get_scopes(set(ips))
+ selected_clients = list(get_selected_clients(scopes['scope']).items())
+
+ return render_template('actions/image_update.html', form=form,
+ selected_clients=selected_clients,
+ scopes=scopes)
+
+
@app.route('/action/reboot', methods=['GET', 'POST'])
@login_required
def action_reboot():