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")})