import sys
from PyQt5.QtWidgets import QWidget, QCheckBox, QApplication, QPushButton, QLineEdit, QTextEdit
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QMainWindow, QLabel
from time import sleep
from Event_Desk import Event, Desk


class Example(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setGeometry(300, 300, 600, 300)
        self.setWindowTitle('Супер пупер Ленин умер')

        self.btn = QPushButton('Добавить событие', self)
        self.btn.resize(150, 40)
        self.btn.move(430, 250)
        self.btn.clicked.connect(self.start)

        self.desk = Desk()

    def start(self):

        self.piu = Piu(self)
        self.piu.show()
        # self.piu.close()

class Piu(QWidget):
    def __init__(self, parent):
        super().__init__()
        self.parent = parent
        self.initUI()

    def initUI(self):
        self.setGeometry(400, 400, 600, 300)
        self.setWindowTitle('параметры события')

        self.btn = QPushButton('фисе', self)
        self.btn.resize(100, 40)
        self.btn.move(100, 250)
        self.btn.clicked.connect(self.close)

        self.input1 = QLineEdit(self)
        self.input1.move(130, 2)

        self.input2 = QLineEdit(self)
        self.input2.move(130, 50)

        self.input3 = QTextEdit(self)
        self.input3.move(130, 100)
        self.input3.resize(400, 140)

        self.num_label1 = QLabel(self)
        self.num_label1.setText("Время начала~: ")
        self.num_label1.move(0, 2)

        self.num_label2 = QLabel(self)
        self.num_label2.setText("Время конца~: ")
        self.num_label2.move(0, 50)

        self.num_label3 = QLabel(self)
        self.num_label3.setText("Описание~: ")
        self.num_label3.move(0, 100)


    def start(self):
        if not self.input1.text().isdigit() or not self.input2.text().isdigit():
            pass
        else:
            event = Event(self.input1.text(), self.input2.text(), None)
            self.parent.desk.add(event)
            self.close()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    ex.show()
    sys.exit(app.exec_())