Merge pull request #1005 from mxatmx/0.2

feat: enable all entities for schedule display #1004

Author: frankrousseau

zou/app/blueprints/crud/task_type.py

@@ -1,6 +1,7 @@
 from flask_jwt_extended import jwt_required
 
 from zou.app.models.task_type import TaskType
+from zou.app.models.schedule_item import ScheduleItem
 from zou.app.services.exception import WrongParameterException
 from zou.app.services import tasks_service
 
@@ -317,6 +318,15 @@ def delete(self, instance_id):
         """
         return super().delete(instance_id)
 
+    def pre_delete(self, instance_dict):
+        """
+        Delete related ScheduleItems before deleting the TaskType.
+        This runs AFTER check_delete_permissions, preventing foreign key
+        constraint violations while maintaining security.
+        """
+        ScheduleItem.query.filter_by(task_type_id=instance_dict["id"]).delete()
+        return instance_dict
+
     def update_data(self, data, instance_id):
         data = super().update_data(data, instance_id)
         name = data.get("name", None)

zou/app/services/schedule_service.py

@@ -55,7 +55,7 @@ def get_task_types_schedule_items(project_id):
     task_types = [
         task_type
         for task_type in task_types
-        if task_type["for_entity"] in ["Asset", "Shot"]
+        if task_type["for_entity"] in ["Asset", "Shot", "Sequence", "Episode", "Edit"]
     ]
     task_type_map = base_service.get_model_map_from_array(task_types)
     schedule_items = set(