from typing import List
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from dao import advisor_dao
from database import get_db
from schema import advisor_schema

#子路由
router = APIRouter()

#查所有
@router.get("/getAllAdvisors", response_model=List[advisor_schema.AdvisorOut],summary="查所有顾问")
def get_all_advisors(db: Session = Depends(get_db)):
    return advisor_dao.get_all(db)

#查单个
@router.get("/getAdvisor/{advisor_id}",response_model=advisor_schema.AdvisorOut,summary="查单个顾问")
def get_advisor(advisor_id:int,db:Session =Depends(get_db)):
    advisor=advisor_dao.get_by_id(db,advisor_id)
    if advisor:
        return advisor
    else:
        raise HTTPException(status_code=404,detail="顾问不存在")


#查指定顾问下的学生信息
@router.get("/advisors/{advisor_id}/students", response_model=List[advisor_schema.StudentOut],summary="查指定顾问下的学生信息")
def get_advisor_students(advisor_id: int, db: Session = Depends(get_db)):
    students = advisor_dao.adv_all_stu(db, advisor_id)
    if students is None:
        raise HTTPException(status_code=404, detail="顾问不存在")
    return students   # 空列表 [] 会正常返回 200
#增加
@router.post("/createAdvisor", response_model=advisor_schema.AdvisorOut,summary="增加顾问")
def create_advisor(advisor: advisor_schema.AdvisorCreate, db: Session = Depends(get_db)):
    return advisor_dao.create_advisor(db, advisor.advisor_name, advisor.phone, advisor.gender)

#删除/恢复
@router.put("/changeAdvisorStatus/{advisor_id}/status",response_model=advisor_schema.AdvisorOut,summary="删除顾问")
def change_advisor_status(advisor_id: int,
                          body: advisor_schema.AdvisorStatus,
                          db: Session = Depends(get_db)):
    advisor = advisor_dao.change_advisor(db, advisor_id, body.flag)
    if advisor is None:
        raise HTTPException(status_code=404, detail="顾问不存在")
    return advisor

#修改
@router.put("/updateAdvisor/{advisor_id}", response_model=advisor_schema.AdvisorOut,summary="修改顾问")
def update_advisor(advisor_id: int,
                   advisor: advisor_schema.AdvisorCreate,
                   db: Session = Depends(get_db)):
    advisors = advisor_dao.update_advisor(db, advisor_id, advisor.advisor_name, advisor.phone, advisor.gender)
    if advisors is None:
        raise HTTPException(status_code=404, detail="顾问不存在")
    return advisors