Updating POST on /api/workergroups (dashboard/app.py)

Author: DasFranck

mrq/dashboard/app.py

@@ -136,9 +136,16 @@ def get_workergroups():
 @requires_auth
 def post_workergroups():
     workergroups = json.loads(request.form["workergroups"])
+
+    # Remove workergroups which hasn't be sent but were present previously (supposed deleted)
+    workergroup_list_json = list(workergroups.keys())
+    workergroup_list_mongo = [document["_id"] for document in connections.mongodb_jobs.mrq_workergroups.find()]
+    workergroup_to_delete_list = list(set(workergroup_list_mongo) - set(workergroup_list_json))
+    for workergroup_id in workergroup_to_delete_list:
+        connections.mongodb_jobs.mrq_workergroups.delete_one({"_id": workergroup_id})
+
     for k, v in workergroups.iteritems():
         connections.mongodb_jobs.mrq_workergroups.update_one({"_id": k}, {"$set": v}, upsert=True)
-
     return jsonify({"status": "ok"})