Flask接口开发--POST接口

以用户的注册和登录为例。

request接收参数

当我们在页面发出一个POST请求，请求传到服务器时，需要如何拿到当前请求的数据呢？在Flask可以使用 request 来接收请求参数。

request导入方式

from flask import request

使用 request 来接收参数的方式有多种，比如

request.form：用于接收表单参数
request.args：用于接收GET参数
request.json：用于接收JSON参数
request.values：获取所有参数（表单参数+GET参数）
request.file：用于接收文件

当然 request 还有不少其他属性，这里就不一一列出了，我们在这里使用了 request.json 和 request.values 来开发POST接口，如果要拿到具体某个参数的值，则是通过 request.json.get(“key”) 或 request.values.get(“key”) 来完成。

完整代码：

from flask import Flask, jsonify, request
app = Flask(__name__)
app.config["JSON_AS_ASCII"] = False  # jsonify返回的中文正常显示
@app.route("/register", methods=['POST'])
def user_register():
    username = request.json.get("username").strip() # 用户名
    password = request.json.get("password").strip() # 密码
    sex = request.json.get("sex", "0").strip() # 性别，默认为0(男性)
    telephone = request.json.get("telephone", "").strip() # 手机号，默认为空
    address = request.json.get("telphone", "").strip() # 地址，默认为空
    if username and password and telephone:
        import re
        if username == "wintest":
            return jsonify({"code": 2002, "msg": "用户名已存在！！！"})
        elif not (sex == "0" or sex == "1"):
            return jsonify({"code": 2003, "msg": "输入的性别只能是 0(男) 或 1(女)！！！"})
        elif not (len(telephone) == 11 and re.match("^1[3,5,7,8]\d{9}$", telephone)):
            return jsonify({"code": 2004, "msg": "手机号格式不正确！！！"})
        else:
            return jsonify({"code": 0, "msg": "恭喜，注册成功！"})
    else:
        return jsonify({"code": 2001, "msg": "用户名/密码/手机号不能为空，请检查！！！"})
@app.route("/login", methods=['POST'])
def user_login():
    username = request.values.get("username")
    password = request.values.get("password")
    if username and password:
        if username == "wintest" and password == "123456":
            return jsonify({"code": 0, "msg": "恭喜，登录成功！"})
        return jsonify({"code": 1002, "msg": "用户名或密码错误！！！"})
    else:
        return jsonify({"code": 1001, "msg": "用户名或密码不能为空！！！"})
if __name__ == '__main__':
    app.run()

用户注册接口

@app.route("/register", methods=['POST'])
def user_register():
    """用户注册"""
    username = request.json.get("username").strip()  # 用户名
    password = request.json.get("password").strip()  # 密码
    sex = request.json.get("sex", "0").strip()  # 性别，默认为0(男性)
    telephone = request.json.get("telephone", "").strip()  # 手机号，默认为空串
    address = request.json.get("address", "").strip()  # 地址，默认为空串
    if username and password and telephone:
        #强制提交查询连接的未提交事务
        db.conn.ping(reconnect=True)
        db.conn.commit()  # 提交当前连接事务
        sql1 = "SELECT username FROM user WHERE username = '{}'".format(username)
        res1 = db.select_db(sql1)
        print("查询到用户名 ==>> {}".format(res1))
        sql2 = "SELECT telephone FROM user WHERE telephone = '{}'".format(telephone)
        res2 = db.select_db(sql2)
        print("查询到手机号 ==>> {}".format(res2))
        if res1:
            return jsonify({"code": 2002, "msg": "用户名已存在，注册失败！！！"})
        elif not (sex == "0" or sex == "1"):
            return jsonify({"code": 2003, "msg": "输入的性别只能是 0(男) 或 1(女)！！！"})
        elif not (len(telephone) == 11 and re.match("^1[3,5,7,8]\d{9}$", telephone)):
            return jsonify({"code": 2004, "msg": "手机号格式不正确！！！"})
        elif res2:
            return jsonify({"code": 2005, "msg": "手机号已被注册！！！"})
        else:
            sql3 = "INSERT INTO user(username, password, role, sex, telephone, address) " \
                  "VALUES('{}', '{}', '1', '{}', '{}', '{}')".format(username, password, sex, telephone, address)
            db.execute_db(sql3)
            print("新增用户信息 ==>> {}".format(sql3))
            return jsonify({"code": 0, "msg": "恭喜，注册成功！"})
    else:
        return jsonify({"code": 2001, "msg": "用户名/密码/手机号不能为空，请检查！！！"})

在这里，我们接收参数是通过 request.json.get(“key”) 来实现，在请求的时候，因为直接在浏览器上输入网址访问资源都是GET方式，所以我们可以通过其他工具来模拟，如Postman、Fiddler都可以。

正则表达式 “^1[3,5,7,8]\d{9}$” 简单分析：

^ 表示一个字符串的开始

1 表示第一个字符是数字1

[3,5,7,8] 表示第二个字符只是数字 3/5/7/8 中的一个

\d{9} 表示后面后面只能匹配数字，且只能有9位数字

$ 表示一个字符串的结束

注意：该正则表达式是一个不完全的手机号验证，在这里只能较简单的验证手机号的合法性。

通过postman发送请求模拟：

点击发送请求：

运行的程序中能看到调用post接口记录

用户登录接口

@app.route("/login", methods=['POST'])
def user_login():
    username = request.values.get("username")
    password = request.values.get("password")
    if username and password:
        if username == "wintest" and password == "123456":
            return jsonify({"code": 0, "msg": "恭喜，登录成功！"})
        return jsonify({"code": 1002, "msg": "用户名或密码错误！！！"})
    else:
        return jsonify({"code": 1001, "msg": "用户名或密码不能为空！！！"})

在这里，我们接收参数是通过 request.values.get(“key”) 来实现，因为使用了 request.values，那么我们的请求头参数中，Content-Type参数就不能用 application/json 了，我们可以使用表单提交的格式：application/x-www-form-urlencoded ，而 body 中参数的格式为：key1=value1&key2=value2 。

通过postman发送请求模拟：

发送接口请求：

OK，以上我们使用Flask开发了2个简单的POST请求接口，但并没有连接到数据库进行操作，比如用户注册/登录接口中，验证的用户名都是固定的，不够灵活。下一节，我们将引入数据库来进一步完善。