From 85be1258e16d4646baf42cbdf486e048e7ca2cdc Mon Sep 17 00:00:00 2001 From: Javier Sánchez Parra Date: Mon, 23 May 2022 10:35:09 +0200 Subject: Use assigned repository on images transfers Otherwise, update/restore fails on images not saved in the same machine which runs ogServer. --- ogcp/views.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/ogcp/views.py b/ogcp/views.py index 9f3d217..cd7213c 100644 --- a/ogcp/views.py +++ b/ogcp/views.py @@ -122,6 +122,13 @@ def get_clients(state_filter=None): return clients +def get_repository(repository_id): + repositories = get_repositories() + [repository] = [repository for repository in repositories + if repository['id'] == repository_id] + return repository + + def get_repositories(): r = g.server.get('/repositories') repositories = r.json()['repositories'] @@ -509,11 +516,12 @@ def action_image_restore(): if not image: flash(_(f'Image to restore was not found'), category='error') return redirect(url_for('commands')) + repository = get_repository(image['repo_id']) payload = {'disk': disk, 'partition': partition, 'name': image['name'], - 'repository': g.server.ip, + 'repository': repository['ip'], 'clients': ips, 'type': form.method.data, 'profile': str(image['software_id']), @@ -1018,12 +1026,13 @@ def action_image_update(): if not image: flash(_('Image to restore was not found'), category='error') return redirect(url_for('commands')) + repository = get_repository(image['repo_id']) payload = {'clients': [ip], 'disk': disk, 'partition': partition, 'code': code, 'name': image['name'], - 'repository': g.server.ip, + 'repository': repository['ip'], 'id': str(image['id']), # Dummy parameters, not used by ogServer on image update. 'group_id': 0, -- cgit v1.2.3-18-g5258