Add rest call to get job logs

lib/job_server.py

@@ -9,7 +9,7 @@ from datetime import datetime
 from zipfile import ZipFile
 
 import requests
-from flask import Flask, request, render_template, send_file, after_this_request
+from flask import Flask, request, render_template, send_file, after_this_request, Response
 from werkzeug.utils import secure_filename
 
 from lib.render_job import RenderJob
@@ -44,6 +44,20 @@ def get_job_status(job_id):
         return f'Cannot find job with ID {job_id}', 400
 
 
+@server.get('/api/job/<job_id>/logs')
+def get_job_logs(job_id):
+    found_job = RenderQueue.job_with_id(job_id)
+    if found_job:
+        log_path = found_job.worker.log_path
+        log_data = None
+        if log_path and os.path.exists(log_path):
+            with open(log_path) as file:
+                log_data = file.read()
+        return Response(log_data, mimetype='text/plain')
+    else:
+        return f'Cannot find job with ID {job_id}', 400
+
+
 @server.get('/api/file_list/<job_id>')
 def get_file_list(job_id):
     found_job = RenderQueue.job_with_id(job_id)

lib/render_job.py

@@ -81,6 +81,22 @@ class RenderJob:
     def stop(self):
         self.worker.stop()
 
+    def time_elapsed(self):
+        # calculate elapsed time
+        elapsed_time = 'Unknown'
+        start_time = self.worker.start_time
+        end_time = self.worker.end_time
+
+        if start_time:
+            if end_time:
+                elapsed_time = str(end_time - start_time)
+            elif self.render_status() == RenderStatus.RUNNING:
+                elapsed_time = str(datetime.now() - start_time)
+        return elapsed_time
+
+    def frame_count(self):
+        return self.worker.total_frames
+
     @classmethod
     def generate_id(cls):
         return str(uuid.uuid4()).split('-')[0]