后端

back_end/server.py

@@ -1,15 +1,12 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Depends, File, UploadFile
-from pydantic import BaseModel
+from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 import random
 import time
-from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import JSONResponse
 import pandas as pd
-# from openpyxl import Workbook, load_workbook
 import math
 import os
-from fastapi import Depends
 
 app = FastAPI()
 
@@ -22,9 +19,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# DATA_PATH = os.path.abspath(__file__) + "\\data\\"
+DATA_PATH = "./back_end/data/"
-DATA_PATH = "./demo/Data/"
 
+# 初始excel信息
 initial_data = {
     '学号': ['*********'],
     '姓名': ['BOSS_K'],
@@ -32,8 +29,7 @@ initial_data = {
 }
 initial_info = {
     'basicProbability': [1.0],
-    # 'lastRow': [0],
+    'maxPoint': [9.9],
-    'maxPoint': [0.0],
     'minPoint': [0.0]
 }
 initial_data_df = pd.DataFrame(initial_data)
@@ -49,9 +45,16 @@ class StudentRequest(BaseModel):
 async def get_file_path(class_name: str):
     file_path = DATA_PATH + class_name + ".xlsx"
     if not os.path.exists(file_path): 
-        raise HTTPException(status_code=403, detail="Not authenticated")
+        raise HTTPException(status_code=403, detail="班级不存在")
     return file_path
 
+#查询班级是否存在
+@app.post("/{class_name}/search-room")
+async def upload_file(class_name: str):
+    file_path = DATA_PATH + class_name + ".xlsx"
+    if not os.path.exists(file_path): 
+        return {"message": "班级不存在"}
+    return {"message": "班级存在"}
 
 # 上传excel
 @app.post("/{class_name}/upload")
@@ -64,12 +67,10 @@ async def upload_file(file_path: str = Depends(get_file_path), file: UploadFile
         # 将 Excel 文件读取为 DataFrame
         df_data = pd.read_excel(file_path, sheet_name='Data')
         df_info = pd.read_excel(file_path, sheet_name='Info')
+        df_info['basicProbability'] = df_info['basicProbability'].astype(float)
         df_upload = pd.read_excel(file.file)
 
-        new_rows = []
+        new_rows = [] # 新增信息
-
-        # maxPoint = df_info.loc[0, 'maxPoint']
-        # minPoint = df_info.loc[0, 'minPoint']
 
         for index, row in df_upload.iterrows():
             id = row['学号']        # 根据列名访问学号
@@ -77,36 +78,29 @@ async def upload_file(file_path: str = Depends(get_file_path), file: UploadFile
             points = 0.0
             if '积分' in row:
                 points = row['积分']
-            # maxPoint = math.max(points, maxPoint)
-            # minPoint = math.min(points, minPoint)
             new_rows.append({'学号': id, '姓名': name, '积分': points})
 
-        df_data = pd.concat([df_data, pd.DataFrame(new_rows)], ignore_index=True)
+        df_data = pd.concat([df_data, pd.DataFrame(new_rows)], ignore_index=True) # 合并
-
+        df_data = df_data.sort_values(by='积分', ascending=False) # 根据 '积分' 列降序排序
-        # 根据 '积分' 列降序排序
-        df_data = df_data.sort_values(by='积分', ascending=False)
 
+        # 更新excel
         with pd.ExcelWriter(file_path, mode='a', engine='openpyxl') as writer:
             del writer.book['Data']  # 删除旧的工作表
             df_data.to_excel(writer, sheet_name='Data', index=False)
-        
-        
         df_info.loc[0, 'basicProbability'] = 1.0
-        # df_info.loc[0, 'maxPoint'] = maxPoint
-        # df_info.loc[0, 'minPoint'] = minPoint
         with pd.ExcelWriter(file_path, mode='a', engine='openpyxl') as writer:
             del writer.book['Info']  # 删除旧的工作表
             df_info.to_excel(writer, sheet_name='Info', index=False)
         
         return {"message": "文件上传成功"}
 
+    # 上传失败
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
 # 创建班级的接口
-@app.get("/create-class/{class_name}")
+@app.post("/create-class/{class_name}")
 async def create_class(class_name: str):
-    # return {"ok":"ok"}
     file_path = DATA_PATH + class_name + ".xlsx"
     if os.path.exists(file_path):
         return {"message": "班级已存在"}
@@ -119,73 +113,80 @@ async def create_class(class_name: str):
     return {"message": "班级创建成功"}
 
 # 随机点名的接口
-@app.get("/{class_name}/random-call")
+@app.post("/{class_name}/random-call")
 async def random_call(file_path: str = Depends(get_file_path)):
     df_data = pd.read_excel(file_path, sheet_name='Data')
     df_info = pd.read_excel(file_path, sheet_name='Info')
 
-    n = len(df_data)
+    n = len(df_data) #学生数量
     if n <= 0:
         raise HTTPException(status_code=404, detail="没有学生可供点名")
     
-    # 进行随机
+    df_info['basicProbability'] = df_info['basicProbability'].astype(float)
     basicProbability = float(df_info.iloc[0]['basicProbability'])
-    # lastRow = int(df_info.iloc[0]['lastRow'])
+
     maxPoint = float(df_info.iloc[0]['maxPoint'])
     minPoint = float(df_info.iloc[0]['minPoint'])
 
     res = {}
-    basicProbability *= 0.8
+    
+    # 进行随机抽取
+    basicProbability *= 0.5
     while (True):
-        # print(basicProbability)
+        # 从下标[0, n-1]中随机选出一名学生, 并获取其当前积分
-        # lastRow = (lastRow + 1) % n
         i = math.floor(random.random() * n)
         points = df_data.iloc[i]['积分']
-        hit_probability = (maxPoint - points + 1) / (maxPoint - minPoint + 20) * basicProbability
+        # 根据当前班级中积分情况，计算该学生被抽中的相对概率, 分数越高抽中概率越低
-        if random.random() <= hit_probability: #成功命中倒霉蛋
+        hit_probability = (maxPoint - points + 1) / (maxPoint - minPoint + 10) * basicProbability
+
+        if random.random() <= hit_probability: #成功命中幸运儿
             res = {"name": str(df_data.iloc[i]['姓名']), 
                    "id": str(df_data.iloc[i]['学号']), 
                    "points": float(points)}
             break
-        basicProbability *= 1.03
+        #未抽到学生时，提高基础概率，防止长时间抽不到人
+        #当前抽到一个名学生期望需要约35(log(1/0.5)/log(1.02))次
+        basicProbability *= 1.02 
 
     # 更新信息
     df_info.loc[0, 'basicProbability'] = basicProbability
-    # df_info.loc[0, 'lastRow'] = lastRow
+
     with pd.ExcelWriter(file_path, mode='a', engine='openpyxl') as writer:
         del writer.book['Info']  # 删除旧的工作表
         df_info.to_excel(writer, sheet_name='Info', index=False)
 
     return res
 
-
 # 修改积分的接口
 @app.post("/{class_name}/change-points")
 async def change_points(student_request: StudentRequest, file_path: str = Depends(get_file_path)):
     
+    #获取当前班级的数据
     df_data = pd.read_excel(file_path, sheet_name='Data')
     df_info = pd.read_excel(file_path, sheet_name='Info')
+    df_data['积分'] = df_data['积分'].astype(float)
+    df_info['maxPoint'] = df_info['maxPoint'].astype(float)
+    df_info['minPoint'] = df_info['minPoint'].astype(float)
 
-    print(df_data)
+    index = -1 # 点到学生的序号
-    index = -1
     for i, row in df_data.iterrows():
-        print(row['学号'], student_request.id)
-        print(row['姓名'], student_request.name)
         if (str(row['学号']) == str(student_request.id) and str(row['姓名']) == str(student_request.name)):
             index = i
             break
     
+    #未找到学生
     if (index < 0):
         raise HTTPException(status_code=400, detail="学号姓名无效")
     
-    df_data.loc[index, '积分'] = student_request.points
+    # 更新信息，并导入excel
+    df_data.loc[index, '积分'] += student_request.points
+    df_info.loc[0, 'maxPoint'] = max(df_info.loc[0, 'maxPoint'], df_data.loc[index, '积分'])
+    df_info.loc[0, 'minPoint'] = min(df_info.loc[0, 'minPoint'], df_data.loc[index, '积分'])
+
     df_data = df_data.sort_values(by='积分', ascending=False)
     with pd.ExcelWriter(file_path, mode='a', engine='openpyxl') as writer:
         del writer.book['Data']  # 删除旧的工作表
         df_data.to_excel(writer, sheet_name='Data', index=False)
-    
-    df_info.loc[0, 'maxPoint'] = max(df_info.loc[0, 'maxPoint'], student_request.points)
-    df_info.loc[0, 'minPoint'] = min(df_info.loc[0, 'minPoint'], student_request.points)
     with pd.ExcelWriter(file_path, mode='a', engine='openpyxl') as writer:
         del writer.book['Info']  # 删除旧的工作表
         df_info.to_excel(writer, sheet_name='Info', index=False)