python nginx不同参数压测脚本

Server：

from flask import Flask, request, jsonify
import json
import re
import os
import subprocess

conf_file_src = '/etc/nginx/nginx.conf'
restart_shell = 'sudo service nginx restart'

app = Flask(__name__)

# 重置conf文件
# conf文件中要测试的值置为 ###arg###
def reset_conf(file_name, data):
    # 打开模板文件
    file = open(file_name, 'r')
    file_temp = file.read()
    for key in data:
        # 替换要调试的参数
        re_model = re.compile(r'###')
        args = re.findall(r'###S' + key + '###', file_temp)

        for arg in args:
            print(arg)
            file_temp = file_temp.replace(arg, data[key], 100)

    # 写入新的配置文件
    file_conf = open(conf_file_src, 'w+')
    file_conf.write(file_temp)
    file_conf.close()


# 重启nginx服务
def reset_service():
    # 重启service
    res = os.popen(restart_shell).readline()
    return 'Result: ' + res

# url : xxx.xxx.xxx.xxx:5000/restart
# 参数：
# data: {'arg1':'v1','arg2':'v1']}
# 返回：转发执行结果
@app.route('/restart', methods=['POST'])
def restart():
    data = json.loads(request.get_data(as_text=True))
    print(data)
    reset_conf("nginx.conf.template", data)
    result = reset_service()
    return result

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

Client：

import requests
import json

url = "http://127.0.0.1:5000/restart"
field = 'sendfile'
data = ["on", "off"]

def reset():
    for s in data:
        #发送请求修改nginx配置文件
        json = '{"' + field +'":"' + s +'"}'
        res = requests.post(url=url, data='{"sendfile":"on"}')
        print(json)
        print(res)
        print(res.content)

        #压力测试
        # wrk ...

if __name__ == '__main__':
    reset()