def test_get_output_folder_path(self):
path = file_tree_service.get_output_folder_path(
self.shot.serialize(),
output_type=self.output_type_cache,
task_type=self.task_type_animation.serialize(),
sep="/")
self.assertEqual(
path, "/simple/productions/export/cosmos_landromat/shots/s01/p01/"
"animation/cache")
def add_path_info(
self,
output_file,
mode,
entity,
output_type,
task_type=None,
name="main",
extension="",
representation="",
separator="/"
):
folder_path = file_tree_service.get_output_folder_path(
entity,
mode=mode,
output_type=output_type,
task_type=task_type,
revision=output_file["revision"],
representation=representation,
name=name,
sep=separator
)
file_name = file_tree_service.get_output_file_name(
entity,
mode=mode,
revision=output_file["revision"],
output_type=output_type,
task_type=task_type,
name=name
)
output_file = files_service.update_output_file(
output_file["id"],
{
"path": "%s%s%s%s" % (
folder_path,
separator,
file_name,
extension
)
}
)
output_file.update({
"folder_path": folder_path,
"file_name": file_name
})
return output_file
def post(self, entity_id):
args = self.get_arguments()
try:
entity = entities_service.get_entity(entity_id)
user_service.check_project_access(entity["project_id"])
user_service.check_entity_access(entity_id)
output_type = files_service.get_output_type(args["output_type_id"])
task_type = tasks_service.get_task_type(args["task_type_id"])
entity = entities_service.get_entity(entity_id)
is_revision_set_by_user = args["revision"] != 0
if not is_revision_set_by_user:
revision = files_service.get_next_output_file_revision(
entity_id, args["name"])
else:
revision = args["revision"]
folder_path = file_tree_service.get_output_folder_path(
entity,
mode=args["mode"],
output_type=output_type,
task_type=task_type,
name=args["name"],
representation=args["representation"],
sep=args["separator"],
revision=args["revision"],
)
file_name = file_tree_service.get_output_file_name(
entity,
mode=args["mode"],
revision=revision,
output_type=output_type,
task_type=task_type,
name=args["name"],
)
except MalformedFileTreeException as exception:
return (
{
"message": str(exception),
"received_data": request.json
},
400,
)
return {"folder_path": folder_path, "file_name": file_name}, 200
def post(self, entity_id):
args = self.get_arguments()
try:
entity = entities_service.get_entity(entity_id)
if not permissions.has_manager_permissions():
user_service.check_has_task_related(entity["project_id"])
output_type = files_service.get_output_type(args["output_type_id"])
task_type = tasks_service.get_task_type(args["task_type_id"])
entity = entities_service.get_entity(entity_id)
is_revision_set_by_user = args["revision"] != 0
if not is_revision_set_by_user:
revision = files_service.get_next_output_file_revision(
entity_id,
args["name"]
)
else:
revision = args["revision"]
folder_path = file_tree_service.get_output_folder_path(
entity,
mode=args["mode"],
output_type=output_type,
task_type=task_type,
name=args["name"],
representation=args["representation"],
sep=args["separator"],
revision=args["revision"]
)
file_name = file_tree_service.get_output_file_name(
entity,
mode=args["mode"],
revision=revision,
output_type=output_type,
task_type=task_type,
name=args["name"]
)
except MalformedFileTreeException:
return {
"error":
"Tree is not properly written, check modes and variables",
"received_data": request.json,
}, 400
return {"folder_path": folder_path, "file_name": file_name}, 200
