source: flexograder/builder/exam_manager.py@ bc820cf

from builder.exam import Exam
from flexoentity import EntityState, EntityType, Domain


class ExamManager:
    def __init__(self):
        self._exams = {}
        self.active_exam_id = None

    def create_exam(self, title: str, author: str = "unknown", domain: str = "EX"):
        """Create and register a new Exam entity."""
        exam = Exam.with_domain(domain)
        exam.author = author
        exam.title = title
        exam._update_fingerprint
        exam_id = exam.flexo_id

        if exam.flexo_id in self._exams:
            raise ValueError(f"Exam {exam.flexo_id} already exists.")

        self._exams[exam_id] = exam
        self.active_exam_id = exam.flexo_id
        return exam
    
    @property
    def exams(self):
        return self._exams
    
    def add_exam(self, exam):
        """Add an externally loaded Exam (e.g., from JSON)."""
        self._exams[exam.flexo_id] = exam
        self.active_exam_id = exam.flexo_id
       
    def list_exams(self):
        return list(self.exams.keys())
    
    def list_titles(self):
        """Return titles for GUI dropdowns."""
        return [exam.title for exam in self._exams.values()]
    
    def get_active(self):
        return self.exams.get(self.active_exam_id)
    
    def get_active_id(self) -> str | None:
        return self.active_exam_id

    def set_active(self, exam_id: str):
        if exam_id not in self._exams:
            raise ValueError(f"No exam with id {exam_id}")
        self.active_exam_id = exam_id

    def set_active_by_title(self, title: str):
        """Convenience for GUI dropdown selection."""
        for exam in self._exams.values():
            if exam.title == title:
                self.active_exam_id = exam.flexo_id.id
                return
        raise ValueError(f"No exam with title '{title}' found")

    def remove_exam(self, exam_id: str) -> bool:
        if exam_id not in self._exams:
            return False
        del self._exams[exam_id]
        if self.active_exam_id == exam_id:
            self.active_exam_id = None
        return True

    # ────────────────────────────────────────────────
    #  Persistence helpers
    # ────────────────────────────────────────────────
    def save_all(self, directory: str):
        """Save all exams to JSON files in a given directory."""
        from pathlib import Path

        directory = Path(directory)
        directory.mkdir(parents=True, exist_ok=True)
        for exam in self._exams.values():
            filename = directory / f"{exam.title}_{exam.flexo_id.id}.json"
            exam.to_json_file(filename)

    def load_all(self, directory: str):
        """Load all exams from JSON files in a given directory."""
        from pathlib import Path
        import json

        directory = Path(directory)
        if not directory.exists():
            return
        for path in directory.glob("*.json"):
            with open(path, "r", encoding="utf-8") as f:
                data = json.load(f)
            exam = Exam.from_dict(data)
            self.add_exam(exam)