Subjob Zip Files (#18)

* Transfer any uploaded zip files to subjobs instead of extracted zips

* Fix Blender naming scheme
This commit is contained in:
2023-06-15 19:21:10 -05:00
committed by GitHub
parent 78a51ffea0
commit 54ec5f0e9c
2 changed files with 9 additions and 7 deletions

View File

@@ -357,7 +357,8 @@ def add_job_handler():
logger.info(f"Import complete for {loaded_project_local_path.split(server.config['UPLOAD_FOLDER'])[-1]}")
# process uploaded zip files
if loaded_project_local_path.lower().endswith('.zip'):
zip_path = loaded_project_local_path if loaded_project_local_path.lower().endswith('.zip') else None
if zip_path:
zip_path = loaded_project_local_path
work_path = os.path.dirname(zip_path)
try:
@@ -387,12 +388,12 @@ def add_job_handler():
for job_data in jobs_list:
try:
# prepare output paths
output_dir = os.path.join(job_dir, job_data.get('name', None) or 'output')
output_dir = os.path.join(job_dir, job_data.get('name') if len(jobs_list) > 1 else 'output')
os.makedirs(output_dir, exist_ok=True)
# get new output path in output_dir
job_data['output_path'] = os.path.join(output_dir, os.path.basename(
job_data.get('name', None) or job_data.get('output_path', None) or loaded_project_local_path
job_data.get('output_path', None) or loaded_project_local_path
))
# create & configure jobs
@@ -409,7 +410,7 @@ def add_job_handler():
# determine if we can / should split the job
if server.config.get('enable_split_jobs', False) and (worker.total_frames > 1) and not worker.parent:
create_subjobs(worker, job_data, loaded_project_local_path)
create_subjobs(worker, job_data, zip_path or loaded_project_local_path)
RenderQueue.add_to_render_queue(worker, force_start=job_data.get('force_start', False))
make_job_ready(worker.id)