Flask接口开发--POST接口

发布于:2025-03-25 ⋅ 阅读:(27) ⋅ 点赞:(0)

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

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请求接口,但并没有连接到数据库进行操作,比如用户注册/登录接口中,验证的用户名都是固定的,不够灵活。下一节,我们将引入数据库来进一步完善。


网站公告

今日签到

点亮在社区的每一天
去签到