jmetert笔记:通过java调用python脚本
jmeter 创建BeanShell 取样器,编写java脚本
import java.io.BufferedReader;
import java.io.InputStreamReader;// 1. 命令里的路径改成自己脚本的路径,-t后面是填需要传入Python脚本的参数,可传入jmeter的变量。
String command = "cmd /c python G://capchat.py -t ${captcha}";
String var;
Runtime rt = Runtime.getRuntime();
Process pr = rt.exec(command);pr.waitFor();
BufferedReader b = new BufferedReader(new InputStreamReader(pr.getInputStream()));
String line = "";
StringBuilder response = new StringBuilder();
while ((line = b.readLine()) != null) {response.append(line);}
String response_data = response.toString();
System.out.println(response_data);
b.close();
var = response.toString();// 2. 定义Jmeter中引用的变量名,jmeter后面可以直接使用${变量名}调用变量;
vars.put("captcha_str",var);// 把var值赋予给参数captcha_str 把var打印到jmeter.log的日志中
`
python 接收jemter 传过来的参数 ,处理并返回
这里使用了ddddocr第三方库识别验证码,如需使用Python版本必须>=3.8
项目地址:https://github.com/sml2h3/ddddocr
安装命令
pip install ddddocr -i https://pypi.tuna.tsinghua.edu.cn/simple/
import os
import re
from PIL import Image
import base64
import ddddocr
from io import BytesIO
import argparse
# 获取jmeter传入的值,然后赋值给变量ticket,变量即可为Python调用
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--ticket")
args = parser.parse_args()
ticket = args.ticket #赋值给变量ticket
context = re.findall('data:image/png;base64,(.*)', ticket) # 正则截取 base字符
lst1 = list(map(lambda x: str(x), context)) # 转换为str
str1 = ''.join(lst1)
image = Image.open(BytesIO(base64.b64decode(str1)))
img = image.convert('L') # 图片转灰度
# 保存验证码图片
img.save(r'\Verify.png')
ocr = ddddocr.DdddOcr()
with open(r'\Verify.png', 'rb') as f:
img_bytes = f.read()
res = ocr.classification(img_bytes) # 获取的验证
print(res) # print 打印参数
os.remove(r'\Verify.png')