חזרה - מערכת רישומי תלמידים

from student import Student
from studentregistry import StudentRegistry
import time
import random

# מטפל במקרי בדיקה לבדיקות מקיפות
test_case = input()

if test_case == "default_test":
    # יצירת תלמידים
    student1 = Student(1001, "Alice Smith")
    student2 = Student(1002, "Bob Johnson")
    student3 = Student(1003, "Charlie Brown")

    # הוספת ציונים
    student1.add_grade("Math", 90)
    student1.add_grade("Science", 85)
    student1.add_grade("English", 92)

    student2.add_grade("Math", 78)
    student2.add_grade("Science", 80)
    student2.add_grade("English", 85)

    student3.add_grade("Math", 95)
    student3.add_grade("Science", 91)
    student3.add_grade("English", 89)

    # הצגת רשומות התלמידים
    print("Student Records:")
    student1.display_record()
    print()
    student2.display_record()
    print()
    student3.display_record()
    print()

    # בדיקת תקינות השם
    try:
        student1.name = "J"  # קצר מדי
    except ValueError as e:
        print(f"Name validation error: {e}")
    print()

    # יצירת מאגר תלמידים והוספת תלמידים
    registry = StudentRegistry()
    registry.add_student(student1)
    registry.add_student(student2)
    registry.add_student(student3)

    # הצגת כל התלמידים
    print("All Students in Registry:")
    registry.display_all()
    print()

    # מציאת התלמיד המצטיין
    top_student = registry.get_top_student()
    print(f"Top Student: {top_student.name} (Average: {top_student.grade_average})")

    # הסרת תלמיד
    registry.remove_student(1002)  # הסרת Bob
    print("\
After removing student 1002:")
    registry.display_all()

elif test_case == "empty_registry":
    # בדיקת מאגר תלמידים ריק
    registry = StudentRegistry()
    top_student = registry.get_top_student()
    if top_student is None:
        print("Top student in empty registry: None")
    print("Displaying all students in empty registry:")
    registry.display_all()  # לא אמור להציג דבר

elif test_case == "validation_tests":
    # בדוק את כל כללי האימות
    student = Student(1001, "Valid Name")
    print(f"Created student with name: {student.name}")
    
    # בדוק שם ריק
    try:
        student.name = ""
    except ValueError as e:
        print(f"Empty name test: {e}")
    
    # בדוק שם בן תו אחד
    try:
        student.name = "A"
    except ValueError as e:
        print(f"Single character name test: {e}")
    
    # בדוק ערך לא בוליאני של enrolled
    try:
        student.enrolled = "yes"
    except ValueError as e:
        print(f"Non-boolean enrolled test: {e}")
    
    # בדוק שינויים תקינים
    student.name = "New Valid Name"
    student.enrolled = False
    print(f"After valid changes - Name: {student.name}, Enrolled: {student.enrolled}")

elif test_case == "grade_calculation":
    # בדוק את חישוב ממוצע הציונים
    student = Student(1001, "Test Student")
    print(f"Student with no grades - Average: {student.grade_average}")
    
    # הוסף ציון אחד
    student.add_grade("Math", 85)
    print(f"Student with one grade - Average: {student.grade_average}")
    
    # הוסף ציונים נוספים
    student.add_grade("Science", 90)
    student.add_grade("English", 95)
    print(f"Student with multiple grades - Average: {student.grade_average}")

elif test_case == "registry_operations":
    # בדוק את כל פעולות הרישום
    registry = StudentRegistry()
    
    # יצירת תלמידים
    student1 = Student(101, "Student One")
    student2 = Student(102, "Student Two")
    
    # הוסף תלמידים
    registry.add_student(student1)
    registry.add_student(student2)
    print("After adding students:")
    registry.display_all()
    
    # קבל תלמיד
    retrieved_student = registry.get_student(101)
    print(f"\
Retrieved student 101: {retrieved_student.name if retrieved_student else None}")
    
    # קבל תלמיד שאינו קיים
    non_existent = registry.get_student(999)
    print(f"Retrieved student 999: {non_existent}")
    
    # הסר תלמיד
    registry.remove_student(101)
    print("\
After removing student 101:")
    registry.display_all()
    
    # הסר תלמיד שאינו קיים (לא אמורה להיזרק שגיאה)
    registry.remove_student(999)
    print("\
After attempting to remove non-existent student 999:")
    registry.display_all()

elif test_case == "large_registry":
    # בדוק ביצועים עם תלמידים רבים
    start_time = time.time()
    registry = StudentRegistry()
    
    # צור 100 תלמידים עם ציונים אקראיים
    for i in range(1, 101):
        student = Student(1000 + i, f"Student {i}")
        # הוסף 5 ציונים אקראיים
        for j in range(5):
            student.add_grade(f"Course {j+1}", random.randint(60, 100))
        registry.add_student(student)
    
    creation_time = time.time() - start_time
    print(f"Time to create 100 students: {creation_time:.4f} seconds")
    
    # מצא את התלמיד המצטיין
    start_time = time.time()
    top_student = registry.get_top_student()
    top_time = time.time() - start_time
    print(f"Time to find top student: {top_time:.4f} seconds")
    print(f"Top student: {top_student.name} with average {top_student.grade_average:.2f}")
    
    # מצא ידנית את 5 התלמידים המצטיינים
    print("\
Top 5 students:")
    students_list = []
    for i in range(1, 101):
        student = registry.get_student(1000 + i)
        students_list.append(student)
    
    # מיין לפי ממוצע הציונים בסדר יורד
    students_list.sort(key=lambda s: s.grade_average, reverse=True)
    
    # הצג את 5 התלמידים המצטיינים
    for i in range(5):
        student = students_list[i]
        print(f"{i+1}. {student.name}: {student.grade_average:.2f}")