from datetime import datetime as dt

from PyQt5 import QtWidgets
from PyQt5.QtCore import Qt
from src.utils.base.base_widgets import BaseMainWindow

class MainWindow(BaseMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.set_style(self)

    def initUI(self) -> None:
        self.tabWidget = QtWidgets.QTabWidget()
        layout = QtWidgets.QVBoxLayout()
        layout.addWidget(self.tabWidget)
        self.setLayout(layout)

    def show_plot_tabs(self, plot_widgets: list[QtWidgets.QWidget]) -> None:
        for plot_widget in plot_widgets:
            tab = QtWidgets.QWidget()
            grid = QtWidgets.QGridLayout()
            grid.addWidget(plot_widget)
            tab.setLayout(grid)
            self.tabWidget.addTab(tab, "SF_trace_" + dt.now().strftime('%Y_%m_%d-%H_%M_%S'))
            self.tabWidget.setCurrentWidget(tab)

    def keyPressEvent(self, a0):
        if a0.key() == Qt.Key_F5:
            self.clear()