CloudflareSpeedTest-GUI.py

import sys
import os
import json
import subprocess
import ctypes
from functools import partial
from PyQt5 import QtWidgets, QtGui, QtCore

def resource_path(relative_path):
    """获取 PyInstaller 打包后资源文件路径"""
    if hasattr(sys, "_MEIPASS"):
        return os.path.join(sys._MEIPASS, relative_path)
    return os.path.join(os.path.abspath("."), relative_path)

APP_ICON = resource_path("app.ico")
CFST_EXE = "cfst.exe"
IP_FILE_NAME = "ip.txt"
SAVED_SETTINGS_FILE = "saved_settings.json"
APP_USER_MODEL_ID = "com.example.cloudflarespeedtest"

def _set_windows_appid(appid):
    try:
        ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(appid)
    except Exception:
        pass


if sys.platform.startswith("win"):
    _set_windows_appid(APP_USER_MODEL_ID)

class MainWin(QtWidgets.QWidget):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("CloudflareSpeedTest_GUI -- 小琳解说")

        # 图标（修复：PyInstaller 下也能找到）
        if os.path.exists(APP_ICON):
            self.setWindowIcon(QtGui.QIcon(APP_ICON))
        else:
            self.setWindowIcon(QtGui.QIcon(resource_path("app.ico")))

        self.setFont(QtGui.QFont("Microsoft YaHei", 10))
        self.resize(500, 520)

        self._build_ui()
        self._load_saved_settings_list()

    def _build_ui(self):
        main_layout = QtWidgets.QVBoxLayout(self)

        params = [
            ("-n", "200", "延迟线程 1-1000"),
            ("-t", "4", "延迟次数"),
            ("-dn", "10", "下载数量"),
            ("-dt", "10", "下载时间(秒)"),
            ("-tp", "443", "端口"),
            ("-url", "https://cf.xiu2.xyz/url", "测速地址"),
            ("-httping", "", "HTTPing 模式 (勾选启用)"),
            ("-httping-code", "200", "HTTP 有效状态码"),
            ("-cfcolo", "HKG,KHH,NRT,LAX", "地区码, HTTPing 模式可用"),
            ("-tl", "9999", "平均延迟上限(ms)"),
            ("-tll", "0", "平均延迟下限(ms)"),
            ("-tlr", "1.00", "丢包上限 0.00-1.00"),
            ("-sl", "0", "下载速度下限 MB/s"),
            ("-p", "10", "显示结果数量"),
            ("-f", "ip.txt", "IP 段文件"),
            ("-ip", "", "指定 IP 段"),
            ("-o", "result.csv", "输出文件"),
            ("-dd", "", "禁用下载测速 (勾选启用)"),
            ("-allip", "", "测速全部 IP (勾选启用)")
        ]

        grid = QtWidgets.QGridLayout()
        grid.setColumnStretch(1, 1)

        self.controls = {}
        row = 0

        for key, default, hint in params:
            cb = QtWidgets.QCheckBox(key)
            cb.setChecked(False)

            if key == "-n":
                widget = QtWidgets.QSpinBox()
                widget.setRange(1, 1000)
                try:
                    widget.setValue(int(default))
                except Exception:
                    widget.setValue(200)
                widget.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
                widget.setEnabled(False)
            else:
                widget = QtWidgets.QLineEdit(default)
                if default == "":
                    widget.setPlaceholderText(hint)
                widget.setEnabled(False)
                widget.setStyleSheet("QLineEdit:disabled { color: gray; }")

            lbl = QtWidgets.QLabel(hint)
            cb.stateChanged.connect(partial(self._on_checkbox_toggled, key))

            grid.addWidget(cb, row, 0)
            grid.addWidget(widget, row, 1)
            grid.addWidget(lbl, row, 2)

            self.controls[key] = (cb, widget)
            row += 1

        main_layout.addLayout(grid)

        save_load_layout = QtWidgets.QGridLayout()
        save_load_layout.setColumnStretch(1, 1)

        save_label = QtWidgets.QLabel("保存设置名称")
        self.save_name_edit = QtWidgets.QLineEdit()
        self.save_name_edit.setPlaceholderText("填写保存设置名称")
        self.save_btn = QtWidgets.QPushButton("保存设置")

        load_label = QtWidgets.QLabel("已保存设置")
        self.load_combo = QtWidgets.QComboBox()
        self.load_combo.setEditable(False)

        sp = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Fixed)
        self.save_name_edit.setSizePolicy(sp)
        self.load_combo.setSizePolicy(sp)

        self.load_btn = QtWidgets.QPushButton("加载设置")
        self.delete_btn = QtWidgets.QPushButton("删除已保存")

        load_btns_layout = QtWidgets.QHBoxLayout()
        load_btns_layout.addWidget(self.load_btn)
        load_btns_layout.addWidget(self.delete_btn)
        load_btns_layout.addStretch()

        save_load_layout.addWidget(save_label, 0, 0)
        save_load_layout.addWidget(self.save_name_edit, 0, 1)
        save_load_layout.addWidget(self.save_btn, 0, 2)

        save_load_layout.addWidget(load_label, 1, 0)
        save_load_layout.addWidget(self.load_combo, 1, 1)
        save_load_layout.addLayout(load_btns_layout, 1, 2)

        # 运行按钮
        self.run_btn = QtWidgets.QPushButton("运行\n测速")
        btn_size = 88
        self.run_btn.setFixedSize(btn_size, btn_size)

        font = QtGui.QFont("Microsoft YaHei", 14)
        font.setBold(True)
        self.run_btn.setFont(font)
        self.run_btn.setStyleSheet("""
            QPushButton {
                background-color: #0078D7;
                color: white;
                border: none;
                font-weight: bold;
                border-radius: 8px;
            }
            QPushButton:pressed {
                background-color: #005A9E;
            }
        """)

        save_load_layout.addWidget(
            self.run_btn, 0, 3, 2, 1, alignment=QtCore.Qt.AlignCenter
        )

        main_layout.addLayout(save_load_layout)

        self.save_btn.clicked.connect(self._on_save_clicked)
        self.load_btn.clicked.connect(self._on_load_clicked)
        self.delete_btn.clicked.connect(self._on_delete_clicked)
        self.run_btn.clicked.connect(self._on_run_clicked)

    def _on_checkbox_toggled(self, key, state):
        cb, widget = self.controls[key]
        enabled = (state == 2)
        widget.setEnabled(enabled)
        if isinstance(widget, QtWidgets.QLineEdit):
            if enabled:
                widget.setStyleSheet("QLineEdit { color: black; }")
            else:
                widget.setStyleSheet("QLineEdit:disabled { color: gray; }")

    def _load_saved_settings_list(self):
        self.load_combo.clear()
        if not os.path.exists(SAVED_SETTINGS_FILE):
            return
        try:
            with open(SAVED_SETTINGS_FILE, "r", encoding="utf-8") as f:
                data = json.load(f)
            names = sorted(data.keys())
            self.load_combo.addItems(names)
        except Exception:
            pass

    def _read_saved_settings(self):
        if not os.path.exists(SAVED_SETTINGS_FILE):
            return {}
        try:
            with open(SAVED_SETTINGS_FILE, "r", encoding="utf-8") as f:
                return json.load(f)
        except Exception:
            return {}

    def _write_saved_settings(self, data):
        try:
            with open(SAVED_SETTINGS_FILE, "w", encoding="utf-8") as f:
                json.dump(data, f, ensure_ascii=False, indent=2)
            return True
        except Exception:
            return False

    def _on_save_clicked(self):
        name = self.save_name_edit.text().strip()
        if not name:
            QtWidgets.QMessageBox.warning(self, "保存失败", "请填写保存设置的名称。")
            return

        settings = {}
        for k, (cb, widget) in self.controls.items():
            if isinstance(widget, QtWidgets.QSpinBox):
                val = widget.value()
            else:
                val = widget.text()
            settings[k] = [cb.isChecked(), val]

        all_saved = self._read_saved_settings()
        all_saved[name] = settings

        ok = self._write_saved_settings(all_saved)
        if ok:
            QtWidgets.QMessageBox.information(self, "保存成功", f"设置已保存为: {name}")
            self._load_saved_settings_list()
            idx = self.load_combo.findText(name)
            if idx >= 0:
                self.load_combo.setCurrentIndex(idx)
        else:
            QtWidgets.QMessageBox.warning(self, "保存失败", "写入保存文件失败。")

    def _on_load_clicked(self):
        name = self.load_combo.currentText().strip()
        if not name:
            QtWidgets.QMessageBox.warning(self, "加载失败", "请先选择一个已保存的设置名称。")
            return

        all_saved = self._read_saved_settings()
        if name not in all_saved:
            QtWidgets.QMessageBox.warning(self, "加载失败", "所选设置不存在或已被删除。")
            self._load_saved_settings_list()
            return

        settings = all_saved[name]

        for k, (cb, widget) in self.controls.items():
            if k in settings:
                checked, val = settings[k]
                cb.setChecked(bool(checked))
                if isinstance(widget, QtWidgets.QSpinBox):
                    try:
                        widget.setValue(int(val))
                    except Exception:
                        pass
                    widget.setEnabled(bool(checked))
                else:
                    widget.setText(str(val))
                    widget.setEnabled(bool(checked))
                    widget.setStyleSheet(
                        "QLineEdit { color: black; }" if widget.isEnabled() else "QLineEdit:disabled { color: gray; }"
                    )

        QtWidgets.QMessageBox.information(self, "加载成功", f"已加载设置: {name}")

    def _on_delete_clicked(self):
        name = self.load_combo.currentText().strip()
        if not name:
            QtWidgets.QMessageBox.warning(self, "删除失败", "请先选择一个已保存的设置名称。")
            return

        all_saved = self._read_saved_settings()
        if name not in all_saved:
            QtWidgets.QMessageBox.warning(self, "删除失败", "所选设置不存在。")
            self._load_saved_settings_list()
            return

        reply = QtWidgets.QMessageBox.question(
            self, "确认删除",
            f"确定要删除已保存设置: {name} ?",
            QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No
        )

        if reply != QtWidgets.QMessageBox.Yes:
            return

        del all_saved[name]
        ok = self._write_saved_settings(all_saved)
        if ok:
            QtWidgets.QMessageBox.information(self, "删除成功", f"已删除: {name}")
            self._load_saved_settings_list()
        else:
            QtWidgets.QMessageBox.warning(self, "删除失败", "删除时写入文件失败。")

    def _find_file_case_insensitive(self, target_name):
        target_lower = target_name.lower()
        for entry in os.listdir("."):
            if entry.lower() == target_lower:
                return entry
        return None

    def _build_cmd_list(self, exe_name):
        cmd_list = [exe_name]

        for k, (cb, widget) in self.controls.items():
            if not cb.isChecked():
                continue

            if k == "-n":
                cmd_list.append(k)
                cmd_list.append(str(widget.value()))
                continue

            if k in ("-httping", "-dd", "-allip"):
                cmd_list.append(k)
                continue

            val = widget.text().strip()
            if val == "":
                continue

            cmd_list.append(k)
            cmd_list.append(val)

        return cmd_list

    def _on_run_clicked(self):
        cfst_actual = self._find_file_case_insensitive(CFST_EXE)
        ip_actual = self._find_file_case_insensitive(IP_FILE_NAME)

        missing = []
        if not cfst_actual:
            missing.append(CFST_EXE)
        if not ip_actual:
            missing.append(IP_FILE_NAME)

        if missing:
            missing_str = "，".join(missing)
            QtWidgets.QMessageBox.warning(
                self, "文件缺失",
                f"未找到必要文件: {missing_str}\n请将缺失文件放在程序同目录后重试。"
            )
            return

        cmd_list = self._build_cmd_list(cfst_actual)
        if not cmd_list:
            cmd_list = [cfst_actual]

        try:
            if os.name == "nt":
                CREATE_NEW_CONSOLE = 0x00000010
                subprocess.Popen(cmd_list, creationflags=CREATE_NEW_CONSOLE)
            else:
                subprocess.Popen(cmd_list)
        except Exception:
            return

if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)

    # 全局图标
    app.setWindowIcon(QtGui.QIcon(APP_ICON))

    w = MainWin()
    w.show()
    sys.exit(app.exec_())