使用python来保存键盘输入情况,可保存到sqlite3数据库

1.代码单次保存最大键盘输入数目是300,全局变量可改

2、在gui界面可以设置单次保存的名字,方便下次查找,录入数据库

  1. gui界面有串口选择按钮,需要有硬件串口转hid模块ch9329的,可以直接发送串口数据来实现模拟键盘输入. 没有硬件的直接注释即可.

好了, 不说多,直接上代码

ch9329的github地址

ch9329 github

python 复制代码
import sys
import threading
import sqlite3
from PyQt5.QtWidgets import (QApplication, QComboBox,QWidget, QPushButton,QHBoxLayout, QVBoxLayout, QLineEdit, QLabel, QTextEdit)
from pynput import keyboard
from serial import Serial  # 串口模块
import serial.tools.list_ports  # 用于列出本地可用的串口


from ch9329 import keyboard as ch9329Keyboard
from ch9329 import mouse
from ch9329.config import get_manufacturer
from ch9329.config import get_product
from ch9329.config import get_serial_number


# 定义全局变量来存储键盘事件
key_events = []
MAX_EVENTS = 5 * 60  # 最大事件数量

# 定义键盘事件处理函数
def on_press(key):
    if len(key_events) < MAX_EVENTS:
        key_events.append(('pressed', str(key)))

def on_release(key):
    if len(key_events) < MAX_EVENTS:
        key_events.append(('released', str(key)))
    if str(key) == 'Key.esc':  # 如果按下的是Esc键,停止监听
        return False

# 启动键盘监听器的函数
def start_listener():
    with keyboard.Listener(on_press=on_press, on_release=on_release) as listener:
        listener.join()

# 保存数据到文件
def save_to_file():
    with open('key_events.txt', 'w') as file:
        for event in key_events:
            file.write(f"{event[0]} {event[1]}\n")

# 将数据保存到 SQLite 数据库
def save_to_database(name):
    if not name:
        return

    # 连接到 SQLite 数据库(如果不存在则创建)
    conn = sqlite3.connect('key_events.db')
    cursor = conn.cursor()

    # 创建表格(如果不存在)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS events (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT,
            event_type TEXT,
            key TEXT
        )
    ''')

    # 插入数据
    for event in key_events:
        cursor.execute('''
            INSERT INTO events (name, event_type, key)
            VALUES (?, ?, ?)
        ''', (name, event[0], event[1]))

    # 提交事务并关闭连接
    conn.commit()
    conn.close()

# 查找数据库中的数据
def find_in_database(name):
    if not name:
        return "Please enter a name."

    # 连接到 SQLite 数据库
    conn = sqlite3.connect('key_events.db')
    cursor = conn.cursor()

    # 查询匹配的记录
    cursor.execute('''
        SELECT * FROM events WHERE name = ?
    ''', (name,))
    results = cursor.fetchall()

    # 关闭连接
    conn.close()

    if results:
        return '\n'.join([f"{row[1]} - {row[2]} - {row[3]}" for row in results])
    else:
        return "No records found."

# 删除数据库中的数据
def delete_from_database(name):
    if not name:
        return "Please enter a name."

    # 连接到 SQLite 数据库
    conn = sqlite3.connect('key_events.db')
    cursor = conn.cursor()

    # 查询并删除匹配的记录
    cursor.execute('''
        SELECT * FROM events WHERE name = ?
    ''', (name,))
    results = cursor.fetchall()

    if results:
        cursor.execute('''
            DELETE FROM events WHERE name = ?
        ''', (name,))
        conn.commit()
        conn.close()
        return f"Deleted {len(results)} records for name: {name}"
    else:
        conn.close()
        return "No records found to delete."

# 串口功能:打开串口
def open_serial(com_port, baud_rate):
    global serial_port
    try:
        serial_port = Serial(com_port, baud_rate)
        return f"Serial port {com_port} opened at {baud_rate} baud."
    except Exception as e:
        return f"Failed to open serial port: {str(e)}"

# 获取本地可用的串口
def get_available_ports():
    ports = serial.tools.list_ports.comports()
    return [port.device for port in ports]

# 串口功能:关闭串口
def close_serial():
    global serial_port
    if serial_port and serial_port.is_open:
        serial_port.close()
        return "Serial port closed."
    else:
        return "No serial port is open."

# GUI 窗口类
class KeyboardListenerApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.parse_res = []
    def initUI(self):
        # 创建 UI 元素
        self.start_button = QPushButton('Start Listening', self)
        self.save_button = QPushButton('Save Data', self)
        self.find_button = QPushButton('Find Data', self)
        self.delete_button = QPushButton('Delete Data', self)  # 增加删除按钮
        self.open_serial_button = QPushButton('Open Serial', self)  # 打开串口按钮
        self.close_serial_button = QPushButton('Close Serial', self)  # 关闭串口按钮
        
        self.name_edit = QLineEdit(self)
        self.name_edit.setPlaceholderText("Enter your name")
        self.query_edit = QLineEdit(self)
        self.query_edit.setPlaceholderText("Enter name to search")
        self.results_area = QTextEdit(self)
        self.results_area.setReadOnly(True)

        self.send_serial_button = QPushButton('SendSerialData', self)  # 发送串口数据按钮
        # 串口选择和波特率选择
        self.com_port_select = QComboBox(self)
        self.baud_rate_select = QComboBox(self)
        self.baud_rate_select.addItems(["9600", "19200", "38400", "57600", "115200"])

        # 获取可用的串口
        self.update_ports()

        # 创建布局
        namelayout = QHBoxLayout()
        layout = QVBoxLayout()
        namelayout.addWidget(QLabel("Name:"))
        namelayout.addWidget(self.name_edit)
        layout.addLayout(namelayout)
        layout.addWidget(self.start_button)
        layout.addWidget(self.save_button)
        Searchnamelayout = QHBoxLayout()
        Searchnamelayout.addWidget(QLabel("Search Name:"))
        Searchnamelayout.addWidget(self.query_edit)
        layout.addLayout(Searchnamelayout)
        layout.addWidget(self.find_button)
        layout.addWidget(self.delete_button)  # 增加删除按钮到布局
        layout.addWidget(QLabel("Results:"))
        layout.addWidget(self.results_area)

        # add串口配置布局
        serial_layout = QHBoxLayout()
        serial_layout.addWidget(QLabel("COM Port:"))
        serial_layout.addWidget(self.com_port_select)
        serial_layout.addWidget(QLabel("Baud Rate:"))
        serial_layout.addWidget(self.baud_rate_select)

        layout.addLayout(serial_layout)
        layout.addWidget(self.open_serial_button)
        layout.addWidget(self.close_serial_button)
        layout.addWidget(self.send_serial_button)

        self.setLayout(layout)

        # 设置窗口属性
        self.setWindowTitle('Keyboard Listener')
        self.setGeometry(300, 300, 600, 400)

        # 绑定按钮事件
        self.start_button.clicked.connect(self.start_listening)
        self.save_button.clicked.connect(self.save_data)
        self.find_button.clicked.connect(self.find_data)
        self.delete_button.clicked.connect(self.delete_data)  # 绑定删除按钮事件
        self.open_serial_button.clicked.connect(self.open_serial)
        self.close_serial_button.clicked.connect(self.close_serial)
        # 绑定按钮事件
        self.send_serial_button.clicked.connect(self.send_serial_data)
    
    def update_ports(self):
        """更新可用的串口列表"""
        available_ports = get_available_ports()
        self.com_port_select.clear()
        self.com_port_select.addItems(available_ports)

        # 如果没有可用串口,禁用"Open Serial"按钮
        if not available_ports:
            self.open_serial_button.setEnabled(False)
        else:
            self.open_serial_button.setEnabled(True)
    
    def _send_serialdata(self,res,port):
        import time
        if res is None:
            return
        if len(res) == 0:
            return
        for i in res:
            ch9329Keyboard.press_and_release(port, i,min_interval=0.2,max_interval=0.3)
            time.sleep(0.5)
        
    def send_serial_data(self):
        global serial_port
        if serial_port and serial_port.is_open:
            """发送串口数据"""
            send_thread = threading.Thread(target=self._send_serialdata, args=(self.parse_res, serial_port), daemon=True)
            send_thread.start()

    def start_listening(self):
        # 创建并启动一个新的线程来运行键盘监听器
        listener_thread = threading.Thread(target=start_listener, daemon=True)
        listener_thread.start()

    def save_data(self):
        name = self.name_edit.text()
        if key_events:
            # 使用线程来保存数据到文件和数据库
            save_thread = threading.Thread(target=self.save_data_in_background, args=(name,), daemon=True)
            save_thread.start()

    def save_data_in_background(self, name):
        # 在后台线程中保存数据
        save_to_file()
        save_to_database(name)
        print("Data saved successfully.")
    def parse_and_execute(self,result):
        res = []
        """解析查找结果并执行按键事件"""
        # 将结果按行拆分
        lines = result.splitlines()
        print("解析查找结果并执行按键事件.")
        for line in lines:
            # 分割每行的内容
            parts = line.split(" - ")
            if len(parts) >= 3:
                action = parts[1]  # "pressed" 或 "released"
                key = parts[2]     # 键值('a' 或者 Key.ctrl_l)
                # 去除多余的引号,例如 'a' -> a
                key = key.strip("'")
                if len(key) >= 3:
                    key = key.strip("Key.")
                # 执行键盘操作
                if action == 'pressed':
                    #print(key)
                    res.append(key)
                    #keyboard_controller.press(key)
                #elif action == 'released':
                    #keyboard_controller.release(key)
            else:
                print("len < 3.")
        return res
    def find_data(self):
        name = self.query_edit.text()
        result = find_in_database(name)
        self.results_area.setText(result)
        self.parse_res = self.parse_and_execute(result)

    def delete_data(self):
        name = self.query_edit.text()
        result = delete_from_database(name)
        self.results_area.setText(result)
    def open_serial(self):
        com_port = self.com_port_select.currentText()
        baud_rate = int(self.baud_rate_select.currentText())
        result = open_serial(com_port, baud_rate)
        self.results_area.setText(result)

    def close_serial(self):
        result = close_serial()
        self.results_area.setText(result)
# 主函数
def main():
    app = QApplication(sys.argv)
    ex = KeyboardListenerApp()
    ex.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()
相关推荐
梧桐树04292 小时前
python常用内建模块:collections
python
Dream_Snowar2 小时前
速通Python 第三节
开发语言·python
蓝天星空3 小时前
Python调用open ai接口
人工智能·python
jasmine s3 小时前
Pandas
开发语言·python
郭wes代码3 小时前
Cmd命令大全(万字详细版)
python·算法·小程序
leaf_leaves_leaf3 小时前
win11用一条命令给anaconda环境安装GPU版本pytorch,并检查是否为GPU版本
人工智能·pytorch·python
夜雨飘零14 小时前
基于Pytorch实现的说话人日志(说话人分离)
人工智能·pytorch·python·声纹识别·说话人分离·说话人日志
404NooFound4 小时前
Python轻量级NoSQL数据库TinyDB
开发语言·python·nosql
天天要nx4 小时前
D102【python 接口自动化学习】- pytest进阶之fixture用法
python·pytest
minstbe4 小时前
AI开发:使用支持向量机(SVM)进行文本情感分析训练 - Python
人工智能·python·支持向量机