接口自动化测试-flask框架

我会带着你远行 2023-10-06 21:23 23阅读 0赞

一、flask框架安装

二、路由

from flask import Flask
    from flask import jsonify
    from flask import request
    
    app = Flask(__name__)  #创建一个Flask对象
    
    users = {'1001':['1001','张三','123456',8900],
             '1002': ['1002', '李四', '123456', 8900],
             '1003': ['1003', '王五', '123456', 8900],
             '1004': ['1004', '赵六', '123456', 8900]}
    @app.route('/users',methods=['GET'])
    def getUsers():
        app.config['JSON_AS_ASCII'] = False
        return jsonify({"code":20000,"message":"success","data":users})
    
    
    
    @app.get("/users/<string:id>")   # <类型:参数>实现动态路由 请求方式http://10.6.10.81:5566/users/1001
    def getUser(id):
        if id in users:
            user_info = users[id]
            return f"尊敬的 {user_info[1]}"
        else:
            return "账户不存在"
    
    @app.get("/users/user/id")  #请求url http://10.6.10.81:5566/users/user/id?id=1004
    def get_user():
        id = request.args.get("id") #args 获取客户端get请求参数account的值
        if id:
            if id in users:
                user_info = users[id]
                return jsonify({"code":20000,"message":"success","data":user_info})
            else:
                return jsonify({"code":20001,"message":"账户不存在"})
        else:
            return jsonify({"code":20002,"message":"参数有错"})
    
    @app.post("/login")
    def login():
        account=request.form.get("account")//form获取post请求方式的值
        password = request.values.get("password")//values获取post或者get请求方式的值
        print("password:"+password)
        if account and password:
            if account in users:
                info = users[account]
                if password == info[2]:
                    return jsonify({"code":20000,"message":"success"})
                else:
                    return jsonify({"code": "20001", "message": "password error"})
            else:
                return jsonify({"code": "20003", "message": "account not exists"})
        else:
            return jsonify({"code":"20002","message":"param error"})
    
    
    if __name__ == "__main__":
        app.config['JSON_AS_ASCII']=False
        app.run(host="0.0.0.0",port=5566,debug=True) #host='0.0.0.0'表示可以使用127.0.0.1，localhost，本机ip

三、返回json数据

from flask import jsonify
    jsonify({})

四、服务端会话保持

*  cookie

模块：make\_response

使用：

1、创建response对象：res=make\_response()

*  session