""" 数学学习软件 - 数据管理模块 负责用户数据存储、题目生成等业务逻辑 """ import json import os import random import re import smtplib from email.mime.text import MIMEText from email.mime.multipart import MIMEMultipart from typing import Dict, List, Tuple, Optional from email_config import get_smtp_config, DEFAULT_SENDER_CONFIG class DataManager: """数据管理类，负责用户信息和题目数据的管理""" def __init__(self): self.users_file = "users.json" self.email_config_file = "email_config.json" self.current_user = None # 邮件配置 self.sender_config = DEFAULT_SENDER_CONFIG self.ensure_data_files() self.load_email_config() def ensure_data_files(self): """确保数据文件存在""" if not os.path.exists(self.users_file): with open(self.users_file, 'w', encoding='utf-8') as f: json.dump({}, f, ensure_ascii=False, indent=2) def load_email_config(self): """加载邮箱配置""" try: if os.path.exists(self.email_config_file): with open(self.email_config_file, 'r', encoding='utf-8') as f: config = json.load(f) self.sender_config.update(config) except Exception as e: print(f"加载邮箱配置失败: {e}") def load_users(self) -> Dict: """加载用户数据""" try: with open(self.users_file, 'r', encoding='utf-8') as f: return json.load(f) except: return {} def save_users(self, users_data: Dict): """保存用户数据""" with open(self.users_file, 'w', encoding='utf-8') as f: json.dump(users_data, f, ensure_ascii=False, indent=2) def generate_verification_code(self) -> str: """生成6位数字验证码""" return str(random.randint(100000, 999999)) def send_verification_email(self, email: str, verification_code: str) -> bool: """发送验证码邮件""" try: # 检查发送邮箱配置 if (self.sender_config["email"] == "your_email@qq.com" or self.sender_config["password"] == "your_app_password"): print("请先配置发送邮箱信息") # 在没有配置邮箱的情况下，返回验证码供测试使用 print(f"测试模式 - 验证码: {verification_code}") return True # 获取SMTP配置 smtp_config = get_smtp_config(self.sender_config["email"]) # 创建邮件内容 msg = MIMEMultipart() msg['From'] = self.sender_config['email'] # QQ邮箱要求简单格式 msg['To'] = email msg['Subject'] = "数学学习软件 - 注册验证码" # 邮件正文 body = f"""欢迎使用数学学习软件！ 您的注册验证码是：{verification_code} 请在软件中输入此验证码完成注册。 验证码有效期为10分钟。 如果您没有申请注册，请忽略此邮件。 祝您学习愉快！ {self.sender_config['name']}""" msg.attach(MIMEText(body, 'plain', 'utf-8')) # 连接SMTP服务器并发送邮件 server = smtplib.SMTP(smtp_config["smtp_server"], smtp_config["smtp_port"]) if smtp_config["use_tls"]: server.starttls() # 启用TLS加密 server.login(self.sender_config["email"], self.sender_config["password"]) text = msg.as_string() server.sendmail(self.sender_config["email"], email, text) server.quit() return True except Exception as e: print(f"邮件发送失败: {e}") # 测试模式：如果邮件发送失败，在控制台显示验证码 print(f"测试模式 - 验证码: {verification_code}") return True # 返回True以便测试 def register_user(self, email: str, username: str) -> bool: """用户注册，发送验证码到邮箱""" users = self.load_users() if email in users: raise ValueError("该邮箱已注册") # 检查用户名是否已存在 for user_data in users.values(): if user_data.get("username") == username: raise ValueError("该用户名已存在") verification_code = self.generate_verification_code() # 发送验证码邮件 if not self.send_verification_email(email, verification_code): raise ValueError("验证码邮件发送失败，请检查邮箱地址") users[email] = { "username": username, "verification_code": verification_code, "verified": False, "password": None } self.save_users(users) return True def verify_registration(self, email: str, code: str) -> bool: """验证注册码""" users = self.load_users() if email not in users: return False if users[email]["verification_code"] == code: users[email]["verified"] = True self.save_users(users) return True return False def validate_password(self, password: str) -> bool: """验证密码格式：6-10位，必须含大小写字母和数字""" if len(password) < 6 or len(password) > 10: return False has_upper = bool(re.search(r'[A-Z]', password)) has_lower = bool(re.search(r'[a-z]', password)) has_digit = bool(re.search(r'\d', password)) return has_upper and has_lower and has_digit def set_password(self, email: str, password: str) -> bool: """设置用户密码""" if not self.validate_password(password): return False users = self.load_users() if email not in users or not users[email]["verified"]: return False users[email]["password"] = password self.save_users(users) return True def login(self, email: str, password: str) -> bool: """用户登录""" users = self.load_users() if email not in users: return False user = users[email] if user["verified"] and user["password"] == password: self.current_user = email return True return False def change_password(self, old_password: str, new_password: str) -> bool: """修改密码""" if not self.current_user: return False users = self.load_users() user = users[self.current_user] if user["password"] != old_password: return False if not self.validate_password(new_password): return False user["password"] = new_password self.save_users(users) return True def logout(self): """用户登出""" self.current_user = None def get_current_username(self) -> str: """获取当前登录用户的用户名""" if not self.current_user: return "" users = self.load_users() if self.current_user in users: return users[self.current_user].get("username", "") return "" class QuestionGenerator: """题目生成器""" def __init__(self): self.difficulty_levels = { "小学": {"range": (1, 100), "operations": ["+", "-", "*"]}, "初中": {"range": (1, 1000), "operations": ["+", "-", "*", "/"]}, "高中": {"range": (1, 10000), "operations": ["+", "-", "*", "/", "**"]} } def generate_question(self, level: str) -> Dict: """生成单个题目""" config = self.difficulty_levels[level] min_val, max_val = config["range"] operations = config["operations"] # 随机选择运算符 operation = random.choice(operations) if operation == "**": # 幂运算特殊处理 base = random.randint(2, 10) exponent = random.randint(2, 4) question = f"{base}^{exponent} = ?" correct_answer = base ** exponent elif operation == "/": # 除法确保整除 divisor = random.randint(2, 20) quotient = random.randint(2, max_val // divisor) dividend = divisor * quotient question = f"{dividend} ÷ {divisor} = ?" correct_answer = quotient else: # 加减乘法 num1 = random.randint(min_val, max_val) num2 = random.randint(min_val, max_val) if operation == "+": question = f"{num1} + {num2} = ?" correct_answer = num1 + num2 elif operation == "-": # 确保结果为正数 if num1 < num2: num1, num2 = num2, num1 question = f"{num1} - {num2} = ?" correct_answer = num1 - num2 else: # multiplication question = f"{num1} × {num2} = ?" correct_answer = num1 * num2 # 生成选项 options = self.generate_options(correct_answer) return { "question": question, "options": options, "correct_answer": correct_answer } def generate_options(self, correct_answer: int) -> List[int]: """生成四个选项，其中一个是正确答案""" options = [correct_answer] # 生成3个错误选项 while len(options) < 4: # 在正确答案附近生成错误选项 offset = random.randint(-abs(correct_answer)//2, abs(correct_answer)//2) if offset == 0: offset = random.choice([-1, 1]) * random.randint(1, 10) wrong_answer = correct_answer + offset if wrong_answer > 0 and wrong_answer not in options: options.append(wrong_answer) # 打乱选项顺序 random.shuffle(options) return options def generate_test_paper(self, level: str, question_count: int) -> List[Dict]: """生成试卷""" questions = [] generated_questions = set() # 用于避免重复题目 attempts = 0 while len(questions) < question_count and attempts < question_count * 10: question_data = self.generate_question(level) question_text = question_data["question"] if question_text not in generated_questions: generated_questions.add(question_text) questions.append(question_data) attempts += 1 return questions class ScoreCalculator: """分数计算器""" @staticmethod def calculate_score(correct_count: int, total_count: int) -> int: """计算分数（百分制）""" if total_count == 0: return 0 return int((correct_count / total_count) * 100) @staticmethod def get_grade_comment(score: int) -> str: """根据分数获取评价""" if score >= 90: return "优秀！继续保持！" elif score >= 80: return "良好！再接再厉！" elif score >= 70: return "中等，还需努力！" elif score >= 60: return "及格，继续加油！" else: return "需要更多练习！"