brett/Zordon
1
0
Fork 0
mirror of https://github.com/blw1138/Zordon.git synced 2025-12-17 16:58:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Generate project file hashes on separate thread

Browse Source
This commit is contained in:
Brett Williams
2022-12-20 13:50:26 -08:00
parent 4ef94dd7b8
commit 5f39fe3d64
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lib/render_job.py
View File

@@ -3,6 +3,7 @@ import hashlib
import json import json
import logging import logging
import os import os
import threading
import uuid import uuid
from datetime import datetime from datetime import datetime
@@ -13,6 +14,11 @@ logger = logging.getLogger()
class RenderJob: class RenderJob:
# Get file hash on bg thread
def __get_file_hash(self):
if os.path.exists(self.worker.input_path):
self.file_hash = hashlib.md5(open(self.worker.input_path, 'rb').read()).hexdigest()
def __init__(self, renderer, input_path, output_path, args, priority=2, owner=None, client=None, notify=None, def __init__(self, renderer, input_path, output_path, args, priority=2, owner=None, client=None, notify=None,
custom_id=None, name=None): custom_id=None, name=None):
self.id = custom_id or self.generate_id() self.id = custom_id or self.generate_id()
@@ -28,6 +34,9 @@ class RenderJob:
self.worker = RenderWorkerFactory.create_worker(renderer, input_path, output_path, args) self.worker = RenderWorkerFactory.create_worker(renderer, input_path, output_path, args)
self.worker.log_path = os.path.join(os.path.dirname(input_path), self.name + '.log') self.worker.log_path = os.path.join(os.path.dirname(input_path), self.name + '.log')
self.file_hash = None
threading.Thread(target=self.__get_file_hash).start() # get file hash on bg thread
def render_status(self): def render_status(self):
if self.scheduled_start and self.worker.status == RenderStatus.NOT_STARTED: if self.scheduled_start and self.worker.status == RenderStatus.NOT_STARTED:
return RenderStatus.SCHEDULED return RenderStatus.SCHEDULED
@@ -46,7 +55,7 @@ class RenderJob:
job_dict = self.__dict__.copy() job_dict = self.__dict__.copy()
job_dict['status'] = self.render_status().value job_dict['status'] = self.render_status().value
job_dict['time_elapsed'] = self.time_elapsed() if type(self.time_elapsed) != str else self.time_elapsed job_dict['time_elapsed'] = self.time_elapsed() if type(self.time_elapsed) != str else self.time_elapsed
job_dict['file_hash'] = self.file_hash if self.file_hash and isinstance(self.file_hash, str) else self.file_hash() job_dict['file_hash'] = self.file_hash
job_dict['percent_complete'] = self.percent_complete() job_dict['percent_complete'] = self.percent_complete()
job_dict['file_list'] = self.file_list() job_dict['file_list'] = self.file_list()
job_dict['worker'] = self.worker.__dict__.copy() job_dict['worker'] = self.worker.__dict__.copy()