def get_schedule_items(project_id):
"""
Return all schedule items for given project. If no schedule item exists
for a given task type, it creates one.
"""
task_types = tasks_service.get_task_types_for_project(project_id)
task_type_map = base_service.get_model_map_from_array(task_types)
schedule_items = set(
ScheduleItem.query.filter_by(project_id=project_id).all())
schedule_item_map = {
str(schedule_item.task_type_id): schedule_item
for schedule_item in schedule_items
}
new_schedule_items = set()
schedule_item_to_remove = set()
for schedule_item in schedule_items:
if schedule_item.task_type_id is not None:
if str(schedule_item.task_type_id) not in task_type_map:
schedule_item_to_remove.add(schedule_item)
for task_type in task_types:
if task_type["id"] not in schedule_item_map:
new_schedule_item = ScheduleItem.create(
project_id=project_id,
start_date=date.today(),
end_date=date.today() + timedelta(days=1),
task_type_id=task_type["id"])
new_schedule_items.add(new_schedule_item)
schedule_items = \
schedule_items.union(new_schedule_items) - schedule_item_to_remove
return sorted(
[schedule_item.present() for schedule_item in schedule_items],
key=lambda x: x["start_date"])
def generate_fixture_schedule_item(self,
task_type_id=None,
object_id=None):
if task_type_id is None:
task_type_id = self.task_type.id
self.schedule_item = ScheduleItem.create(
project_id=self.project.id,
task_type_id=self.task_type.id,
object_id=object_id)
return self.schedule_item.serialize()
def get_entity_schedule_items(
project_id, task_type_id, to_create, to_create_map, existing_schedule_items
):
schedule_item_map = {
str(schedule_item.object_id): schedule_item
for schedule_item in existing_schedule_items
}
new_schedule_items = set()
schedule_item_to_remove = set()
for schedule_item in existing_schedule_items:
if schedule_item.object_id is not None:
if str(schedule_item.object_id) not in to_create_map:
schedule_item_to_remove.add(schedule_item)
for entity in to_create:
if entity["id"] not in schedule_item_map:
new_schedule_item = ScheduleItem.create(
project_id=project_id,
start_date=date.today(),
end_date=date.today() + timedelta(days=1),
object_id=entity["id"],
task_type_id=task_type_id,
)
events.emit(
"schedule-item:new",
{"schedule_item_id": str(new_schedule_item.id)},
)
new_schedule_items.add(new_schedule_item)
schedule_items = (
existing_schedule_items.union(new_schedule_items)
- schedule_item_to_remove
)
results = []
for schedule_item in schedule_items:
result = schedule_item.present()
result["name"] = to_create_map[result["object_id"]]["name"]
results.append(result)
return sorted(results, key=lambda x: x["name"])
def setUp(self):
super(ScheduleItemTestCase, self).setUp()
self.generate_fixture_department()
self.generate_fixture_task_type()
self.generate_shot_suite()
ScheduleItem.create(project_id=self.project.id,
task_type_id=self.task_type.id,
start_date=date.today(),
end_date=date.today() + timedelta(days=1))
ScheduleItem.create(project_id=self.project.id,
task_type_id=self.task_type_animation.id,
start_date=date.today(),
end_date=date.today() + timedelta(days=1))
ScheduleItem.create(project_id=self.project.id,
task_type_id=self.task_type_layout.id,
start_date=date.today(),
end_date=date.today() + timedelta(days=1))