Import Shots for mu

operators/imports.py

@@ -2,21 +2,21 @@
 from pathlib import Path
 
 import bpy
-from bpy.types import Operator
+from bpy.types import Operator, UIList
 from bpy.props import (CollectionProperty, BoolProperty, EnumProperty, StringProperty)
 
 from vse_toolbox.constants import (EDITS, EDIT_SUFFIXES, MOVIES, MOVIE_SUFFIXES,
     SOUNDS, SOUND_SUFFIXES)
 
 from vse_toolbox.sequencer_utils import (clean_sequencer, import_edit, import_movie,
-    import_sound, get_strips)
+    import_sound, get_strips, get_channel_index)
        
 from vse_toolbox.bl_utils import get_scene_settings
 from vse_toolbox.file_utils import install_module
 
 
 class VSETB_OT_auto_select_files(Operator):
-    bl_idname = "import.auto_select_files"
+    bl_idname = "vse_toolbox.auto_select_files"
     bl_label = "Auto Select"
     bl_description = "Auto Select Files"
     bl_options = {"REGISTER", "UNDO"}
@@ -107,7 +107,7 @@ class VSETB_OT_import_files(Operator):
         layout.use_property_decorate = False
 
         col = layout.column(align=True)
-        col.operator('import.auto_select_files', text='Auto Select')
+        col.operator('vse_toolbox.auto_select_files', text='Auto Select')
 
         row = layout.row(heading="Import Edit", align=True)
         row.prop(self, 'import_edit')
@@ -190,9 +190,102 @@ class VSETB_OT_import_files(Operator):
 
         return {"FINISHED"}
 
+class VSETB_UL_import_task(UIList):
+    def draw_item(self, context, layout, data, item, icon, active_data,
+                  active_propname, index):
+
+        layout.prop(item, 'import_enabled', text='')
+        layout.label(text=item.name)
+        
+class VSETB_OT_import_shots(Operator):
+    bl_idname = "vse_toolbox.import_shots"
+    bl_label = "Import Shots"
+    bl_description = "Import Shots for disk or tracker"
+    bl_options = {"REGISTER", "UNDO"}
+
+    @classmethod
+    def poll(cls, context):
+        return True
+
+    def execute(self, context):
+        scn = context.scene
+        settings = get_scene_settings()
+        project = settings.active_project
+
+        from gadget import Project
+
+        p = Project.from_env()
+        p.shots.fetch()
+        
+        tasks = [t for t in project.task_types if t.import_enabled]
+        for i, task_type in enumerate(tasks):
+            channel_index = 9 + i
+
+            scn.sequence_editor.channels[channel_index].name = task_type.name
+
+        # Create Channels:
+        scn.sequence_editor.channels[9]
+
+        for strip in get_strips('Shots'):
+            for i, task_type in enumerate(tasks):
+                shot = p.shots[strip.name]
+                versions = shot.tasks[task_type.name].outputs[0].versions.fetch()
+
+                if versions:
+                    print(versions[-1])
+                
+                    imported_strip = import_movie(versions[-1].path)
+                    imported_strip.frame_start = strip.frame_start
+
+                    if imported_strip.frame_final_duration != strip.frame_final_duration:
+                        print(f'strip {strip.name} has different duration')
+                        imported_strip.color_tag = 'COLOR_03'
+                    
+                    imported_strip.frame_final_end = strip.frame_final_end
+
+                    imported_strip.channel = get_channel_index(task_type.name)
+                    print(get_channel_index(task_type.name), imported_strip.channel)
+                else:
+                    print(f'No movie for {strip.name} of task {task_type.name}')
+
+        # for task_type in tasks:
+        #     for strip in get_strips(task_type.name):
+        #         strip.channel = get_channel_index(task_type.name)
+
+
+        return {'FINISHED'}
+
+    def draw(self, context):
+        settings = get_scene_settings()
+        project = settings.active_project
+
+        layout = self.layout
+        col = layout.column(align=False)
+        col.use_property_split = True
+        col.use_property_decorate = False
+
+        row = col.row(align=True)
+        row.prop(project, "import_source", text='Source', expand=True)
+
+        row = col.row(align=True)
+        row.prop(project, 'import_task', text='Import Tasks')
+
+        if project.import_task == 'SELECTED':
+            col.use_property_split = False
+            col.template_list("VSETB_UL_import_task", "import_task", project, "task_types", project, "task_type_index", rows=8)
+
+    def invoke(self, context, event):
+        scn = context.scene
+        settings = get_scene_settings()
+        project = settings.active_project
+        
+        return context.window_manager.invoke_props_dialog(self)
+
 classes = (
+    VSETB_UL_import_task,
     VSETB_OT_auto_select_files,
     VSETB_OT_import_files,
+    VSETB_OT_import_shots,
 )
 
 def register():