软件测试学习笔记丨Flask框架-请求与响应

本文转自测试人社区，原文链接：https://ceshiren.com/t/topic/23408

请求方法

from flask import Flask

app = Flask(__name__)


@app.route("/cases", methods=["get"])
def get_case():
    return {"code": 0, "msg": "get success"}


# post请求不能直接通过url访问,可以使用postman或者requests库操作
@app.route("/case", methods=["post"])
def post_case():
    return {"code": 0, "msg": "post success"}


if __name__ == '__main__':
    app.run()

处理请求数据

注意：是request对象（不是requests）

json可以处理一些post请求的数据

from flask import Flask, request

from log_utils import logger

app = Flask(__name__)


# get
@app.route("/login/", methods=["get"])
def hogwarts():
    logger.info(f"请求参数为{request.args}")
    result = request.args
    a = result.get('a')
    b = result.get('b')
    print(a, b)
    return {"code": 0, "msg": "get success"}


# post请求的json数据
@app.route("/regist", methods=["post"])
def post_regist():
    logger.info(f"请求参数为{request.json}")
    return {"code": 0, "msg": "post success"}


# form表单
@app.route("/put", methods=["put"])
def put_regist():
    name = request.form.get('name')
    p = request.form.get('password')
    email = request.form.get('email')
    logger.info(f"请求参数为{request.form}")
    logger.info(f"name：{name},password：{p},email：{email}")
    return {"code": 0, "msg": "post success"}


# 文件上传
@app.route("/file", methods=["post"])
def post_file():
    obj = request.files.get("file")
    logger.info(obj)
    filename = obj.filename
    logger.info(f"文件名为：{filename}")
    # 将接口接收的文件保存在本地
    obj.save("./1.txt")
    return {"code": 0, "msg": "post success"}


if __name__ == '__main__':
    app.run()

由于postman对文件上传有一些bug，所以写了测试代码上传文件:

import requests


def test_file():
    url = "http://127.0.0.1:5000/file"
    file = {
        "file": open("C:/Users/Administrator/Desktop/1.txt", 'rb')
    }
    r = requests.post(url=url, files=file)
    assert r.status_code == 200

处理响应信息

返回的响应类型：
- 文本：直接return
- 元组
- json
- html
- 额外数据

返回元组有三种情况：

（response，status）
（response，headers）
（response，status，headers）

@app.route("/tuple")
def tuple_result():
    return "hogwarts", 200, {"test": "demo"}

返回json：

直接返回dict会转换成json
使用jsonify()方法，通过参数传入键值对

@app.route("/dict")
def json_result():
    return {'status': 0}


@app.route("/json")
def json_re():
    return jsonify({'status': 0})
    # 也可以通过参数指定
    # return jsonify(status=1,name="joker")

返回html–render_template(文件名称.html)

使用模板渲染技术

注意：html文件必须在同级名为templates的目录下

@app.route("/html")
def html_demo():
    return render_template("demo.html")

设置额外数据–make_response()：

设置cookie
设置响应头信息等

@app.route("/")
def index():
    # 设置返回页面模板
    res = make_response(render_template('demo.html'))
    # 设置cookie
    res.set_cookie('username', 'password')
    # 设置请求头
    res.headers['hogwarts'] = 'joker'
    return res

在这里插入图片描述

软件测试学习笔记丨Flask框架-请求与响应

请求方法

处理请求数据

处理响应信息

推荐学习

悦读