from core.factories import UserFactory
from users.factories import SchoolClassFactory
from users.models import School, SchoolRole, UserSchoolRole, DEFAULT_SCHOOL_ID


def create_school_with_users(school_name):
    school = School.create_school(school_name, id=DEFAULT_SCHOOL_ID)

    teacher_role = SchoolRole.objects.get_default_teacher_role_for_school(school)
    teacher = UserFactory(username='teacher')
    UserSchoolRole.objects.create(user=teacher, school_role=teacher_role)

    student_role = SchoolRole.objects.get_default_student_role_for_school(school)
    for i in range(1, 7):
        student = UserFactory(username='student{}'.format(i))
        UserSchoolRole.objects.create(user=student, school_role=student_role)
        SchoolClassFactory(users=[teacher, student], school=school)