From 373b214cff663feebc05641ee023fb98b5d4cb91 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=80=C3=AE=C3=97=C3=9A=C3=95=C3=B1?= <lizz556@163.com>
Date: Mon, 29 Aug 2022 15:45:42 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8A=A5=E8=A1=A8=E5=88=86=E7=BB=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/api_v1/endpoints/interview.py | 40 ++++++++++++++++++-------------
 schemas/interview_modes.py        |  2 +-
 2 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/api/api_v1/endpoints/interview.py b/api/api_v1/endpoints/interview.py
index 5ed650d..d8c077c 100644
--- a/api/api_v1/endpoints/interview.py
+++ b/api/api_v1/endpoints/interview.py
@@ -888,24 +888,30 @@ async def find_mode(
 ) -> schemas.Msg:
     """查找hr所有分组"""
     data = await crud.api_interview_modes.get_interview_modes(db, data_in)
-    new_data = []
-    if data:
-        for i in data:
-            new_i = deepcopy(i)
-            tables = i['mode_list']
-            if tables:
-                new_tables = []
-                table_data = await crud.api_interview_tables.get_tables(db, {'table_id': {'$in': tables}})
-                for table in table_data:
-                    new_table = {
-                        'title': table['title'],
-                        'url': table['url']
-                    }
-                    new_tables.append(new_table)
-                new_i['mode_list'] = new_tables
-            new_data.append(new_i)
+    return schemas.Msg(code=200, msg='ok', data=data)
 
-    return schemas.Msg(code=200, msg='ok', data=new_data)
+
+@router.post("/find_mode_tables")
+async def find_mode_tables(
+        request: Request,
+        data_in: schemas.UpdateModes,
+        db: AsyncIOMotorDatabase = Depends(get_database),
+) -> schemas.Msg:
+    """获取一条hr分组包含的报表"""
+    mode_data = await crud.api_interview_modes.get_one_mode(db, data_in.mode_id)
+    new_tables = []
+    if mode_data:
+        tables = mode_data['mode_list']
+        if tables:
+            table_data = await crud.api_interview_tables.get_tables(db, {'table_id': {'$in': tables}})
+            for table in table_data:
+                new_table = {
+                    'table_id': table['table_id'],
+                    'title': table['title'],
+                    'url': table['url']
+                }
+                new_tables.append(new_table)
+    return schemas.Msg(code=200, msg='ok', data=new_tables)
 
 
 @router.post("/update_mode")
diff --git a/schemas/interview_modes.py b/schemas/interview_modes.py
index 5368fb1..edff878 100644
--- a/schemas/interview_modes.py
+++ b/schemas/interview_modes.py
@@ -28,4 +28,4 @@ class FindModes(BaseModel):
 # 更新分组
 class UpdateModes(BaseModel):
     mode_id: str  # 模板唯一id
-    update_data: dict  # 更新的数据
+    update_data: dict = {}  # 更新的数据