Flask-Restful 学习总结笔记

Flask-Restful

一、安装：

pip install flask-restful

完成安装后就可以引入常用类或模块

from flask_restful import Api, Resource， reqparse, abort

二、案例：

from flask import Flask, request
from flask_restful import Resource, Api

app = Flask(__name__)
api = Api(app)

# 定义一个全局字典
USER_LIST = {
    '1': {'name': 'Tom'},
    '2': {'name': 'Candy'}
}


class HelloWorld(Resource):

    def get(self):
        return USER_LIST

    def post(self):
        # 添加数据
        user_id = int(max(USER_LIST.keys())) + 1
        # 将 int 类型 user_id 转为 str
        user_id = '%i' % user_id
        USER_LIST[user_id] = {'name': request.form['name']}
        return USER_LIST


api.add_resource(HelloWorld, '/users')

if __name__ == '__main__':
    app.run(debug=True)

从案例可以看出，Restful 扩展通过 api.add_resource() 方法来添加路由，其中方法的第一个参数是视图类名（该类继承自Resource类），其函数成员函数定义了不同的 HTTP 请求方法的逻辑；第二个参数定义了 URL 路径。

运行上面案例，可以通过 http://127.0.0.1:5000/users 来访问，GET 请求时会显示全局变量 USER_LIST 中的内容，POST 请求时会在USER_LIST 中添加一项，并返回刚添加的项。如果在POST 请求中找不到 name 字段，则返回400 Bad Request 错误。由于类UserList 没有定义put 和delete 函数，所以在PUT 或DELETE 请求时会返回405 Method Not Allowed 错误。

另外，路由支持多路径：

api.add_resource(HelloWorld, ‘/userlist’, ‘/users’)

这样访问 http://127.0.0.1:5000/userlist 和 http://127.0.0.1:5000/users 效果是一样的。

三、带参数的请求

class HelloWorld(Resource):

    def get(self, user_id):
        return USER_LIST[user_id]

    def put(self, user_id):
        USER_LIST[user_id] = {'name': request.form['name']}
        return USER_LIST[user_id]

    def delete(self, user_id):
        del USER_LIST[user_id]
        return USER_LIST

api.add_resource(HelloWorld, '/users/<user_id>', '/userlist')

四、参数解析

在 POST 和 PUT 请求中，直接访问 form 表单并验证的工作有些麻烦，Flask-RESTful 提供了 reqparse 库来简化：

parse = reqparse.RequestParser()
parse.add_argument('name', type=str)
class HelloWorld(Resource):
	def put(self, user_id):
        # 使用 strict=True 调用 parse_args 能够确保当请求包含你的解析器中未定义的参数的时候会抛出一个异常。
        args = parse.parse_args(strict=True)
        USER_LIST[user_id] = {'name': args['name']}
        return USER_LIST[user_id]
api.add_resource(HelloWorld, '/users/<user_id>', '/userlist')

五、类视图方法返回值

Flask-RESTful 支持视图方法多种类型的返回值。同 Flask 一样，你可以返回任一迭代器，它将会被转换成一个包含原始 Flask 响应对象的响应。Flask-RESTful 也支持使用多个返回值来设置响应代码和响应头，如下所示:

class Todo1(Resource):
    def get(self):
        # Default to 200 OK
        return {'task': 'Hello world'}

class Todo2(Resource):
    def get(self):
        # Set the response code to 201
        return {'task': 'Hello world'}, 201

class Todo3(Resource):
    def get(self):
        # Set the response code to 201 and return custom headers
        return {'task': 'Hello world'}, 201, {'Etag': 'some-opaque-string'}

六、完整案例

from flask import Flask
from flask_restful import Resource, Api, reqparse, abort

app = Flask(__name__)
api = Api(app)

TODOS = {
    'todo1': {'task': 'Build an Api'},
    'todo2': {'task': '??????'},
    'todo3': {'task': 'profit!'}
}


def abort_if_todo_doesnot_exists(todo_id):
    if todo_id not in TODOS:
        abort(404, message='todo {} does not exists.'.format(todo_id))


parse = reqparse.RequestParser()
parse.add_argument('task', type=str)


class Todo(Resource):
    def get(self, todo_id):
        abort_if_todo_doesnot_exists(todo_id)
        return TODOS[todo_id]

    def delete(self, todo_id):
        abort_if_todo_doesnot_exists(todo_id)
        del TODOS[todo_id]
        return TODOS

    def put(self, todo_id):
        args = parse.parse_args()
        task = {'task': args['task']}
        TODOS[todo_id] = task
        return task, 201


class TodoList(Resource):
    def get(self):
        return TODOS

    def post(self):
        args = parse.parse_args()
        todo_id = int(max(TODOS.keys()).lstrip('todo')) + 1
        todo_id = "todo%i" % todo_id
        TODOS[todo_id] = {'task': args['task']}
        return TODOS[todo_id]


api.add_resource(TodoList, '/todos')
api.add_resource(Todo, '/todos/<todo_id>')

if __name__ == '__main__':
    app.run(debug=True)