顾问模块合并,都合完了哦,庆祝一下!!!

This commit is contained in:
pzc
2026-09-13 23:06:01 +08:00
parent 109f9c7862
commit c392cd90e6
6 changed files with 162 additions and 13 deletions
+71
View File
@@ -0,0 +1,71 @@
from fastapi import HTTPException
from sqlalchemy.orm import Session
from model.wl_advisor_model import Advisor
from model.wl_student_model import Student
from scheme.wl_advisor_scheme import AdvisorCreate, AdvisorUpdate
# ---- 增 ----
def create_advisor(db: Session, data: AdvisorCreate) -> Advisor:
db_advisor = Advisor(**data.model_dump())
db.add(db_advisor)
db.commit()
db.refresh(db_advisor)
return db_advisor
# ---- 查(多条件 + 分页)----
def list_advisors(db: Session, adv_name=None, adv_phone=None, adv_region=None,
adv_gender=None, skip: int = 0, limit: int = 100):
q = db.query(Advisor).filter(Advisor.is_deleted == 0)
if adv_name:
q = q.filter(Advisor.adv_name.like(f"%{adv_name}%"))
if adv_phone:
q = q.filter(Advisor.adv_phone.like(f"%{adv_phone}%"))
if adv_region:
q = q.filter(Advisor.adv_region.like(f"%{adv_region}%"))
if adv_gender:
q = q.filter(Advisor.adv_gender == adv_gender)
# 后录进来的顾问排前面,方便前端看到最新数据
return q.order_by(Advisor.adv_id.desc()).offset(skip).limit(limit).all()
# ---- 根据编号查单个顾问 ----
def get_advisor_by_id(db: Session, adv_id: int) -> Advisor | None:
return db.query(Advisor).filter(
Advisor.adv_id == adv_id,
Advisor.is_deleted == 0,
).first()
# ---- 改 ----
def update_advisor(db: Session, adv_id: int, data: AdvisorUpdate) -> Advisor | None:
db_advisor = get_advisor_by_id(db, adv_id)
if not db_advisor:
return None
# exclude_unset:前端传了什么就改什么,没传的字段保持原值
for k, v in data.model_dump(exclude_unset=True).items():
setattr(db_advisor, k, v)
db.commit()
db.refresh(db_advisor)
return db_advisor
# ---- 逻辑删除 ----
def delete_advisor(db: Session, adv_id: int) -> bool:
db_advisor = get_advisor_by_id(db, adv_id)
if not db_advisor:
return False
# 名下还挂着学生就不给删,处理方式和班级模块 delete_class 保持一致
student = db.query(Student).filter(
Student.advisor_id == adv_id,
Student.is_deleted == 0,
).first()
if student:
raise HTTPException(status_code=400, detail="该顾问名下还有学生,不能删除!")
db_advisor.is_deleted = 1
db.commit()
return True