fga-eps-mds/2020.2-Anunbis

from ..model.student import Student
from ..ext.database import db
from sqlalchemy.exc import IntegrityError
from . import course_services, post_services, auth_services
from ..model.post import AgreeStudentPost, DisagreeStudentPost
from sqlalchemy import func


def get(**kwargs):
    return Student.query.filter_by(**kwargs).first()


def get_by_email(email):
    return Student.query.filter(
        func.lower(Student.email) == email.lower()
    ).one_or_none()


def delete(student_bd):
    delete_feedbacks(student_bd)
    delete_posts(student_bd)
    db.session.delete(student_bd)
    db.session.commit()


def delete_posts(student_bd):
    for post in student_bd.posts:
        post_services.delete(post)


def delete_feedbacks(student_bd):
    for agree in AgreeStudentPost.query.filter_by(
        reg_student=student_bd.reg_student
    ).all():
        db.session.delete(agree)
    for disagree in DisagreeStudentPost.query.filter_by(
        reg_student=student_bd.reg_student
    ).all():
        db.session.delete(disagree)
    db.session.commit()


def register_student(student):
    validate, status_code = __validate_student_relationship(student)
    if validate:
        return validate, status_code

    try:
        student_bd = Student(**student)
        db.session.add(student_bd)
        db.session.commit()
        auth_services.verify_email(student_bd)
        return {"message": "Student successfully registered!"}, 201
    except IntegrityError:
        return {"message": "Student already registered"}, 409


def modify_student(student_db, student_new):
    student_db.password = student_new.get("password")
    db.session.commit()
    return {"message": "Student successfully changed!"}, 200


def __validate_student_relationship(student):
    if course_services.get_course_id(student.get("id_course")) is None:
        return {"message": "Course not found!"}, 404

    return None, 200