当前位置：首页 > article >正文

程序代码篇---SQLite数据库存储信息

article 2025/3/25 14:02:58

文章目录

前言
1.代码
2.代码改进说明
- 2.1OCR识别线程
- 2.2权限整合
3.摄像头集成
4.自动去重添加
5.界面功能扩展
- 5.1手动添加快递表单
- 5.2数据表格更新
- 5.3管理员操作菜单
6.运行准备
- 6.1安装依赖
- 6.2摄像头权限
- 6.3如果需要，将用户加入video组
7.扩展功能建议
- 7.1识别优化
- 7.2性能优化
- 7.3记录查看筛选
- 7.4数据导出
8.功能
9.运行效果

前言

本文简单介绍了SQLite的使用，存储信息并进行实时更新。

1.代码

import sys
import cv2
import sqlite3
import numpy as np
from datetime import datetime
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer
from PyQt5.QtGui import QImage, QPixmap
from PyQt5.QtWidgets import (QApplication, QWidget, QLabel, QLineEdit, QPushButton,
                             QVBoxLayout, QHBoxLayout, QMessageBox, QTableWidget,
                             QTableWidgetItem, QHeaderView, QComboBox)

from paddleocr import PaddleOCR

#初始化PaddleOCR（使用中英文超轻量模型）
ocr = PaddleOCR(use_angle_cls=True, lang='ch', use_gpu=False)

#======================
#摄像头OCR识别线程
#======================
class OCRThread(QThread):
    result_signal = pyqtSignal(str)  # 识别结果信号

    def __init__(self):
        super().__init__()
        self.running = False
        self.cap = cv2.VideoCapture(0)  # 打开默认摄像头

    def run(self):
        self.running = True
        while self.running:
            ret, frame = self.cap.read()
            if ret:
                # OCR识别
                results = ocr.ocr(frame, cls=True)
                for line in results:
                    if line and len(line) >= 1:
                        text = line[0][1][0]  # 提取识别文本
                        if len(text) >= 10:  # 假设快递单号长度至少10位
                            self.result_signal.emit(text)
                            break

    def stop(self):
        self.running = False
        self.cap.release()

#======================
#数据库管理类（带权限控制）
#======================
class ExpressManager:
    def __init__(self, current_user):
        self.conn = sqlite3.connect('express_delivery.db')
        self.current_user = current_user
        self.is_admin = self._check_admin()

    def _check_admin(self):
        cursor = self.conn.execute("SELECT role FROM users WHERE username=?", 
                                  (self.current_user,))
        return cursor.fetchone()[0] == 'admin'

    # 其他数据库方法保持之前实现，添加权限检查
    def delete_express(self, tracking_num):
        if not self.is_admin:
            raise PermissionError("需要管理员权限")
        # 删除逻辑...

#======================
#主界面
#======================
class MainWindow(QWidget):
    def __init__(self, username):
        super().__init__()
        self.username = username
        self.manager = ExpressManager(username)
        self.init_ui()
        self.init_camera()

    def init_ui(self):
        self.setWindowTitle(f"快递管理系统 - {self.username}")
        self.setGeometry(100, 100, 800, 600)

        # 创建控件
        self.table = QTableWidget()
        self.table.setColumnCount(4)
        self.table.setHorizontalHeaderLabels(["快递单号", "用户名", "加急状态", "创建时间"])
        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        # 摄像头显示区域
        self.cam_label = QLabel()
        self.cam_label.setFixedSize(640, 480)

        # 控制按钮
        self.btn_start = QPushButton("开始扫描")
        self.btn_stop = QPushButton("停止扫描")
        self.btn_add = QPushButton("手动添加")
        
        # 管理员功能
        if self.manager.is_admin:
            self.btn_delete = QPushButton("删除记录")
            self.btn_update = QPushButton("修改状态")

        # 布局
        control_layout = QHBoxLayout()
        control_layout.addWidget(self.btn_start)
        control_layout.addWidget(self.btn_stop)
        control_layout.addWidget(self.btn_add)

        main_layout = QVBoxLayout()
        main_layout.addLayout(control_layout)
        main_layout.addWidget(self.cam_label)
        main_layout.addWidget(self.table)

        self.setLayout(main_layout)
        self.load_data()

    def init_camera(self):
        self.timer = QTimer()
        self.timer.timeout.connect(self.update_frame)
        self.ocr_thread = OCRThread()
        self.ocr_thread.result_signal.connect(self.handle_ocr_result)

    def handle_ocr_result(self, text):
        """处理OCR识别结果"""
        # 去重检查
        existing = self.manager.search_express(tracking_num=text)
        if not existing:
            self.manager.add_express(text, self.username)
            self.load_data()

    def update_frame(self):
        """更新摄像头画面"""
        ret, frame = self.ocr_thread.cap.read()
        if ret:
            img = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            h, w, ch = img.shape
            q_img = QImage(img.data, w, h, QImage.Format_RGB888)
            self.cam_label.setPixmap(QPixmap.fromImage(q_img))

    # 其他方法：load_data, 按钮事件处理等...

#======================
#登录窗口
#======================
class LoginWindow(QWidget):
    def __init__(self):
        super().__init__()
        # 保持之前的登录逻辑...
        
    def handle_login_success(self, username, role):
        self.main_window = MainWindow(username)
        self.main_window.show()
        self.close()

#======================
#运行程序
#======================
if __name__ == "__main__":
    app = QApplication(sys.argv)
    
    # 初始化数据库
    init_database()
    
    window = LoginWindow()
    window.show()
    sys.exit(app.exec_())

2.代码改进说明

2.1OCR识别线程

class OCRThread(QThread):
    result_signal = pyqtSignal(str)

使用独立线程处理摄像头和OCR识别
通过信号传递识别结果，避免阻塞GUI主线程

2.2权限整合

class ExpressManager:
    def __init__(self, current_user):
        self.is_admin = self._check_admin()

在数据库操作类中增加权限检查
管理员才能执行删除/修改操作

3.摄像头集成

def init_camera(self):
    self.timer = QTimer()
    self.timer.timeout.connect(self.update_frame)

使用QTimer定时更新摄像头画面
OpenCV捕获视频流，转换为Qt图像格式显示

4.自动去重添加

def handle_ocr_result(self, text):
    existing = self.manager.search_express(tracking_num=text)
    if not existing:
        self.manager.add_express(text, self.username)

自动检查快递单号是否已存在
只添加新快递单号到数据库

5.界面功能扩展

5.1手动添加快递表单

class AddExpressDialog(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("手动添加")
        layout = QFormLayout()
        self.tracking_input = QLineEdit()
        self.urgent_combo = QComboBox()
        self.urgent_combo.addItems(["普通", "加急"])
        layout.addRow("快递单号:", self.tracking_input)
        layout.addRow("加急状态:", self.urgent_combo)
        # 添加确认按钮...

5.2数据表格更新

def load_data(self):
    self.table.setRowCount(0)
    data = self.manager.search_express()
    for row, record in enumerate(data):
        self.table.insertRow(row)
        self.table.setItem(row, 0, QTableWidgetItem(record[0]))
        # 填充其他列...

5.3管理员操作菜单

if self.manager.is_admin:
    menu = self.table.createContextMenuPolicy()
    menu.addAction("修改状态").triggered.connect(self.edit_record)
    menu.addAction("删除记录").triggered.connect(self.delete_record)

6.运行准备

6.1安装依赖

sudo apt install python3-opencv
pip install paddleocr paddlepaddle pyqt5

6.2摄像头权限

#检查视频设备权限
ls -l /dev/video*

6.3如果需要，将用户加入video组

sudo usermod -aG video $USER

7.扩展功能建议

7.1识别优化

添加图像预处理（灰度化、二值化）
使用ROI区域限定识别范围
添加后处理验证快递单号格式

7.2性能优化

#在OCRThread中设置识别间隔
def run(self):
    while self.running:
        # 每0.5秒识别一次
        time.sleep(0.5)
        # 识别逻辑...

7.3记录查看筛选

self.filter_combo = QComboBox()
self.filter_combo.addItems(["全部", "加急件", "今日新增"])
self.filter_combo.currentIndexChanged.connect(self.apply_filter)

7.4数据导出

def export_to_csv(self):
    with open('export.csv', 'w') as f:
        writer = csv.writer(f)
        writer.writerow(["单号", "用户", "加急", "时间"])
        for row in self.manager.search_express():
            writer.writerow(row)