Flask-restful,marshal_with +嵌套数据

时间:2015-06-04 22:16:21

标签: python flask flask-restful

我已经坚持了一段时间了。我的问题是我需要能够使用marshal_with并验证来自POST的嵌套字段。我的测试看起来像这样:

def test_user_can_apply_with_multiple_dogs(self):

    data = {
        # User data
        'registration_type': "guest",
        'first_name': 'Alex',
        'last_name': 'Daro',
        'phone_number': '805-910-9198',
        'street': '13950 NW Passage',
        'street2': '#208',
        'city': 'Marina del Rey',
        'state': 'CA',
        'zipcode': '90292',
        'photo': 'test_image.png',
        'how_did_you_hear': 0,
        #Dog data
        'pets': [
            {
                'dog_photo': "dog.png",
                'name': 'Genghis Khan',
                'breed': 'Shih Tzu',
                'age': 'Puppy',
                'size': 'Small',
            },
            {
                'dog_photo': "dog2.png",
                'name': 'Archibald',
                'breed': 'Great Dane',
                'age': 'Adult',
                'size': 'Extra Large',
            },
        ]
    }

    resp = self.client.post('/profile/registration', data=json.dumps(data))
    self.assertEqual(resp.status_code, 200)

我的端点类看起来像这样:

nested_fields = {
    'dog_photo': fields.String,
    'name': fields.String,
    'breed': fields.String,
    'age': fields.String, 
    'size': fields.String, 
}

profile_fields = {
    # 'user_email':fields.String,
    'token': fields.String,
    'registration_type': fields.String,
    'first_name': fields.String,
    'last_name': fields.String,
    'phone_number': fields.String,
    'street': fields.String,
    'street2': fields.String,
    'city': fields.String,
    'state': fields.String,
    'zipcode': fields.Integer,
    'photo': fields.String,
    'how_did_you_hear': fields.String,
    #Dog data
    'pets': fields.Nested(nested_fields)

}
class GuestProfile(Resource):
    @marshal_with(profile_fields)
    def post(self):
        # User data
        parser = reqparse.RequestParser()
        parser.add_argument('registration_type', type=str)
        parser.add_argument('first_name', type=str, required=True, help="First Name cannot be blank.")
        parser.add_argument('last_name', type=str, required=True, help="Last Name cannot be blank.")
        parser.add_argument('phone_number', type=str, required=True, help="Phone Number cannot be blank.")
        parser.add_argument('street', type=str, required=True, help="Street cannot be blank.")
        parser.add_argument('street2', type=str)
        parser.add_argument('city', type=str, required=True, help="City cannot be blank.")
        parser.add_argument('state', type=str, required=True, help="State cannot be blank.")
        parser.add_argument('zipcode', type=str, required=True, help="Zipcode cannot be blank.")
        parser.add_argument('photo', type=str, required=True, help="Please select a photo.")
        parser.add_argument('how_did_you_hear', type=str, required=True, help="How did you hear about us cannot be "
                                                                              "blank.")
        parser.add_argument('pets', type=str)
        kwargs = parser.parse_args()
        print kwargs, "KWWW" 

kwargs ['pet']总是以None的形式出现。有人有主意吗?

2 个答案:

答案 0 :(得分:2)

Here's an example in the docs of how to make a custom parser type.

基本上,您定义了一个函数:

  • 获取从请求中提取的参数的原始值
  • 如果解析或验证失败,则引发public ActionResult AdministrationPortal() { Person p_model = (Person)TempData["_person"]; return View(p_model); }
  • 如果验证超过
  • ,则返回参数

与您的问题相关的基本示例:

ValueError

正如你可能猜到的那样,这很快就会变得笨拙和讨厌。 Flask-RESTful中的请求解析不是为处理嵌套数据而设计的。它适用于查询字符串参数,标头,基本JSON等,并且可以配置为处理嵌套数据。但是,如果你要做很多事情,请在路上避免一些痛苦,并调查像Marshmallow这样的编组库。

答案 1 :(得分:1)

您错过了' Content-Type'标题以及action='append'为您的宠物参数here所述。

我在上面测试了您的代码,只需添加' Content-Type'和规定的行动,它适用于Python 2.7和3.

现在问题是它返回了一个字符串列表,因为你将pet参数指定为type=str。为了得到一个dicts列表,你必须编写一个自定义解析器类型(正如@Josh正确指出的那样)。见下面的例子:

实施例

def pet_parser(pet):
    # You can do other things here too as suggested in @Josh's repsonse
    return pet

在资源中:

parser.add_argument('pets', type=pet_parser, action='append')

并在您的测试功能中:

headers = [('Content-Type', 'application/json')]
self.client.post('url', data=data, headers=headers)