Blender Kitsu: Add Import Edit Render Operator for Shots #236

Merged
Nick Alberelli merged 6 commits from TinyNick/blender-studio-pipeline:feature/import_edit_export into main 2024-02-19 21:02:44 +01:00
5 changed files with 24 additions and 24 deletions
Showing only changes of commit 09dec7d779 - Show all commits

View File

@ -4,32 +4,32 @@ import re
from pathlib import Path
def edit_export_get_latest(context: bpy.types.Context):
"""Find latest export in editorial export directory"""
def edit_render_get_latest(context: bpy.types.Context):
"""Find latest render in editorial render directory"""
files_list = edit_exports_get_all(context)
files_list = edit_renders_get_all(context)
if len(files_list) >= 1:
files_list = sorted(files_list, reverse=True)
return files_list[0]
return None
def edit_exports_get_all(context: bpy.types.Context):
"""Find latest export in editorial export directory"""
def edit_renders_get_all(context: bpy.types.Context):
"""Find latest render in editorial render directory"""
addon_prefs = prefs.addon_prefs_get(context)
edit_export_path = Path(addon_prefs.edit_export_dir)
edit_render_path = Path(addon_prefs.edit_export_dir)
files_list = [
f
for f in edit_export_path.iterdir()
for f in edit_render_path.iterdir()
if f.is_file()
and edit_export_is_valid_edit_name(addon_prefs.edit_export_file_pattern, f.name)
and edit_render_is_valid_edit_name(addon_prefs.edit_export_file_pattern, f.name)
]
return files_list
def edit_export_is_valid_edit_name(file_pattern: str, filename: str) -> bool:
def edit_render_is_valid_edit_name(file_pattern: str, filename: str) -> bool:
"""Verify file name matches file pattern set in preferences"""
# Prevents un-expected matches
file_pattern = re.escape(file_pattern)
@ -40,13 +40,13 @@ def edit_export_is_valid_edit_name(file_pattern: str, filename: str) -> bool:
return False
def edit_export_import_latest(
def edit_render_import_latest(
context: bpy.types.Context, shot
) -> list[bpy.types.Sequence]: # TODO add info to shot
"""Loads latest export from editorial department"""
"""Loads latest render from editorial department"""
addon_prefs = prefs.addon_prefs_get(context)
strip_channel = 1
latest_file = edit_export_get_latest(context)
latest_file = edit_render_get_latest(context)
if not latest_file:
return None
# Check if Kitsu server returned empty shot

View File

@ -8,7 +8,7 @@ from ..types import Task, TaskStatus
from ..playblast.core import override_render_path, override_render_format
from . import opsdata
from ..logger import LoggerFactory
from .core import edit_export_import_latest, edit_exports_get_all, edit_export_get_latest
from .core import edit_render_import_latest, edit_renders_get_all, edit_render_get_latest
logger = LoggerFactory.getLogger()
@ -19,7 +19,7 @@ class KITSU_OT_edit_render_publish(bpy.types.Operator):
bl_description = (
"Renders current VSE Edit as .mp4"
"Saves the set version to disk and uploads it to Kitsu with the specified "
"comment and task type. Overrides some render settings during export. "
"comment and task type. Overrides some render settings during render. "
)
task_status: bpy.props.EnumProperty(name="Task Status", items=cache.get_all_task_statuses_enum)
@ -253,7 +253,7 @@ class KITSU_OT_edit_render_import_latest(bpy.types.Operator):
def check_if_latest_edit_render_is_imported(self, context: Context) -> bool:
# Check if latest edit render is already loaded.
for strip in self._existing_edit_renders:
latest_edit_render_path = edit_export_get_latest(context)
latest_edit_render_path = edit_render_get_latest(context)
if self.compare_strip_to_path(strip, latest_edit_render_path):
self._latest_render_name = latest_edit_render_path.name
return True
@ -278,7 +278,7 @@ class KITSU_OT_edit_render_import_latest(bpy.types.Operator):
addon_prefs = prefs.addon_prefs_get(context)
# Get paths to all edit renders
all_edit_render_paths = edit_exports_get_all(context)
all_edit_render_paths = edit_renders_get_all(context)
if all_edit_render_paths == []:
self.report(
{"WARNING"},
@ -302,7 +302,7 @@ class KITSU_OT_edit_render_import_latest(bpy.types.Operator):
# Import new edit render
shot = cache.shot_active_get()
strips = edit_export_import_latest(context, shot)
strips = edit_render_import_latest(context, shot)
if strips is None:
self.report({"WARNING"}, f"Loaded Latest Editorial Render failed to import!")

View File

@ -43,17 +43,17 @@ def init_edit_render_file_model(
addon_prefs = prefs.addon_prefs_get(context)
kitsu_props = context.scene.kitsu
edit_export_dir = Path(addon_prefs.edit_export_dir)
edit_render_dir = Path(addon_prefs.edit_export_dir)
# Is None if invalid.
if addon_prefs.edit_export_dir == "" or not edit_export_dir.exists():
if addon_prefs.edit_export_dir == "" or not edit_render_dir.exists():
logger.error(
"Failed to initialize edit render file model. Invalid path. Check addon preferences"
)
return
EDIT_RENDER_FILE_MODEL.reset()
EDIT_RENDER_FILE_MODEL.root_path = edit_export_dir
EDIT_RENDER_FILE_MODEL.root_path = edit_render_dir
if not EDIT_RENDER_FILE_MODEL.versions:
EDIT_RENDER_FILE_MODEL.append_item("v001")

View File

@ -41,7 +41,7 @@ from .auth.ops import (
)
from .context.ops import KITSU_OT_con_productions_load
from .lookdev.prefs import LOOKDEV_preferences
from .edit.core import edit_export_get_latest
from .edit.core import edit_render_get_latest
logger = LoggerFactory.getLogger()
@ -613,7 +613,7 @@ class KITSU_addon_preferences(bpy.types.AddonPreferences):
@property
def is_editorial_dir_valid(self) -> bool:
if edit_export_get_latest(bpy.context) is None:
if edit_render_get_latest(bpy.context) is None:
logger.error(
"Failed to initialize editorial export file model. Invalid path/pattern. Check addon preferences"
)

View File

@ -16,7 +16,7 @@ from .core import (
)
from ..context import core as context_core
from ..edit.core import edit_export_import_latest
from ..edit.core import edit_render_import_latest
from .file_save import save_shot_builder_file
from .template import replace_workspace_with_template
from .assets import get_shot_assets
@ -236,7 +236,7 @@ class KITSU_OT_build_new_shot(bpy.types.Operator):
link_task_type_output_collections(shot, task_type)
if bkglobals.LOAD_EDITORIAL_REF.get(task_type_short_name):
edit_export_import_latest(context, shot)
edit_render_import_latest(context, shot)
# Run Hooks
hooks_instance = Hooks()