scripts-blender/addons/blender_kitsu/bkglobals.py

@@ -118,3 +118,15 @@ OUTPUT_COL_LINK_MAPPING = {
     "smear_to_mesh": None,
     "storyboard": None,
 }
+
+LOAD_EDITORIAL_REF = {
+    "anim": True,
+    "comp": False,
+    "fx": False,
+    "layout": True,
+    "lighting": False,
+    "previz": False,
+    "rendering": False,
+    "smear_to_mesh": False,
+    "storyboard": False,
+}

scripts-blender/addons/blender_kitsu/shot_builder_2/editorial.py

@@ -0,0 +1,80 @@
+import bpy
+from .. import prefs
+from pathlib import Path
+import re
+
+
+def editorial_export_get_latest(
+    context: bpy.types.Context, shot
+) -> list[bpy.types.Sequence]:  # TODO add info to shot
+    """Loads latest export from editorial department"""
+    addon_prefs = prefs.addon_prefs_get(context)
+    strip_channel = 1
+    latest_file = editorial_export_check_latest(context)
+    if not latest_file:
+        return None
+    # Check if Kitsu server returned empty shot
+    if shot.id == '':
+        return None
+    strip_filepath = latest_file.as_posix()
+    strip_frame_start = addon_prefs.shot_builder_frame_offset
+
+    scene = context.scene
+    if not scene.sequence_editor:
+        scene.sequence_editor_create()
+    seq_editor = scene.sequence_editor
+    movie_strip = seq_editor.sequences.new_movie(
+        latest_file.name,
+        strip_filepath,
+        strip_channel + 1,
+        strip_frame_start,
+        fit_method="FIT",
+    )
+    sound_strip = seq_editor.sequences.new_sound(
+        latest_file.name,
+        strip_filepath,
+        strip_channel,
+        strip_frame_start,
+    )
+    new_strips = [movie_strip, sound_strip]
+
+    # Update shift frame range prop.
+    frame_in = shot.data.get("frame_in")
+    frame_3d_start = shot.data.get("3d_start")
+    frame_3d_offset = frame_3d_start - addon_prefs.shot_builder_frame_offset
+    edit_export_offset = addon_prefs.edit_export_frame_offset
+
+    # Set sequence strip start kitsu data.
+    for strip in new_strips:
+        strip.frame_start = (
+            -frame_in + (strip_frame_start * 2) + frame_3d_offset + edit_export_offset
+        )
+    return new_strips
+
+
+def editorial_export_check_latest(context: bpy.types.Context):
+    """Find latest export in editorial export directory"""
+    addon_prefs = prefs.addon_prefs_get(context)
+
+    edit_export_path = Path(addon_prefs.edit_export_dir)
+
+    files_list = [
+        f
+        for f in edit_export_path.iterdir()
+        if f.is_file()
+        and editorial_export_is_valid_edit_name(
+            addon_prefs.edit_export_file_pattern, f.name
+        )
+    ]
+    if len(files_list) >= 1:
+        files_list = sorted(files_list, reverse=True)
+        return files_list[0]
+    return None
+
+
+def editorial_export_is_valid_edit_name(file_pattern: str, filename: str) -> bool:
+    """Verify file name matches file pattern set in preferences"""
+    match = re.search(file_pattern, filename)
+    if match:
+        return True
+    return False

scripts-blender/addons/blender_kitsu/shot_builder_2/ops.py

@@ -14,6 +14,8 @@ from .core import (
     link_task_type_output_collections,
 )
 
+from .editorial import editorial_export_get_latest
+
 active_project = None
 
 
@@ -164,6 +166,9 @@ class KITSU_OT_build_new_shot(bpy.types.Operator):
         # Link External Output Collections
         link_task_type_output_collections(shot, task_short_name)
 
+        if bkglobals.LOAD_EDITORIAL_REF.get(task_short_name):
+            editorial_export_get_latest(context, shot)
+
         print("Create shot with the following details")  # TODO Remove
         print(f"Seq Name: '{seq.name}' Seq ID: '{self.seq_id}'")
         print(f"Shot Name: '{shot.name}' Shot ID: '{self.shot_id}'")