ADD file via upload

main.py

@@ -0,0 +1,106 @@
+import sys
+import time
+from PySide6 import QtCore, QtGui, QtWidgets
+from PySide6.QtWidgets import QApplication, QMainWindow, QMessageBox
+from login import Ui_Login
+from readercontrol import Ui_Readercontrol
+from connect import *
+from systemcontrol import Ui_systemcontrol
+from bookcontrol import Ui_bookcontrol
+
+
+cursor, conn = connect()
+
+
+class Readerui(QtWidgets.QMainWindow, Ui_Readercontrol):
+    def __init__(self, parent=None):
+        super(Readerui, self).__init__()
+        self.setupUi(self)
+
+
+class bookadminui(QtWidgets.QMainWindow, Ui_bookcontrol):
+    def __init__(self, parent=None):
+        super(bookadminui, self).__init__()
+        self.setupUi(self)
+
+
+class sysadminui(QtWidgets.QMainWindow, Ui_systemcontrol):
+    def __init__(self, parent=None):
+        super(sysadminui, self).__init__()
+        self.setupUi(self)
+
+
+class MyMainForm(QMainWindow, Ui_Login):
+    def __init__(self, parent=None):
+        super(MyMainForm, self).__init__()
+        self.setupUi(self)
+        self.exitbt.clicked.connect(self.exit)
+        self.loginbt.clicked.connect(self.login)
+    def exit(self):
+        rec_code = QMessageBox.question(self, "确认", "您确认要退出吗？", QMessageBox.Yes | QMessageBox.No)
+        if rec_code != 65536:
+            self.close()
+    def login(self):
+        ID = self.userline.text()
+        PW = self.pwline.text()
+        if ID == '' or PW == '':
+            QMessageBox.warning(self, "警告", "请输入用户名或密码", QMessageBox.Yes)
+        else:
+            # 读者登录
+            if self.idbox.currentText() == '读者':
+                sql = 'select * from readers where ID = "%s" and password="%s"' % (ID, PW)
+                res = cursor.execute(sql)
+                if res:
+                    logintime = time.strftime("%Y-%m-%d", time.localtime())
+                    sql = 'select * from loginrecord where time="%s"' % logintime
+                    res = cursor.execute(sql)
+                    logined = cursor.fetchall()
+                    if res:
+                        last = logined[-1]
+                        number = last[-1]
+                        num = number + 1
+                        sql = 'INSERT INTO loginrecord(ID,time,number) VALUES(%s,"%s",%d)' % (ID, logintime, num)
+                        cursor.execute(sql)
+                        conn.commit()
+                    else:
+                        sql = 'INSERT INTO loginrecord(ID,time,number) VALUES(%s,"%s",%d)' % (ID, logintime, 1)
+                        cursor.execute(sql)
+                        conn.commit()
+                    self.read = Readerui()
+                    self.read.show()
+                    self.hide()
+                    # self.close()
+                else:
+                    QMessageBox.warning(self, "警告", "密码错误！", QMessageBox.Yes)
+
+            elif self.idbox.currentText() == '图书管理员':
+                type = '图书管理员'
+                sql = 'select * from workers where ID = "%s" and password="%s" and type="%s" ' % (ID, PW, type)
+                res = cursor.execute(sql)
+                if res:
+                    self.bookadmin = bookadminui()
+                    self.bookadmin.show()
+                    # self.close()
+                    self.hide()
+                    pass
+                else:
+                    QMessageBox.warning(self, "警告", "密码错误，请重新输入！", QMessageBox.Yes)
+
+            elif self.idbox.currentText() == '系统管理员':
+                type = '系统管理员'
+                sql = 'select * from workers where ID = "%s" and password="%s" and type="%s"' % (ID, PW, type)
+                res = cursor.execute(sql)
+                # 进行判断
+                if res:
+                    self.sysadmin = sysadminui()
+                    self.sysadmin.show()
+                    # self.close()
+                    self.hide()
+                else:
+                    QMessageBox.warning(self, "警告", "密码错误，请重新输入！", QMessageBox.Yes)
+
+
+app = QApplication(sys.argv)
+myWindow = MyMainForm()
+myWindow.show()
+sys.exit(app.exec())