import login_ui
from main import *


class LoginDialog(QtGui.QMainWindow, login_ui.Ui_Dialog):
    def __init__(self):
        QtGui.QMainWindow.__init__(self)
        Ui_Dialog.__init__(self)
        self.setupUi(self)
        self.setWindowTitle(u"登录")
        self.login_button.clicked.connect(self.login_button_clicked)
        self.main_window = None

    def login_button_clicked(self):
        self.check_login()

    def check_login(self):
        username = self.username_lineedit.text().toUtf8().data()
        password = self.password_lineedit.text().toUtf8().data()
        db_api = ReadAPI()
        api_response = db_api.login(username, password)
        if api_response.status_code == requests.codes.ok:
            if eval(api_response.text)['state'] == 0:
                self.close()
                self.load_main(username)
            else:
                self.state_label.setText(u"用户名或密码错误！")
        else:
            self.state_label.setText(u"未知错误！")

    def load_main(self, username):
        self.main_window = MainUI(username)
        self.main_window.show()

    def accept(self):
        pass

    def reject(self):
        pass

if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    login = LoginDialog()
    login.show()
    sys.exit(app.exec_())