Flask - restful输出的基本使用

准备

• 创建一个friend表

class Friend(db.Model):
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    uid = db.Column(db.Integer, db.ForeignKey('user.id'))
    fid = db.Column(db.Integer, db.ForeignKey('user.id'))

• 用户和用户的朋友都在一张表格里面找

查找用户的朋友

class FriendResource(Resource):
    def get(self, id):
        friends = Friend.query.filter(Friend.uid == id).all()
        user = User.query.get(id)

        friend_list = []
        for friend in friends:
            user = User.query.get(friend.fid)
            friend_list.append(user)

        # 要返回的数据
        data = {
            'user' : user.username,
            'num': len(friends),
            'friends': friend_list
        }
        return data

• 这里User对象不是json可序列化的, 所以会报错 TypeError: Object of type User is not JSON serializable

marshal()

• marshal() 函数可以创建一个套叠结构.
• 使用marshal() 函数, 需要传递两个参数, 第一个是数据, 第二个是结构, 通过marshal()自动将数据转换为套叠结构, 并以定义的结构展示

data = {
	'user' : user.username,
	'num': len(friends),
	'friends': marshal(friend_list, user_fields)
}

第二种方式

• 使用marshal_with()
• 朋友列表是一个套叠形式展现的, 可以使用List(列表)形式展现.
  • 'friends': fields.List()
• 如果直接返回不能有自定义的对象 , 如User.
• 列表里面是很多user, 可以使用fields.Nested()指定用什么格式展现列表
  • 'friends': fields.List(fields.Nested(user_fields))