from datetime import datetime as dt

from PyQt5 import QtWidgets
from PyQt5.QtCore import Qt

from gui.widgets import mainForm


class MainWindow(QtWidgets.QWidget, mainForm.Ui_Form):
    def __init__(self):
        super().__init__()
        self.setupUi(self)

    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.tabWidget.clear()