flask框架启服务+json格式入参+postman获取上个接口的token作为下个接口的入参+关联接口【多测师_王sir】

import json
import os
import time

from flask import Flask, request, make_response, jsonify, render_template
from werkzeug.utils import secure_filename

app = Flask(__name__)
user_token_dic = {
    'user01': '3b6754f00bb0063071c5b71ce2b56b4ed0ce56a63493e785bea85b74c41ce200'
    , 'user02': 'dcdc7286f6822dc5a4b54d2ce53f40ae0e562d7fc06d64b8dd332f7d528c7c67'
    , 'user03': '04c4fba92b3451b16daedbbc7dd4cec84b8a88333f3da60626c6abbfdc829a02'
    , 'user04': '3c3a7fa64c34e2530b708ec9bdeedd49e0098bfd3239b1cd58985feeb38531b8'
    , 'user05': 'abb9d47918055902a8a6f3f8e86cd962d37d49647e9b51bdd7f7d67ff95f9864'
    , 'user06': '6b2b8f34dd4ad1812a76a7037ed23d9ebc0189070320fa32dc610a79852419c7'
    , 'user07': 'aac287a1eeb39d1983aaa732a7a5f33a64918adb1e88da6b344d2090eaed7239'
    , 'user08': '3d0c7ff2ad3063bc0e93befeef895def90290114ee590ca09b60f8e4476aca27'
    , 'user09': 'a0a9941e29cebdd732a2a57411f14d45e14bc3061d66c2e1c30eca421ed0b02d'
    , 'user10': '34467677a88fc7966807c26ced0972cc775be9e6983d4bfbd2a0981a4350dcfd'
}
menu_json = {
    "code": "200",
    "breakfast": [
        {
            "menu_nunber": "01",
            "menu_price": 5.50,
            "menu_name": "小笼包"
        },
        {
            "menu_nunber": "02",
            "menu_price": 3.00,
            "menu_name": "八宝粥"
        },
        {
            "menu_nunber": "03",
            "menu_price": 1.50,
            "menu_name": "油条"
        },
        {
            "menu_nunber": "04",
            "menu_price": 1.00,
            "menu_name": "茶叶蛋"
        },
        {
            "menu_nunber": "05",
            "menu_price": 1.50,
            "menu_name": "豆包"
        },
        {
            "menu_nunber": "06",
            "menu_price": 2.00,
            "menu_name": "烧饼"
        }
    ],
    "lunch": [
        {
            "menu_nunber": "07",
            "menu_price": 17.00,
            "menu_name": "宫保鸡丁"
        },
        {
            "menu_nunber": "08",
            "menu_price": 25.00,
            "menu_name": "锅包肉"
        },
        {
            "menu_nunber": "09",
            "menu_price": 35.00,
            "menu_name": "糖醋排骨"
        },
        {
            "menu_nunber": "10",
            "menu_price": 30.00,
            "menu_name": "蒜台炒肉片"
        },
        {
            "menu_nunber": "11",
            "menu_price": 25.50,
            "menu_name": "酸辣土豆丝"
        },
        {
            "menu_nunber": "12",
            "menu_price": 32.00,
            "menu_name": "红烧茄子"
        }
    ],
    "dinner": [
        {
            "menu_nunber": "13",
            "menu_price": 21.00,
            "menu_name": "辣椒炒肉"
        },
        {
            "menu_nunber": "14",
            "menu_price": 15.00,
            "menu_name": "豌豆肉片汤"
        },
        {
            "menu_nunber": "15",
            "menu_price": 26.00,
            "menu_name": "原味五花肉卷"
        },
        {
            "menu_nunber": "16",
            "menu_price": 20.00,
            "menu_name": "醋溜白菜"
        },
        {
            "menu_nunber": "17",
            "menu_price": 25.50,
            "menu_name": "地三鲜"
        },
        {
            "menu_nunber": "18",
            "menu_price": 39.00,
            "menu_name": "红烧排骨"
        }
    ]
}

server_internal_error_data = {
    "code": "500",
    "message": "Server internal error."
}

UPLOAD_FOLDER = 'upload'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
basedir = os.path.abspath(os.path.dirname(__file__))
ALLOWED_EXTENSIONS = set(['txt', 'png', 'jpg', 'xls', 'JPG', 'PNG', 'xlsx', 'gif', 'GIF', 'doc', 'docx', 'ppt', 'pptx'])


@app.route("/api/v1/user/login", methods=['POST','GET'])
def login():
    """登录接口，输入参数格式：
        "authRequest": {
            "userName": "[username]",
            "password": "[password]"
        }

        :return
        Success : http code 200
        {
            "code": "200",
            "message": "login success",
            "access_token": "3b6754f00bb0063071c5b71ce2b56b4ed0ce56a63493e785bea85b74c41ce200"
        }

        Fail : http code 401
        {
            "code": "401",
            "message": "login fail"
        }
    """
    try:
        raw_data = request.get_data(as_text=True)
        data = json.loads(raw_data)
        username = data.get("authRequest").get("userName")
        password = data.get("authRequest").get("password")

        if username not in user_token_dic.keys() or password != 'pwd':
            print('Username or password error!')
            login_fail_resp_data = {
                "code": "401",
                "message": "login fail"
            }
            return make_response(jsonify(login_fail_resp_data), '401')

        login_succ_resp_data = {
            "code": "200",
            "message": "login success",
            "access_token": user_token_dic.get(username)
        }

        return make_response(jsonify(login_succ_resp_data), '200')
    except Exception:
        return make_response(jsonify(server_internal_error_data), '500')


@app.route('/api/v1/menu/list', methods=['GET'])
def list():
    """菜单浏览接口

    :return: 菜单列表
    """
    access_token = request.headers.get("access_token")

    if access_token is None:
        login_fail_resp_data = {
            "code": "401",
            "message": "Please login first."
        }
        return make_response(jsonify(login_fail_resp_data), '401')

    if access_token not in user_token_dic.values():
        print('access_token error, please re-login.')
        login_fail_resp_data = {
            "code": "401",
            "message": "Unknown user info, please re-login."
        }
        return make_response(jsonify(login_fail_resp_data), '401')

    category = request.args.get("type")

    if category is not None:
        return make_response(jsonify(menu_json.get(category)), 200)

    return make_response(jsonify(menu_json), 200)


@app.route("/api/v1/menu/confirm", methods=['POST'])
def confirm():
    """下单接口，输入参数格式：
        header = {'access_token' : ''}
        data:
        {
            "order_list": [
                {
                    "menu_nunber" : "01",
                    "number" : 1
                },
                {
                    "menu_nunber" : "03",
                    "number" : 2
                },
                {
                    "menu_nunber" : "04",
                    "number" : 1
                },
                {
                    "menu_nunber" : "05",
                    "number" : 3
                }
            ]
        }

        :return
        Success : http code 200
        {
            "code": "200",
            "message": "Order success.",
            "total": 7
        }

        not login : http code 401
        {
            "code": "401",
            "message": "Please login first."
        }
    """
    try:
        access_token = request.headers.get("access_token")
        if access_token is None:
            login_fail_resp_data = {
                "code": "401",
                "message": "Please login first."
            }
            return make_response(jsonify(login_fail_resp_data), '401')

        raw_data = request.get_data(as_text=True)
        order_list = json.loads(raw_data).get("order_list")
        total = 0
        for order in order_list:
            total = total + order.get("number")
            time.sleep(0.6)

        order_success_resp_data = {
            "code": "200",
            "message": "Order success.",
            "total": total
        }
        return make_response(jsonify(order_success_resp_data), '200')
    except Exception:
        return make_response(jsonify(server_internal_error_data), '500')


@app.route("/api/v1/user/logout", methods=['DELETE'])
def logout():
    """用户注销接口，输入参数格式：
        header = {'access_token' : ''}
    """
    try:
        access_token = request.headers.get("access_token")

        if access_token not in user_token_dic.values():
            print('access_token error, logout failed.')
            login_fail_resp_data = {
                "code": "401",
                "message": "Unknown user info, logout fail."
            }
            return make_response(jsonify(login_fail_resp_data), '401')

        login_succ_resp_data = {
            "code": "200",
            "message": "logout success"
        }

        return make_response(jsonify(login_succ_resp_data), '200')
    except Exception:
        return make_response(jsonify(server_internal_error_data), '500')


# suffix validation
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS


# upload file testing page
@app.route('/', methods=['GET'], strict_slashes=False)
def indexpage():
    return render_template('index.html')


# upload file
@app.route('/uploadfile', methods=['POST'], strict_slashes=False)
def api_upload():
    file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
    if not os.path.exists(file_dir):
        os.makedirs(file_dir)
    f = request.files['file']  # get the file name

    if f and allowed_file(f.filename):  # Verify the file type
        fname = secure_filename(f.filename)
        ext = fname.rsplit('.', 1)[1]
        unix_time = time.time()
        new_filename = str(unix_time) + '.' + ext
        f.save(os.path.join(file_dir, new_filename))
        print(f'{f.filename} upload success!')
        return jsonify({'code': 200, 'result': 'success', 'msg': f'file {f.filename} upload success'})
    else:
        print(f'{f.filename} upload failed!')
        return jsonify(
            {'code': 500, 'result': 'failed', 'msg': f'file {f.filename} upload fail, please check the file type'})


if __name__ == "__main__":
    app.run(port=9091, debug=True, host='0.0.0.0')

    # Demo for https, password = 1234
    # app.run('0.0.0.0', debug=True, port=8100, ssl_context=(f'{os.path.abspath(os.curdir)}/cert_files/server.crt'
    # , f'{os.path.abspath(os.curdir)}/cert_files/server.key')