def init_data():
projects_service.get_open_status()
projects_service.get_closed_status()
print("Project status initialized.")
assets_service.get_or_create_type("Characters")
assets_service.get_or_create_type("Props")
assets_service.get_or_create_type("Environment")
assets_service.get_or_create_type("FX")
assets_service.get_or_create_type("Camera")
print("Asset types initialized.")
shots_service.get_episode_type()
shots_service.get_sequence_type()
shots_service.get_shot_type()
print("Shot types initialized.")
modeling = tasks_service.get_or_create_department("Modeling")
animation = tasks_service.get_or_create_department("Animation")
fx = tasks_service.get_or_create_department("FX")
compositing = tasks_service.get_or_create_department("Compositing")
concept = tasks_service.get_or_create_department("Concept")
layout = tasks_service.get_or_create_department("Layout")
tasks_service.get_or_create_task_type(concept, "Concept", "#8D6E63", 1)
tasks_service.get_or_create_task_type(modeling, "Texture", "#64B5F6", 2)
tasks_service.get_or_create_task_type(modeling, "Modeling", "#78909C", 3)
tasks_service.get_or_create_task_type(animation, "Setup", "#9CCC65", 4)
tasks_service.get_or_create_task_type(concept, "Storyboard", "#43A047", 1,
True)
tasks_service.get_or_create_task_type(layout, "Layout", "#7CB342", 2, True)
tasks_service.get_or_create_task_type(animation, "Animation", "#009688", 3,
True)
tasks_service.get_or_create_task_type(compositing, "Lighting", "#F9A825",
4, True)
tasks_service.get_or_create_task_type(fx, "FX", "#26C6DA", 5, True)
tasks_service.get_or_create_task_type(compositing, "Render", "#F06292", 6,
True)
tasks_service.get_or_create_task_type(compositing, "Compositing",
"#ff5252", 7, True)
print("Task types initialized.")
tasks_service.get_or_create_status("Todo", "todo", "#f5f5f5")
tasks_service.get_or_create_status("Work In Progress", "wip", "#3273dc")
tasks_service.get_or_create_status("Waiting For Approval",
"wfa",
"#ab26ff",
is_reviewable=True)
tasks_service.get_or_create_status("Retake",
"retake",
"#ff3860",
is_reviewable=True)
tasks_service.get_or_create_status("Done", "done", "#22d160", is_done=True)
print("Task status initialized.")
def test_get_todos(self):
task_id = self.task.id
shot_task_id = self.shot_task.id
path = "data/user/tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 0)
self.assign_user(task_id)
self.assign_user(shot_task_id)
path = "data/user/tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 2)
tasks_service.update_task(
shot_task_id,
{
"task_status_id":
tasks_service.get_or_create_status(
"Done", "done", "#22d160", is_done=True)["id"]
},
)
path = "data/user/tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 1)
def setUp(self):
super(TaskRoutesTestCase, self).setUp()
self.generate_fixture_project_status()
self.generate_fixture_project()
self.generate_fixture_asset_type()
self.generate_fixture_asset()
self.generate_fixture_sequence()
self.generate_fixture_shot()
self.generate_fixture_department()
self.generate_fixture_task_type()
self.generate_fixture_task_status()
self.generate_fixture_person()
self.generate_fixture_assigner()
self.generate_fixture_task_status_wip()
self.generate_fixture_task_status_retake()
self.generate_fixture_task_status_done()
self.todo_status = tasks_service.get_or_create_status("Todo")
self.asset_id = str(self.asset.id)
self.shot_id = str(self.shot.id)
self.task_type_id = str(self.task_type.id)
self.project_id = str(self.project.id)
self.wip_status_id = self.task_status_wip.id
self.retake_status_id = self.task_status_retake.id
self.done_status_id = self.task_status_done.id
self.person_id = self.person.id
def test_status_automation_to_status(self):
wip_status = tasks_service.get_or_create_status(
"Work In Progress", "wip", "#3273dc")
comments_service.create_comment(
self.person.id,
self.task_concept.id,
str(self.task_status_done.id),
"Test",
[],
{},
None,
)
self.assertEqual(str(self.task_modeling.task_status_id),
wip_status["id"])
def test_get_done_tasks_for_person(self):
self.generate_fixture_task()
self.task_id = self.task.id
tasks = self.get("/data/persons/%s/done-tasks" % self.person.id)
self.assertEqual(len(tasks), 0)
done_status = tasks_service.get_or_create_status("Done",
"done",
"#22d160",
is_done=True)
tasks_service.update_task(self.task_id,
{"task_status_id": done_status["id"]})
tasks = self.get("/data/persons/%s/done-tasks" % self.person.id)
self.assertEqual(len(tasks), 1)
def setUp(self):
super(RouteCreateTasksTestCase, self).setUp()
self.generate_fixture_project_status()
self.generate_fixture_project()
self.generate_fixture_entity_type()
self.generate_fixture_entity()
self.generate_fixture_sequence()
self.generate_fixture_shot()
self.generate_fixture_department()
self.generate_fixture_task_type()
self.generate_fixture_task_status()
self.generate_fixture_person()
self.generate_fixture_assigner()
self.todo_status = tasks_service.get_or_create_status("Todo")
self.entity_id = str(self.entity.id)
self.shot_id = str(self.shot.id)
self.task_type_id = str(self.task_type.id)
def test_get_done_tasks(self):
task_id = self.task.id
path = "data/user/done-tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 0)
self.assign_user(task_id)
path = "data/user/done-tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 0)
done_status = tasks_service.get_or_create_status("Done",
"done",
"#22d160",
is_done=True)
tasks_service.update_task(task_id,
{"task_status_id": done_status["id"]})
path = "data/user/done-tasks/"
tasks = self.get(path)
self.assertEqual(len(tasks), 1)
def test_get_status(self):
task_status = tasks_service.get_or_create_status("WIP", "wip")
self.assertEqual(task_status["name"], "WIP")
def init_data():
"""
Put the minimum required data into the database to start with it.
"""
projects_service.get_open_status()
projects_service.get_closed_status()
print("Project status initialized.")
assets_service.get_or_create_asset_type("Characters")
assets_service.get_or_create_asset_type("Props")
assets_service.get_or_create_asset_type("Environment")
assets_service.get_or_create_asset_type("FX")
print("Asset types initialized.")
shots_service.get_episode_type()
shots_service.get_sequence_type()
shots_service.get_shot_type()
print("Shot types initialized.")
modeling = tasks_service.get_or_create_department("Modeling")
animation = tasks_service.get_or_create_department("Animation")
fx = tasks_service.get_or_create_department("FX")
compositing = tasks_service.get_or_create_department("Compositing")
concept = tasks_service.get_or_create_department("Concept")
layout = tasks_service.get_or_create_department("Layout")
tasks_service.get_or_create_task_type(concept, "Concept", "#8D6E63", 1)
tasks_service.get_or_create_task_type(modeling, "Modeling", "#78909C", 2)
tasks_service.get_or_create_task_type(modeling, "Shading", "#64B5F6", 3)
tasks_service.get_or_create_task_type(animation, "Rigging", "#9CCC65", 4)
tasks_service.get_or_create_task_type(
concept,
"Storyboard",
"#43A047",
priority=1,
for_shots=True,
for_entity="Shot",
)
tasks_service.get_or_create_task_type(
layout,
"Layout",
"#7CB342",
priority=2,
for_shots=True,
for_entity="Shot",
)
tasks_service.get_or_create_task_type(
animation,
"Animation",
"#009688",
priority=3,
for_shots=True,
for_entity="Shot",
)
tasks_service.get_or_create_task_type(
compositing,
"Lighting",
"#F9A825",
priority=4,
for_shots=True,
for_entity="Shot",
)
tasks_service.get_or_create_task_type(fx,
"FX",
"#26C6DA",
priority=5,
for_shots=True,
for_entity="Shot")
tasks_service.get_or_create_task_type(
compositing,
"Rendering",
"#F06292",
priority=6,
for_shots=True,
for_entity="Shot",
)
tasks_service.get_or_create_task_type(
compositing,
"Compositing",
"#ff5252",
priority=7,
for_shots=True,
for_entity="Shot",
)
print("Task types initialized.")
tasks_service.get_or_create_status("Todo", "todo", "#f5f5f5")
tasks_service.get_or_create_status("Work In Progress", "wip", "#3273dc")
tasks_service.get_or_create_status("Waiting For Approval", "wfa",
"#ab26ff")
tasks_service.get_or_create_status("Retake",
"retake",
"#ff3860",
is_retake=True)
tasks_service.get_or_create_status("Done", "done", "#22d160", is_done=True)
print("Task status initialized.")
