test
This commit is contained in:
+55
-105
@@ -1,165 +1,115 @@
|
||||
# 导入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
|
||||
from schemas.employment import EmploymentCreate, EmploymentUpdate, EmploymentResponse
|
||||
|
||||
|
||||
def create_employment(db: Session, obj_in: EmploymentCreate) -> Optional[Employment]:
|
||||
"""
|
||||
新增就业信息：一个学生只能存在一条就业记录
|
||||
:param db: 数据库会话对象，由FastAPI依赖注入传入
|
||||
:param obj_in: 新增的入参对象 EmploymentCreate
|
||||
:return: 成功返回ORM对象，发生任何异常返回None
|
||||
"""
|
||||
def create_employment(db, obj_in):
|
||||
"""新增就业信息：一个学生只能存在一条就业记录"""
|
||||
try:
|
||||
# model_dump()把pydantic对象转字典；**解包字典，赋值给Employment模型字段，创建ORM对象
|
||||
# obj_in.model_dump()把pydantic请求对象转为字典，解包传给Employment生成ORM对象
|
||||
db_obj = Employment(**obj_in.model_dump())
|
||||
# 将对象加入会话，此时还没有写入数据库
|
||||
# add：把对象加入数据库会话，此时还没写入数据库
|
||||
db.add(db_obj)
|
||||
# 提交事务，真正执行insert SQL写入数据库
|
||||
# commit：提交事务，真正写入mysql数据库
|
||||
db.commit()
|
||||
# db.refresh：从数据库刷新对象，回填数据库自动生成的id等字段到db_obj
|
||||
# refresh：从数据库刷新对象，拿到数据库自动生成的id、默认时间等字段
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
# model_validate：把sqlalchemy ORM对象解析进pydantic响应模型
|
||||
# model_dump：把pydantic模型转为普通python字典返回给上层api
|
||||
return EmploymentResponse.model_validate(db_obj).model_dump()
|
||||
except Exception:
|
||||
# 只要出现任意异常，执行回滚，撤销本次会话所有未提交操作，防止脏数据残留
|
||||
# 发生任何异常，执行回滚，撤销本次会话中未提交的数据库改动
|
||||
db.rollback()
|
||||
# 异常情况返回None，上层api可以判断新增失败
|
||||
return None
|
||||
|
||||
|
||||
def get_employment_by_id(db: Session, eid: int) -> Optional[Employment]:
|
||||
"""
|
||||
根据主键id查询就业详情，过滤逻辑删除的数据
|
||||
:param db: 数据库会话
|
||||
:param eid:就业记录主键id
|
||||
:return:查到返回ORM对象；没找到/异常返回None
|
||||
"""
|
||||
def get_employment_by_id(db, eid):
|
||||
"""根据主键id查询就业详情，过滤逻辑删除"""
|
||||
try:
|
||||
# 查询Employment表，条件：id等于eid，并且is_deleted=0（没有被逻辑删除）
|
||||
return db.query(Employment).filter(
|
||||
db_obj = db.query(Employment).filter(
|
||||
Employment.id == eid,
|
||||
Employment.is_deleted == 0
|
||||
).first() # first()获取第一条结果，没有匹配返回None
|
||||
).first()
|
||||
if not db_obj:
|
||||
return None
|
||||
return EmploymentResponse.model_validate(db_obj).model_dump()
|
||||
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
|
||||
"""
|
||||
def get_employment_by_student_id(db, student_id):
|
||||
"""根据学生id查询就业信息，过滤逻辑删除"""
|
||||
try:
|
||||
return db.query(Employment).filter(
|
||||
db_obj = db.query(Employment).filter(
|
||||
Employment.student_id == student_id,
|
||||
Employment.is_deleted == 0
|
||||
).first()
|
||||
if not db_obj:
|
||||
return None
|
||||
return EmploymentResponse.model_validate(db_obj).model_dump()
|
||||
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
|
||||
"""
|
||||
def get_employment_list(db, company_name=None, salary_min=None, salary_max=None, skip=0, limit=20):
|
||||
"""就业列表查询；支持公司名模糊、薪资范围筛选；只查询is_deleted=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()统计符合条件的总记录数，用于前端分页
|
||||
# count()统计满足条件总条数，用于分页total
|
||||
total = query.count()
|
||||
# offset跳过skip条，limit最多取limit条，执行查询拿到结果列表
|
||||
# offset偏移量（跳过多少条），limit每页多少条，执行查询拿到结果列表
|
||||
records = query.offset(skip).limit(limit).all()
|
||||
return records, total
|
||||
# 列表推导式：循环每一条ORM记录，全部转为字典，得到字典列表
|
||||
dict_list = [EmploymentResponse.model_validate(item).model_dump() for item in records]
|
||||
# 返回字典列表 + 总条数
|
||||
return dict_list, total
|
||||
except Exception:
|
||||
# 异常返回空列表、总条数0
|
||||
# 异常返回空列表，总条数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
|
||||
"""
|
||||
def update_employment(db, eid, obj_in):
|
||||
"""修改就业信息，只更新传入的字段"""
|
||||
try:
|
||||
# 先根据id查询这条记录（自动过滤逻辑删除）
|
||||
db_obj = get_employment_by_id(db, eid)
|
||||
# 如果记录不存在，直接返回None
|
||||
if not db_obj:
|
||||
# 根据id查询未删除的记录
|
||||
db_obj_raw = db.query(Employment).filter(
|
||||
Employment.id == eid,
|
||||
Employment.is_deleted == 0
|
||||
).first()
|
||||
# 记录不存在直接返回None
|
||||
if db_obj_raw is None:
|
||||
return None
|
||||
|
||||
# exclude_unset=True：只拿到前端实际传过来的字段，没有传的字段不会出现在字典中
|
||||
# exclude_unset=True：只取出前端实际传过来的字段，不会带上没传的None字段
|
||||
update_dict = obj_in.model_dump(exclude_unset=True)
|
||||
|
||||
# 循环字典，把每一个key‑value赋值给ORM对象属性
|
||||
for key, value in update_dict.items():
|
||||
setattr(db_obj, key, value)
|
||||
|
||||
# 提交事务，把修改写入数据库
|
||||
setattr(db_obj_raw, key, value)
|
||||
db.commit()
|
||||
# 刷新对象，同步数据库最新数据到对象
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
db.refresh(db_obj_raw)
|
||||
return EmploymentResponse.model_validate(db_obj_raw).model_dump()
|
||||
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
|
||||
"""
|
||||
def delete_employment_logic(db, eid):
|
||||
"""逻辑删除，只修改is_deleted=1，不做物理删除"""
|
||||
try:
|
||||
# 查询这条记录，过滤已逻辑删除
|
||||
db_obj = get_employment_by_id(db, eid)
|
||||
# 记录不存在返回False
|
||||
if not db_obj:
|
||||
db_obj_raw = db.query(Employment).filter(
|
||||
Employment.id == eid,
|
||||
Employment.is_deleted == 0
|
||||
).first()
|
||||
if db_obj_raw is None:
|
||||
return False
|
||||
# 修改逻辑删除标记为1
|
||||
db_obj.is_deleted = 1
|
||||
# 提交事务，把修改写入数据库
|
||||
db_obj_raw.is_deleted = 1
|
||||
db.commit()
|
||||
return True
|
||||
except Exception:
|
||||
# 异常回滚，返回False
|
||||
db.rollback()
|
||||
return False
|
||||
|
||||
Reference in New Issue
Block a user