1 files changed, 61 insertions, 1 deletions
diff --git a/ogcp/views.py b/ogcp/views.py
index 77bad42..c24cb73 100644
--- a/ogcp/views.py
+++ b/ogcp/views.py
@@ -15,7 +15,7 @@ from ogcp.forms.action_forms import (
     GenericForm, SelectClientForm, ImageUpdateForm, ImportClientsForm,
     ServerForm, DeleteRepositoryForm, RepoForm, FolderForm, CacheForm,
     ClientMoveForm, RunScriptForm, ImageConfigForm, ImageFetchForm,
-    ServerConfigurationForm
+    ServerConfigurationForm, SetRepoForm
 )
 from flask_login import (
     current_user, LoginManager,
@@ -2416,6 +2416,66 @@ def action_oglive():
                                selected_clients=selected_clients)
 
 
+def get_clients_repo_dictionary(ips, server, repositories):
+    repos = {}
+    repo_id_to_name = {repo["id"]: repo["name"] for repo in repositories}
+
+    for ip in ips:
+        r = server.get('/client/info', payload={"client": [ip]})
+        if not r:
+            raise ServerError
+        if r.status_code != requests.codes.ok:
+            raise ServerErrorCode
+        resp = r.json()
+        repo_name = repo_id_to_name[resp['repo_id']]
+        if repo_name not in repos:
+            repos[repo_name] = [ip]
+        else:
+            repos[repo_name].append(ip)
+    return repos
+
+@app.route('/action/repo/set', methods=['GET', 'POST'])
+@login_required
+def action_repo_set():
+    form = SetRepoForm(request.form)
+    if request.method == 'POST':
+        ips = form.ips.data.split(' ')
+        payload = {'clients': ips, 'id': int(form.repo.data)}
+        server = get_server_from_clients(ips)
+        r = server.post('/client/repo', payload)
+        if not r:
+            return ogserver_down('commands')
+        if r.status_code == requests.codes.ok:
+            flash(_('Repo set ogLive request sent successfully'),
+                  category='info')
+        else:
+            flash(_('Ogserver replied with status code not ok'),
+                  category='error')
+        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(list(ips))
+        try:
+            repositories = get_repositories(server)
+            repos_set = get_clients_repo_dictionary(ips, server, repositories)
+        except ServerError:
+            return ogserver_down('commands')
+        except ServerErrorCode:
+            return ogserver_error('commands')
+
+        form.repo.choices = [(repo["id"], repo["name"]) for repo in repositories]
+
+        scopes, clients = get_scopes(set(ips))
+        selected_clients = list(get_selected_clients(scopes['scope']).items())
+        return render_template('actions/repo_set.html', repos_set=repos_set, form=form, scopes=scopes,
+                               selected_clients=selected_clients)
+
+
 @app.route('/action/image/create', methods=['GET', 'POST'])
 @login_required
 def action_image_create():