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)