upload to kitsu

operators/operators.py

@@ -98,13 +98,18 @@ def get_task_status_items(self, context):
     project = settings.active_project
 
     status_items = [('CURRENT', 'Current', '')]
-    status_items += [(t.name, t.name, '') for t in project.task_statuses]
+
+    if project:
+        status_items += [(t.name, t.name, '') for t in project.task_statuses]
     return status_items
 
 def get_task_type_items(self, context):
     settings = get_scene_settings()
     project = settings.active_project
 
+    if not project:
+        return [('NONE', 'None', '')]
+
     return [(t.name, t.name, '') for t in project.task_types]
 
 class VSETB_OT_upload_to_tracker(Operator):
@@ -158,7 +163,7 @@ class VSETB_OT_upload_to_tracker(Operator):
         #self.report({'ERROR'}, f'Export not implemented yet.')
         prefs = get_addon_prefs()
         settings = get_scene_settings()
-        #project = settings.active_project
+        project = settings.active_project
         tracker = prefs.tracker
 
         status = self.status
@@ -170,6 +175,10 @@ class VSETB_OT_upload_to_tracker(Operator):
             shot_name = strip.name
             sequence = tracker.get_sequence(sequence_name)
 
+
+            metadata = strip.vsetb_strip_settings.metadata.to_dict()
+            print(metadata)
+
             if not sequence:
                 self.report({"INFO"}, f'Create sequence {sequence_name} in Kitsu')
                 sequence = tracker.new_sequence(sequence_name)
@@ -180,16 +189,29 @@ class VSETB_OT_upload_to_tracker(Operator):
                 shot = tracker.new_shot(shot_name, sequence=sequence)
 
             task = tracker.get_task(self.task, entity=shot)
-            
-            print(task)
-            tracker.new_comment(task, comment=self.comment, status=status)
-            #status = tracker.get_task(shot, self.task)
-            # if self.add_preview:
-            #     strip_movie_path = Path(get_strip_render_path(strip, template))
-            #     if path.exists():
+            if not task:
+                task = tracker.new_task(shot, task_type=self.task)
 
-            #     else:
-            #         self.report({"WARNING"}, f"{strip_movie_path} not exist, skipped")
+            # print('\n', 'task comment')
+            # print(task['last_comment'])
+
+            preview = None
+            if self.add_preview:
+                preview = Path(get_strip_render_path(strip, project.render_template))
+                if not preview.exists():
+                    preview = None
+
+                elif task['last_comment'].get('previews'):
+                    if self.preview_mode == 'REPLACE':
+                        tracker.remove_comment(task['last_comment'])
+                    elif self.preview_mode == 'ONLY_NEW':
+                        preview = None
+
+            if status != 'CURRENT' or preview:
+                tracker.new_comment(task, comment=self.comment, status=status, preview=preview)
+
+            if self.custom_data:
+                tracker.update_data(shot, strip.vsetb_strip_settings.metadata.to_dict())
 
 
         return {"FINISHED"}
@@ -444,7 +466,9 @@ class VSETB_OT_load_projects(Operator):
         settings = get_scene_settings()
         prefs = get_addon_prefs()
         tracker = prefs.tracker
-        
+
+        old_project_name = settings.project_name.replace(' ', '_').upper()
+       
         settings.projects.clear()
         tracker.connect()
 
@@ -478,11 +502,20 @@ class VSETB_OT_load_projects(Operator):
             for asset_type_data in tracker.get_asset_types(project_data):
                asset_type = project.asset_types.add()
                asset_type.name = asset_type_data['name']
+
+        for project in settings.projects:
+            print(project.name, project.name.replace(' ', '_').upper(), old_project_name)
+            if project.name.replace(' ', '_').upper() == old_project_name:
+
+                print('Restore Project Name')
+                settings.project_name = project.name
+        #else:
+        #    print('Could Not restore Project Name')
+
+        #if settings.active_project:
+        #    settings.active_project.set_strip_metadata()
         
-        if settings.active_project:
-            settings.active_project.set_strip_metadata()
-        
-        bpy.ops.vse_toolbox.load_assets()
+        #bpy.ops.vse_toolbox.load_assets()
 
         return {'FINISHED'}