From efcfbf596bfd9f5f623e2e05a92e140fd27f1838 Mon Sep 17 00:00:00 2001
From: hyb <kk_huangyangbo@163.com>
Date: Thu, 08 Jan 2026 08:57:48 +0000
Subject: [PATCH] 新增认证校验工具类,可实现将明文加密后,直接获取token,避免认证过期 已更新华东师范大学二期的压测脚本实现自动获取最新token功能,其他脚本后续更新

---
 测试组/脚本/造数脚本2/华东师范大学二期/并发创建笼架.py |   32 ++++++++++++++++++++++++--------
 1 files changed, 24 insertions(+), 8 deletions(-)

diff --git "a/\346\265\213\350\257\225\347\273\204/\350\204\232\346\234\254/\351\200\240\346\225\260\350\204\232\346\234\2542/\345\215\216\344\270\234\345\270\210\350\214\203\345\244\247\345\255\246\344\272\214\346\234\237/\345\271\266\345\217\221\345\210\233\345\273\272\347\254\274\346\236\266.py" "b/\346\265\213\350\257\225\347\273\204/\350\204\232\346\234\254/\351\200\240\346\225\260\350\204\232\346\234\2542/\345\215\216\344\270\234\345\270\210\350\214\203\345\244\247\345\255\246\344\272\214\346\234\237/\345\271\266\345\217\221\345\210\233\345\273\272\347\254\274\346\236\266.py"
index 5d73ec0..edadb6d 100644
--- "a/\346\265\213\350\257\225\347\273\204/\350\204\232\346\234\254/\351\200\240\346\225\260\350\204\232\346\234\2542/\345\215\216\344\270\234\345\270\210\350\214\203\345\244\247\345\255\246\344\272\214\346\234\237/\345\271\266\345\217\221\345\210\233\345\273\272\347\254\274\346\236\266.py"
+++ "b/\346\265\213\350\257\225\347\273\204/\350\204\232\346\234\254/\351\200\240\346\225\260\350\204\232\346\234\2542/\345\215\216\344\270\234\345\270\210\350\214\203\345\244\247\345\255\246\344\272\214\346\234\237/\345\271\266\345\217\221\345\210\233\345\273\272\347\254\274\346\236\266.py"
@@ -10,31 +10,47 @@
 """
 import sys
 import os
-# 将上一级目录加入模块搜索路径
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 import asyncio
 import aiohttp
 import time
 import traceback
 import datetime
 from tqdm import tqdm
-from Util.random_util import RandomUtil
-from Util.dingtalk_helper import DingTalkHelper
+
+def get_parent_directory(file_path, levels=1):
+    """获取指定层级的父目录"""
+    path = os.path.abspath(file_path)
+    for _ in range(levels):
+        path = os.path.dirname(path)
+    return path
+parent_dir = get_parent_directory(__file__, 5)  # 获取上五级目录
+sys.path.append(parent_dir)
+
+from 测试组.脚本.造数脚本2.Util import TokenValidator, DingTalkHelper, RandomUtil, RequestRecord, LoadTestReportGenerator
 
 
 # --- 配置 ---
 ACCESS_TOKEN = '4625f6690acd9347fae5b3a05af598be63e73d604b933a9b3902425b8f136d4d'
 SECRET = 'SEC3b6937550bd297b5491855f6f40c2ff1b41bc8c495e118ba9848742b1ddf8f19'
 
+# 账号密码配置
+username = "gly"
+password = "Baoyi@1341"
+# 创建获取token实例
+token_validator = TokenValidator()
+
+domain = "http://192.168.6.190:5561"
+token = token_validator.get_token(domain, username, password)
+
 apiname = "创建动物房笼架"
-url = "http://192.168.6.190:5561/api/base/room/shelf/save"
+url = f"{domain}/api/base/room/shelf/save"
 headers = {
-    "token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJleHAiOjE3NjYyMTk5MzIsInVzZXJuYW1lIjoiZ2x5In0.fN-yJ0SVrEd_GyIvPmaBL2nx1M3a1jloSOsXyu3E5_w",
+    "token": token,
     "Content-Type": "application/json"
 }
 
 NUM_WORKERS = 100
-TOTAL_REQUESTS = 40000
+TOTAL_REQUESTS = 2
 MAX_RETRIES = 3
 REQUEST_TIMEOUT = 60
 OUTPUT_DIR = './load_test_report'
@@ -154,7 +170,7 @@
     try:
         import importlib.util
         script_dir = os.path.dirname(os.path.abspath(__file__))
-        report_path = os.path.join(script_dir, 'H:\\项目\\造数脚本\\Util\\stress_test_report_generator.py')
+        report_path = os.path.join(script_dir, 'H:\\项目\\archive\\测试组\\脚本\\造数脚本2\\Util\\stress_test_report_generator.py')
         if os.path.exists(report_path):
             spec = importlib.util.spec_from_file_location('report_module', report_path)
             report_module = importlib.util.module_from_spec(spec)

--
Gitblit v1.9.1