def generate_fixture_task_type(self):
self.task_type = TaskType(name="Shaders",
color="#FFFFFF",
department_id=self.department.id)
self.task_type.save()
self.task_type_animation = TaskType(
name="Animation",
color="#FFFFFF",
department_id=self.department_animation.id)
self.task_type_animation.save()
def test_get_task_dicts_for_entity_utf8(self):
start_date = fields.get_date_object("2017-02-20")
due_date = fields.get_date_object("2017-02-28")
real_start_date = fields.get_date_object("2017-02-22")
self.working_file.delete()
self.output_file.delete()
self.task.delete()
self.task_type = TaskType(name="Modélisation",
color="#FFFFFF",
department_id=self.department.id)
self.task_type.save()
self.task = Task(name="Première Tâche",
project_id=self.project.id,
task_type_id=self.task_type.id,
task_status_id=self.task_status.id,
entity_id=self.asset.id,
assignees=[self.person],
assigner_id=self.assigner.id,
duration=50,
estimation=40,
start_date=start_date,
due_date=due_date,
real_start_date=real_start_date)
self.task.save()
tasks = tasks_service.get_task_dicts_for_entity(self.asset.id)
self.assertEqual(len(tasks), 1)
self.assertEqual(tasks[0]["name"], u"Première Tâche")
self.assertEqual(tasks[0]["task_type_name"], u"Modélisation")
def save_task_type(self, department, data):
task_type = TaskType.get_by(shotgun_id=data["shotgun_id"])
data["department_id"] = department.id
if task_type is None:
task_type = TaskType.get_by(name=data["name"],
for_entity=data["for_entity"])
if task_type is None:
task_type = TaskType(**data)
task_type.save()
current_app.logger.info("Task Type created: %s" % task_type)
else:
existing_task_type = TaskType.get_by(
name=data["name"],
for_entity=data["for_entity"],
department_id=data["department_id"],
)
if existing_task_type is not None:
data.pop("name", None)
data.pop("for_entity", None)
data.pop("department_id", None)
task_type.update(data)
tasks_service.clear_task_type_cache(str(task_type.id))
current_app.logger.info("Task Type updated: %s" % task_type)
return task_type
def get_or_create_task_type(
department,
name,
color="#888888",
priority=1,
for_shots=False,
for_entity="Asset",
short_name="",
shotgun_id=None
):
"""
Create a new task type if it doesn't exist. If it exists, it returns the
type from database.
"""
task_type = TaskType.get_by(name=name)
if task_type is None:
task_type = TaskType(
name=name,
short_name=short_name,
department_id=department["id"],
color=color,
priority=priority,
for_shots=for_shots,
shotgun_id=shotgun_id
)
task_type.save()
return task_type.serialize()
def get_or_create_task_type(departmemt, name, color="#888888"):
task_type = TaskType.get_by(name=name)
if task_type is None:
task_type = TaskType(
name=name,
department_id=departmemt.id,
color=color
)
task_type.save()
return task_type
def get_or_create_task_type(department,
name,
color="#888888",
priority=1,
for_shots=False):
task_type = TaskType.get_by(name=name)
if task_type is None:
task_type = TaskType(name=name,
department_id=department["id"],
color=color,
priority=priority,
for_shots=for_shots)
task_type.save()
return task_type.serialize()
def save_task_type(self, department, data):
task_type = TaskType.get_by(shotgun_id=data["shotgun_id"])
data["department_id"] = department.id
if task_type is None:
task_type = TaskType.get_by(name=data["name"])
if task_type is None:
task_type = TaskType(**data)
task_type.save()
current_app.logger.info("Task Type created: %s" % task_type)
else:
task_type.update(data)
current_app.logger.info("Task Type updated: %s" % task_type)
return task_type
