def test_get_or_create_software(self): self.assertIsNone(Software.get_by(name="Maya")) software = files_service.get_or_create_software("Maya", "may", ".ma") self.assertIsNotNone(Software.get_by(name="Maya")) software_again = files_service.get_or_create_software( "Maya", "may", ".ma") self.assertEqual(software["id"], software_again["id"])
def get_arguments(self): person = persons_service.get_current_user() maxsoft = files_service.get_or_create_software("3ds Max", "max", ".max") parser = reqparse.RequestParser() parser.add_argument( "name", help="The asset name is required.", required=True ) parser.add_argument("description", default="") parser.add_argument("mode", default="working") parser.add_argument("comment", default="") parser.add_argument("person_id", default=person["id"]) parser.add_argument("software_id", default=maxsoft["id"]) parser.add_argument("revision", default=0, type=int) parser.add_argument("sep", default="/") args = parser.parse_args() return ( args["name"], args["mode"], args["description"], args["comment"], args["person_id"], args["software_id"], args["revision"], args["sep"], )
def get_arguments(self): maxsoft = files_service.get_or_create_software("3ds Max", "max", ".max") parser = reqparse.RequestParser() parser.add_argument("name", default="main") parser.add_argument("mode", default="working") parser.add_argument("software_id", default=maxsoft["id"]) parser.add_argument("comment", default="") parser.add_argument("revision", default=0) parser.add_argument("sep", default="/") args = parser.parse_args() return (args["name"], args["mode"], args["software_id"], args["comment"], args["revision"], args["sep"])
def get_folder_from_software(software, field="name"): if software is None: software = files_service.get_or_create_software( "3dsmax", "max", ".max") return software[field]