def get_context():
if permissions.has_admin_permissions():
projects = projects_service.open_projects()
else:
projects = get_open_projects()
asset_types = assets_service.get_asset_types()
custom_actions = custom_actions_service.get_custom_actions()
status_automations = status_automations_service.get_status_automations()
persons = persons_service.get_persons(
minimal=not permissions.has_manager_permissions()
)
notification_count = get_unread_notifications_count()
project_status_list = projects_service.get_project_statuses()
departments = tasks_service.get_departments()
task_types = tasks_service.get_task_types()
task_status_list = tasks_service.get_task_statuses()
search_filters = get_filters()
return {
"asset_types": asset_types,
"custom_actions": custom_actions,
"status_automations": status_automations,
"departments": departments,
"notification_count": notification_count,
"persons": persons,
"project_status": project_status_list,
"projects": projects,
"task_types": task_types,
"task_status": task_status_list,
"search_filters": search_filters,
}
def get_context():
if permissions.has_admin_permissions():
projects = projects_service.open_projects()
else:
projects = get_open_projects()
asset_types = assets_service.get_asset_types()
custom_actions = custom_actions_service.get_custom_actions()
persons = persons_service.get_persons()
notifications = get_last_notifications()
project_status_list = projects_service.get_project_statuses()
task_types = tasks_service.get_task_types()
task_status_list = tasks_service.get_task_statuses()
search_filters = get_filters()
return {
"asset_types": asset_types,
"custom_actions": custom_actions,
"notifications": notifications,
"persons": persons,
"project_status": project_status_list,
"projects": projects,
"task_types": task_types,
"task_status": task_status_list,
"search_filters": search_filters,
}
def prepare_import(self, project_id):
self.episodes = {}
self.entity_types = {}
self.descriptor_fields = self.get_descriptor_field_map(
project_id, "Asset")
project = projects_service.get_project(project_id)
self.is_tv_show = projects_service.is_tv_show(project)
if self.is_tv_show:
episodes = shots_service.get_episodes_for_project(project_id)
self.episodes = {
episode["name"]: episode["id"]
for episode in episodes
}
self.created_assets = []
self.task_types_in_project_for_assets = (
TaskType.query.join(ProjectTaskTypeLink).filter(
ProjectTaskTypeLink.project_id == project_id).filter(
TaskType.for_entity == "Asset"))
self.task_statuses = {
status["id"]: [status[n] for n in ("name", "short_name")]
for status in tasks_service.get_task_statuses()
}
self.current_user_id = get_current_user()["id"]
self.task_types_for_ready_for_map = {
task_type.name: str(task_type.id)
for task_type in TaskType.query.join(ProjectTaskTypeLink).filter(
ProjectTaskTypeLink.project_id == project_id).filter(
TaskType.for_entity == "Shot").all()
}
def patch_set_done_flag_on_task_status():
"""
Patch to run after upgrade from 0.4.8 or lower to 0.4.9 or superior.
"""
for task_status in tasks_service.get_task_statuses():
tasks_service.update_task_status(
task_status["id"],
{"is_done": task_status["short_name"] == "done"})
def prepare_import(self, project_id):
self.episodes = {}
self.sequences = {}
self.descriptor_fields = self.get_descriptor_field_map(
project_id, "Shot"
)
project = projects_service.get_project(project_id)
self.is_tv_show = projects_service.is_tv_show(project)
self.created_shots = []
self.task_types_in_project_for_shots = (
TaskType.query.join(ProjectTaskTypeLink)
.filter(ProjectTaskTypeLink.project_id == project_id)
.filter(TaskType.for_entity == "Shot")
)
self.task_statuses = {
status["id"]: [status[n] for n in ("name", "short_name")]
for status in get_task_statuses()
}
self.current_user_id = get_current_user()["id"]
def patch_set_done_flag_on_task_status():
for task_status in tasks_service.get_task_statuses():
tasks_service.update_task_status(
task_status["id"],
{"is_done": task_status["short_name"] == "done"})