def generate_fixture_project_standard(self):
self.project_standard = Project.create(
name="Big Buck Bunny", project_status_id=self.open_status.id
)
self.project_standard.update(
{"file_tree": file_tree_service.get_tree_from_file("default")}
)
def generate_fixture_project(self, name="Cosmos Landromat"):
self.project = Project.create(name=name,
project_status_id=self.open_status.id)
self.project_id = self.project.id
self.project.update(
{"file_tree": file_tree_service.get_tree_from_file("simple")})
return self.project
def generate_fixture_project_no_preview_tree(self):
self.project_no_preview_tree = Project.create(
name="Agent 327", project_status_id=self.open_status.id
)
self.project_no_preview_tree.update(
{"file_tree": file_tree_service.get_tree_from_file("no_preview")}
)
def test_get_folder_path_template(self):
tree = file_tree_service.get_tree_from_file("simple")
template = file_tree_service.get_folder_path_template(
tree, "working", self.shot.serialize())
self.assertEqual(template, tree["working"]["folder_path"]["shot"])
template = file_tree_service.get_folder_path_template(
tree, "working", self.asset.serialize())
self.assertEqual(template, tree["working"]["folder_path"]["asset"])
template = file_tree_service.get_folder_path_template(
tree, "working", self.sequence.serialize())
self.assertEqual(template, tree["working"]["folder_path"]["sequence"])
def test_get_file_name_template(self):
tree = file_tree_service.get_tree_from_file("default")
template = file_tree_service.get_file_name_template(
tree, "working", self.shot.serialize())
self.assertEqual(template, tree["working"]["file_name"]["shot"])
template = file_tree_service.get_file_name_template(
tree, "working", self.asset.serialize())
self.assertEqual(template, tree["working"]["file_name"]["asset"])
template = file_tree_service.get_file_name_template(
tree, "working", self.sequence.serialize())
self.assertEqual(template, tree["working"]["file_name"]["sequence"])
def post(self, project_id):
tree_name = self.get_arguments()
try:
user_service.check_project_access(project_id)
tree = file_tree_service.get_tree_from_file(tree_name)
project = projects_service.update_project(project_id,
{"file_tree": tree})
except WrongFileTreeFileException:
abort(400, "Selected tree is not available")
return project
def test_import_projects(self):
self.projects = self.load_fixture("projects")
self.assertEqual(len(self.projects), 2)
self.projects = self.get("data/projects")
self.assertEqual(len(self.projects), 2)
self.project_status = self.get("data/project-status")
self.project_status = [
x for x in self.project_status if x["name"] == "Active"
]
active_id = self.project_status[0]["id"]
self.assertEqual(self.projects[0]["project_status_id"], active_id)
tree = file_tree_service.get_tree_from_file(DEFAULT_FILE_TREE)
self.assertDictEqual(self.projects[0]["file_tree"], tree)
def import_entry(self, data):
project = Project.get_by(shotgun_id=data["shotgun_id"])
if project is None:
project = Project(**data)
tree_name = current_app.config["DEFAULT_FILE_TREE"]
project.file_tree = file_tree_service.get_tree_from_file(tree_name)
project.save()
current_app.logger.info("Project created: %s" % project)
else:
project.update(data)
current_app.logger.info("Project updated: %s" % project)
return project
def import_entry(self, data):
project = Project.get_by(shotgun_id=data["shotgun_id"])
if project is None:
project = Project(**data)
tree_name = current_app.config["DEFAULT_FILE_TREE"]
try:
project.file_tree = file_tree_service.get_tree_from_file(
tree_name)
except WrongFileTreeFileException:
current_app.logger.error(
"Can't find default file to set project file tree. Set "
"an empty file tree instead.")
project.file_tree = {}
project.save()
current_app.logger.info("Project created: %s" % project)
else:
project.update(data)
current_app.logger.info("Project updated: %s" % project)
return project
def test_get_root_path(self):
tree = file_tree_service.get_tree_from_file("simple")
path = file_tree_service.get_root_path(tree, "working", "/")
self.assertEqual(path, "/simple/productions/")
def test_get_tree_from_file(self):
simple_tree = file_tree_service.get_tree_from_file("simple")
self.assertIsNotNone(simple_tree["working"])
def generate_fixture_project(self):
self.project = Project(name="Cosmos Landromat",
project_status_id=self.open_status.id)
self.project.save()
self.project.update(
{"file_tree": file_tree_service.get_tree_from_file("simple")})
