import sys
import math
from PyQt5 import QtWidgets
import ui


class ExampleApp(QtWidgets.QMainWindow, ui.Ui_Form):
    def __init__(self):
        # Это здесь нужно для доступа к переменным, методам
        # и т.д. в файле design.py
        super().__init__()
        self.setupUi(self)  # Это нужно для инициализации нашего дизайна

        self.btn1.clicked.connect(lambda: self.btn1_action(btn=self.btn1))
        self.btn2.clicked.connect(lambda: self.btn1_action(btn=self.btn2))
        self.btn3.clicked.connect(lambda: self.btn1_action(btn=self.btn3))
        self.btn4.clicked.connect(lambda: self.btn1_action(btn=self.btn4))
        self.btn5.clicked.connect(lambda: self.btn1_action(btn=self.btn5))
        self.btn6.clicked.connect(lambda: self.btn1_action(btn=self.btn6))
        self.btn7.clicked.connect(lambda: self.btn1_action(btn=self.btn7))
        self.btn8.clicked.connect(lambda: self.btn1_action(btn=self.btn8))
        self.btn9.clicked.connect(lambda: self.btn1_action(btn=self.btn9))
        self.btn0.clicked.connect(lambda: self.btn1_action(btn=self.btn0))
        self.btn_clear.clicked.connect(self.btn_clear_action)
        self.btn_sqrt.clicked.connect(self.sqrt_action)
        self.btn_minus.clicked.connect(self.minus_action)
        self.btn_eq.clicked.connect(self.eq_action)

    def btn1_action(self, btn):
        if self.table.value() != 0:
            self.table.display(str(self.table.intValue()) + btn.text())
        else:
            self.table.display(btn.text())

    def btn_clear_action(self):
        self.table.display(0)

    def sqrt_action(self):
        self.table.display(math.sqrt(self.table.value()))

    def minus_action(self):
        self.btn_minus.setEnabled(False)
        self.previous_num = self.table.value()
        self.table.display(0)

    def eq_action(self):
        if not self.btn_minus.isEnabled():
            self.table.display(self.previous_num - self.table.value())
            self.btn_minus.setEnabled(True)


def main():
    app = QtWidgets.QApplication(sys.argv)  # Новый экземпляр QApplication
    window = ExampleApp()  # Создаём объект класса ExampleApp
    window.show()  # Показываем окно
    app.exec_()  # и запускаем приложение


if __name__ == '__main__':  # Если мы запускаем файл напрямую, а не импортируем
    main()  # то запускаем функцию main()