ShotGrid publish update

wgergely · Jan 15, 2024 · e7c8d22 · e7c8d22
1 parent e30a534
commit e7c8d22
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/bookmarks/shotgun/sg_publish_clip.py b/bookmarks/shotgun/sg_publish_clip.py
@@ -349,19 +349,22 @@ def save_changes(self):
                     )
 
                 # Publish steps for creating the Cut, CutInfo and Version entities
-                # Get the entity type
-                if 'task_entity' in data and data['task_entity']:
-                    entity = data['task_entity']
-                elif 'asset_entity' in data and data['asset_entity']:
-                    entity = data['asset_entity']
+                if 'asset_entity' in data and data['asset_entity']:
+                    asset_entity = data['asset_entity']
                 else:
                     raise RuntimeError('No asset or task entity found to associate with the publish.')
 
+                if 'task_entity' in data and data['task_entity']:
+                    task_entity = data['task_entity']
+                else:
+                    task_entity = None
+
                 version_data = {
                     'project': data['project_entity'],
                     'code': data['name'],
                     'description': data['description'],
-                    'entity': entity,
+                    'entity': asset_entity,
+                    'sg_task': task_entity,
                     'user': user_entity
                 }
                 version = sg.create('Version', version_data)
@@ -382,7 +385,8 @@ def save_changes(self):
 
                 cut_data = {
                     'project': data['project_entity'],
-                    'entity': entity,
+                    'entity': asset_entity,
+                    'sg_task': task_entity,
                     'description': data['description'],
                     'version': version
                 }
@@ -418,7 +422,7 @@ def save_changes(self):
             # Publish step for copying local files to their destinations
             try:
                 common.show_message(
-                    'Publishing Published File',
+                    'Creating a Published File',
                     body='Please wait while the file is being published.',
                     message_type=None,
                     buttons=[],
@@ -430,7 +434,8 @@ def save_changes(self):
                     'project': data['project_entity'],
                     'code': data['name'],
                     'description': data['description'],
-                    'entity': data['asset_entity'],
+                    'entity': asset_entity,
+                    'sg_task': task_entity,
                     'version': version,
                     'published_file_type': data['published_file_type_entity'],
                     'path': {