Merge pull request 'Nanfangyu0824' (#36) from nanfangyu0824 into main

Reviewed-on: #36

api/employment.py

@@ -1,19 +1,16 @@
 from fastapi import APIRouter, Depends, HTTPException, Query
 from sqlalchemy.orm import Session
 from core.database import get_db
-from dao.employment import (
-    create_employment,
-    get_employment_by_id,
-    get_employment_by_student_id,
-    get_employment_list,
-    update_employment,
-    delete_employment_logic
-)
-from schemas.employment import (
-    EmploymentCreate,
-    EmploymentUpdate,
-    EmploymentResponse
-)
+from dao.employment import (create_employment,
+                            get_employment_by_id,
+                            get_employment_by_student_id,
+                            get_employment_list,
+                            update_employment,
+                            delete_employment_logic)
+
+from schemas.employment import (EmploymentCreate,
+                                EmploymentUpdate,
+                                EmploymentResponse)
 
 employment_router = APIRouter(prefix="/employments", tags=["就业管理模块"])
 

dao/employment.py

@@ -6,31 +6,22 @@ from schemas.employment import EmploymentCreate, EmploymentUpdate, EmploymentRes
 def create_employment(db, obj_in):
     """新增就业信息：一个学生只能存在一条就业记录"""
     try:
-        # obj_in.model_dump()把pydantic请求对象转为字典，解包传给Employment生成ORM对象
         db_obj = Employment(**obj_in.model_dump())
-        # add：把对象加入数据库会话，此时还没写入数据库
-        db.add(db_obj)
-        # commit：提交事务，真正写入mysql数据库
-        db.commit()
-        # refresh：从数据库刷新对象，拿到数据库自动生成的id、默认时间等字段
-        db.refresh(db_obj)
-        # model_validate：把sqlalchemy ORM对象解析进pydantic响应模型
-        # model_dump：把pydantic模型转为普通python字典返回给上层api
+        db.add(db_obj)                      # add：把对象加入数据库会话，此时还没写入数据库
+        db.commit()                         # commit：提交事务，真正写入mysql数据库
+        db.refresh(db_obj)                  # refresh：从数据库刷新对象，拿到数据库自动生成的id、默认时间等字段
         return EmploymentResponse.model_validate(db_obj).model_dump()
+        # model_validate：把sqlalchemy ORM对象解析进pydantic响应模型
     except Exception:
-        # 发生任何异常，执行回滚，撤销本次会话中未提交的数据库改动
         db.rollback()
-        # 异常情况返回None，上层api可以判断新增失败
         return None
 
 
 def get_employment_by_id(db, eid):
     """根据主键id查询就业详情，过滤逻辑删除"""
     try:
-        db_obj = db.query(Employment).filter(
-            Employment.id == eid,
-            Employment.is_deleted == 0
-        ).first()
+        db_obj = db.query(Employment).filter(Employment.id == eid
+                                             ,Employment.is_deleted == 0).first()
         if not db_obj:
             return None
         return EmploymentResponse.model_validate(db_obj).model_dump()
@@ -41,10 +32,8 @@ def get_employment_by_id(db, eid):
 def get_employment_by_student_id(db, student_id):
     """根据学生id查询就业信息，过滤逻辑删除"""
     try:
-        db_obj = db.query(Employment).filter(
-            Employment.student_id == student_id,
-            Employment.is_deleted == 0
-        ).first()
+        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()
@@ -78,16 +67,11 @@ def get_employment_list(db, company_name=None, salary_min=None, salary_max=None,
 def update_employment(db, eid, obj_in):
     """修改就业信息，只更新传入的字段"""
     try:
-        # 根据id查询未删除的记录
-        db_obj_raw = db.query(Employment).filter(
-            Employment.id == eid,
-            Employment.is_deleted == 0
-        ).first()
-        # 记录不存在直接返回None
+        db_obj_raw = db.query(Employment).filter(Employment.id == eid
+                                                ,Employment.is_deleted == 0).first()
         if db_obj_raw is None:
             return None
-        # exclude_unset=True：只取出前端实际传过来的字段，不会带上没传的None字段
-        update_dict = obj_in.model_dump(exclude_unset=True)
+        update_dict = obj_in.model_dump(exclude_unset=True)# exclude_unset=True：只取出前端实际传过来的字段，不会带上没传的None字段
         for key, value in update_dict.items():
             setattr(db_obj_raw, key, value)
         db.commit()
@@ -101,10 +85,8 @@ def update_employment(db, eid, obj_in):
 def delete_employment_logic(db, eid):
     """逻辑删除，只修改is_deleted=1，不做物理删除"""
     try:
-        db_obj_raw = db.query(Employment).filter(
-            Employment.id == eid,
-            Employment.is_deleted == 0
-        ).first()
+        db_obj_raw = db.query(Employment).filter(Employment.id == eid
+                                                 ,Employment.is_deleted == 0).first()
         if db_obj_raw is None:
             return False
         db_obj_raw.is_deleted = 1

main.py

@@ -29,7 +29,8 @@ app.include_router(statistics_router, prefix="/statistics")
 from api.score import scores_router
 app.include_router(scores_router, tags=['成绩接口'])
 #-------------南方宇---------------
-
+from api.employment import employment_router
+app.include_router(employment_router, prefix="/employment")
 #-----------主程序----------------
 
 if __name__ == "__main__":

models/employment.py

@@ -1,13 +1,13 @@
 from sqlalchemy import *
 from sqlalchemy.orm import declarative_base,sessionmaker
 from datetime import datetime
-from core.database import Base, engine
+from core.database import Base
 
 # 为了符合三范式，就业表只保存 student_id，不保存学生姓名和班级名称。查询返回时通过关联 students 和 classes 带出学生姓名、班级名称。
 
 class Employment(Base):
-    __tablename__ = "employment_info_detail"                              # 实际数据库的表名字
-    id = Column(Integer, primary_key=True                   # 主键
+    __tablename__ = "employment_info_detail"                   # 实际数据库的表名字
+    id = Column(Integer, primary_key=True                      # 主键
                 , autoincrement=True                           # 声明是自增主键
                 ,comment="就业信息主键id")
     student_id = Column(Integer, ForeignKey("student_info_detail.id")  # 外键 -> students.id

schemas/employment.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field,ConfigDict
 from datetime import date, datetime
 from decimal import Decimal
 
@@ -36,3 +36,4 @@ class EmploymentResponse(BaseModel):
     remark: str | None
     created_at: datetime | None
     updated_at: datetime | None
+    model_config = ConfigDict(from_attributes=True) #允许pydantic从对象属性里取值