diff --git a/StatCalc/api/statcalc_api.py b/StatCalc/api/statcalc_api.py index ad32a4e..d82bf7b 100644 --- a/StatCalc/api/statcalc_api.py +++ b/StatCalc/api/statcalc_api.py @@ -62,14 +62,14 @@ from students.model.students_model import Students from class_management.model.class_management_model import ClassInfo @statcalc_api.get('/statcalc/top',summary='就业薪资Top5') -def salarytop5(db=Depends(get_db)): +def salarytop5(num:int,db=Depends(get_db)): q = db.query(Employments, Company.employment_company, Students.name.label('sname'), ClassInfo.name.label('class_num')) \ .join(Company, Employments.company_id == Company.id) \ .join(Students, Employments.stuid == Students.id) \ .join(ClassInfo, Employments.class_id == ClassInfo.id) \ .filter(Employments.is_deleted == 0) \ .order_by(Employments.employment_salary.desc()) \ - .limit(5).all() + .limit(num).all() # q是4元组 (Employments对象, company_name, sname, class_num) return [{'学生姓名': row[2], '学生班级编号': row[3], '就业时间': str(row[0].offer_recived_time) if row[0].offer_recived_time else None, '公司': row[1]} for row in q]