使用Flask构建简单的RESTful服务

我们现在的一个项目是使用Django来构建，说来也是基于技术扩展的考虑，我对于Django里面大而全的一些组件还是持有保守态度，所以虽然项目用了Django，但是对于很多组件的使用都是尽可能少用或者不用，这样一来虽然前期清苦些，但是从现在来看，由于依赖很低，我可以匹配很多种其他的方案。

当然对我来说我格外喜欢Django的ORM方案，这个我对比了下Flask方向的ORM方案SQLAlchemy,Django的感觉要更好一些。

而在RESTful的方向上，Django自身的第三方实现rest_framework也不错，现在看起来很简单的概念和实现，用了很长一段时间才彻底理解。

而反过来看下Flask的RESTful方案，其实也有一定的借鉴意义。

Flask比Django要简练的多，我们来看一个最简单的Flask例子，比如开启一个web服务，打印出hello world

代码如下：

from flask import Flask

app = Flask(__name__)

@app.route('/')

def hello_world():

return 'Hello World!'

if __name__ == '__main__':

app.run('192.168.56.102')

最后的IP是我的服务器IP，可以根据需要替换。

使用如下的方式来运行，服务即可开启。

[root@dev01 flask]# python hello.py

* Running on http://192.168.56.102:5000/ (Press CTRL+C to quit)

192.168.56.1 - - [19/May/2018 22:29:22] "GET / HTTP/1.1" 200 -

192.168.56.1 - - [19/May/2018 22:29:22] "GET /favicon.ico HTTP/1.1" 404 -

192.168.56.1 - - [19/May/2018 22:29:22] "GET /favicon.ico HTTP/1.1" 404 -

得到的结果如下：

到了这里，如果是做过Python Web开发的同学很可能说，这不是忽悠我呢，Python自身的技术也可以实现，本身Python就实现了一个SimpleHttpServer，可以参见之前的一篇文章：如果理解Python web开发技术

里面的代码看起来更简洁：

#!/usr/bin/env python
#coding:utf-8

from wsgiref.simple_server import make_server
def RunServer(environ, start_response):
    start_response('200 OK', [('Content-Type', 'text/html')])
    return '<h1>Hello, wsgi!</h1>'

if __name__ == '__main__':
    httpd = make_server('', 8000, RunServer)
    print "Serving HTTP on port 8000..."
    httpd.serve_forever()

我们来看看一个相对完整的Flask代码示例，内容参考了：https://www.jianshu.com/p/6ac1cab17929

代码如下：

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# by vellhe 2017/7/9

from flask import Flask, abort, request, jsonify

app = Flask(__name__)

# 测试数据暂时存放

tasks = []

@app.route('/add_task/', methods=['POST'])

def add_task():

#需要自己来维护这个列表结构

if not request.json or 'id' not in request.json or 'info' not in request.json:

abort(400)

task = {

'id': request.json['id'],

'info': request.json['info']

}

tasks.append(task)

return jsonify({'result': 'success'})

@app.route('/get_task/', methods=['GET'])

def get_task():

if not request.args or 'id' not in request.args:

# 没有指定id则返回全部

return jsonify(tasks)

else:

task_id = request.args['id']

task = filter(lambda t: t['id'] == int(task_id), tasks)

return jsonify(task) if task else jsonify({'result': 'not found'})

if __name__ == "__main__":

# 将host设置为192.168.56.102，则外网用户也可以访问到这个服务

app.run(host="192.168.56.102", port=8383, debug=True)

这个程序的一个难点就是如何在浏览器中模拟这个POST请求，当然可以使用postman来做，windows端可以参考这个链接下载：

https://app.getpostman.com/app/download/win64

第二个难点就是对于JSON的处理，里面还是有很多的参考之处。总体感觉虽然可以实现，但是所有的细节都需要自己来控制，比如输入结构，输出结构，信息查找的匹配，还有url和方法的映射。总是感觉有些拖泥带水，拖累太重，you can you up的感觉。

使用RESTful的方式，在Flask里面就是引入这个模块即可，相对来说比较简洁和轻量。

要安装flask_restful的话，一个命令即可。

pip install flask_restful

比如我有个需求，做一个基本的任务管理需求，可以分为两类功能，对任务理和任务列表管理，任务管理包括查看任务，增加,删除任务，都是对应单一的任务。任务列表就是对多个任务信息的查询和添加。

这种需求，其实基本的处理单元是任务，每一层级都可以不断的细化。尽可能对于应用层面来说更加透明，比如我就开放一个url: todos完成任务列表的管理，可以查看任务列表，添加任务信息。

todos/todo1 完成对单一任务的管理，比如添加修改，删除。

这个例子可能听起来不是很清晰，我举一个生活中的例子，比如你去一个游乐园，只有一个服务台，充值，退卡，办卡的需求都是在这一个柜台办理。这个时候需求可能是这几类：

比如你给他100块钱，那就意味着你要办卡，并且充值相应的金额（可能还要扣除工本费）

比如你给他100块钱，并且带卡，那就意味着你要充值

比如你给他一张卡，那就意味着你要退钱。

所以上述的需求，输入可能很简单，但是对应的业务场景可能截然不同。所以上述的代码要实现这个需求，逻辑还是比较复杂，而且不够清晰。

RESTful里面的一个优势就是可以基于class来构建不同的需求接口，可能对外开放的url是一个统一入口，但是可以在这个基础上进行细化。比如url: /todos和/todos/todo1

写一个程序脚本来实现：

#!/usr/bin/env python

# -*- coding: utf-8 -*-

from flask import Flask

from flask_restful import reqparse, abort, Api, Resource

app = Flask(__name__)

api = Api(app)

TODOS = {

'todo1': {'task': 'test1'},

'todo2': {'task': 'test2'},

'todo3': {'task': 'test3'},

}

def abort_if_todo_doesnt_exist(todo_id):

if todo_id not in TODOS:

abort(404, message="Todo {} doesn't exist".format(todo_id))

parser = reqparse.RequestParser()

parser.add_argument('task')

# Todo

# shows a single todo item and lets you delete a todo item

class Todo(Resource):

def get(self, todo_id):

abort_if_todo_doesnt_exist(todo_id)

return TODOS[todo_id]

def delete(self, todo_id):

abort_if_todo_doesnt_exist(todo_id)

del TODOS[todo_id]

return '', 204

def put(self, todo_id):

args = parser.parse_args()

task = {'task': args['task']}

TODOS[todo_id] = task

return task, 201

# TodoList

# shows a list of all todos, and lets you POST to add new tasks

class TodoList(Resource):

def get(self):

return TODOS

def post(self):

args = parser.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], 201

##

## Actually setup the Api resource routing here

##

api.add_resource(TodoList, '/todos')

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

if __name__ == '__main__':

app.run('192.168.56.102',debug=True)

调用情况如下：

如果是查看某一个task的信息，可以直接输入即可匹配。

看一下代码其实会发现，这里注册了两个API,这里和上面程序的不同就在于里面使用了Resource做了封装，如果我要添加一个逻辑，其实也是很方便的。不需要堆砌一大堆的if-else