Add refetch button to assignment list

Author: KrKOo

nbgrader/server_extensions/assignment_list/handlers.py

@@ -193,7 +193,7 @@ def list_courses(self):
 
         return retvalue
 
-    def fetch_assignment(self, course_id, assignment_id):
+    def fetch_assignment(self, course_id, assignment_id, replace_missing_files=False):
         with self.get_assignment_dir_config() as config:
             try:
                 config = self.load_config()
@@ -205,7 +205,9 @@ def fetch_assignment(self, course_id, assignment_id):
                 fetch = ExchangeFactory(config=config).FetchAssignment(
                     coursedir=coursedir,
                     authenticator=authenticator,
-                    config=config)
+                    config=config,
+                    replace_missing_files=replace_missing_files
+                    )
                 fetch.start()
 
             except:
@@ -310,22 +312,22 @@ def post(self, action):
         except web.MissingArgumentError:
             data = self.get_json_body()
 
+        assignment_id = data['assignment_id']
+        course_id = data['course_id']
+
         if action == 'fetch':
-            assignment_id = data['assignment_id']
-            course_id = data['course_id']
             self.manager.fetch_assignment(course_id, assignment_id)
             self.finish(json.dumps(self.manager.list_assignments(course_id=course_id)))
+        elif action == 'fetch_missing':
+            self.manager.fetch_assignment(course_id, assignment_id, replace_missing_files=True)
+            self.finish(json.dumps(self.manager.list_assignments(course_id=course_id)))
         elif action == 'submit':
-            assignment_id = data['assignment_id']
-            course_id = data['course_id']
             output = self.manager.submit_assignment(course_id, assignment_id)
             if output['success']:
                 self.finish(json.dumps(self.manager.list_assignments(course_id=course_id)))
             else:
                 self.finish(json.dumps(output))
         elif action == 'fetch_feedback':
-            assignment_id = data['assignment_id']
-            course_id = data['course_id']
             self.manager.fetch_feedback(course_id, assignment_id)
             self.finish(json.dumps(self.manager.list_assignments(course_id=course_id)))
 
@@ -367,7 +369,7 @@ def get(self):
 #-----------------------------------------------------------------------------
 
 
-_assignment_action_regex = r"(?P<action>fetch|submit|fetch_feedback)"
+_assignment_action_regex = r"(?P<action>fetch|fetch_missing|submit|fetch_feedback)"
 
 default_handlers = [
     (r"/assignments", AssignmentListHandler),

src/assignment_list/assignmentlist.ts

@@ -293,6 +293,36 @@ class Assignment {
 
       }
     } else if (this.data.status == 'fetched') {
+        var refetchButton = document.createElement('button');
+        refetchButton.classList.add('btn', 'btn-danger', 'btn-xs');
+        refetchButton.setAttribute('data-bs-toggle', 'tooltip');
+        refetchButton.setAttribute('data-bs-placement', 'top');
+        refetchButton.setAttribute('style', 'background:#d43f3a; margin-left:5px');
+        refetchButton.setAttribute('title', 'If you broke any of your assignment files and you want to redownload them, delete those files and click this button to refetch the original version of those files.')
+        container.append(refetchButton);
+        
+        refetchButton.innerText = 'Refetch';
+        refetchButton.onclick = async function(){
+          refetchButton.innerText = 'Refetching...';
+          refetchButton.setAttribute('disabled', 'disabled');
+          const dataToSend = { course_id: that.data['course_id'], assignment_id: that.data['assignment_id']};
+          try {
+            const reply = await requestAPI<any>('assignments/fetch_missing', {
+              body: JSON.stringify(dataToSend),
+              method: 'POST'
+            });
+
+            that.on_refresh(reply);
+
+          } catch (reason) {
+            remove_children(container);
+            container.innerText = 'Error refetching assignment.';
+            console.error(
+              `Error on POST /assignment_list/fetch_missing ${dataToSend}.\n${reason}`
+            );
+          }
+        }
+
         button.innerText = "Submit";
         button.onclick = async function(){
           button.innerText = 'submitting...';