使用Flask-marshmallow和烧瓶放置复杂的物体

时间:2015-01-22 20:02:56

标签: python json rest flask

我正在开发API并使用烧瓶 - 奶油和棉花糖来学习烧瓶,我想知道是否可以一次发布和序列化复杂对象列表。

这是我目前的代码:

class Post(db.Document):
    created_at = db.DateTimeField(default=datetime.now, required=True)
    title = db.StringField(max_length=255, required=True)
    slug = db.StringField(max_length=255, required=True)
    body = db.StringField(required=True)
    comments = db.ListField(db.EmbeddedDocumentField('Comment'))

    meta = {
        'allow_inheritance': True,
        'indexes': ['-created_at', 'slug'],
        'ordering': ['-created_at']
    }


class Comment(db.EmbeddedDocument):
    created_at = db.DateTimeField(default=datetime.now, required=True)
    body = db.StringField(verbose_name="Comment", required=True)
    author = db.StringField(verbose_name="Name", max_length=255, required=True)


class PostSerializer(Serializer):
    id = fields.String()
    class Meta:
        additional = ("created_at", "title", "slug", "body", "comments")


# just a test for multiple data
class PostViewList(Resource):
    def __init__(self):
        self.reqparse = reqparse.RequestParser()
        self.reqparse.add_argument('posts', type = str, action='append',
                                   required = True, help = 'No posts given', location='json')

    def post(self):
        args = self.reqparse.parse_args()

        serializer = PostSerializer(many=True)
        result = serializer.load(args['posts'])

        return args["posts"], 201

api.add_resource(PostViewList, '/posts')

现在的方式,我在“serializer.load”上收到错误,因为帖子不被视为对象,而是字符串。如果我不尝试序列化并返回“args ['posts']”它会显示我发布的整个字符串。

我尝试发布以下Json:

{"posts":[ { "title":"asd", "slug": "asd", "body" : "asd" }, { "title":"qwe", "slug": "qwe", "body" : "qwe" }]}

现在的方式,我还没有发送评论列表,因为我甚至无法处理我没有尝试过的过程。

1 个答案:

答案 0 :(得分:2)

好的,过了一段时间我才发现它。

我已经在PostSerializer上实现了make_object,并且也为注释声明了一个序列化器,在帖子中嵌套了很多:

class CommentsSerializer(Serializer):
    class Meta:
        fields = ("created_at", "body", "author")

class PostSerializer(Serializer):
    id = fields.String()
    title = fields.String(required=True)
    comments = fields.Nested(CommentsSerializer, many=True)

    def make_object(self, data):
        return Post(**data)

    class Meta:
        additional = ("slug", "body", "created_at")

然后,在post方法中,只需从烧瓶中获取json,验证并序列化所有内容

class PostViewList(Resource):
    def post(self):
        if not request.get_json():
            return bad_request('No input data provided')    
        content_input = request.get_json().get("posts")    
        serializer = PostSerializer(many=True)
        errors = serializer.validate(content_input)
        if errors:
            return jsonify(errors), 400    
        result = serializer.load(content_input)    
        r = Post.objects.insert(result.data)
        return PostSerializer(r, many=True).data, 201