From 11151b412244b7d6b6a575612d694b159f249f34 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Mon, 13 Jul 2020 16:24:38 +0200 Subject: [PATCH] added possibility to add task entity type custom attributes --- .../ftrack/actions/action_create_cust_attrs.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/pype/modules/ftrack/actions/action_create_cust_attrs.py b/pype/modules/ftrack/actions/action_create_cust_attrs.py index f3bba1a9b4..593da28cb0 100644 --- a/pype/modules/ftrack/actions/action_create_cust_attrs.py +++ b/pype/modules/ftrack/actions/action_create_cust_attrs.py @@ -455,10 +455,20 @@ class CustomAttributes(BaseAction): # Prepare data of entity specific attributes for entity_type, cust_attr_datas in cust_attr_def.items(): - for key, cust_attr_data in cust_attr_datas.items(): - cust_attr_data["key"] = key - cust_attr_data["entity_type"] = entity_type - attrs_data.append(cust_attr_data) + if entity_type.lower() != "task": + for key, cust_attr_data in cust_attr_datas.items(): + cust_attr_data["key"] = key + cust_attr_data["entity_type"] = entity_type + attrs_data.append(cust_attr_data) + continue + + # Task should have nested level for object type + for object_type, _cust_attr_datas in cust_attr_datas.items(): + for key, cust_attr_data in _cust_attr_datas.items(): + cust_attr_data["key"] = key + cust_attr_data["entity_type"] = entity_type + cust_attr_data["object_type"] = object_type + attrs_data.append(cust_attr_data) # Process prepared data for cust_attr_data in attrs_data: