main.pyw

import datetime
import sys

from PyQt5.QtCore import Qt, QIODevice
from PyQt5.QtSerialPort import QSerialPort, QSerialPortInfo
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QMessageBox, QWidget, QPushButton, QComboBox, QLabel, QTextEdit, QHBoxLayout, QCheckBox, QGridLayout, QScrollArea

baudRate = 500000           # Установите нужную скорость порта
newFilterState = False      # Состояние новых Filter Checkbox

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("PixelDebugTools (by @ItMan7145, @DragonKnight)")
        self.setGeometry(100, 100, 900, 700)

        self.central_widget = QWidget(self)
        self.setCentralWidget(self.central_widget)

        self.layout = QVBoxLayout(self.central_widget)

        # --- Управление портом ---
        self.port_layout = QHBoxLayout()
        self.layout.addLayout(self.port_layout)

        self.port_label = QLabel("Выберите порт:", self)
        self.port_layout.addWidget(self.port_label)

        self.port_combo_box = QComboBox(self)
        self.port_layout.addWidget(self.port_combo_box)

        self.refresh_button = QPushButton("Обновить порты", self)
        self.port_layout.addWidget(self.refresh_button)
        self.refresh_button.clicked.connect(self.refresh_ports)

        self.connect_button = QPushButton("Подключиться", self)
        self.port_layout.addWidget(self.connect_button)
        self.connect_button.clicked.connect(self.connect_button_clicked)

        self.disconnect_button = QPushButton("Отключиться", self)
        self.port_layout.addWidget(self.disconnect_button)
        self.disconnect_button.setEnabled(False)
        self.disconnect_button.clicked.connect(self.disconnect_button_clicked)

        # --- Динамические фильтры ---
        filter_widget = QWidget()
        self.filter_layout = QGridLayout(filter_widget)
        self.filter_layout.setContentsMargins(0, 0, 0, 0)
        self.filter_layout.setAlignment(Qt.AlignTop | Qt.AlignLeft)

        scroll = QScrollArea()
        scroll.setWidgetResizable(True)
        scroll.setWidget(filter_widget)
        scroll.setMaximumHeight(150)

        self.layout.addWidget(QLabel("Фильтр префиксов:"))
        self.layout.addWidget(scroll)

        self.all_checkbox = QCheckBox("ALL", self)
        self.all_checkbox.setChecked(False)
        self.filter_layout.addWidget(self.all_checkbox, 0, 0)

        # Словарь: префикс → чекбокс
        self.prefix_checkboxes = {}  # str -> QCheckBox
        self.next_row = 1
        self.next_col = 0
        self.columns = 7

        # --- Лог ---
        self.layout.addWidget(QLabel("Логи:"))
        self.log_text_edit = QTextEdit(self)
        self.log_text_edit.setReadOnly(True)
        self.layout.addWidget(self.log_text_edit)

        self.clear_button = QPushButton("Очистить лог")
        self.layout.addWidget(self.clear_button)
        self.clear_button.clicked.connect(self.clear_log)

        self.serial = None
        self.refresh_ports()

    def refresh_ports(self):
        """Обновление списка портов с красивым отображением"""
        self.port_combo_box.clear()

        ports = QSerialPortInfo.availablePorts()
        if not ports:
            self.port_combo_box.addItem("Нет доступных портов")
            return

        for port in ports:
            port_name = port.portName()

            # Формируем читаемое описание
            parts = [port_name]
            if port.description():
                parts.append(port.description().strip())
            #if port.manufacturer():
            #    parts.append(f"({port.manufacturer().strip()})")

            display_text = ": ".join(parts[:2])
            if len(parts) > 2:
                display_text += f" {parts[2]}"

            # Добавляем в комбо-бокс: отображаемый текст + данные = реальное имя порта
            self.port_combo_box.addItem(display_text, port_name)

    def connect_button_clicked(self):
        # Получаем реальное имя порта из данных элемента
        port_name = self.port_combo_box.currentData()
        if not port_name:
            QMessageBox.warning(self, "Предупреждение", "Выберите порт.")
            return

        if not isinstance(baudRate, int) or baudRate <= 0:
            QMessageBox.critical(self, "Ошибка", "Установите правильную скорость порта (baudRate).")
            return

        try:
            self.serial = QSerialPort()
            self.serial.setPortName(port_name)
            self.serial.setBaudRate(baudRate)
            self.serial.readyRead.connect(self.read_from_serial)

            if self.serial.open(QIODevice.ReadWrite):
                self.connect_button.setEnabled(False)
                self.disconnect_button.setEnabled(True)
                self.refresh_button.setEnabled(False)
                self.port_combo_box.setEnabled(False)
                self.port_label.setText(f"Подключен: {port_name}")
                self.log_text_edit.append(f"[INFO] Подключено к {port_name}\n")
            else:
                QMessageBox.critical(self, "Ошибка", "Не удалось открыть порт.")
                self.serial = None
        except Exception as e:
            QMessageBox.critical(self, "Ошибка", f"Ошибка подключения: {e}")
            self.serial = None

    def disconnect_button_clicked(self):
        if self.serial and self.serial.isOpen():
            self.serial.close()
            self.serial = None

        self.connect_button.setEnabled(True)
        self.disconnect_button.setEnabled(False)
        self.refresh_button.setEnabled(True)
        self.port_combo_box.setEnabled(True)
        self.port_label.setText("Выберите порт:")
        self.log_text_edit.append("[INFO] Отключено\n")

    def read_from_serial(self):
        if not (self.serial and self.serial.isOpen()):
            return

        while self.serial.canReadLine():
            data = self.serial.readLine().data()
            try:
                text = data.decode('utf-8', errors='ignore').rstrip('\r\n')
            except:
                text = data.decode('ascii', errors='ignore').rstrip('\r\n')

            if text:
                self.process_data(text)

    def extract_prefix(self, line: str) -> str | None:
        """
        Извлекает префикс из строки вида: +PREFIX\tmessage
        Возвращает PREFIX (без + и без \t...)
        """
        if not line.startswith('+'):
            return None

        # Ищем позицию табуляции
        tab_pos = line.find('\t')
        if tab_pos == -1:
            return None

        prefix_part = line[1:tab_pos]  # всё после + и до \t
        # Префикс состоит из букв, цифр и _
        if not prefix_part.replace('_', '').replace('-', '').isalnum():
            return None

        return prefix_part

    def add_prefix_checkbox(self, prefix: str):
        """Добавляет чекбокс для нового префикса"""
        if prefix in self.prefix_checkboxes:
            return

        checkbox = QCheckBox(prefix, self)
        checkbox.setChecked(newFilterState)

        row = self.next_row
        col = self.next_col
        self.filter_layout.addWidget(checkbox, row, col)
        self.prefix_checkboxes[prefix] = checkbox

        self.next_col += 1
        if self.next_col >= self.columns:
            self.next_col = 0
            self.next_row += 1

    def process_data(self, raw_line: str):
        """Обрабатывает входящую строку"""
        timestamped = f'{datetime.datetime.now().strftime("%H:%M:%S.%f")[:-3]}:    {raw_line}'

        # Извлекаем префикс и добавляем в фильтры при необходимости
        prefix = self.extract_prefix(raw_line)
        if prefix:
            self.add_prefix_checkbox(prefix)

        # Решение о выводе в лог
        if self.all_checkbox.isChecked():
            self.log_text_edit.append(timestamped)
            return

        # Если ALL выключен — проверяем по префиксам
        if prefix and prefix in self.prefix_checkboxes:
            if self.prefix_checkboxes[prefix].isChecked():
                self.log_text_edit.append(timestamped)
        else:
            # Если префикс не распознан — показываем только при ALL (уже обработано выше)
            pass

    def clear_log(self):
        self.log_text_edit.clear()

    def closeEvent(self, event):
        if self.serial and self.serial.isOpen():
            self.serial.close()
        super().closeEvent(event)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())