views/reports_window.py - master - manokuzCS/Web_Scanner_SQL_XSS_CSRF

views/reports_window.py
234 строки9 KB
Manokuz
29.10.2025 (Fixes)
29 окт 2025, 22:55
29 окт 2025, 22:55602d6a5
"""
Окно для просмотра и экспорта отчетов
views/reports_window.py
"""

from PyQt5.QtWidgets import (QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
                            QLabel, QTableWidget, QTableWidgetItem, QPushButton, 
                            QComboBox, QMessageBox, QDateEdit)
from PyQt5.QtCore import Qt, QDate
from PyQt5.QtGui import QFont
from typing import Dict, List, Optional
from utils.database import db
from utils.logger import logger
from utils.export_utils import ExportUtils

class ReportsWindow(QMainWindow):
    """Окно для просмотра и экспорта отчетов"""

    def __init__(self, user_id: int, parent: Optional[QWidget] = None):
        super().__init__(parent)
        self.user_id = user_id
        self.setWindowTitle("📋 Отчеты")
        self.setGeometry(100, 100, 1200, 800)

        self.setup_ui()
        self.load_reports()

    def setup_ui(self):
        """Настройка интерфейса"""
        central_widget = QWidget()
        self.setCentralWidget(central_widget)

        main_layout = QVBoxLayout()

        # Заголовок
        title_label = QLabel("📋 Отчеты о сканировании")
        title_font = QFont()
        title_font.setPointSize(14)
        title_font.setBold(True)
        title_label.setFont(title_font)
        title_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
        main_layout.addWidget(title_label)

        # Панель фильтров
        filters_layout = QHBoxLayout()

        # Фильтр по дате начала
        start_date_label = QLabel("Начальная дата:")
        filters_layout.addWidget(start_date_label)

        self.start_date = QDateEdit()
        self.start_date.setCalendarPopup(True)
        self.start_date.setDate(QDate.currentDate().addMonths(-1))
        filters_layout.addWidget(self.start_date)

        # Фильтр по дате окончания
        end_date_label = QLabel("Конечная дата:")
        filters_layout.addWidget(end_date_label)

        self.end_date = QDateEdit()
        self.end_date.setCalendarPopup(True)
        self.end_date.setDate(QDate.currentDate())
        filters_layout.addWidget(self.end_date)

        # Кнопка применения фильтра
        apply_filter_btn = QPushButton("Применить фильтр")
        apply_filter_btn.clicked.connect(self.apply_filter)
        filters_layout.addWidget(apply_filter_btn)

        # Кнопка сброса фильтра
        reset_filter_btn = QPushButton("Сбросить фильтр")
        reset_filter_btn.clicked.connect(self.reset_filter)
        filters_layout.addWidget(reset_filter_btn)

        filters_layout.addStretch()

        # Кнопка экспорта
        export_btn = QPushButton("Экспортировать отчет")
        export_btn.clicked.connect(self.export_report)
        filters_layout.addWidget(export_btn)

        main_layout.addLayout(filters_layout)

        # Таблица отчетов
        self.reports_table = QTableWidget()
        self.reports_table.setColumnCount(7)
        self.reports_table.setHorizontalHeaderLabels([
            "ID сканирования", "URL", "Дата", "Тип уязвимости", 
            "Параметр", "Серьезность", "Подробности"
        ])
        self.reports_table.setSelectionBehavior(QTableWidget.SelectRows)  # type: ignore
        main_layout.addWidget(self.reports_table)

        # Панель экспорта
        export_layout = QHBoxLayout()

        export_format_label = QLabel("Формат экспорта:")
        export_layout.addWidget(export_format_label)

        self.export_format = QComboBox()
        self.export_format.addItems(["HTML", "PDF", "JSON", "CSV"])
        export_layout.addWidget(self.export_format)

        export_layout.addStretch()

        main_layout.addLayout(export_layout)

        # Кнопка закрытия
        close_btn = QPushButton("Закрыть")
        close_btn.clicked.connect(self.close_window)
        main_layout.addWidget(close_btn)

        central_widget.setLayout(main_layout)
        
    def close_window(self) -> None:
        """Закрытие окна"""
        self.close()

    def load_reports(self):
        """Загрузка отчетов из базы данных"""
        try:
            conn = db.get_db_connection()
            cursor = conn.cursor()

            start_date_str = self.start_date.date().toString("yyyy-MM-dd")
            end_date_str = self.end_date.date().toString("yyyy-MM-dd")

            cursor.execute("""
                SELECT scan_id, url, timestamp, vulnerability_type, 
                       parameter, severity, details
                FROM scan_results 
                WHERE user_id = ? AND DATE(timestamp) BETWEEN ? AND ?
                ORDER BY timestamp DESC
                LIMIT 500
            """, (self.user_id, start_date_str, end_date_str))

            rows = cursor.fetchall()
            conn.close()

            self.reports_table.setRowCount(len(rows))

            for i, row in enumerate(rows):
                scan_id = str(row[0])
                url = row[1] if row[1] else "N/A"
                timestamp = row[2] if row[2] else "N/A"
                vuln_type = row[3] if row[3] else "N/A"
                parameter = row[4] if row[4] else "N/A"
                severity = row[5] if row[5] else "N/A"
                details = row[6] if row[6] else "N/A"

                self.reports_table.setItem(i, 0, QTableWidgetItem(scan_id))
                self.reports_table.setItem(i, 1, QTableWidgetItem(url))
                self.reports_table.setItem(i, 2, QTableWidgetItem(timestamp))
                self.reports_table.setItem(i, 3, QTableWidgetItem(vuln_type))
                self.reports_table.setItem(i, 4, QTableWidgetItem(parameter))
                self.reports_table.setItem(i, 5, QTableWidgetItem(severity))
                self.reports_table.setItem(i, 6, QTableWidgetItem(details))

            # Настраиваем ширину колонок
            self.reports_table.resizeColumnsToContents()

        except Exception as e:
            logger.error(f"Ошибка при загрузке отчетов: {e}")

    def apply_filter(self):
        """Применение фильтра к отчетам"""
        try:
            self.load_reports()
        except Exception as e:
            logger.error(f"Ошибка при применении фильтра: {e}")

    def reset_filter(self):
        """Сброс фильтра"""
        try:
            self.start_date.setDate(QDate.currentDate().addMonths(-1))
            self.end_date.setDate(QDate.currentDate())
            self.load_reports()
        except Exception as e:
            logger.error(f"Ошибка при сбросе фильтра: {e}")

    def export_report(self):
        """Экспорт отчета"""
        try:
            # Получаем формат экспорта
            format_name = self.export_format.currentText()
            file_extension = format_name.lower()

            # Собираем данные для экспорта
            reports_data: List[Dict[str, str]] = []

            for row in range(self.reports_table.rowCount()):
                # Безопасно получаем значения из ячеек
                def get_item_text(col: int) -> str:
                    item = self.reports_table.item(row, col)
                    return item.text() if item else "N/A"
                
                report_data = {
                    'scan_id': get_item_text(0),
                    'url': get_item_text(1),
                    'timestamp': get_item_text(2),
                    'vulnerability_type': get_item_text(3),
                    'parameter': get_item_text(4),
                    'severity': get_item_text(5),
                    'details': get_item_text(6)
                }
                reports_data.append(report_data)

            if not reports_data:
                QMessageBox.warning(self, "Предупреждение", "Нет данных для экспорта")
                return

            # Экспортируем данные
            success = ExportUtils.export_data(
                self, 
                reports_data, 
                format_name, 
                file_extension, 
                self.user_id
            )

            if success:
                QMessageBox.information(
                    self, 
                    "Успех", 
                    f"Отчет успешно экспортирован в формате {format_name}"
                )

        except Exception as e:
            logger.error(f"Ошибка при экспорте отчета: {e}")
            QMessageBox.critical(
                self, 
                "Ошибка", 
                f"Не удалось экспортировать отчет: {str(e)}"
            )