1/dao/wl_student_dao.py

from sqlalchemy.orm import Session
from sqlalchemy import func
from model.wl_student_model import student
from model.wl_class_model import Class   # 你同学写的
from scheme.wl_student_scheme import StudentCreate, StudentUpdate


# ---- 学号生成 ----
def generate_stu_no(db: Session, class_id: int, seq: int = None) -> str:
    #非测试就解除下面5行测试，注释date_str = "20260912"
    # cls = db.query(class_).filter_by(class_id=class_id).first()
    # if not cls or not cls.start_time:
    #     raise ValueError("班级不存在或未设置开班时间")
    # date_str = cls.start_time.strftime("%Y%m%d")
    # 直接写死一个日期字符串用于测试
    date_str = "20260912"

    if seq is None:
        count = db.query(func.count(student.stu_id)) \
                  .filter(student.class_id == class_id).scalar() or 0
        seq = count + 1
    return f"{date_str}{seq:03d}"


# ---- 增 ----
def create_student(db: Session, data: StudentCreate) -> student:
    stu_no = generate_stu_no(db, data.class_id)
    stu = student(stu_no=stu_no, **data.model_dump())
    db.add(stu)
    db.commit()
    db.refresh(stu)
    return stu


# ---- 查（多条件）----
def search_students(db: Session, stu_no=None, stu_name=None,
                    class_id=None, advisor_id=None, status=None):
    q = db.query(student).filter(student.is_deleted == 0)   # 若加了逻辑删除字段
    if stu_no:
        q = q.filter(student.stu_no.like(f"%{stu_no}%"))
    if stu_name:
        q = q.filter(student.stu_name.like(f"%{stu_name}%"))
    if class_id:
        q = q.filter(student.class_id == class_id)
    if advisor_id:
        q = q.filter(student.advisor_id == advisor_id)
    if status:
        q = q.filter(student.status == status)
    return q.all()


def get_student_by_no(db: Session, stu_no: str) -> student | None:
    return db.query(student).filter(
        student.stu_no == stu_no,
        student.is_deleted == 0
    ).first()


# ---- 改 ----
def update_student(db: Session, stu_no: str, data: StudentUpdate) -> student | None:
    stu = get_student_by_no(db, stu_no)
    if not stu:
        return None
    for k, v in data.model_dump(exclude_unset=True).items():
        setattr(stu, k, v)
    db.commit()
    db.refresh(stu)
    return stu


# ---- 逻辑删 ----
def delete_student(db: Session, stu_no: str) -> bool:
    stu = get_student_by_no(db, stu_no)
    if not stu:
        return False
    stu.is_deleted = 1
    db.commit()
    return True