from datetime import datetime

from fastapi import HTTPException,Depends
from sqlalchemy import func
from model.all_model import Student_Model,WlScore,Employment_info
from util.database import get_db

# 实现根据年龄查询学生信息的功能
def select_age(min_age,max_age,session=Depends(get_db)):
    try:
        if min_age is None and max_age is None:
            raise HTTPException(status_code=400, detail="请至少传入一个参数")
        session = session.query(Student_Model)
        if min_age is not None:
            session = session.filter(Student_Model.age >= min_age,Student_Model.delete_status == 0)
        if max_age is not None:
            session = session.filter(Student_Model.age <= max_age,Student_Model.delete_status == 0)
        return session.all()
    except HTTPException:
        raise
    except Exception:
        raise HTTPException(status_code=500,detail="查询异常")

# 实现统计每个班级的学员总数以及男女生的总数的功能
def select_all(session=Depends(get_db)):
    print('1234')
    try:
        l1 = session.query(Student_Model.class_id,func.count(1)).group_by(Student_Model.class_id,Student_Model.gender).all()
        return l1
    except Exception:
        raise HTTPException(status_code=500,detail="查询异常")

# 实现根据成绩查询学员信息的功能
def select_score(min_score,max_score,session=Depends(get_db)):
    try:
        if min_score is None and max_score is None:
            raise HTTPException(status_code=400, detail="请至少传入一个参数")
        session = session.query(Student_Model,WlScore.score,WlScore.score)\
            .join(WlScore,WlScore.stu_id == Student_Model.stu_id,WlScore.delete_status==0,WlScore.delete_status == 0)
        if min_score is not None:
            session = session.filter(WlScore.score >= min_score)
        if min_score is not None:
            session = session.filter(WlScore.score <= max_score)
        return session.all()
    except HTTPException:
        raise
    except Exception:
        raise HTTPException(status_code=500,detail="查询异常")

#实现查询每个学生的就业时长
def select_emp_duration(stu_id:int,session):
    try:
       l2 = session.query(Employment_info,Student_Model.name.label('stu_name'),).join(Student_Model,stu_id==Student_Model.stu_id).filter(Employment_info.stu_id == stu_id,Employment_info.delete_status == 0).all()
       list=[]
       today=datetime.today()
       for i in l2:
          duration_days = (today - i.start_date).days
       list.append({Employment_info.stu_id:i.stu_id,

                    Employment_info.offer_issuance_date:i.offer_issuance_date,
                    Employment_info.company_name:i.company_name})