使用Flask+MongoDB实现基于REST的接口简单操作

#coding=utf-8
#mongo.py
from flask import Flask,abort
from flask import jsonify
from flask import request
from flask_pymongo import PyMongo


app = Flask(__name__)

app.config[‘MONGO_DBNAME‘] = ‘rest‘
app.config[‘MONGO_URI‘] = ‘mongodb://172.18.252.20:27017/rest‘  #如果部署在本上，其中ip地址可填127.0.0.1

mongo = PyMongo(app)

@app.route(‘/login‘, methods=[‘GET‘])
def get_all_users():
  star = mongo.db.userInfo.find()
  output = []
  for s in star:
    output.append({‘name‘ : s[‘name‘], ‘pwd‘ : s[‘pwd‘]})
  return jsonify({‘result‘ : output})


@app.route(‘/register‘, methods=[‘POST‘])
def add_user():
  star = mongo.db.userInfo
  name = request.json[‘name‘]
  pwd = request.json[‘pwd‘]
  star_id = star.insert({‘name‘: name, ‘pwd‘: pwd})
  new_star = star.find_one({‘_id‘: star_id })
  output = {‘name‘ : new_star[‘name‘], ‘pwd‘ : new_star[‘pwd‘]}
  return jsonify({‘result‘ : output})

@app.route(‘/modify/<string:name>‘, methods=[‘PUT‘])
def update_user(name):
    user = mongo.db.userInfo.find({"name":name})
    output = []
    for s in user:
      output.append({‘name‘: s[‘name‘], ‘pwd‘: s[‘pwd‘]})
    if len(output) == 0:
      abort(404)
    mongo.db.userInfo.update({"name":name},{‘$set‘:{"name":"LZ111"}})
    return jsonify({‘result‘: output})

@app.route(‘/delete/<string:name>‘, methods=[‘DELETE‘])
def delete_user(name):
    user = mongo.db.userInfo.find({"name": name})
    output = []
    for s in user:
      output.append({‘name‘: s[‘name‘], ‘pwd‘: s[‘pwd‘]})
    if len(output) == 0:
      abort(404)
    mongo.db.userInfo.remove({‘name‘: name})
    return jsonify({‘result‘: True})



if __name__ == ‘__main__‘:
    # app.run(host = ‘0.0.0.0‘, port = 80, debug = True)
    app.run()

如果使用tornado,新建一个tornado_server.py文件，添加以下代码，作为Flask服务启动部署，也可以实现相关功能：

#coding=utf-8
#tornado_server.py
from tornado.wsgi import WSGIContainer
from tornado.httpserver import HTTPServer
from tornado.ioloop import IOLoop

from mongo import app


http_server = HTTPServer(WSGIContainer(app))
# http_server.listen(9000)  #此时，开启端口9000，默认本机127.0.0.1的IP地址
http_server.bind(80, "172.18.252.20")# 开启端口为9000，172.18.252.20为内网ip地址,也可以设置为0.0.0.0，
http_server.start(1)
IOLoop.instance().start()

看完上面代码，点击运行，即可实现相关接口功能了。既然是接口，GET方式还好说，如果是POST方式，使用浏览器就不那么好操作了，此时我们需要使用一款接口测试软件，在这里我推荐使用Postman（PS：Postman百度经验介绍，Postman官网下载链接 , 附加百度云盘WIN64位下载链接：https://pan.baidu.com/s/12t09OMcrezgNY3_JfYFx8g 密码：z1vk）,如果有同学使用基于Linux相关系统，推荐使用CURL来实现。

下面就正式开始使用Postman测试Get、Post、Put和Delete方式接口：

Get方式，选中下图中下拉框GET方式，直接输入URL，然后点击Send蓝色按钮即可得到请求数据。

技术图片