a13rollcall/data_manager.py

import os
from datetime import datetime
import copy


class DataManager:
    def __init__(self, config_manager):
        self.config_mgr = config_manager
        self.stats = {}
        self.text_progress = {}

    def _parse_student_line(self, line):
        line = line.strip()
        if not line:
            return None
        parts = line.split(maxsplit=1)
        if len(parts) == 2 and parts[0].isdigit():
            name = parts[1].strip()
        else:
            name = line
        if name.isdigit() or not name:
            return None
        return name

    def load_students(self):
        file_path = self.config_mgr.get("files", "students")
        students = []
        error = ""
        if not os.path.exists(file_path):
            error = f"学生名单文件 {file_path} 不存在，可在设置中创建"
            return students, error
        try:
            with open(file_path, "r", encoding="utf-8") as f:
                for line in f:
                    name = self._parse_student_line(line)
                    if name:
                        students.append(name)
            if not students:
                error = "学生名单为空，请检查文件内容"
        except Exception as e:
            error = f"读取学生名单失败：{str(e)}"
        return students, error

    def load_texts(self):
        file_path = self.config_mgr.get("files", "texts")
        texts = []
        error = ""
        if not os.path.exists(file_path):
            error = f"课文库文件 {file_path} 不存在，可在设置中创建"
            return texts, error
        try:
            current_title = None
            current_paragraphs = []
            current_para = ""
            with open(file_path, "r", encoding="utf-8") as f:
                for line in f:
                    line = line.rstrip("\n")
                    if line.startswith("【") and line.endswith("】"):
                        if current_title is not None:
                            if current_para.strip():
                                current_paragraphs.append(current_para.strip())
                            texts.append({
                                "title": current_title,
                                "paragraphs": current_paragraphs
                            })
                        current_title = line[1:-1]
                        current_paragraphs = []
                        current_para = ""
                    elif line.strip() == "":
                        if current_para.strip():
                            current_paragraphs.append(current_para.strip())
                            current_para = ""
                    else:
                        if current_para:
                            current_para += "\n" + line
                        else:
                            current_para = line
                if current_title is not None:
                    if current_para.strip():
                        current_paragraphs.append(current_para.strip())
                    texts.append({
                        "title": current_title,
                        "paragraphs": current_paragraphs
                    })
        except Exception as e:
            error = f"读取课文库失败：{str(e)}"
        return texts, error

    def save_record(self, student_name, text_title, paragraph_info, status, remark="课堂抽查"):
        file_path = self.config_mgr.get("files", "records")
        time_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        line = f"{time_str} {student_name} {text_title} {paragraph_info} {status} {remark}\n"
        try:
            with open(file_path, "a", encoding="utf-8") as f:
                f.write(line)
        except Exception:
            pass

    def get_student_status(self, name):
        return self.config_mgr.get("student_status", name) or "未背过"

    def set_student_status(self, name, status):
        self.config_mgr.set(status, "student_status", name)
        self.config_mgr.trigger_hooks("status_change", name, status)
        self.config_mgr.save_config()
        self.update_status_stats(name, status)

    def get_student_weight(self, name):
        w = self.config_mgr.get("student_weights", name)
        return w if w is not None else 1

    def set_student_weight(self, name, weight, skip_save=False):
        self.config_mgr.set(weight, "student_weights", name)
        if not skip_save:
            self.config_mgr.save_config()

    def reset_all_weights(self, students):
        for name in students:
            self.set_student_weight(name, 1, skip_save=True)
        self.config_mgr.save_config()

    def load_stats(self):
        file_path = self.config_mgr.get("files", "stats")
        self.stats = {}
        if not os.path.exists(file_path):
            return
        try:
            with open(file_path, "r", encoding="utf-8") as f:
                for line in f:
                    line = line.strip()
                    if not line:
                        continue
                    parts = line.split()
                    if len(parts) < 6:
                        continue
                    try:
                        name = parts[0]
                        draw_count = int(parts[1])
                        mastered = int(parts[2])
                        familiar = int(parts[3])
                        unlearned = int(parts[4])
                        score = int(parts[5])
                        self.stats[name] = {
                            "draw_count": draw_count,
                            "mastered": mastered,
                            "familiar": familiar,
                            "unlearned": unlearned,
                            "score": score
                        }
                    except (ValueError, IndexError):
                        continue
        except Exception:
            self.stats = {}

    def save_stats(self):
        file_path = self.config_mgr.get("files", "stats")
        try:
            with open(file_path, "w", encoding="utf-8") as f:
                for name, data in self.stats.items():
                    line = f"{name} {data['draw_count']} {data['mastered']} {data['familiar']} {data['unlearned']} {data['score']}\n"
                    f.write(line)
        except Exception:
            pass

    def _init_stats_for_new_students(self, students):
        for name in students:
            if name not in self.stats:
                self.stats[name] = {
                    "draw_count": 0,
                    "mastered": 0,
                    "familiar": 0,
                    "unlearned": 0,
                    "score": 100
                }

    def increment_draw_count(self, name):
        if name not in self.stats:
            self._init_stats_for_new_students([name])
        self.stats[name]["draw_count"] += 1
        self.save_stats()

    def update_status_stats(self, name, status):
        if name not in self.stats:
            self._init_stats_for_new_students([name])
        data = self.stats[name]
        if status == "已背过":
            data["mastered"] += 1
            data["score"] += 10
        elif status == "未背熟":
            data["familiar"] += 1
            data["score"] -= 5
        elif status == "未背过":
            data["unlearned"] += 1
            data["score"] -= 15
        self.save_stats()

    def get_all_stats(self):
        return copy.deepcopy(self.stats)

    def get_ranking(self):
        ranking = [(name, data["score"]) for name, data in self.stats.items()]
        ranking.sort(key=lambda x: x[1], reverse=True)
        return ranking

    def get_text_progress(self, text_title):
        return self.config_mgr.get("text_progress", text_title) or 0

    def set_text_progress(self, text_title, index):
        self.config_mgr.set(index, "text_progress", text_title)
        self.config_mgr.save_config()