From 540c3b0e47c858701f38f1adf808c91472bd7771 Mon Sep 17 00:00:00 2001
From: scout <1134087124@qq.com>
Date: Wed, 4 Dec 2024 11:53:38 +0800
Subject: [PATCH] init
---
requirements.txt | 6 +
start.bat | 7 +
workspace_monitor.py | 425 +++++++++++++++++++++++++++++++++++++++++++
3 files changed, 438 insertions(+)
create mode 100644 requirements.txt
create mode 100644 start.bat
create mode 100644 workspace_monitor.py
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..6f27204
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+opencv-python==4.8.1.78
+mediapipe==0.10.8
+PyQt6==6.6.1
+PyQt6-Qt6==6.6.1
+PyQt6-sip==13.6.0
+PyQt6-WebEngine==6.6.0
\ No newline at end of file
diff --git a/start.bat b/start.bat
new file mode 100644
index 0000000..71946b0
--- /dev/null
+++ b/start.bat
@@ -0,0 +1,7 @@
+@echo off
+echo Installing/Updating requirements...
+"C:\Users\user\AppData\Local\Microsoft\WindowsApps\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\python.exe" -m pip install -r requirements.txt
+
+echo Starting application...
+"C:\Users\user\AppData\Local\Microsoft\WindowsApps\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\python.exe" workspace_monitor.py
+pause
\ No newline at end of file
diff --git a/workspace_monitor.py b/workspace_monitor.py
new file mode 100644
index 0000000..da21a80
--- /dev/null
+++ b/workspace_monitor.py
@@ -0,0 +1,425 @@
+import cv2
+import time
+import ctypes
+import sys
+import traceback
+import logging
+import os
+import mediapipe as mp
+from datetime import datetime
+from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
+ QHBoxLayout, QPushButton, QLabel, QSpinBox,
+ QStyle, QProgressBar, QSystemTrayIcon, QMenu)
+from PyQt6.QtCore import Qt, QTimer
+from PyQt6.QtGui import QImage, QPixmap, QIcon, QFont
+
+# 设置日志
+logging.basicConfig(
+ level=logging.DEBUG,
+ format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+ handlers=[
+ logging.FileHandler('app.log'),
+ logging.StreamHandler()
+ ]
+)
+logger = logging.getLogger(__name__)
+
+def handle_exception(exc_type, exc_value, exc_traceback):
+ logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
+ sys.__excepthook__(exc_type, exc_value, exc_traceback)
+
+sys.excepthook = handle_exception
+
+try:
+ logger.info("Starting application...")
+
+ class WorkspaceMonitor(QMainWindow):
+ def __init__(self):
+ try:
+ super().__init__()
+ logger.info("Initializing WorkspaceMonitor...")
+ self.setWindowTitle("自动锁屏")
+ self.setMinimumSize(1000, 700)
+
+ # 系统托盘
+ self.tray_icon = QSystemTrayIcon(self)
+ self.tray_icon.setIcon(self.style().standardIcon(QStyle.StandardPixmap.SP_ComputerIcon))
+ self.create_tray_menu()
+ self.tray_icon.show()
+
+ # 初始化变量
+ self.running = False
+ self.timeout = 10
+ logger.info("Initializing MediaPipe...")
+ self.mp_face_detection = mp.solutions.face_detection
+ self.mp_drawing = mp.solutions.drawing_utils
+ self.face_detection = self.mp_face_detection.FaceDetection(
+ model_selection=1,
+ min_detection_confidence=0.5
+ )
+
+ logger.info("Opening camera...")
+ self.cap = cv2.VideoCapture(0)
+ if not self.cap.isOpened():
+ logger.error("Failed to open camera!")
+ raise Exception("无法打开摄像头")
+
+ self.last_face_time = time.time()
+ self.no_face_duration = 0
+
+ self.setup_ui()
+ logger.info("Initialization complete.")
+
+ except Exception as e:
+ logger.error(f"Error in WorkspaceMonitor initialization: {str(e)}")
+ logger.error(traceback.format_exc())
+ raise
+
+ def setup_ui(self):
+ central_widget = QWidget()
+ self.setCentralWidget(central_widget)
+ main_layout = QVBoxLayout(central_widget)
+
+ # 标题
+ title_label = QLabel("自动锁屏")
+ title_label.setFont(QFont('Arial', 16, QFont.Weight.Bold))
+ title_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+ main_layout.addWidget(title_label)
+
+ # 摄像头预览
+ preview_container = QWidget()
+ preview_container.setStyleSheet("background-color: #f0f0f0; border-radius: 10px;")
+ preview_layout = QVBoxLayout(preview_container)
+
+ self.camera_label = QLabel()
+ self.camera_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+ self.camera_label.setMinimumHeight(480)
+ preview_layout.addWidget(self.camera_label)
+
+ main_layout.addWidget(preview_container)
+
+ # 控制面板
+ control_panel = QWidget()
+ control_panel.setStyleSheet("""
+ QWidget {
+ background-color: #ffffff;
+ border-radius: 10px;
+ padding: 10px;
+ }
+ QPushButton {
+ background-color: #4CAF50;
+ color: white;
+ border-radius: 5px;
+ padding: 8px;
+ min-width: 100px;
+ }
+ QPushButton:hover {
+ background-color: #45a049;
+ }
+ QLabel {
+ color: #333333;
+ }
+ """)
+
+ control_layout = QHBoxLayout(control_panel)
+
+ # 开始/停止按钮
+ self.start_stop_btn = QPushButton("开始监控")
+ self.start_stop_btn.clicked.connect(self.toggle_monitoring)
+ control_layout.addWidget(self.start_stop_btn)
+
+ # 超时设置
+ timeout_container = QWidget()
+ timeout_layout = QHBoxLayout(timeout_container)
+
+ timeout_label = QLabel("锁屏等待时间:")
+ self.timeout_spinbox = QSpinBox()
+ self.timeout_spinbox.setRange(5, 300)
+ self.timeout_spinbox.setValue(10)
+ self.timeout_spinbox.setSuffix(" 秒")
+ self.timeout_spinbox.valueChanged.connect(self.update_timeout)
+
+ timeout_layout.addWidget(timeout_label)
+ timeout_layout.addWidget(self.timeout_spinbox)
+ control_layout.addWidget(timeout_container)
+
+ # 状态显示
+ status_container = QWidget()
+ status_layout = QVBoxLayout(status_container)
+
+ self.status_label = QLabel("状态: 未开始监控")
+ self.status_label.setStyleSheet("""
+ QLabel {
+ font-weight: bold;
+ font-size: 14px;
+ color: #333333;
+ margin-bottom: 5px;
+ }
+ """)
+
+ self.confidence_label = QLabel("置信度: 0%")
+ self.confidence_label.setStyleSheet("""
+ QLabel {
+ color: #666666;
+ margin-bottom: 5px;
+ }
+ """)
+
+ # 进度条容器
+ progress_container = QWidget()
+ progress_container.setFixedHeight(45)
+ progress_container.setStyleSheet("""
+ QWidget {
+ background-color: transparent;
+ padding: 5px;
+ }
+ """)
+ progress_layout = QVBoxLayout(progress_container)
+ progress_layout.setContentsMargins(0, 0, 0, 0)
+
+ # 进度条
+ self.progress_bar = QProgressBar()
+ self.progress_bar.setRange(0, 100)
+ self.progress_bar.setValue(0)
+ self.progress_bar.setFormat("离开时间: 0s")
+ self.progress_bar.setMinimumWidth(250)
+ self.progress_bar.setFixedHeight(20)
+ self.set_progress_bar_style()
+
+ progress_layout.addWidget(self.progress_bar)
+
+ status_layout.addWidget(self.status_label)
+ status_layout.addWidget(self.confidence_label)
+ status_layout.addWidget(progress_container)
+
+ control_layout.addWidget(status_container)
+
+ main_layout.addWidget(control_panel)
+
+ self.timer = QTimer()
+ self.timer.timeout.connect(self.update_frame)
+
+ def create_tray_menu(self):
+ menu = QMenu()
+ show_action = menu.addAction("显示主窗口")
+ show_action.triggered.connect(self.show)
+ quit_action = menu.addAction("退出")
+ quit_action.triggered.connect(self.close)
+ self.tray_icon.setContextMenu(menu)
+ self.tray_icon.activated.connect(self.tray_icon_activated)
+
+ def tray_icon_activated(self, reason):
+ if reason == QSystemTrayIcon.ActivationReason.DoubleClick:
+ self.show()
+
+ def update_frame(self):
+ # 检查摄像头是否连接
+ if not self.cap.isOpened():
+ self.handle_camera_disconnected()
+ return
+
+ ret, frame = self.cap.read()
+ if not ret:
+ self.handle_camera_disconnected()
+ return
+
+ frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+ results = self.face_detection.process(frame_rgb)
+
+ max_confidence = 0
+ current_time = time.time()
+
+ if results.detections:
+ for detection in results.detections:
+ confidence = detection.score[0]
+ max_confidence = max(max_confidence, confidence)
+
+ bbox = detection.location_data.relative_bounding_box
+ h, w, _ = frame.shape
+ x = int(bbox.xmin * w)
+ y = int(bbox.ymin * h)
+ width = int(bbox.width * w)
+ height = int(bbox.height * h)
+
+ cv2.rectangle(frame, (x, y), (x + width, y + height), (0, 255, 0), 2)
+ cv2.putText(frame, f"置信度: {confidence:.2f}",
+ (x, y-10), cv2.FONT_HERSHEY_SIMPLEX,
+ 0.5, (0, 255, 0), 2)
+
+ self.last_face_time = current_time
+ self.no_face_duration = 0
+ self.status_label.setText("状态: 检测到人脸")
+ self.confidence_label.setText(f"置信度: {max_confidence*100:.1f}%")
+ self.progress_bar.setValue(0)
+ self.progress_bar.setFormat("离开时间: 0s")
+ self.set_progress_bar_style("#4CAF50")
+ else:
+ self.no_face_duration = current_time - self.last_face_time
+ self.status_label.setText(f"状态: 未检测到人脸 ({int(self.no_face_duration)}秒)")
+ self.confidence_label.setText("置信度: 0%")
+
+ # 计算进度并更新进度条
+ progress = min(int(self.no_face_duration / self.timeout * 100), 100)
+ self.progress_bar.setValue(progress)
+ self.progress_bar.setFormat(f"离开时间: {int(self.no_face_duration)}s")
+
+ # 根据进度设置颜色
+ if progress > 70:
+ self.set_progress_bar_style("#FF4444") # 红色
+ elif progress > 30:
+ self.set_progress_bar_style("#FFA726") # 橙色
+ else:
+ self.set_progress_bar_style("#4CAF50") # 绿色
+
+ if self.no_face_duration > self.timeout:
+ self.status_label.setText(f"状态: 锁定屏幕 - {datetime.now()}")
+ self.tray_icon.showMessage("工位监控", "系统即将锁定", QSystemTrayIcon.MessageIcon.Warning)
+ self.lock_windows()
+ # 重置所有状态
+ self.last_face_time = time.time()
+ self.no_face_duration = 0
+ self.progress_bar.setValue(0)
+ self.progress_bar.setFormat("离开时间: 0s")
+ self.set_progress_bar_style("#4CAF50") # 重置为绿色
+ time.sleep(1)
+
+ rgb_display = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+ h, w, ch = rgb_display.shape
+ bytes_per_line = ch * w
+ qt_image = QImage(rgb_display.data, w, h, bytes_per_line, QImage.Format.Format_RGB888)
+ scaled_pixmap = QPixmap.fromImage(qt_image).scaled(
+ self.camera_label.size(),
+ Qt.AspectRatioMode.KeepAspectRatio
+ )
+ self.camera_label.setPixmap(scaled_pixmap)
+
+ def toggle_monitoring(self):
+ if not self.running:
+ self.running = True
+ self.start_stop_btn.setText("停止监控")
+ self.start_stop_btn.setStyleSheet("background-color: #ff4444;")
+ self.status_label.setText("状态: 正在监控")
+ self.timer.start(30)
+ self.tray_icon.showMessage("工位监控", "监控已开始", QSystemTrayIcon.MessageIcon.Information)
+ else:
+ self.running = False
+ self.start_stop_btn.setText("开始监控")
+ self.start_stop_btn.setStyleSheet("background-color: #4CAF50;")
+ self.status_label.setText("状态: 已停止监控")
+ self.timer.stop()
+ self.tray_icon.showMessage("工位监控", "监控已停止", QSystemTrayIcon.MessageIcon.Information)
+
+ def update_timeout(self, value):
+ self.timeout = value
+
+ def lock_windows(self):
+ ctypes.windll.user32.LockWorkStation()
+
+ def closeEvent(self, event):
+ if self.running:
+ self.tray_icon.showMessage("工位监控", "程序已最小化系统托盘",
+ QSystemTrayIcon.MessageIcon.Information)
+ self.hide()
+ event.ignore()
+ else:
+ self.running = False
+ self.timer.stop()
+ self.cap.release()
+ self.tray_icon.hide()
+ event.accept()
+
+ def handle_camera_disconnected(self):
+ if self.running:
+ self.running = False
+ self.timer.stop()
+ self.start_stop_btn.setText("开始监控")
+ self.start_stop_btn.setStyleSheet("background-color: #4CAF50;")
+ self.status_label.setText("状态: 摄像头已断开")
+ self.confidence_label.setText("置信度: 0%")
+ self.progress_bar.setValue(0)
+ self.progress_bar.setFormat("摄像头未连接")
+ self.tray_icon.showMessage(
+ "工位监控",
+ "摄像头已断开，监控已停止",
+ QSystemTrayIcon.MessageIcon.Warning
+ )
+
+ # 尝试重新打开摄像头
+ self.try_reconnect_camera()
+
+ def try_reconnect_camera(self):
+ if self.cap.isOpened():
+ self.cap.release()
+
+ # 创建定时器尝试重新连接
+ self.reconnect_timer = QTimer()
+ self.reconnect_timer.timeout.connect(self.attempt_camera_reconnect)
+ self.reconnect_timer.start(2000) # 每2秒尝试重连
+
+ def attempt_camera_reconnect(self):
+ self.cap = cv2.VideoCapture(0)
+ if self.cap.isOpened():
+ self.reconnect_timer.stop()
+ self.status_label.setText("状态: 摄像头已重新连接")
+ self.progress_bar.setFormat("离开时间: %vs")
+ self.tray_icon.showMessage(
+ "工位监控",
+ "摄像头已重新连接，请手动开启监控",
+ QSystemTrayIcon.MessageIcon.Information
+ )
+
+ def set_progress_bar_style(self, color="#4CAF50"):
+ base_style = """
+ QProgressBar {
+ border: 1px solid #E0E0E0;
+ border-radius: 4px;
+ text-align: center;
+ background-color: #F5F5F5;
+ padding: 1px;
+ font-size: 12px;
+ }
+ QProgressBar::chunk {
+ background-color: %s;
+ border-radius: 3px;
+ margin: 0px;
+ }
+ """
+ self.progress_bar.setStyleSheet(base_style % color)
+
+ def main():
+ try:
+ logger.info("Starting main function...")
+ app = QApplication(sys.argv)
+ app.setStyle('Fusion')
+ window = WorkspaceMonitor()
+ window.show()
+ logger.info("Application started successfully.")
+ sys.exit(app.exec())
+ except Exception as e:
+ logger.error(f"Error in main function: {str(e)}")
+ logger.error(traceback.format_exc())
+ raise
+
+ if __name__ == "__main__":
+ try:
+ main()
+ except Exception as e:
+ logger.error(f"Fatal error: {str(e)}")
+ logger.error(traceback.format_exc())
+ # 显示错误消息框
+ if QApplication.instance() is None:
+ app = QApplication(sys.argv)
+ from PyQt6.QtWidgets import QMessageBox
+ msg = QMessageBox()
+ msg.setIcon(QMessageBox.Icon.Critical)
+ msg.setText("程序发生错误")
+ msg.setInformativeText(str(e))
+ msg.setWindowTitle("错误")
+ msg.setDetailedText(traceback.format_exc())
+ msg.exec()
+ sys.exit(1)
+
+except Exception as e:
+ logger.error(f"Fatal error outside main: {str(e)}")
+ logger.error(traceback.format_exc())
+ sys.exit(1)
\ No newline at end of file