XM_project/管理节点/node.py

import base64
import datetime
import sys
from itertools import product

import requests

import message.message_pb2 as message
from PyQt5.QtWidgets import QApplication, QSizePolicy, QLabel, QWidget, QListWidgetItem, QGridLayout

import json
import threading
import time
from functools import partial
from PyQt5.QtCore import pyqtSignal, Qt, QTimer, QThread, QSize
from PyQt5.QtGui import QCursor, QFont, QColor, QPixmap, QIcon, QImage
from PyQt5.QtWidgets import QPushButton, QMainWindow, QMenu, QMessageBox

import message.message_pb2 as message
import google.protobuf.text_format as tf
import re
import ui.spaceUi as sui
import db_helper.db_operation as spmng
from config import MQ_CONFIG as mq_config
import async_communication as cmt

from config import PARKING_CONFIG as park_config

class MainWindow(QMainWindow, sui.Ui_MainWindow):

    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)
        self.setupUi(self)

        font = QFont()
        font.setFamily('宋体')
        font.setBold(False)
        font.setPointSize(17)
        self.parking_listWidget.setFont(font)

        self.pushButton_1480.setIcon(QIcon(QPixmap("./images/1480.png")))
        self.pushButton_1500.setIcon(QIcon(QPixmap("./images/1500.png")))
        self.pushButton_2050.setIcon(QIcon(QPixmap("./images/2050.png")))

        if park_config.unit == 11:
            self.unit_label.setText("A-1")
        elif park_config.unit == 12:
            self.unit_label.setText("A-2")
        elif park_config.unit == 13:
            self.unit_label.setText("A-3")
        elif park_config.unit == 14:
            self.unit_label.setText("A-4")
        elif park_config.unit == 21:
            self.unit_label.setText("B-1")
        elif park_config.unit == 22:
            self.unit_label.setText("B-2")
        elif park_config.unit == 23:
            self.unit_label.setText("B-3")
        elif park_config.unit == 24:
            self.unit_label.setText("B-4")
        elif park_config.unit == 25:
            self.unit_label.setText("B-5")
        elif park_config.unit == 31:
            self.unit_label.setText("C-1")
        elif park_config.unit == 32:
            self.unit_label.setText("C-2")
        self.btn_positions = [(i, j) for i in range(14, -1, -1) for j in range(6)]
        self.space_dict = {}
        self.space_is_init = False
        self.show_model = self.car_number_rbtn_2.isChecked()

        self.rmq_statu_dict = {}
        self.command_dict = {}



        self.timer = QTimer()
        self.timer.timeout.connect(self.Switch)
        self.timer.start(200)

        self.find_btn.clicked.connect(self.find_btn_click)
    def getBackGroundColor(self, color):
        if color == "red":
            return "background-color:rgb(255,70,70)"
        elif color == "gray":
            return "background-color:rgb(168,168,168)"
        elif color == "green":
            return "background-color:rgb(0,180,0)"
        elif color == "yellow":
            return "background-color:rgb(248,239,71)"
        elif color == "pastel":
            return "background-color:rgb(241,158,194)"


    def get_name(self,key):
        if key == 'id':
            return '下标'
        elif key == 'table_id':
            return '表编号'
        elif key == 'floor':
            return '楼层'
        elif key == 'height':
            return '限高'
        elif key == 'car_number':
            return '车牌号'
        elif key == 'statu':
            return '车位状态'
        elif key == 'plate_color':
            return '号牌颜色'
        elif key == 'plate_type':
            return '号牌类型'
        elif key == 'plate_confidence':
            return '号牌可信度'
        elif key == 'recognition_time':
            return '识别时间'
        elif key == 'primary_key':
            return '唯一码'
        else:
            return '未知'
    def show_parking_details(self,park_dict):
        self.parking_listWidget.clear()
        for poit in park_dict.items():
            item = QListWidgetItem()
            item.setSizeHint(QSize(50, 42))
            name = self.get_name(poit[0])
            if name == '未知':
                continue
            value = poit[1]
            if name == '楼层':
                value = value - 2
            item.setText(name + "：" + str(value))
            self.parking_listWidget.addItem(item)
        res = g_space.query_space_in_car_number(park_dict['car_number'])
        self.plate_full_image_label.setPixmap(QPixmap(""))
        self.plate_clip_image_label.setPixmap(QPixmap(""))
        if len(res) > 0:
            try:
                rep = requests.get(res[0]['plate_full_image'])
                pix = QPixmap()
                pix.loadFromData(rep.content)
                pix = pix.scaled(self.plate_full_image_label.size(), Qt.KeepAspectRatio)
                self.plate_full_image_label.setPixmap(pix)
            except Exception as e:
                print('全景图：'+str(e.args))
            try:
                rep = requests.get(res[0]['plate_clip_image'])
                pix = QPixmap()
                pix.loadFromData(rep.content)
                pix = pix.scaled(self.plate_clip_image_label.size(), Qt.KeepAspectRatio)
                self.plate_clip_image_label.setPixmap(pix)
            except Exception as e:
                print('特写图：'+str(e.args))

    def find_btn_click(self):
        text = self.find_lineEdit.text()
        if text == '':
            QMessageBox.question(None, "提示", "请输入要查询的内容",
                                 QMessageBox.Ok)
            return
        if self.car_number_rbtn.isChecked():
            park_dict = g_space.query_space_in_car_number(text)
            if len(park_dict) > 0:
                self.show_parking_details(park_dict[0])
            else:
                QMessageBox.question(None, "查询结果", "车牌号:" + text + "  未查询到数据！",
                                     QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容
        elif self.key_rbtn.isChecked():
            park_dict = g_space.query_space_in_primary_key(text)
            if len(park_dict) > 0:
                self.show_parking_details(park_dict[0])
            else:
                QMessageBox.question(None, "查询结果", "唯一码:" + text + "  未查询到数据！",
                                     QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容
        elif self.parking_rbtn.isChecked():
            park_dict = g_space.query_space_in_unit_and_table_id(park_config.unit,text)
            if len(park_dict) > 0:
                self.show_parking_details(park_dict[0])

            else:
                QMessageBox.question(None, "查询结果", "车位id:" + text + "  未查询到数据！",
                                     QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容

        return

    def park_btn_left_click(self):
        sender = self.sender()
        tool_tip = sender.toolTip()
        dict = json.loads(tool_tip)
        self.show_parking_details(dict)



    def park_btn_right_click(self):
        menu = QMenu(self)
        sender = self.sender()
        tool_tip = sender.toolTip()
        dict = json.loads(tool_tip)
        action = menu.addAction('启用车位')
        action.triggered.connect(partial(self.btn_opeSpace, dict, 0))
        action = menu.addAction('停用车位')
        action.triggered.connect(partial(self.btn_disableSpace, dict, 10))
        action = menu.addAction('清除车位')
        action.triggered.connect(partial(self.btn_clearSpace, dict))
        action = menu.addAction('取出车辆')
        action.triggered.connect(partial(self.btn_pickUp, dict))
        menu.exec_(QCursor.pos())

    def btn_disableSpace(self, dict, statu):
        res = g_space.update_space_status(dict["id"], statu)
        if res == 0:
            QMessageBox.question(None, '提示', '更新车位失败！请检查是否点击错误！',
                                 QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容

    def btn_opeSpace(self, dict, statu):
        res = g_space.update_space_status(dict["id"], statu)
        if res == 0:
            QMessageBox.question(None, '提示', '更新车位失败！请检查是否点击错误！',
                                 QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容

    def btn_clearSpace(self, dict):
        res = g_space.clear_space_data(dict["id"])
        if res == 0:
            QMessageBox.question(None, '提示', '更新车位失败！请检查是否点击错误！',
                                 QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容

    def btn_pickUp(self, dict):
        table = message.pick_table()
        table.primary_key = dict["primary_key"]
        g_rabbitmq.publish(mq_config.mq_command_exchange_name, "user_pick_command_request_port", tf.MessageToString(table, as_utf8=True))
        QMessageBox.question(None, '提示', '取车消息发送成功！',
                             QMessageBox.Ok)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容

    def get_btn_show(self, park):
        font = QFont()
        font.setFamily('宋体')
        font.setPointSize(21)
        if park["serial_id"] in park_config.channel_ex[park_config.unit]["入口"]:
            text_str = "入   口"
            font.setBold(True)
        elif park["serial_id"] in park_config.channel_ex[park_config.unit]["出口"]:
            text_str = "出   口"
            font.setBold(True)
        elif park["statu"] == 2 and park["car_number"] is not None and self.car_number_rbtn_2.isChecked():
            text_str = " "+park["car_number"]
            font.setPointSize(17)
        else:
            text_str = "     车位:%d\n     楼层:%d" % (park["table_id"], park["floor"] - 2)
            font.setPointSize(10)

        stylesheet = ""
        if park["statu"] == 10:
            stylesheet = "background-color:rgb(255,130,130);pressed{padding-top:2px;padding-left:2px;};text-align:left;"
        elif park["statu"] == 2 and park["car_number"] is not None:
            stylesheet = "background-color:rgb(248,239,71);pressed{padding-top:2px;padding-left:2px;};text-align:left;"
        elif park["statu"] == 6 or park["statu"] == 7:
            stylesheet = "background-color:rgb(0,162,232);pressed{padding-top:2px;padding-left:2px;};"
        elif park["statu"] == 0:
            stylesheet = "background-color:rgb(195,195,195);pressed{padding-top:2px;padding-left:2px;};text-align:left;"

        icon = QIcon()  # 设置按钮图标

        if park["height"] == 1.48:
            icon = QIcon(QPixmap('./images/1480.png'))  # 设置按钮图标
        elif park["height"] == 1.50:
            icon = QIcon(QPixmap('./images/1500.png'))  # 设置按钮图标
        elif park["height"] == 2.05:
            icon = QIcon(QPixmap('./images/2050.png'))  # 设置按钮图标
        tool_tip = '{"id":%d,\n"table_id":%d,\n"floor":%d, \n"height":%f,\n"car_number":"%s" , \n"statu":%d,\n"plate_color":"%s" ,\n"plate_type":"%s" ,\n"plate_confidence":"%s",\n"recognition_time":"%s",\n"primary_key":"%s"}' % (
            park["id"], (-1 if park["table_id"] is None else park["table_id"]),
            park["floor"], (0 if park["height"] is None else park["height"]),
            park["car_number"],park["statu"],park["plate_color"],
            park["plate_type"],park["plate_confidence"],park["recognition_time"],park["primary_key"])
        return font, text_str, stylesheet, tool_tip, icon

    def Switch(self):

        # 绘制车位按钮
        space_dict = g_space.query_space_in_unit(park_config.unit)
        if len(space_dict) < 0:
            return
        is_show_car_number = self.car_number_rbtn_2.isChecked()
        if self.space_dict != space_dict:
            if self.space_is_init == False:
                self.space_dict = space_dict
                self.space_is_init = True
                is_drawIn = False
                is_drawOut = False
                for park, position in zip(self.space_dict, self.btn_positions):
                    btn = QPushButton()
                    font, text_str, stylesheet, tool_tip, icon = self.get_btn_show(park)
                    btn.setFont(font)
                    btn.setLayout(QGridLayout())
                    btn.setText(text_str)
                    btn.setStyleSheet(stylesheet)
                    btn.setToolTip(tool_tip)
                    btn.setObjectName(str(park["table_id"]))
                    btn.setIcon(icon)
                    btn.setIconSize(QSize(30,30))
                    if text_str == "入   口" and is_drawIn is False:
                        self.unit_gridLayout.addWidget(btn, *position, 1, len(park_config.channel_ex[park_config.unit]["入口"]))
                        is_drawIn = True
                    elif text_str == "出   口" and is_drawOut is False:
                        self.unit_gridLayout.addWidget(btn, *position, 1, len(park_config.channel_ex[park_config.unit]["出口"]))
                        is_drawOut = True
                    elif text_str != "出   口" and text_str != "入   口":
                        btn.setContextMenuPolicy(Qt.CustomContextMenu)
                        btn.customContextMenuRequested.connect(self.park_btn_right_click)
                        btn.clicked.connect(self.park_btn_left_click)
                        self.unit_gridLayout.addWidget(btn, *position)
            else:
                for park1, park2 in zip(self.space_dict, space_dict):
                    if park1 != park2:
                        font, text_str, stylesheet, tool_tip,icon = self.get_btn_show(park2)
                        results = self.findChild(QPushButton, str(park2["table_id"]))
                        results.setFont(font)
                        results.setText(text_str)
                        results.setStyleSheet(stylesheet)
                        results.setToolTip(tool_tip)

                self.space_dict = space_dict
        if self.show_model != is_show_car_number:
            self.show_model = is_show_car_number
            for park in self.space_dict:
                if park["statu"] == 2:
                    font, text_str, stylesheet, tool_tip,icon = self.get_btn_show(park)
                    results = self.findChild(QPushButton, str(park["table_id"]))
                    results.setFont(font)
                    results.setText(text_str)
                    results.setStyleSheet(stylesheet)
                    results.setToolTip(tool_tip)

    def closeEvent(self, event):
        results = QMessageBox.question(self, '退出', '你确定要退出吗?', QMessageBox.Yes | QMessageBox.No,
                                       QMessageBox.No)  # "退出"代表的是弹出框的标题,"你确认退出.."表示弹出框的内容
        if results == QMessageBox.Yes:
            g_rabbitmq.close()
            event.accept()  # 接受关闭事件

        else:
            event.ignore()  # 忽略关闭事件


if __name__ == '__main__':
    g_rabbitmq = cmt.RabbitAsyncCommunicator(mq_config.mq_ip, mq_config.mq_port, mq_config.mq_user, mq_config.mq_password)
    g_rabbitmq.Init(None, None)
    g_rabbitmq.start()
    g_space = spmng.DBOperation()

    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.showMaximized()
    mainWindow.plate_full_image_label.setMaximumSize(mainWindow.plate_full_image_label.size())
    mainWindow.plate_clip_image_label.setMaximumSize(mainWindow.plate_clip_image_label.size())

    sys.exit(app.exec_())