from dotenv import load_dotenv

from constants import STUDENTS
from utils.db import get_test_by_student_class_book

load_dotenv()

passed_students = {}

for class_name, students in STUDENTS.items():
    for student in students:
        tests = get_test_by_student_class_book(student, class_name, "Іван Франко - Захар Беркут")
        unique_models = set()
        for test in tests:
            unique_models.add(test["model"])
        print("Amount of unique models:", len(unique_models))
        if len(unique_models) >= 5:
            if class_name not in passed_students:
                passed_students[class_name] = []
            passed_students[class_name].append(student)

print(f"Склали 5 тестів")
for class_name, students in passed_students.items():
    print(f"Клас: {class_name}")
    for student in students:
        print(f"  {student}")
print(f"\n\n")


not_passed_students = {}

for class_name, students in STUDENTS.items():
    for student in students:
        tests = get_test_by_student_class_book(student, class_name, "Іван Франко - Захар Беркут")
        unique_models = set()
        for test in tests:
            unique_models.add(test["model"])
        print("Amount of unique models:", len(unique_models))
        if len(unique_models) < 5:
            if class_name not in not_passed_students:
                not_passed_students[class_name] = []
            not_passed_students[class_name].append(student)

print(f"Не cклали 5 тестів")
for class_name, students in not_passed_students.items():
    print(f"Клас: {class_name}")
    for student in students:
        print(f"  {student}")