diff --git a/api/api_v1/endpoints/data_mana.py b/api/api_v1/endpoints/data_mana.py index 3f127e8..54a4027 100644 --- a/api/api_v1/endpoints/data_mana.py +++ b/api/api_v1/endpoints/data_mana.py @@ -89,35 +89,24 @@ async def edit_data_attr( # return schemas.Msg(code=0, msg='ok', data=data_in) #在gametoos同步区服了,所以不需要这段代码 -# @router.post("/add_select_map") -# async def add_select_map( -# game: str, -# url: str, -# #file: bytes = File(...), -# db: AsyncIOMotorDatabase = Depends(get_database), -# current_user: schemas.UserDB = Depends(deps.get_current_user) -# ) -> schemas.Msg: -# """添加游戏区服信息选择映射""" - # dfs = pd.read_excel(file, engine='openpyxl', sheet_name=None) - # for attr_name, df in dfs.items(): - # #将id这列转换成字符串类型 - # if len(df) >0: - # df['id'] = df['id'].astype(str) - # map_ = df.to_dict('records') - # data_in = schemas.SelectMap(game=game, attr_name=attr_name, map_=map_) - # await crud.select_map.save(db, data_in) - # return schemas.Msg(code=0, msg='ok', data=1) - # response = requests.get(url) - # data = json.loads(response.text) - # attr_name = 'svrindex' - # map_list = [] - # for k, v in data.items(): - # map_dict = {} - # map_dict[k] = v['servername'] - # map_list.append(map_dict) - # data_in = schemas.SelectMap(game=game, attr_name=attr_name, map_=map_list) - # await crud.select_map.save(db, data_in) - # return schemas.Msg(code=0, msg='ok', data=1) +@router.post("/add_select_map") +async def add_select_map( + game: str, + file: bytes = File(...), + db: AsyncIOMotorDatabase = Depends(get_database), + current_user: schemas.UserDB = Depends(deps.get_current_user) +) -> schemas.Msg: + """添加游戏区服信息选择映射""" + dfs = pd.read_excel(file, engine='openpyxl', sheet_name=None) + for attr_name, df in dfs.items(): + #将id这列转换成字符串类型 + if len(df) >0: + df['id'] = df['id'].astype(str) + map_ = df.to_dict('records') + data_in = schemas.SelectMap(game=game, attr_name=attr_name, map_=map_) + await crud.select_map.save(db, data_in) + return schemas.Msg(code=0, msg='ok', data=1) + @router.get("/select_list") async def select_list(