api,dao

PythonProject/dao/employment_dao.py

@@ -1,7 +1,8 @@
-from fastapi import HTTPException
+from fastapi import HTTPException, APIRouter
 from model.all_model import Employment_info, ClassManagement, Student_Model
+from datetime import datetime
 
-
+emp_api = APIRouter()
 def get_emp_dao(stu_id=None,class_id=None,company_name=None,min_salary=None,max_salary=None,db=None):
     try:
         q = db.query(Employment_info).\
@@ -41,18 +42,25 @@ def get_emp_dao(stu_id=None,class_id=None,company_name=None,min_salary=None,max_
 
 def add_emp_dao(o,db):
     try:
+        stu = db.query(Student_Model).filter(Student_Model.stu_id == o['stu_id'], Student_Model.delete_status == 0).first()
+        if not stu:
+            raise HTTPException(status_code=404, detail='该学生不存在,无法添加就业信息')
+
+        exist = db.query(Employment_info).filter(Employment_info.stu_id == o['stu_id'],
+                                                 Employment_info.delete_status == 0).first()
+        if exist:
+            raise HTTPException(status_code=409, detail='该学生已有就业信息，请勿重复增加！')
+
         e1 = Employment_info(**o)
         db.add(e1)
         db.commit()
-        return e1
+        return o
+    except HTTPException:
+        raise
     except Exception as e:
         db.rollback()
         raise HTTPException(status_code=500, detail=f'新增就业信息失败: {e}')
 
-def check_stu_dao(stu_id,db):
-    stu = db.query(Student_Model).filter(Student_Model.stu_id == stu_id,Student_Model.delete_status == 0).first()
-    return stu
-
 def update_emp_dao(stu_id,o,db):
     try:
         rows = db.query(Employment_info).filter(Employment_info.stu_id == stu_id,Employment_info.delete_status == 0).first()
@@ -65,7 +73,9 @@ def update_emp_dao(stu_id,o,db):
 
 def delete_emp_dao(stu_id,db):
     try:
-        rows = db.query(Employment_info).filter(Employment_info.stu_id == stu_id,Employment_info.delete_status == 0).delete()
+        rows = db.query(Employment_info)\
+            .filter(Employment_info.stu_id == stu_id,Employment_info.delete_status == 0)\
+            .update({'delete_status':1,'delete_time': datetime.now()})
         db.commit()
     except:
         db.rollback()