python之restful api（flask）获取数据

需要用到谷歌浏览器的扩展程序 Advanced Rest Client进行模拟请求

1.直接上代码

from flask import Flask
from flask import request
from flask import abort
from flask import jsonify
import re

app = Flask(__name__)

@app.route('/')
def index():
    return 'hello world'


@app.route('/message/<message_id>',methods=['GET'])
def get_message(message_id):
    if message_id == 'all':
        search = 'all'
    else:
        if re.match(r'^d+$',message_id):
            search = int(message_id)
        else:
            abort(400)

    messages = [{'id':1,'site':'baidu'},{'id':2,'site':'taobao'}]
    res = {}
    
    if search == 'all':
        res['result'] = messages
        res['status'] = 'success'
    else:
        for item in messages:
            if item['id'] == search:
                res['result'] = [item]
                res['status'] = 'success'
                break
        if 'result' not in res:
            res['status'] = 'error'
            res['content'] = 'id is not in range'
        
    if request.headers['Content-Type'] == 'application/json':
        return jsonify(res),{'Content-Type':'application/json'}
    elif request.headers['Content-Type'] == 'text/xml':
        if 'result' not in res:
            return '''<?xml version="1.0"?>
<objects>
    <status>%s</status>
    <content>%s</content>
</objects>
'''%(res['status'],res['content']),{'Content-Type':'text/xml'}
        else:
            res_head = '<?xml version="1.0"?><objects><result>'
            res_foot = '</result><status>success</status></objects>'
            res_body = ''
            for item in res['result']:
                res_body += '<item id="{id}"><id>{id}</id><site>{site}</site></item>'.format(id=item['id'],site=item['site'])
            res_whole = res_head + res_body + res_foot
            return res_whole,{'Content-Type':'text/xml'}
    else:
        return str(res)
    


if __name__ == '__main__':
    app.run(port=5100)

代码解释

　　restful api请求路由：/message/<message_id>，message_id为 all 或者 整型数字　　

　　数据源：messages = [{'id':1,'site':'baidu'},{'id':2,'site':'taobao'}]

　　代码中的abort：触发400错误即报Bad Request　　The browser (or proxy) sent a request that this server could not understand. ，当message_id不是all 也不是整型数字时

　　注意一点：Advanced Rest Client 测试请求时不要在Headers指定Accept（因为它可以接受多种类型），而应该指定Content-Type

　　返回值没有用到状态码，而是通过判断返回值中的status是success还是error

2.直接上测试结果图

 

 

注：有不懂的可以私信博主本人