scripts-blender/addons/blender_kitsu/types.py

@@ -25,6 +25,8 @@ from typing import Any, Dict, List, Optional, Union, Tuple, TypeVar
 
 import gazu
 from .logger import LoggerFactory
+from . import bkglobals
+
 
 logger = LoggerFactory.getLogger()
 
@@ -565,6 +567,12 @@ class Shot(Entity):
         gazu.shot.update_shot(asdict(self))
         return self
 
+    def get_shot_task_name(self, task_type: TaskType) -> str:  #
+        return f"{self.name}{bkglobals.FILE_DELIMITER}{task_type.get_short_name()}"
+
+    def get_output_collection_name(self, task_type: TaskType) -> str:
+        return f"{self.get_shot_task_name(task_type)}{bkglobals.FILE_DELIMITER}output"
+
     def update_data(self, data: Dict[str, Any]) -> Shot:
         gazu.shot.update_shot_data(asdict(self), data=data)
         if not self.data:
@@ -714,6 +722,11 @@ class TaskType(Entity):
             if t["for_entity"] == "Sequence"
         ]
 
+    def get_short_name(self) -> str:
+        for key, value in bkglobals.SHOT_TASK_MAPPING.items():
+            if value == self.name:
+                return key
+
     def __bool__(self) -> bool:
         return bool(self.id)