Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2025-12-25 13:24:54 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

OP-3684 - fix for new publisher

Browse source 
New publisher expects frames in file names in '.0000.' format, AE by default provides ('_0000.'). Locally rendered files need to be renamed to appropriate format.
This commit is contained in:
Petr Kalis 2022-08-02 18:13:39 +02:00
parent 6d48611626
commit 80b6ef981a
1 changed files with 35 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

38
openpype/hosts/aftereffects/plugins/publish/extract_local_render.py
View file

@ -1,7 +1,8 @@
import os import os
import sys import sys
import six import six
import re
import shutil
import openpype.api import openpype.api
from openpype.hosts.aftereffects.api import get_stub from openpype.hosts.aftereffects.api import get_stub
@ -22,15 +23,26 @@ class ExtractLocalRender(openpype.api.Extractor):
# pull file name from Render Queue Output module # pull file name from Render Queue Output module
render_q = stub.get_render_info() render_q = stub.get_render_info()
stub.render(staging_dir) stub.render(staging_dir)
render_q_file_name = render_q.file_name
if not render_q: if not render_q:
raise ValueError("No file extension set in Render Queue") raise ValueError("No file extension set in Render Queue")
_, ext = os.path.splitext(os.path.basename(render_q.file_name)) _, ext = os.path.splitext(os.path.basename(render_q_file_name))
ext = ext[1:] ext = ext[1:]
replace_frames_format = self._get_replace_format(render_q_file_name)
first_file_path = None first_file_path = None
files = [] files = []
self.log.info("files::{}".format(os.listdir(staging_dir)))
for file_name in os.listdir(staging_dir): for file_name in os.listdir(staging_dir):
_, found_ext = os.path.splitext(file_name)
if found_ext[1:] != ext:
continue
if replace_frames_format:
file_name = self._translate_frames(file_name,
replace_frames_format,
staging_dir)
files.append(file_name) files.append(file_name)
if first_file_path is None: if first_file_path is None:
first_file_path = os.path.join(staging_dir, first_file_path = os.path.join(staging_dir,
@ -78,3 +90,23 @@ class ExtractLocalRender(openpype.api.Extractor):
"stagingDir": staging_dir, "stagingDir": staging_dir,
"tags": ["thumbnail"] "tags": ["thumbnail"]
}) })
def _translate_frames(self, file_name, replace_frames_format, staging_dir):
orig_file_name = file_name
found_frames = re.search(replace_frames_format, file_name)
if found_frames:
new_frames = found_frames.group(0).replace('_', '.')
file_name = file_name.replace(found_frames.group(0), new_frames)
shutil.move(os.path.join(staging_dir, orig_file_name),
os.path.join(staging_dir, file_name))
return file_name
def _get_replace_format(self, file_name):
# replace delimiter for frames to one integrate is expecting (.0000.)
# returns frame format to be replaced
hashes_found = re.search(r"(_%5B[#]*%5D.)", file_name)
if hashes_found:
hashes = re.sub("[^#]", '', hashes_found.group(0))
return "_[0-9]{{{0}}}.".format(len(hashes))