Python使用pynput模拟实现键盘自动输入工具

pip install pyqt5 pynput

class CountdownWindow(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowFlags(Qt.WindowStaysOnTopHint | Qt.FramelessWindowHint)
        self.setStyleSheet("background-color: black;")
        
        # 72pt超大红色数字
        font = QFont("Arial", 72, QFont.Bold)
        self.countdown_label.setFont(font)

class TypeThread(QThread):
    def run(self):
        keyboard = Controller()
        for char in self.text:
            if not self._is_running:  # 安全停止检查点
                break
            keyboard.type(char)  # 非阻塞式键入

from PyQt5.QtWidgets import (QApplication, QMainWindow, QVBoxLayout, QHBoxLayout, 
                            QPushButton, QLabel, QTextEdit, QWidget, QFileDialog, 
                            QMessageBox, QSpinBox, QCheckBox, QDialog)
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer
from PyQt5.QtGui import QFont
from pynput.keyboard import Controller
import time
import sys

class CountdownWindow(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("倒计时")
        self.setWindowFlags(Qt.WindowStaysOnTopHint | Qt.FramelessWindowHint)
        self.setStyleSheet("background-color: black;")
        
        layout = QVBoxLayout()
        self.countdown_label = QLabel("5")
        self.countdown_label.setAlignment(Qt.AlignCenter)
        self.countdown_label.setStyleSheet("color: red;")
        font = QFont("Arial", 72, QFont.Bold)
        self.countdown_label.setFont(font)
        
        layout.addWidget(self.countdown_label)
        self.setLayout(layout)
        
        # 设置窗口大小并居中
        self.resize(200, 200)
        self.move_to_center()
    
    def move_to_center(self):
        screen = QApplication.primaryScreen().geometry()
        x = (screen.width() - self.width()) // 2
        y = (screen.height() - self.height()) // 2
        self.move(x, y)
    
    def update_countdown(self, seconds):
        self.countdown_label.setText(str(seconds))
        if seconds <= 0:
            self.close()

class TypeThread(QThread):
    update_signal = pyqtSignal(str)
    finished_signal = pyqtSignal()

    def __init__(self, text):
        super().__init__()
        self.text = text
        self._is_running = True

    def run(self):
        keyboard = Controller()
        for char in self.text:
            if not self._is_running:
                break
            keyboard.type(char)
        self.finished_signal.emit()

    def stop(self):
        self._is_running = False

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("自动输入工具")
        self.setGeometry(100, 100, 600, 400)
        
        # 初始化线程和倒计时窗口
        self.thread = None
        self.countdown_timer = QTimer()
        self.countdown_seconds = 5
        self.countdown_window = None
        
        # 创建主部件和布局
        main_widget = QWidget()
        main_layout = QVBoxLayout()
        
        # 标题
        title = QLabel("📋 自动输入工具")
        title.setAlignment(Qt.AlignCenter)
        title.setStyleSheet("font-size: 20px; font-weight: bold;")
        main_layout.addWidget(title)
        
        # 说明
        info = QLabel("🔍 请将文本粘贴到下方或从文件导入")
        info.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(info)
        
        # 文本编辑区
        self.text_edit = QTextEdit()
        self.text_edit.setPlaceholderText("📝 请在此粘贴要输入的文本...")
        main_layout.addWidget(self.text_edit)
        
        # 控制区域
        control_layout = QHBoxLayout()
        
        # 窗口置顶复选框
        self.topmost_check = QCheckBox("📌 置顶")
        self.topmost_check.stateChanged.connect(self.toggle_topmost)
        control_layout.addWidget(self.topmost_check)
        
        # 按钮
        btn_layout = QHBoxLayout()
        
        self.load_btn = QPushButton("📂 导入文件")
        self.load_btn.clicked.connect(self.load_from_file)
        btn_layout.addWidget(self.load_btn)
        
        self.start_btn = QPushButton("📝 开始输入")
        self.start_btn.clicked.connect(self.start_typing)
        btn_layout.addWidget(self.start_btn)
        
        self.stop_btn = QPushButton("⚡ 停止输入")
        self.stop_btn.clicked.connect(self.stop_typing)
        self.stop_btn.setEnabled(False)
        btn_layout.addWidget(self.stop_btn)
        
        control_layout.addLayout(btn_layout)
        main_layout.addLayout(control_layout)
        
        # 状态栏
        self.status_bar = QLabel("🟢 准备就绪")
        self.status_bar.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(self.status_bar)
        
        main_widget.setLayout(main_layout)
        self.setCentralWidget(main_widget)
        
        # 样式
        self.setStyleSheet("""
            QPushButton {
                padding: 8px;
                font-size: 14px;
            }
            QTextEdit {
                font-size: 14px;
            }
        """)
        
        # 连接倒计时定时器
        self.countdown_timer.timeout.connect(self.update_countdown)
    
    def toggle_topmost(self, state):
        if state == Qt.Checked:
            self.setWindowFlags(self.windowFlags() | Qt.WindowStaysOnTopHint)
        else:
            self.setWindowFlags(self.windowFlags() & ~Qt.WindowStaysOnTopHint)
        self.show()
    
    def start_countdown(self):
        self.countdown_seconds = 5
        self.countdown_window = CountdownWindow(self)
        self.countdown_window.show()
        self.countdown_timer.start(1000)  # 1秒间隔
        self.status_bar.setText(f"⏳ 倒计时: {self.countdown_seconds}秒...")  # 更新状态栏
    
    def update_countdown(self):
        self.countdown_seconds -= 1
        self.countdown_window.update_countdown(self.countdown_seconds)
        self.status_bar.setText(f"⏳ 倒计时: {self.countdown_seconds}秒...")  # 更新状态栏
        
        if self.countdown_seconds <= 0:
            self.countdown_timer.stop()
            QTimer.singleShot(500, self.countdown_window.close)  # 稍后关闭窗口
            self._start_typing_after_delay()
    
    def load_from_file(self):
        file_path, _ = QFileDialog.getOpenFileName(self, "📂 打开文本文件", "", "文本文件 (*.txt);;所有文件 (*)")
        if file_path:
            try:
                # 尝试用UTF-8编码打开，如果失败则用GBK编码
                try:
                    with open(file_path, 'r', encoding='utf-8') as f:
                        data = f.read()
                except UnicodeDecodeError:
                    try:
                        with open(file_path, 'r', encoding='gbk') as f:
                            data = f.read()
                    except UnicodeDecodeError:
                        with open(file_path, 'r', encoding='utf-8', errors='replace') as f:
                            data = f.read()
                
                self.text_edit.setPlainText(data)
                self.status_bar.setText(f"✅ 成功从 {file_path} 加载文本")
            except Exception as e:
                QMessageBox.critical(self, "❌ 错误", f"无法读取文件: {str(e)}")
                self.status_bar.setText("❌ 文件加载失败")
    
    def start_typing(self):
        if not self.text_edit.toPlainText().strip():
            QMessageBox.warning(self, "⚠️ 警告", "请输入或导入文本内容！")
            return
            
        self.status_bar.setText("⏳ 准备开始输入...")
        self.start_btn.setEnabled(False)
        self.stop_btn.setEnabled(True)
        self.load_btn.setEnabled(False)
        
        # 开始倒计时
        self.start_countdown()
    
    def _start_typing_after_delay(self):
        text = self.text_edit.toPlainText()
        
        self.thread = TypeThread(text)
        self.thread.finished_signal.connect(self.typing_finished)
        self.thread.start()
        
        self.status_bar.setText("⌨️ 正在输入中...请勿操作键盘！")
    
    def stop_typing(self):
        if self.thread and self.thread.isRunning():
            self.thread.stop()
            self.thread.wait()
        
        # 如果倒计时还在进行也停止
        if self.countdown_timer.isActive():
            self.countdown_timer.stop()
            if self.countdown_window:
                self.countdown_window.close()
        
        self.typing_finished()
        self.status_bar.setText("🟡 输入已停止")
    
    def typing_finished(self):
        self.start_btn.setEnabled(True)
        self.stop_btn.setEnabled(False)
        self.load_btn.setEnabled(True)
        self.status_bar.setText("✅ 输入完成！")
    
    def closeEvent(self, event):
        # 停止所有正在进行的操作
        if self.thread and self.thread.isRunning():
            self.thread.stop()
            self.thread.wait()
        
        if self.countdown_timer.isActive():
            self.countdown_timer.stop()
        
        if self.countdown_window:
            self.countdown_window.close()
        
        event.accept()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())