164 lines
6.2 KiB
Python
164 lines
6.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import annotations
|
|
import sys, os, io, csv, time, datetime
|
|
from typing import Callable, Optional
|
|
|
|
# PyQt5
|
|
from PyQt5 import QtWidgets, uic
|
|
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer, QSize
|
|
from PyQt5.QtGui import QImage, QPixmap, QKeySequence
|
|
from PyQt5.QtWidgets import (
|
|
QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
|
|
QLabel, QPushButton, QComboBox, QTextEdit, QFileDialog, QMessageBox,
|
|
QGroupBox, QGridLayout, QDialog, QFormLayout, QSpinBox, QCheckBox,
|
|
QLineEdit, QTableWidget, QTableWidgetItem,QDialogButtonBox,QShortcut, QHeaderView
|
|
)
|
|
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
import cv2
|
|
from PIL import Image
|
|
import serial
|
|
import serial.tools.list_ports
|
|
|
|
# ---------- 协议导入----------
|
|
from Shared_CODE.FaceRecognitionProtocol import (
|
|
build_reset, build_uvc_view, build_face_view, build_verify,
|
|
build_enroll_itg_single, build_delete_all, build_get_all_userid,build_delete_user,
|
|
MID_REPLY, MID_NOTE, CMD_ENROLL, CMD_ENROLL_ITG,
|
|
parse_reply, parse_note
|
|
)
|
|
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
|
|
ui_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../Shared_UI'))
|
|
users_ui_file_path = os.path.join(ui_path, "DialogUsers.ui")
|
|
|
|
CSV_FILE = os.path.join(ui_path, "users.csv")
|
|
# -------------------- CSV 工具 --------------------"
|
|
|
|
def load_users():
|
|
users = []
|
|
try:
|
|
with open(CSV_FILE, "r", encoding="utf-8") as f:
|
|
reader = csv.reader(f)
|
|
for row in reader:
|
|
if len(row) >= 2:
|
|
user = {"user_id": row[0], "user_name": row[1]}
|
|
user["created_at"] = row[2] if len(row) >= 3 else ""
|
|
users.append(user)
|
|
except FileNotFoundError:
|
|
pass
|
|
return users
|
|
|
|
|
|
def save_users_list(users):
|
|
with open(CSV_FILE, "w", newline="", encoding="utf-8") as f:
|
|
w = csv.writer(f)
|
|
for u in users:
|
|
w.writerow([u.get("user_id", ""), u.get("user_name", ""), u.get("created_at", "")])
|
|
|
|
|
|
def save_user(user_id: int, user_name: str) -> bool:
|
|
users = load_users()
|
|
for u in users:
|
|
if str(u["user_id"]) == str(user_id):
|
|
return False
|
|
created_at = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
users.append({"user_id": str(user_id), "user_name": user_name, "created_at": created_at})
|
|
save_users_list(users)
|
|
return True
|
|
|
|
class UserManageDialog(QDialog):
|
|
def __init__(self, parent=None, send_func=None):
|
|
super().__init__(parent)
|
|
self.send_func = send_func
|
|
uic.loadUi(users_ui_file_path, self)
|
|
|
|
self.btn_delete.clicked.connect(self.delete_selected)
|
|
self.btn_refresh.clicked.connect(self.refresh)
|
|
self.btn_get.clicked.connect(self.get_from_device)
|
|
self.btn_del_all.clicked.connect(self.delete_all_users)
|
|
self.refresh()
|
|
|
|
header = self.table.horizontalHeader()
|
|
# 用户ID列固定宽度
|
|
self.table.setColumnWidth(0, 100)
|
|
# 用户名列自适应内容
|
|
header.setSectionResizeMode(1, QHeaderView.ResizeToContents)
|
|
# 注册时间列填充剩余空间
|
|
header.setSectionResizeMode(2, QHeaderView.Stretch)
|
|
|
|
self.message_timer = QTimer()
|
|
self.message_timer.timeout.connect(self.close_dialog_timeout)
|
|
|
|
self.setWindowTitle("消息提示")
|
|
self.setWindowFlag(Qt.FramelessWindowHint)
|
|
button_box : QDialogButtonBox = self.buttonBox
|
|
ok_button = button_box.button(QDialogButtonBox.Ok)
|
|
if ok_button:
|
|
ok_button.setText('确定')
|
|
else:
|
|
button_box.addButton(QDialogButtonBox.Ok)
|
|
#定义4个快捷键, Key_Enter, Key_Escape经常被各类程序模块使用, 用Key_End, 与Key_Home来代替
|
|
QShortcut(QKeySequence(Qt.Key_PageDown), self, activated=self.key_enter_process)
|
|
QShortcut(QKeySequence(Qt.Key_PageUp), self, activated=self.key_escape_process)
|
|
QShortcut(QKeySequence(Qt.Key_End), self, activated=self.key_enter_process)
|
|
QShortcut(QKeySequence(Qt.Key_Home), self, activated=self.key_escape_process)
|
|
|
|
QShortcut(QKeySequence(Qt.Key_Return), self, activated=self.key_enter_process) # 普通回车
|
|
|
|
def refresh(self):
|
|
users = load_users()
|
|
self.table.setRowCount(len(users))
|
|
for r, u in enumerate(users):
|
|
self.table.setItem(r, 0, QTableWidgetItem(str(u.get("user_id", ""))))
|
|
self.table.setItem(r, 1, QTableWidgetItem(u.get("user_name", "")))
|
|
self.table.setItem(r, 2, QTableWidgetItem(u.get("created_at", "")))
|
|
|
|
def delete_selected(self):
|
|
row = self.table.currentRow()
|
|
if row < 0:
|
|
QMessageBox.warning(self, "提示", "删除选择用户")
|
|
return
|
|
uid = self.table.item(row, 0).text()
|
|
uname = self.table.item(row, 1).text()
|
|
try:
|
|
uid_int = int(uid)
|
|
self.send_func(build_delete_user(uid_int))
|
|
except:
|
|
pass
|
|
users = [u for u in load_users() if str(u["user_id"]) != uid]
|
|
save_users_list(users)
|
|
self.refresh()
|
|
QMessageBox.information(self, "提示", f"用户 {uname}(ID={uid}) 已删除")
|
|
|
|
def get_from_device(self):
|
|
self.send_func(build_get_all_userid())
|
|
|
|
def delete_all_users(self):
|
|
self.send_func(build_delete_all())
|
|
save_users_list([])
|
|
self.refresh()
|
|
QMessageBox.information(self, "提示", "已请求删除所有用户并清空本地记录")
|
|
|
|
def key_enter_process(self):
|
|
button_box : QDialogButtonBox = self.buttonBox
|
|
select_button = button_box.button(QDialogButtonBox.Ok)
|
|
if select_button:
|
|
select_button.click()
|
|
|
|
def key_escape_process(self):
|
|
button_box : QDialogButtonBox = self.buttonBox
|
|
select_button = button_box.button(QDialogButtonBox.Cancel)
|
|
if select_button:
|
|
select_button.click()
|
|
|
|
#消息超时
|
|
def close_dialog_timeout(self):
|
|
self.key_enter_process()
|
|
|
|
if __name__ == '__main__':
|
|
app = QApplication(sys.argv)
|
|
dialog = UserManageDialog()
|
|
dialog.exec()
|
|
sys.exit(0)
|