settings: task types enum entity

openpype/settings/entities/__init__.py

@@ -103,6 +103,7 @@
  EnumEntity,
  AppsEnumEntity,
  ToolsEnumEntity,
+ TaskTypeEnumEntity,
  ProvidersEnum
 )
 
@@ -154,6 +155,7 @@
  "EnumEntity",
  "AppsEnumEntity",
  "ToolsEnumEntity",
+ "TaskTypeEnumEntity",
  "ProvidersEnum",
 
  "ListEntity",

openpype/settings/entities/enum_entity.py

@@ -219,6 +219,40 @@ def set_override_state(self, *args, **kwargs):
  self._current_value = new_value
 
 
+class TaskTypeEnumEntity(BaseEnumEntity):
+ schema_types = ["task-types-enum"]
+
+ def _item_initalization(self):
+ self.multiselection = True
+ self.value_on_not_set = []
+ self.enum_items = []
+ self.valid_keys = set()
+ self.valid_value_types = (list, )
+ self.placeholder = None
+
+ def _get_enum_values(self):
+ from ..lib import get_default_anatomy_settings
+ anatomy_settings = get_default_anatomy_settings()
+
+ valid_keys = set()
+ enum_items = []
+ for task_type, _task_attr in anatomy_settings["tasks"].items():
+ enum_items.append({task_type: task_type})
+ valid_keys.add(task_type)
+
+ return enum_items, valid_keys
+
+ def set_override_state(self, *args, **kwargs):
+ super(TaskTypeEnumEntity, self).set_override_state(*args, **kwargs)
+
+ self.enum_items, self.valid_keys = self._get_enum_values()
+ new_value = []
+ for key in self._current_value:
+ if key in self.valid_keys:
+ new_value.append(key)
+ self._current_value = new_value
+
+
 class ProvidersEnum(BaseEnumEntity):
  schema_types = ["providers-enum"]