DoubaoTeam/dao/employment.py

# 导入sqlalchemy会话对象，数据库会话，所有数据库操作都依靠Session
from sqlalchemy.orm import Session
# 类型提示：可选对象、列表
from typing import Optional, List
# 导入数据库ORM模型
from models.employment import Employment
# 导入pydantic入参模型
from schemas.employment import EmploymentCreate, EmploymentUpdate


def create_employment(db: Session, obj_in: EmploymentCreate) -> Optional[Employment]:
    """
    新增就业信息：一个学生只能存在一条就业记录
    :param db: 数据库会话对象，由FastAPI依赖注入传入
    :param obj_in: 新增的入参对象 EmploymentCreate
    :return: 成功返回ORM对象，发生任何异常返回None
    """
    try:
        # model_dump()把pydantic对象转字典；**解包字典，赋值给Employment模型字段，创建ORM对象
        db_obj = Employment(**obj_in.model_dump())
        # 将对象加入会话，此时还没有写入数据库
        db.add(db_obj)
        # 提交事务，真正执行insert SQL写入数据库
        db.commit()
        # db.refresh：从数据库刷新对象，回填数据库自动生成的id等字段到db_obj
        db.refresh(db_obj)
        return db_obj
    except Exception:
        # 只要出现任意异常，执行回滚，撤销本次会话所有未提交操作，防止脏数据残留
        db.rollback()
        return None


def get_employment_by_id(db: Session, eid: int) -> Optional[Employment]:
    """
    根据主键id查询就业详情，过滤逻辑删除的数据
    :param db: 数据库会话
    :param eid:就业记录主键id
    :return:查到返回ORM对象；没找到/异常返回None
    """
    try:
        # 查询Employment表，条件：id等于eid，并且is_deleted=0（没有被逻辑删除）
        return db.query(Employment).filter(
            Employment.id == eid,
            Employment.is_deleted == 0
        ).first()  # first()获取第一条结果，没有匹配返回None
    except Exception:
        return None


def get_employment_by_student_id(db: Session, student_id: int) -> Optional[Employment]:
    """
    根据学生id查询就业信息，过滤逻辑删除
    :param db:数据库会话
    :param student_id:学生主键id
    :return:查到返回ORM对象，否则None
    """
    try:
        return db.query(Employment).filter(
            Employment.student_id == student_id,
            Employment.is_deleted == 0
        ).first()
    except Exception:
        return None


def get_employment_list(
    db: Session,
    company_name: Optional[str] = None,
    salary_min: Optional[float] = None,
    salary_max: Optional[float] = None,
    skip: int = 0,
    limit: int = 20
) -> tuple[List[Employment], int]:
    """
    就业列表查询；支持公司名模糊、薪资范围筛选；只查询is_deleted=0正常数据
    :param db:数据库会话
    :param company_name:公司名称，模糊查询，不传不做过滤
    :param salary_min:薪资下限，不传不做过滤
    :param salary_max:薪资上限，不传不做过滤
    :param skip:分页偏移量，跳过多少条
    :param limit:分页每页最多取多少条
    :return:元组 (记录列表,总条数)；异常返回 [],0
    """
    try:
        # 构建基础查询对象，只查询未被逻辑删除的数据
        query = db.query(Employment).filter(Employment.is_deleted == 0)

        # 如果传入公司名，增加模糊匹配条件，%代表通配符
        if company_name:
            query = query.filter(Employment.company_name.like(f"%{company_name}%"))

        # 如果传入最低薪资，增加薪资>=下限条件
        if salary_min is not None:
            query = query.filter(Employment.salary >= salary_min)

        # 如果传入最高薪资，增加薪资<=上限条件
        if salary_max is not None:
            query = query.filter(Employment.salary <= salary_max)

        # count()统计符合条件的总记录数，用于前端分页
        total = query.count()
        # offset跳过skip条，limit最多取limit条，执行查询拿到结果列表
        records = query.offset(skip).limit(limit).all()
        return records, total
    except Exception:
        # 异常返回空列表、总条数0
        return [], 0


def update_employment(db: Session, eid: int, obj_in: EmploymentUpdate) -> Optional[Employment]:
    """
    修改就业信息，只更新前端传入的字段
    :param db:数据库会话
    :param eid:待修改记录主键id
    :param obj_in:修改入参对象 EmploymentUpdate
    :return:成功返回更新后的ORM对象；记录不存在/异常返回None
    """
    try:
        # 先根据id查询这条记录（自动过滤逻辑删除）
        db_obj = get_employment_by_id(db, eid)
        # 如果记录不存在，直接返回None
        if not db_obj:
            return None

        # exclude_unset=True：只拿到前端实际传过来的字段，没有传的字段不会出现在字典中
        update_dict = obj_in.model_dump(exclude_unset=True)

        # 循环字典，把每一个key‑value赋值给ORM对象属性
        for key, value in update_dict.items():
            setattr(db_obj, key, value)

        # 提交事务，把修改写入数据库
        db.commit()
        # 刷新对象，同步数据库最新数据到对象
        db.refresh(db_obj)
        return db_obj
    except Exception:
        # 发生异常回滚，放弃修改
        db.rollback()
        return None


def delete_employment_logic(db: Session, eid: int) -> bool:
    """
    逻辑删除，只修改is_deleted=1，**不做物理删除db.delete()**，符合项目文档要求
    :param db:数据库会话
    :param eid:待删除记录id
    :return:成功返回True；记录不存在/异常返回False
    """
    try:
        # 查询这条记录，过滤已逻辑删除
        db_obj = get_employment_by_id(db, eid)
        # 记录不存在返回False
        if not db_obj:
            return False
        # 修改逻辑删除标记为1
        db_obj.is_deleted = 1
        # 提交事务，把修改写入数据库
        db.commit()
        return True
    except Exception:
        # 异常回滚，返回False
        db.rollback()
        return False