import psycopg2 from login import * from PyQt5.QtWidgets import QApplication, QMainWindow import sys from interface import * from activemanager import ActivityManager # 导入活动管理类 import config from activearrangement import Active_arrangement from store import Store NONE = None class LoginWindow(QMainWindow): def __init__(self): super().__init__() # 调用父类的初始化方法 self.ui = Ui_LoginWindow() # 创建一个 UI 界面的实例 self.ui.setupUi(self) # 设置 UI 界面 # 去掉窗口边框 self.setWindowFlag(QtCore.Qt.FramelessWindowHint) # 设置窗口为无边框模式 self.setAttribute(QtCore.Qt.WA_TranslucentBackground) # 设置窗口背景为透明 # 创建阴影效果 self.shadow = QtWidgets.QGraphicsDropShadowEffect(self) # 创建一个阴影效果对象 self.shadow.setOffset(0, 0) # 设置阴影偏移量为 (0, 0) self.shadow.setBlurRadius(10) # 设置阴影模糊半径为 10 像素 self.shadow.setColor(QtCore.Qt.black) # 设置阴影颜色为黑色 self.ui.frame.setGraphicsEffect(self.shadow) # 将阴影效果应用到指定的框架 (frame) # 设置样式表，去掉边框并设置透明背景 self.setStyleSheet("background-color: rgba(255, 255, 255, 0); border: none;") # 设置窗口背景为完全透明，并去掉边框 # 强制窗口固定大小 # self.setFixedSize(self.size()) # （注释掉的代码）如果需要，可以取消注释以固定窗口大小为当前大小 self.ui.pushButton_Login.clicked.connect(lambda: self.ui.stackedWidget_2.setCurrentIndex(0)) self.ui.pushButton_Register.clicked.connect(lambda: self.ui.stackedWidget_2.setCurrentIndex(1)) self.ui.PushButtom_L_sure.clicked.connect(self.login) self.ui.QPushButtom_R_sure.clicked.connect(self.register) self.show() # 显窗口 def register(self): account = self.ui.LineEdit_R_account.text() password = self.ui.LineEdit_R_password.text() password1 = self.ui.LineEdit_R_password1.text() global user_now if len(account) == 0 or len(password ) == 0 or len(password1) == 0: self.ui.stackedWidget.setCurrentIndex(3) elif password1 != password: self.ui.stackedWidget.setCurrentIndex(1) else: conn = psycopg2.connect(database="DataMY", user="postgres", password="822520", host="127.0.0.1", port="5432") cur = conn.cursor() # 查询现有账户 cur.execute("SELECT accounts FROM users") existing_accounts = cur.fetchall() # 获取所有账户 account_list = [row[0] for row in existing_accounts] # 提取账户名列表 if account in account_list: self.ui.stackedWidget.setCurrentIndex(6) # 账户已存在 else: # 插入新账户，并获取新用户的userid cur.execute("INSERT INTO users (accounts, passwords) VALUES (%s, %s) RETURNING userid", (account, password)) new_user_id = cur.fetchone()[0] # 获取新插入的用户ID # 插入用户积分，假设初始积分为 0 initial_points = 0 cur.execute("INSERT INTO user_points (user_id, total_points) VALUES (%s, %s)", (new_user_id, initial_points)) conn.commit() # 提交更改 self.ui.stackedWidget.setCurrentIndex(2) # 注册成功页面 def login(self): # 获取用户输入的账户和密码 account = self.ui.LineEdit_L_account.text() password = self.ui.LineEdit_L_password.text() # 检查账户或密码是否为空 if len(account) == 0 or len(password) == 0: self.ui.stackedWidget.setCurrentIndex(3) # 显示错误消息，提示账户或密码不能为空 return try: # 连接数据库 conn = psycopg2.connect(database="DataMY", user="postgres", password="822520", host="127.0.0.1", port="5432") cur = conn.cursor() # 查询用户账户和密码 cur.execute("SELECT userid, accounts, passwords FROM users WHERE accounts = %s", (account,)) row = cur.fetchone() if row: user_id, db_account, db_password = row # 验证账户和密码是否匹配 if account == db_account and password == db_password: config.user_now = user_id # 将用户ID存储在全局变量中 self.win = MainWindow() # 打开主窗口 self.win.show() # 显示主窗口 self.close() # 关闭登录窗口 else: self.ui.stackedWidget.setCurrentIndex(7) # 显示错误消息，提示账户或密码错误 else: self.ui.stackedWidget.setCurrentIndex(7) # 显示错误消息，提示账户不存在 conn.commit() except Exception as e: print(f"Error logging in: {e}") finally: # 关闭数据库连接 if conn: conn.close() class MainWindow(QMainWindow): def __init__(self): super().__init__() self.ui = Ui_MainWindow() # 假设 Ui_MainWindow 已经定义 self.ui.setupUi(self) # 设置窗口属性 self.setWindowFlag(QtCore.Qt.FramelessWindowHint) self.setAttribute(QtCore.Qt.WA_TranslucentBackground) self.setStyleSheet("background-color: rgba(255, 255, 255, 0); border: none;") # 创建活动管理实例 self.activity_manager = ActivityManager(self.ui) self.activity_arrangement = Active_arrangement(self.ui) self.Store = Store(self.ui) # 初始化拖动相关变量 self.mouse_press_pos = None # 连接信号和槽 self.setup_signals() self.activity_manager.load_data() # 使用活动管理类加载数据 self.Store.setup_ui() self.Store.load_data() self.Store.load_user_info() self.show() # 显示窗口 def setup_signals(self): self.ui.pushButton_active_manage.clicked.connect(lambda: self.ui.stackedWidget.setCurrentIndex(0)) self.ui.pushButton_active_arrange.clicked.connect(lambda: self.ui.stackedWidget.setCurrentIndex(1)) self.ui.pushButton_home.clicked.connect(lambda: self.ui.stackedWidget.setCurrentIndex(2)) self.ui.pushButton_profile.clicked.connect(lambda: self.ui.stackedWidget.setCurrentIndex(4)) self.ui.pushButton_store.clicked.connect(lambda: (self.ui.stackedWidget.setCurrentIndex(3), self.Store.load_user_info())) self.ui.pushButton_logout.clicked.connect(self.logout) self.ui.pushButton_m_sure.clicked.connect(self.change_password) self.ui.pushButton_creat_active.clicked.connect(self.activity_manager.create_activity) # 更新为活动管理方法 self.ui.pushButton_correct_active.clicked.connect(self.activity_manager.update_activity) # 更新为活动管理方法 self.ui.pushButton_delete_active.clicked.connect( self.activity_manager.delete_activity) # 传递活动ID self.ui.pushButton_baoming.clicked.connect(self.activity_manager.register_activity) self.ui.active_table.cellClicked.connect(self.activity_manager.handle_double_click) # 连接双击事件 self.ui.pushButton_4.clicked.connect(self.activity_manager.search) self.ui.pushButton_5.clicked.connect(self.activity_manager.view_registered_activities) self.ui.pushButton_6.clicked.connect(self.activity_manager.load_data) def mousePressEvent(self, event): if event.button() == QtCore.Qt.LeftButton and self.ui.frame_4.geometry().contains(event.pos()): self.mouse_press_pos = event.globalPos() - self.pos() event.accept() def mouseMoveEvent(self, event): if self.mouse_press_pos is not None: self.move(event.globalPos() - self.mouse_press_pos) event.accept() def mouseReleaseEvent(self, event): if event.button() == QtCore.Qt.LeftButton: self.mouse_press_pos = NONE event.accept() def logout(self): global user_now self.close() self.login = LoginWindow() # 假设 LoginWindow 已经定义 user_now = '' def change_password(self): global user_now password = self.ui.lineEdit_password.text() confirm_password = self.ui.lineEdit_password2.text() if not password or not confirm_password: self.ui.stackedWidget_2.setCurrentIndex(1) # 显示错误消息 return if password == confirm_password: try: with psycopg2.connect(database="DataMY", user="postgres", password="822520", host="127.0.0.1", port="5432") as conn: with conn.cursor() as cur: cur.execute("UPDATE users SET passwords = %s WHERE accounts = %s", (password, user_now)) conn.commit() self.ui.stackedWidget_2.setCurrentIndex(2) # 显示成功消息 except Exception as e: print(f"Error changing password: {e}") self.ui.stackedWidget_2.setCurrentIndex(1) if __name__ == '__main__': app = QApplication(sys.argv) win = LoginWindow() sys.exit(app.exec_())