我正在使用带有烧瓶休息和webargs的Flask(使用棉花糖作为后端)。 目前,我可以用这个拉入我想要的字段:
class AddGroup(Resource):
args = {
'name': fields.Str(missing=None),
'phone': fields.Str(missing=None),
}
@use_args(args)
def get(self, args):
name = args['name'].strip()
# ... some GET-related code ...
@use_args(args)
def post(self, args):
name = args['name'].strip()
# ... some POST-related code ...
目前为止,一切都好。 但我真正想做的是确保args['name']
进入各种方法("post"、"get 等")并且已经去除了空格,所以我不必每次都手动处理每个变量。 (剥离空格只是一个例子 - 它可能是其他一些简单或复杂的转换)
有没有一种方法(通过覆盖 String 字段、定义我自己的字段或其他方式)允许在args
到达类方法之前对其进行预处理?
webargs
正在使用marshmallow
来制作架构,因此您可以使用pre_load
或post_load
。棉花糖文档中有"slugify"字符串的例子:
from marshmallow import Schema, fields, pre_load
class UserSchema(Schema):
name = fields.Str()
slug = fields.Str()
@pre_load
def slugify_name(self, in_data, **kwargs):
in_data['slug'] = in_data['slug'].lower().strip().replace(' ', '-')
return in_data
schema = UserSchema()
result = schema.load({'name': 'Steve', 'slug': 'Steve Loria '})
result['slug'] # => 'steve-loria'
您可以在此处查看详细文档:https://marshmallow.readthedocs.io/en/latest/extending.html#pre-processing-and-post-processing-methods
以下自定义字段类(作为组合):
from marshmallow import fields
class Trim(fields.Field):
def __init__(self, inner, *args, **kwargs):
self.inner = inner
super().__init__(*args, **kwargs)
def _bind_to_schema(self, field_name, parent):
super()._bind_to_schema(field_name, parent)
self.inner._bind_to_schema(field_name, parent)
def _deserialize(self, value, *args, **kwargs):
value = value.strip()
return self.inner._deserialize(value, *args, **kwargs)
def _serialize(self, *args, **kwargs):
return self.inner._serialize(*args, **kwargs)
使用示例:
from marshmallow import Schema, fields
class SomeSchema(Schema):
name = Trim(fields.String(), validate=validate.Length(max=100))
email = Trim(fields.Email(), required=True)
从这里开始的想法