跳到主要内容

Flask 用于 API 的方法视图

Web API 的工作通常与 HTTP 动词紧密相关,所以这使得实现这样一个基于 ​MethodView​ 类的 API 很有意义。也就是说,你会注意到 大多数时候, API 需要不同的 URL 规则来访问相同的方法视图。譬如,想象一种 情况,你在 web 上暴露一个用户对象:

URLHTTP 方法描述
/users/GET获得全部用户的列表
/users/POST创建一个新用户
/users/<id>GET显示某个用户
/users/<id>PUT更新某个用户
/users/<id>DELETE删除某个用户

那么,你会想用 ​MethodView​ 做什么?诀窍是利用你可以 对相同的视图提供多个规则的事实。

让我们假设这时视图看起来是这个样子:

class UserAPI(MethodView):

def get(self, user_id):
if user_id is None:
# return a list of users
pass
else:
# expose a single user
pass

def post(self):
# create a new user
pass

def delete(self, user_id):
# delete a single user
pass

def put(self, user_id):
# update a single user
pass

如此,我们怎样把它挂载到路由系统中?添加两条规则,并且为每条规则显式地 指出 HTTP 方法:

user_view = UserAPI.as_view('user_api')
app.add_url_rule('/users/', defaults={'user_id': None},
view_func=user_view, methods=['GET',])
app.add_url_rule('/users/', view_func=user_view, methods=['POST',])
app.add_url_rule('/users/<int:user_id>', view_func=user_view,
methods=['GET', 'PUT', 'DELETE'])

如果你有许多看起来类似的 API ,你可以重构上述的注册代码:

def register_api(view, endpoint, url, pk='id', pk_type='int'):
view_func = view.as_view(endpoint)
app.add_url_rule(url, defaults={pk: None},
view_func=view_func, methods=['GET',])
app.add_url_rule(url, view_func=view_func, methods=['POST',])
app.add_url_rule('%s<%s:%s>' % (url, pk_type, pk), view_func=view_func,
methods=['GET', 'PUT', 'DELETE'])

register_api(UserAPI, 'user_api', '/users/', pk='user_id')