Python+PyQt5编写一个微信多开小工具

# 推荐使用Anaconda创建虚拟环境
conda create -n wechat python=3.8
conda install -c anaconda pyqt=5.15

class WeChatFinder(QThread):
    def search_registry(self):
        # 从HKCU读取安装路径
        with winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Software\Tencent\WeChat") as key:
            ...
    
    def search_path(self, path):
        # 使用os.walk进行深度搜索
        for root, dirs, files in os.walk(path):
            if "WeChat.exe" in files:
                return os.path.join(root, "WeChat.exe")

def launch_wechat_instance(self, exe_path):
    try:
        # 根据模式选择启动方式
        if mode == "sandbox":
            subprocess.Popen(["sandboxie", exe_path])
        else:
            subprocess.Popen([exe_path], creationflags=subprocess.CREATE_NEW_PROCESS_GROUP)
    except subprocess.CalledProcessError as e:
        logging.error(f"进程创建失败: {e.output.decode('gbk')}")

import sys
import os
import winreg
import logging
import ctypes
import json
from PyQt5.QtWidgets import (QApplication, QWidget, QVBoxLayout, QHBoxLayout, 
                            QPushButton, QLabel, QLineEdit, QSpinBox, 
                            QListWidget, QMessageBox, QProgressBar, QFileDialog,
                            QCheckBox, QComboBox)
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QSettings
from PyQt5.QtGui import QIcon, QFont

class WeChatFinder(QThread):
    found_wechat = pyqtSignal(str)
    search_complete = pyqtSignal()
    progress_update = pyqtSignal(int)
    
    def __init__(self):
        super().__init__()
        self._is_running = True
        
    def run(self):
        try:
            # 从注册表查找
            self.search_registry()
            
            # 从常见路径查找
            common_paths = self.get_common_paths()
            total_paths = len(common_paths)
            
            for i, path in enumerate(common_paths):
                if not self._is_running:
                    break
                self.search_path(path)
                self.progress_update.emit(int((i+1)/total_paths*100))
                
        except Exception as e:
            logging.error(f"搜索微信时出错: {str(e)}", exc_info=True)
        finally:
            self.search_complete.emit()
    
    def search_registry(self):
        try:
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Software\Tencent\WeChat") as key:
                install_path, _ = winreg.QueryValueEx(key, "InstallPath")
                if install_path:
                    exe_path = os.path.join(install_path, "WeChat.exe")
                    if os.path.exists(exe_path):
                        self.found_wechat.emit(exe_path)
        except WindowsError:
            pass
    
    def search_path(self, path):
        if os.path.exists(path):
            for root, dirs, files in os.walk(path):
                if not self._is_running:
                    break
                if "WeChat.exe" in files:
                    exe_path = os.path.join(root, "WeChat.exe")
                    if self.verify_wechat_exe(exe_path):
                        self.found_wechat.emit(exe_path)
                    return
    
    def get_common_paths(self):
        return [
            os.path.expanduser("~") + "\\AppData\\Local\\Tencent\\WeChat",
            "C:\\Program Files (x86)\\Tencent\\WeChat",
            "D:\\Program Files\\Tencent\\WeChat",
            "C:\\Program Files\\Tencent\\WeChat",
            "D:\\Program Files (x86)\\Tencent\\WeChat",
            os.path.join(os.environ.get("ProgramFiles", ""), "Tencent", "WeChat"),
            os.path.join(os.environ.get("ProgramFiles(x86)", ""), "Tencent", "WeChat")
        ]
    
    def verify_wechat_exe(self, exe_path):
        # 这里可以添加验证微信签名等安全检查
        return True
    
    def stop(self):
        self._is_running = False

class WeChatLauncher(QWidget):
    def __init__(self):
        super().__init__()
        self.settings = QSettings("WeChatMultiLauncher", "Settings")
        self.init_logging()
        self.initUI()
        self.load_settings()
        self.start_wechat_finder()
        
    def init_logging(self):
        logging.basicConfig(
            filename='wechat_launcher.log',
            level=logging.INFO,
            format='%(asctime)s - %(levelname)s - %(message)s'
        )
    
    def initUI(self):
        self.setWindowTitle('🐧 微信多开助手')
        self.setWindowIcon(QIcon('wechat.ico'))
        self.setFixedSize(700, 600)
        
        main_layout = QVBoxLayout()
        main_layout.setSpacing(15)
        main_layout.setContentsMargins(20, 20, 20, 20)
        
        # 标题
        title_label = QLabel('🐧 微信多开助手')
        title_label.setFont(QFont('Microsoft YaHei', 18, QFont.Bold))
        title_label.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(title_label)
        
        # 分隔线
        main_layout.addWidget(self.create_separator())
        
        # 微信路径部分
        main_layout.addLayout(self.create_path_section())
        
        # 多开设置部分
        main_layout.addLayout(self.create_settings_section())
        
        # 高级选项
        main_layout.addLayout(self.create_advanced_section())
        
        # 进度条
        self.progress_bar = QProgressBar()
        self.progress_bar.setRange(0, 100)
        self.progress_bar.setValue(0)
        self.progress_bar.setTextVisible(True)
        self.progress_bar.hide()
        main_layout.addWidget(self.progress_bar)
        
        # 按钮部分
        main_layout.addLayout(self.create_button_section())
        
        # 状态栏
        self.status_label = QLabel('⏳ 正在搜索微信安装路径...')
        self.status_label.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(self.status_label)
        
        self.setLayout(main_layout)
        
    def create_separator(self):
        separator = QLabel('━' * 50)
        separator.setAlignment(Qt.AlignCenter)
        return separator
    
    def create_path_section(self):
        path_layout = QVBoxLayout()
        
        path_title = QLabel('🔍 微信路径:')
        path_title.setFont(QFont('Microsoft YaHei', 10))
        path_layout.addWidget(path_title)
        
        self.path_list = QListWidget()
        self.path_list.setStyleSheet("QListWidget { border: 1px solid #ccc; border-radius: 5px; }")
        path_layout.addWidget(self.path_list)
        
        # 手动输入路径
        manual_layout = QHBoxLayout()
        manual_label = QLabel('📌 手动指定路径:')
        self.manual_path_input = QLineEdit()
        self.manual_path_input.setPlaceholderText('输入WeChat.exe完整路径...')
        browse_btn = QPushButton('浏览...')
        browse_btn.clicked.connect(self.browse_for_wechat)
        manual_layout.addWidget(manual_label)
        manual_layout.addWidget(self.manual_path_input)
        manual_layout.addWidget(browse_btn)
        path_layout.addLayout(manual_layout)
        
        return path_layout
    
    def create_settings_section(self):
        settings_layout = QHBoxLayout()
        
        settings_label = QLabel('⚙️ 多开设置:')
        settings_layout.addWidget(settings_label)
        
        count_label = QLabel('启动数量:')
        self.count_spin = QSpinBox()
        self.count_spin.setRange(1, 20)
        self.count_spin.setValue(2)
        settings_layout.addWidget(count_label)
        settings_layout.addWidget(self.count_spin)
        
        self.minimize_check = QCheckBox('启动后最小化')
        settings_layout.addWidget(self.minimize_check)
        
        settings_layout.addStretch()
        
        return settings_layout
    
    def create_advanced_section(self):
        advanced_layout = QHBoxLayout()
        
        advanced_label = QLabel('🔧 高级选项:')
        advanced_layout.addWidget(advanced_label)
        
        mode_label = QLabel('多开模式:')
        self.mode_combo = QComboBox()
        self.mode_combo.addItems(['普通多开', '沙盒模式', '不同配置'])
        advanced_layout.addWidget(mode_label)
        advanced_layout.addWidget(self.mode_combo)
        
        self.admin_check = QCheckBox('以管理员身份运行')
        advanced_layout.addWidget(self.admin_check)
        
        advanced_layout.addStretch()
        
        return advanced_layout
    
    def create_button_section(self):
        button_layout = QHBoxLayout()
        
        self.launch_btn = QPushButton('🚀 启动微信')
        self.launch_btn.setStyleSheet("QPushButton { background-color: #07C160; color: white; font-weight: bold; }")
        self.launch_btn.clicked.connect(self.launch_multiple_wechats)
        button_layout.addWidget(self.launch_btn)
        
        refresh_btn = QPushButton('🔄 重新搜索')
        refresh_btn.clicked.connect(self.start_wechat_finder)
        button_layout.addWidget(refresh_btn)
        
        settings_btn = QPushButton('⚙️ 设置')
        settings_btn.clicked.connect(self.show_settings)
        button_layout.addWidget(settings_btn)
        
        exit_btn = QPushButton('❌ 退出')
        exit_btn.clicked.connect(self.close)
        button_layout.addWidget(exit_btn)
        
        return button_layout
    
    def load_settings(self):
        # 加载保存的设置
        self.count_spin.setValue(self.settings.value("launch_count", 2, type=int))
        self.minimize_check.setChecked(self.settings.value("minimize", False, type=bool))
        self.admin_check.setChecked(self.settings.value("admin", False, type=bool))
        self.mode_combo.setCurrentIndex(self.settings.value("mode", 0, type=int))
        
        last_path = self.settings.value("last_path", "")
        if last_path:
            self.manual_path_input.setText(last_path)
    
    def save_settings(self):
        # 保存当前设置
        self.settings.setValue("launch_count", self.count_spin.value())
        self.settings.setValue("minimize", self.minimize_check.isChecked())
        self.settings.setValue("admin", self.admin_check.isChecked())
        self.settings.setValue("mode", self.mode_combo.currentIndex())
        
        selected_path = self.get_selected_path()
        if selected_path:
            self.settings.setValue("last_path", selected_path)
    
    def start_wechat_finder(self):
        self.path_list.clear()
        self.status_label.setText('⏳ 正在搜索微信安装路径...')
        self.progress_bar.show()
        self.progress_bar.setValue(0)
        
        self.finder = WeChatFinder()
        self.finder.found_wechat.connect(self.add_wechat_path)
        self.finder.search_complete.connect(self.on_search_complete)
        self.finder.progress_update.connect(self.progress_bar.setValue)
        self.finder.start()
        
    def add_wechat_path(self, path):
        if path not in [self.path_list.item(i).text() for i in range(self.path_list.count())]:
            self.path_list.addItem(path)
            
    def on_search_complete(self):
        self.progress_bar.hide()
        if self.path_list.count() == 0:
            self.status_label.setText('⚠️ 未找到微信安装路径，请手动指定')
        else:
            self.status_label.setText(f'✅ 找到 {self.path_list.count()} 个微信安装路径')
            self.path_list.setCurrentRow(0)
    
    def browse_for_wechat(self):
        file_path, _ = QFileDialog.getOpenFileName(
            self, 
            "选择WeChat.exe", 
            "", 
            "Executable Files (*.exe)"
        )
        
        if file_path:
            self.manual_path_input.setText(file_path)
            if file_path not in [self.path_list.item(i).text() for i in range(self.path_list.count())]:
                self.path_list.addItem(file_path)
                self.path_list.setCurrentRow(self.path_list.count()-1)
    
    def get_selected_path(self):
        if self.path_list.currentItem():
            return self.path_list.currentItem().text()
        elif self.manual_path_input.text() and os.path.exists(self.manual_path_input.text()):
            return self.manual_path_input.text()
        return None
    
    def is_admin(self):
        try:
            return ctypes.windll.shell32.IsUserAnAdmin()
        except:
            return False
    
    def launch_multiple_wechats(self):
        selected_path = self.get_selected_path()
        if not selected_path:
            self.show_error("请先选择或输入有效的微信路径！")
            return
        
        if self.admin_check.isChecked() and not self.is_admin():
            self.show_error("请以管理员身份运行此程序！")
            return
        
        count = self.count_spin.value()
        self.show_loading(f"正在启动 {count} 个微信实例...")
        
        success_count = 0
        for i in range(count):
            if self.launch_wechat_instance(selected_path):
                success_count += 1
                self.progress_bar.setValue(int((i+1)/count*100))
                QApplication.processEvents()
            else:
                break
        
        self.show_success(f"已启动 {success_count} 个微信实例")
        if self.minimize_check.isChecked():
            self.showMinimized()
        
        self.save_settings()
    
    def launch_wechat_instance(self, exe_path):
        try:
            if self.mode_combo.currentIndex() == 1:  # 沙盒模式
                # 这里可以添加沙盒启动逻辑
                subprocess.Popen([exe_path])
            elif self.mode_combo.currentIndex() == 2:  # 不同配置
                # 这里可以添加不同配置启动逻辑
                subprocess.Popen([exe_path])
            else:  # 普通模式
                subprocess.Popen([exe_path])
            return True
        except Exception as e:
            logging.error(f"启动微信失败: {str(e)}", exc_info=True)
            self.show_error(f"启动微信失败: {str(e)}")
            return False
    
    def show_error(self, message):
        QMessageBox.critical(self, "错误", message)
        self.status_label.setText(f'❌ {message}')
        self.progress_bar.hide()
    
    def show_loading(self, message):
        self.status_label.setText(f'⏳ {message}')
        self.progress_bar.show()
        self.progress_bar.setValue(0)
    
    def show_success(self, message):
        self.status_label.setText(f'✅ {message}')
        self.progress_bar.hide()
    
    def show_settings(self):
        # 这里可以扩展为更详细的设置对话框
        QMessageBox.information(self, "设置", "更多设置功能将在未来版本中添加")
    
    def closeEvent(self, event):
        self.save_settings()
        if hasattr(self, 'finder') and self.finder.isRunning():
            self.finder.stop()
            self.finder.wait()
        event.accept()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    app.setStyle('Fusion')
    
    # 设置全局字体
    font = QFont('Microsoft YaHei', 10)
    app.setFont(font)
    
    launcher = WeChatLauncher()
    launcher.show()
    sys.exit(app.exec_())