2025-12-01

2026-03-17 14:58:51 -06:00
parent 183e865f8b
commit 4b82b57113
6846 changed files with 954887 additions and 162606 deletions
@@ -0,0 +1,94 @@
+# SPDX-FileCopyrightText: 2021 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from typing import Tuple
+
+import bpy
+
+from .. import bkglobals
+from ..types import Sequence, Project, Shot
+from ..logger import LoggerFactory
+import gazu
+
+logger = LoggerFactory.getLogger()
+
+
+def shot_meta(strip: bpy.types.Strip, shot: Shot) -> None:
+    # Update shot info.
+
+    shot.name = strip.kitsu.shot_name
+    shot.description = strip.kitsu.shot_description
+    shot.data["frame_in"] = strip.frame_final_start
+    shot.data["frame_out"] = strip.frame_final_end
+    shot.data["3d_start"] = strip.kitsu_3d_start
+    shot.nb_frames = strip.frame_final_duration
+    shot.data["fps"] = bkglobals.FPS
+
+    # If user changed the sequence the shot belongs to
+    # (can only be done by operator not by hand).
+    if strip.kitsu.sequence_id != shot.sequence_id:
+        sequence = Sequence.by_id(strip.kitsu.sequence_id)
+        shot.sequence_id = sequence.id
+        shot.parent_id = sequence.id
+        shot.sequence_name = sequence.name
+
+    # Update on server.
+    shot.update()
+    logger.info("Pushed meta to shot: %s from strip: %s", shot.name, strip.name)
+
+
+def new_shot(
+    strip: bpy.types.Strip, sequence: Sequence, project: Project, add_tasks=False
+) -> Shot:
+    frame_range = (strip.frame_final_start, strip.frame_final_end)
+    shot = project.create_shot(
+        sequence,
+        strip.kitsu.shot_name,
+        nb_frames=strip.frame_final_duration,
+        frame_in=frame_range[0],
+        frame_out=frame_range[1],
+        data={"fps": bkglobals.FPS, "3d_start": bkglobals.FRAME_START},
+    )
+
+    if add_tasks:
+        create_intial_tasks(shot, project)
+
+    # Update description, no option to pass that on create.
+    if strip.kitsu.shot_description:
+        shot.description = strip.kitsu.shot_description
+        shot.update()
+
+    # Set project name locally, will be available on next pull.
+    shot.project_name = project.name
+    logger.info("Pushed create shot: %s for project: %s", shot.name, project.name)
+    return shot
+
+
+def new_sequence(strip: bpy.types.Strip, project: Project) -> Sequence:
+    sequence = project.create_sequence(
+        strip.kitsu.sequence_name,
+        strip.kitsu.episode_id
+    )
+    logger.info(
+        "Pushed create sequence: %s for project: %s", sequence.name, project.name
+    )
+    return sequence
+
+
+def delete_shot(strip: bpy.types.Strip, shot: Shot) -> str:
+    result = shot.remove()
+    logger.info(
+        "Pushed delete shot: %s for project: %s",
+        shot.name,
+        shot.project_name or "Unknown",
+    )
+    strip.kitsu.clear()
+    return result
+
+
+def create_intial_tasks(shot: Shot, project: Project):
+    shot_entity = gazu.shot.get_shot(shot.id)
+    for task_type in gazu.task.all_task_types_for_project(project.id):
+        if task_type["for_entity"] == "Shot":
+            gazu.task.new_task(shot_entity, task_type)