Flask-Restplus：如何建模字符串或对象？

Question

在 Flask-Restplus 中，我需要对属性值建模，该属性值可能字符串列表或对象列表。

就是这样：

{
    'my_attribute': [
         'value1',
          'value2'
     ]
}

或者它可以如下所示：

{
    'my_attribute': [
        {
             'name': 'value1',
              'foo': 'something'
         },
         {
              'name': 'value2',
               'foo': 'something else'
          }
     ]
}

我应该如何在 Flask-Restplus 中对其进行建模 api.model？

Answer 1

我刚刚自己弄明白了。简而言之，创建一个发出自己的 JSON 模式的自定义字段 class。反过来，模式使用 oneOf 类型来指定这是一个字符串还是一个对象。

from flask_restplus import fields

element_object = api.model('Element_Object', {
    'name': fields.String(),
    'foo': fields.String()
})

class StringOrObjectElement(fields.Nested):
    __schema_type__ = ['string','object']

    def output(self, key, obj):
        if isinstance(obj, str):
            if key == 'name':
                return obj
            else:
                return 'default_value'
        return super().output(key, obj)

    def schema(self):
        schema_dict = super().schema()
        schema_dict.pop('type')
        nested_ref = schema_dict.pop('$ref')
        schema_dict['oneOf'] = [
            {
                'type': 'string'
            },
            {
                '$ref': nested_ref
            }
        ]
        return schema_dict

root_object = api.model('Root_Object', {
    'my_attribute': fields.List(fields.StringOrObjectElement(element_object))

Flask-Restplus：如何建模字符串或对象？

Flask-Restplus: how to model string or object?

python

flask

flask-restplus