本文转自测试人社区,原文链接: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)

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

  • 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