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()