Mathlearn/src/ui/login_window.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
登录窗口
处理用户注册和登录功能
"""

import tkinter as tk
from tkinter import messagebox, ttk
import sys
import os

# 添加父目录到路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from user_manager import UserManager
from ui.register_window import RegisterWindow
from ui.main_window import MainWindow

class LoginWindow:
    """登录窗口类"""
    
    def __init__(self):
        self.user_manager = UserManager()
        self.setup_ui()
    
    def setup_ui(self):
        """设置UI界面"""
        self.window = tk.Tk()
        self.window.title("数学学习软件 - 登录")
        self.window.geometry("400x300")
        self.window.resizable(False, False)
        
        # 居中显示
        self.center_window()
        
        # 创建主框架
        main_frame = ttk.Frame(self.window, padding="20")
        main_frame.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
        
        # 标题
        title_label = ttk.Label(main_frame, text="数学学习软件", font=("Arial", 16, "bold"))
        title_label.grid(row=0, column=0, columnspan=2, pady=(0, 20))
        
        # 邮箱输入
        ttk.Label(main_frame, text="邮箱:").grid(row=1, column=0, sticky=tk.W, pady=5)
        self.email_var = tk.StringVar()
        self.email_entry = ttk.Entry(main_frame, textvariable=self.email_var, width=30)
        self.email_entry.grid(row=1, column=1, pady=5, padx=(10, 0))
        
        # 密码输入
        ttk.Label(main_frame, text="密码:").grid(row=2, column=0, sticky=tk.W, pady=5)
        self.password_var = tk.StringVar()
        self.password_entry = ttk.Entry(main_frame, textvariable=self.password_var, 
                                       show="*", width=30)
        self.password_entry.grid(row=2, column=1, pady=5, padx=(10, 0))
        
        # 按钮框架
        button_frame = ttk.Frame(main_frame)
        button_frame.grid(row=3, column=0, columnspan=2, pady=20)
        
        # 登录按钮
        login_btn = ttk.Button(button_frame, text="登录", command=self.login)
        login_btn.grid(row=0, column=0, padx=5)
        
        # 注册按钮
        register_btn = ttk.Button(button_frame, text="注册", command=self.open_register)
        register_btn.grid(row=0, column=1, padx=5)
        
        # 修改密码按钮
        change_pwd_btn = ttk.Button(button_frame, text="修改密码", command=self.open_change_password)
        change_pwd_btn.grid(row=0, column=2, padx=5)
        
        # 绑定回车键
        self.window.bind('<Return>', lambda e: self.login())
        
        # 设置焦点
        self.email_entry.focus()
    
    def center_window(self):
        """窗口居中显示"""
        self.window.update_idletasks()
        width = self.window.winfo_width()
        height = self.window.winfo_height()
        x = (self.window.winfo_screenwidth() // 2) - (width // 2)
        y = (self.window.winfo_screenheight() // 2) - (height // 2)
        self.window.geometry(f'{width}x{height}+{x}+{y}')
    
    def login(self):
        """处理登录"""
        email = self.email_var.get().strip()
        password = self.password_var.get().strip()
        
        if not email or not password:
            messagebox.showerror("错误", "请输入邮箱和密码")
            return
        
        success, message = self.user_manager.login(email, password)
        
        if success:
            messagebox.showinfo("成功", message)
            self.window.destroy()
            # 打开主窗口
            main_window = MainWindow(email)
        else:
            messagebox.showerror("错误", message)
    
    def open_register(self):
        """打开注册窗口"""
        self.window.destroy()
        register_window = RegisterWindow()
    
    def open_change_password(self):
        """打开修改密码窗口"""
        email = self.email_var.get().strip()
        if not email:
            messagebox.showerror("错误", "请先输入邮箱")
            return
        
        if email not in self.user_manager.users:
            messagebox.showerror("错误", "用户不存在")
            return
        
        # 创建修改密码对话框
        self.change_password_dialog(email)
    
    def change_password_dialog(self, email):
        """修改密码对话框"""
        dialog = tk.Toplevel(self.window)
        dialog.title("修改密码")
        dialog.geometry("350x200")
        dialog.resizable(False, False)
        dialog.transient(self.window)
        dialog.grab_set()
        
        # 居中显示
        dialog.update_idletasks()
        x = (dialog.winfo_screenwidth() // 2) - (175)
        y = (dialog.winfo_screenheight() // 2) - (100)
        dialog.geometry(f'350x200+{x}+{y}')
        
        main_frame = ttk.Frame(dialog, padding="20")
        main_frame.pack(fill=tk.BOTH, expand=True)
        
        # 原密码
        ttk.Label(main_frame, text="原密码:").grid(row=0, column=0, sticky=tk.W, pady=5)
        old_password_var = tk.StringVar()
        old_password_entry = ttk.Entry(main_frame, textvariable=old_password_var, 
                                      show="*", width=25)
        old_password_entry.grid(row=0, column=1, pady=5, padx=(10, 0))
        
        # 新密码
        ttk.Label(main_frame, text="新密码:").grid(row=1, column=0, sticky=tk.W, pady=5)
        new_password_var = tk.StringVar()
        new_password_entry = ttk.Entry(main_frame, textvariable=new_password_var, 
                                      show="*", width=25)
        new_password_entry.grid(row=1, column=1, pady=5, padx=(10, 0))
        
        # 确认新密码
        ttk.Label(main_frame, text="确认新密码:").grid(row=2, column=0, sticky=tk.W, pady=5)
        confirm_password_var = tk.StringVar()
        confirm_password_entry = ttk.Entry(main_frame, textvariable=confirm_password_var, 
                                          show="*", width=25)
        confirm_password_entry.grid(row=2, column=1, pady=5, padx=(10, 0))
        
        def change_password():
            old_password = old_password_var.get().strip()
            new_password = new_password_var.get().strip()
            confirm_password = confirm_password_var.get().strip()
            
            if not all([old_password, new_password, confirm_password]):
                messagebox.showerror("错误", "请填写所有字段")
                return
            
            if new_password != confirm_password:
                messagebox.showerror("错误", "两次输入的新密码不一致")
                return
            
            success, message = self.user_manager.change_password(email, old_password, new_password)
            
            if success:
                messagebox.showinfo("成功", message)
                dialog.destroy()
            else:
                messagebox.showerror("错误", message)
        
        # 按钮
        button_frame = ttk.Frame(main_frame)
        button_frame.grid(row=3, column=0, columnspan=2, pady=20)
        
        ttk.Button(button_frame, text="确定", command=change_password).grid(row=0, column=0, padx=5)
        ttk.Button(button_frame, text="取消", command=dialog.destroy).grid(row=0, column=1, padx=5)
        
        # 设置焦点
        old_password_entry.focus()