Demo2.py

from PIL import Image, ImageDraw
from pyzbar.pyzbar import decode
import pyautogui
import time
import cv2
import numpy as np
import sys
from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QMessageBox, QVBoxLayout, QHBoxLayout, QLabel, QDialog, QLineEdit, QSystemTrayIcon, QMenu
from PyQt5.QtGui import QIcon
from PyQt5.QtCore import Qt, QPoint
import qtawesome as qta
import qdarkstyle
import qrcode
import pyperclip

# 全局变量用于鼠标选择
drawing = False
start_point = (-1, -1)
end_point = (-1, -1)
selected = False
img_copy = None

def mouse_callback(event, x, y, flags, param):
    global drawing, start_point, end_point, selected, img_copy, img_cv
    
    if event == cv2.EVENT_LBUTTONDOWN:
        drawing = True
        start_point = (x, y)
        selected = False
    elif event == cv2.EVENT_MOUSEMOVE:
        if drawing:
            end_point = (x, y)
            img_copy = img_cv.copy()
            cv2.rectangle(img_copy, start_point, end_point, (0, 255, 0), 2)
            cv2.imshow("选择区域", img_copy)
    elif event == cv2.EVENT_LBUTTONUP:
        drawing = False
        end_point = (x, y)
        selected = True
        cv2.rectangle(img_copy, start_point, end_point, (0, 255, 0), 2)
        cv2.imshow("选择区域", img_copy)

def take_screenshot():
    """
    先截取全屏，然后在图片上选择自定义区域
    """
    global img_cv, img_copy, selected, drawing, start_point, end_point
    
    # 重置全局变量
    drawing = False
    start_point = (-1, -1)
    end_point = (-1, -1)
    selected = False
    img_copy = None
    
    screenshot = pyautogui.screenshot()
    
    # 将PIL Image转换为OpenCV格式
    img_array = np.array(screenshot)
    img_cv = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
    img_copy = img_cv.copy()
    
    cv2.namedWindow("选择区域")
    cv2.setMouseCallback("选择区域", mouse_callback)
    cv2.imshow("选择区域", img_cv)
    
    
    while True:
        cv2.waitKey(1)
        if selected:
            break
    
    cv2.destroyAllWindows()
    
    if start_point != (-1, -1) and end_point != (-1, -1):
        x = min(start_point[0], end_point[0])
        y = min(start_point[1], end_point[1])
        w = abs(end_point[0] - start_point[0])
        h = abs(end_point[1] - start_point[1])
        
        if w > 0 and h > 0:
            cropped_img = screenshot.crop((x, y, x + w, y + h))
            return cropped_img
        else:
            print("选择区域无效")
            return None
    else:
        print("未选择区域")
        return None

def decode_qr_code(img):
    """
    解码二维码图片，返回URL
    """
    try:
        # 解码二维码
        decoded_objects = decode(img)
        if decoded_objects:
            # 返回第一个解码结果的数据（通常是URL）
            return decoded_objects[0].data.decode('utf-8')
        else:
            return "未检测到二维码"
    except Exception as e:
        return f"解码失败: {str(e)}"

class CustomTitleBar(QWidget):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent
        self.layout = QHBoxLayout()
        self.layout.setContentsMargins(0, 0, 0, 0)
        
        self.title = QLabel("二维码工具")
        self.title.setStyleSheet("color: white; padding-left: 10px; font-weight: bold;")
        
        self.min_button = QPushButton(qta.icon('fa5s.minus'), "")
        self.close_button = QPushButton(qta.icon('fa5s.times'), "")
        
        self.min_button.clicked.connect(self.parent.showMinimized)
        self.close_button.clicked.connect(self.parent.close)
        
        self.layout.addWidget(self.title)
        self.layout.addStretch()
        self.layout.addWidget(self.min_button)
        self.layout.addWidget(self.close_button)
        
        self.setLayout(self.layout)
        
class CustomMessageBox(QDialog):
    def __init__(self, title, text):
        super().__init__()
        self.setWindowTitle(title)
        self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint)
        
        self.layout = QVBoxLayout()
        
        self.title_bar = CustomTitleBar(self)
        self.layout.addWidget(self.title_bar)
        
        self.text_edit = QLineEdit(text)
        self.text_edit.setReadOnly(True)
        self.layout.addWidget(self.text_edit)
        
        self.setLayout(self.layout)
        
        self.old_pos = None

    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
            self.old_pos = event.globalPos()

    def mouseMoveEvent(self, event):
        if self.old_pos:
            delta = QPoint(event.globalPos() - self.old_pos)
            self.move(self.x() + delta.x(), self.y() + delta.y())
            self.old_pos = event.globalPos()

    def mouseReleaseEvent(self, event):
        self.old_pos = None

class QRDecoder(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint)
        self.initUI()
        self.initTrayIcon()
        self.old_pos = None

    def initUI(self):
        self.title_bar = CustomTitleBar(self)
        self.resize(80, 80)  # 设置窗口初始大小
        
        # 初始化位置
        screen_geo = QApplication.desktop().availableGeometry()
        self.move(screen_geo.width() - self.width() - 120, 80)
        
        layout = QVBoxLayout()
        layout.setContentsMargins(0, 0, 0, 0)
        layout.addWidget(self.title_bar)
        
        # 解码按钮
        decode_icon = qta.icon('fa5s.qrcode')
        decode_button = QPushButton(decode_icon, ' 解码二维码', self)
        decode_button.clicked.connect(self.run_program)
        layout.addWidget(decode_button)
        
        # 生成按钮
        generate_icon = qta.icon('fa5s.barcode')
        generate_button = QPushButton(generate_icon, ' 生成二维码', self)
        generate_button.clicked.connect(self.generate_qr)
        layout.addWidget(generate_button)
        
        self.setLayout(layout)

    def initTrayIcon(self):
        self.tray_icon = QSystemTrayIcon(self)
        self.tray_icon.setIcon(QIcon('icon.png'))
        
        tray_menu = QMenu()
        # show_action = tray_menu.addAction("显示")
        quit_action = tray_menu.addAction("退出")
        
        # show_action.triggered.connect(self.show)
        quit_action.triggered.connect(QApplication.instance().quit)
        
        self.tray_icon.setContextMenu(tray_menu)
        self.tray_icon.show()

    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
            self.old_pos = event.globalPos()

    def mouseMoveEvent(self, event):
        if self.old_pos:
            delta = QPoint(event.globalPos() - self.old_pos)
            self.move(self.x() + delta.x(), self.y() + delta.y())
            self.old_pos = event.globalPos()

    def mouseReleaseEvent(self, event):
        self.old_pos = None

    def run_program(self):
        """
        截图并解码
        """
        img = take_screenshot()
        if img:
            url = decode_qr_code(img)
            msg = CustomMessageBox('解码结果', url)
            msg.exec_()
        else:
            QMessageBox.warning(self, '错误', '截图失败')

    def generate_qr(self):
        """
        生成二维码：读取剪贴板内容并展示
        """
        try:
            content = pyperclip.paste()
            if content:
                img = qrcode.make(content)
                img.show()  # 自动展示二维码图片
            else:
                QMessageBox.warning(self, '错误', '剪贴板为空')
        except Exception as e:
            QMessageBox.warning(self, '错误', f'生成失败: {str(e)}')

def create_icon():
    """
    创建图标文件
    """
    try:

        img = Image.new('RGB', (64, 64), (20, 20, 20)) 
        draw = ImageDraw.Draw(img)
        
        draw.rectangle([10, 10, 30, 30], fill="white")
        draw.rectangle([34, 10, 54, 30], fill="white")
        draw.rectangle([10, 34, 30, 54], fill="white")
        draw.rectangle([38, 38, 42, 42], fill="white")
        draw.rectangle([46, 46, 50, 50], fill="white")
        
        img.save('icon.png')

    except Exception as e:
        print(f"图标创建失败: {e}")

def main():
    create_icon()
    app = QApplication(sys.argv)
    app.setWindowIcon(QIcon('icon.png'))
    app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())
    ex = QRDecoder()
    ex.show()
    sys.exit(app.exec_())

if __name__ == "__main__":
    main()