from fastapi import Depends,HTTPException

from util.database import get_db

from model.all_model import *

# 根据id查询单条有效成绩
def get_score_by_id_dao(score_id:int, db):
    return db.query(WlScore).filter(
        WlScore.score_id == score_id
        ,WlScore.delete_status==0
    ).first()

# 根据stu_id+exam_order查重（新增、修改时校验重复）
def get_score_by_stu_exam_order_dao(stu_id:str,exam_order:int,db):
    return db.query(WlScore).filter(WlScore.stu_id==stu_id
                                    ,WlScore.exam_order==exam_order
                                    ,WlScore.delete_status==0).first()

def get_grade_dao( stu_id:int , db):

    r1 =(db.query( WlScore.score_id
                  ,WlScore.stu_id
                  ,Student_Model.stu_name
                  ,WlScore.exam_order
                  ,WlScore.score
                   )
         .join(Student_Model,WlScore.stu_id == Student_Model.stu_id)
         .filter(WlScore.stu_id == stu_id,WlScore.delete_status == 0,Student_Model.delete_status == 0)
         .all())
    if r1:
        data_list = []
        for row in r1:
            data_list.append({
                "score_id": row.score_id,
                "stu_id": row.stu_id,
                "stu_name": row.stu_name,
                "exam_order": row.exam_order,
                "score": row.score
            })
        return data_list
    else:
        raise HTTPException(status_code=404,detail="学生不存在")

def add_grade_dao(g,db):
    try:
        o1 = WlScore(**g)
        db.add(o1)
        db.commit()
        return o1
    except Exception:
        db.rollback()
        raise HTTPException(status_code=500,detail="学生添加失败")

def update_grade_dao(score_id:int, update_data, db):
    try:
        rows = db.query(WlScore).filter(
            WlScore.score_id == score_id,
            WlScore.delete_status == 0
        ).update(update_data)
    except Exception:
        db.rollback()
        return False
    else:
        db.commit()
        return rows

def delete_grade_dao(stu_id:int,db):
    try:
        rows = db.query(WlScore).filter(
            WlScore.stu_id == stu_id,
            WlScore.delete_status == 0
        ).update({'delete_status':1,'delete_time':datetime.now()})
    except:
        db.rollback()
        return False
    else:
        db.commit()
        return rows