52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import os
|
||
import shutil
|
||
import subprocess
|
||
import datetime
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from core.settings import LOG_SOURCE, LOG_BACKUP_DIR, ALLURE_TEMP, REPORT_DIR
|
||
|
||
# netstat -ano | findstr :4723
|
||
# taskkill /PID 12345 /F
|
||
# 日志自动清理
|
||
def _clean_old_logs(backup_dir, keep_count=10):
|
||
files = sorted(Path(backup_dir).glob("pytest_*.log"), key=os.path.getmtime)
|
||
while len(files) > keep_count:
|
||
file_to_remove = files.pop(0)
|
||
try:
|
||
os.remove(file_to_remove)
|
||
except OSError as e:
|
||
print(f"清理旧日志失败 {file_to_remove}: {e}")
|
||
|
||
|
||
def main():
|
||
try:
|
||
# 2. 执行 Pytest
|
||
# 建议保留你之前配置的 -s -v 等参数
|
||
# 注意：-x 表示遇到错误立即停止，如果是全量回归建议去掉 -x
|
||
pytest.main(["test_cases", "-x", "-v", f"--alluredir={ALLURE_TEMP}"])
|
||
|
||
# 3. 生成报告
|
||
if ALLURE_TEMP.exists():
|
||
# 使用 subprocess 替代 os.system，更安全且跨平台兼容性更好
|
||
subprocess.run(f'allure generate {ALLURE_TEMP} -o {REPORT_DIR} --clean', shell=True, check=False)
|
||
except Exception as e:
|
||
print(f"自动化测试执行过程中发生异常: {e}")
|
||
|
||
finally:
|
||
# 4. 备份日志 (无论测试是否崩溃都执行)
|
||
if LOG_SOURCE.exists():
|
||
now = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
backup_path = LOG_BACKUP_DIR / f"pytest_{now}.log"
|
||
shutil.copy2(LOG_SOURCE, backup_path)
|
||
print(f"日志已备份至: {backup_path}")
|
||
_clean_old_logs(LOG_BACKUP_DIR)
|
||
else:
|
||
print("未找到原始日志文件，跳过备份。")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|