Allow jobs to choose when to save log output

server/autograder.py

@@ -178,6 +178,9 @@ def autograde_assignment(assignment_id):
     ]
     num_tasks = len(tasks)
 
+    if not num_tasks:
+        return "No submissions to grade"
+
     def retry_task(task):
         if task.retries >= MAX_RETRIES:
             logger.error('Did not receive a score for backup {} after {} retries'.format(
@@ -194,8 +197,10 @@ def retry_task(task):
 
         graded = len([task for task in tasks
             if task.status in (GradingStatus.DONE, GradingStatus.FAILED)])
-        logger.info('Graded {:>4}/{} ({:>5.1f}%)'.format(
-            graded, num_tasks, 100 * graded / num_tasks))
+
+        logger.critical('Graded {:>4}/{} ({:>5.1f}%)'.format(
+                        graded, num_tasks, 100 * graded / num_tasks))
+
         if graded == num_tasks:
             break
 

server/jobs/__init__.py

@@ -22,7 +22,7 @@ def handle(self, record):
         self.counter += 1
         super().handle(record)
         print(record.message)
-        if (self.counter % self.log_every) == 0:
+        if record.levelno >= logging.CRITICAL or not (self.counter % self.log_every):
             self.job.log = self.contents
             db.session.commit()
 

server/jobs/example.py

@@ -13,6 +13,8 @@ def test_job(duration=0, should_fail=False, make_file=False):
     logger = jobs.get_job_logger()
 
     logger.info('Starting...')
+    logger.critical('This job will sleep for {} seconds'.format(duration))
+
     time.sleep(duration)
     if should_fail:
         1/0

server/jobs/moss.py

@@ -44,7 +44,7 @@ def submit_to_moss(moss_id=None, file_regex=".*", assignment_id=None, language=N
                           .order_by(Backup.created.desc())
                           .all())
 
-    logger.info("Retreived {} final submissions".format(len(subm_keys)))
+    logger.info("Retrieved {} final submissions".format(len(subm_keys)))
     # TODO: Customize the location of the tmp writing (especially useful during dev)
 
     with tempfile.TemporaryDirectory() as tmp_dir:
@@ -124,7 +124,7 @@ def submit_to_moss(moss_id=None, file_regex=".*", assignment_id=None, language=N
                    .format(lang=language, templates=templates,
                            folder=' '.join(all_student_files)))
 
-        logger.info("Running {}".format(command[:100] + ' ...'))
+        logger.critical("Running {}".format(command[:100] + ' ...'))
 
         try:
             process = subprocess.check_output(shlex.split(command),

tests/test_job.py

@@ -44,7 +44,7 @@ def test_job(self):
         job = Job.query.get(job_id)
         self.assertEqual(job.status, 'finished')
         self.assertFalse(job.failed)
-        self.assertEqual(job.log, 'Starting...\nFinished!\n')
+        self.assertEqual(job.log, 'Starting...\nThis job will sleep for 0 seconds\nFinished!\n')
 
     def test_failing_job(self):
         job_id = self.start_test_job(should_fail=True)