Adding assignment-IDs to plagiarism update-batch endpoint call.

Author: krulis-martin

recodex/api.py

@@ -365,7 +365,8 @@ def create_shadow_assignment_points(self, assignment_id, user_id, points,
 
     def update_shadow_assignment_points(self, points_id, points, note,
                                         awarded_at=None):
-        return self.post("/shadow-assignments/points/{}".format(points_id), data={
+        return self.post("/shadow-assignments/points/{}".format(points_id),
+                         data={
             'points': points,
             'note': note,
             'awardedAt': awarded_at,
@@ -377,13 +378,23 @@ def delete_shadow_assignment_points(self, points_id):
     # Plagiarisms
 
     def create_plagiarism_batch(self, tool, tool_params=None):
-        return self.post('/plagiarism', data={"detectionTool": tool, "detectionToolParams": tool_params})
+        return self.post('/plagiarism', data={
+            "detectionTool": tool,
+            "detectionToolParams": tool_params,
+        })
 
-    def update_plagiarism_batch(self, id, upload_completed):
-        return self.post('/plagiarism/{}'.format(id), data={"uploadCompleted": upload_completed})
+    def update_plagiarism_batch(self, id, upload_completed=None,
+                                assignments=None):
+        data = {}
+        if upload_completed is not None:
+            data["uploadCompleted"] = upload_completed
+        if assignments is not None:
+            data["assignments"] = assignments
+        return self.post('/plagiarism/{}'.format(id), data=data)
 
     def add_plagiarism_detected_similarity(self, id, solution_id, data):
-        return self.post('/plagiarism/{}/{}'.format(id, solution_id), data=data)
+        return self.post('/plagiarism/{}/{}'.format(id, solution_id),
+                         data=data)
 
     # Misc
 

recodex/plugins/plagiarisms/cli.py

@@ -28,13 +28,24 @@ def create_batch(api: ApiClient, tool, tool_params):
 
 @cli.command()
 @click.argument("id")
-@click.option("--upload-completed/--upload-reopen", "completed", default=True)
+@click.option("--upload-completed/--upload-reopen", "completed", default=None)
+@click.option("--assignments", "assignments", is_flag=True)
 @pass_api_client
-def update_batch(api: ApiClient, id, completed):
+def update_batch(api: ApiClient, id, completed, assignments):
     """
-    Update a plagiarism detection batch (whether its upload has been completed)
+    Update a plagiarism detection batch (whether its upload has been completed).
+    Mark assignments as checked by the batch.
     """
-    api.update_plagiarism_batch(id, completed)
+    ids = None
+    if assignments:
+        ids = []
+        data = sys.stdin.read()
+        for token in data.split():
+            token = token.strip()
+            if token:
+                ids.append(token)
+
+    api.update_plagiarism_batch(id, completed, ids)
 
 
 @cli.command()