import sys
import admin
import pymssql
from MainForm import Ui_MainForm
from PyQt5.QtWidgets import *


class EmptyError(Exception):
    def __str__(self):
        return self.ui.Line_login.text() == "" and self.ui.Line_password.text() == ""


class Main(QWidget):
    def __init__(self):
        super(Main, self).__init__()
        self.conn = None

        self.fadmin = None
        self.fuser = None

        self.ui = Ui_MainForm()
        self.ui.setupUi(self)

        self.ui.Log.clicked.connect(self.btn_login_clicked)
        self.show()

        self.ui.Exit.clicked.connect(self.btn_exit_clicked)

    def btn_login_clicked(self):
        try:
            self.conn = pymssql.connect(host='MYPC', user=self.ui.Line_login.text(), password=self.ui.Line_password.text(), database='PersonDepartment')
            self.conn.autocommit(True)

            raise EmptyError(self)

            if self.ui.Line_login.text() == "admin" and self.ui.Line_password.text() == "admin":
                self.fadmin = admin(self.conn)
            elif self.ui.Line_login.text() == "user" and self.ui.Line_password.text() == "user":
                print("user")
                #self.fadmin = UserForm(self.conn)

            self.ui.close()
        except EmptyError:
            self.ui.label_status.setText("Invalid login or password")
        except pymssql.OperationalError:
            self.ui.label_status.setText("Invalid login or password")

    def btn_exit_clicked(self):
        self.ui.close()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = Main()

    sys.exit(app.exec_())








from AdminForm import Ui_AdminForm
from PyQt5.QtWidgets import *


class admin(QWidget):
    def __init__(self, conn):
        super(admin, self).__init__()

        self.ui = Ui_AdminForm()
        self.ui.setupUi(self)

        self.show()

        self.ui.Exit_Admin_Mode.clicked.connect(self.btn_exit_clicked)

    def btn_exit_clicked(self):
        self.ui.close()