Merge remote-tracking branch 'origin/feature/integrate_version_to_tas…

…k' into 1.1.0

quad_pyblish_module/plugins/hiero/publish/integrate_version_to_task.py

@@ -0,0 +1,79 @@
+from pyblish import api
+
+
+class IntegrateVersionToTask(api.ContextPlugin):
+ """Integrate Version To Task"""
+
+ order = api.IntegratorOrder + 10.1
+ hosts = ["hiero"]
+ families = ["clip"]
+ label = "Integrate Version To Task"
+ optional = False
+
+ def get_all_task_types(self, project):
+ """Get all task types from the project schema.
+
+ :param project: The Ftrack project Entity.
+ :type project: ftrack Project Object
+ :return: A dictionary containing all task types.
+ :rtype: dict<ftrackTaskType>
+
+ """
+ tasks = {}
+ proj_template = project['project_schema']
+ temp_task_types = proj_template['_task_type_schema']['types']
+
+ for type in temp_task_types:
+ if type['name'] not in tasks:
+ tasks[type['name']] = type
+
+ return tasks
+
+ def process(self, context):
+ """Process the Pyblish context.
+
+ :param context: The Pyblish context.
+ :type context: pyblish.api.Context
+ """
+ # Get Data we need
+ task_type = context.data.get('task').lower()
+ ftrack_session = context.data.get('ftrackSession')
+ ftrack_project = context.data.get('ftrackProject')
+ ftrack_task_types = self.get_all_task_types(ftrack_project)
+
+ if ftrack_session is None:
+ self.log.info('Ftrack session is not created.')
+ return
+
+ # Build Plugin dict list to analyze
+ to_analyze_lst = []
+ for result_dict in context.data.get('results'):
+ if 'Collect OTIO Review' in result_dict['plugin'].label:
+ to_analyze_lst.append(result_dict)
+
+ for to_analyze in to_analyze_lst:
+ ftrack_shot_version = to_analyze['instance'].data['ftrackEntity']
+ ftrack_asset_versions = to_analyze['instance'].data['ftrackIntegratedAssetVersions']
+
+ # Check if confo task already exists in ftrack
+ is_task_exist = ftrack_session.query(
+ 'Task where name is "{0}" and parent.id is {1}'.format(task_type, ftrack_shot_version['id'])).first()
+ if not is_task_exist:
+ new_task = ftrack_session.create(
+ 'Task',
+ {
+ 'name': task_type,
+ 'parent': ftrack_shot_version,
+ 'type': ftrack_task_types[task_type]
+ }
+ )
+ else:
+ new_task = is_task_exist
+
+ ftrack_session.commit()
+
+ # Link Publish version to Task
+ for asset_version in ftrack_asset_versions:
+ asset_version['task'] = new_task
+
+ ftrack_session.commit()