From 1e43b15cfb627ec5eb3b89fdb72d809caa52108f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E4=BC=9F?= <250213850@qq.com>
Date: Wed, 31 Aug 2022 16:35:31 +0800
Subject: [PATCH] =?UTF-8?q?1.=E6=B7=BB=E5=8A=A0token=E9=AA=8C=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/api_v1/endpoints/interview.py | 135 ++++++++++++++++++++----------
 1 file changed, 89 insertions(+), 46 deletions(-)

diff --git a/api/api_v1/endpoints/interview.py b/api/api_v1/endpoints/interview.py
index ea4fd43..df07549 100644
--- a/api/api_v1/endpoints/interview.py
+++ b/api/api_v1/endpoints/interview.py
@@ -101,6 +101,7 @@ async def interview_find(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据查询 """
     await interview.init()
@@ -118,6 +119,7 @@ async def interview_home_find(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据主页初筛查询 """
     await interview.init()
@@ -135,6 +137,7 @@ async def interview_screen_find(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据复筛查询 """
     await interview.init()
@@ -152,6 +155,7 @@ async def interview_exam_find(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据面试阶段查询 """
     await interview.init()
@@ -169,6 +173,7 @@ async def interview_stage_num(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据offer阶段查询 """
     await interview.init()
@@ -203,6 +208,7 @@ async def interview_waite_in_find(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据待入职阶段查询 """
     await interview.init()
@@ -222,6 +228,7 @@ async def interview_update(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据更新 """
     await interview.init()
@@ -240,6 +247,7 @@ async def interview_insert(
         request: Request,
         interview: InterviewDo = Depends(InterviewDo),
         db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据写入 """
     await interview.init()
@@ -255,7 +263,8 @@ async def interview_insert(
 async def file_to_hw(
         request: Request,
         file: UploadFile = File(...),
-        db: CKDrive = Depends(get_ck_db)
+        db: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ 简历上传到华为云 """
     path_data = os.getcwd() + '/jianli'  # 当前文件所在的目录
@@ -353,6 +362,7 @@ async def file_to_hw(
 async def interview_file_insert(
         request: Request,
         data_in: schemas.Filenames,
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """ interview面试数据导入 """
     path_data = os.getcwd() + '/jianli'  # 当前文件所在的目录
@@ -489,6 +499,7 @@ async def add_job(
         request: Request,
         data_in: schemas.Ins_Job,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """新增职位"""
     await crud.jobs.insert_job(db, data_in)
@@ -500,6 +511,7 @@ async def find_job(
         request: Request,
         data_in: schemas.Find_job,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取单个职位信息"""
     res = await crud.jobs.find_job(db, data_in.job_id)
@@ -590,7 +602,8 @@ async def get_job(
         request: Request,
         data_in: schemas.Jobs = None,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        ckdb: CKDrive = Depends(get_ck_db)
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取职位名称"""
     # 获取对应条件的职位
@@ -630,7 +643,9 @@ async def get_job(
 async def download_interview(
         request: Request,
         data_in: schemas.Timesinter,
-        ckdb: CKDrive = Depends(get_ck_db)):
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
+):
     """导出所有面试安排"""
     sql = f"""SELECT interview_type,interview_stage,star_time,end_time,name,phone,job_name,hr_name,interview_name,
     interview_sign FROM HR.resumes WHERE toDate(star_time) >= '{data_in.start_time}' and toDate(star_time) <= '{data_in.end_time}'"""
@@ -664,6 +679,7 @@ async def add_remark(
         request: Request,
         data_in: schemas.BaseRemark,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """添加备注"""
     await crud.api_interview_remark.insert_remark(db, data_in)
@@ -675,6 +691,7 @@ async def find_remark(
         request: Request,
         data_in: schemas.FindRemark,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """查找备注"""
     data = await crud.api_interview_remark.get_interview_remarks(db, data_in)
@@ -685,7 +702,8 @@ async def find_remark(
 async def find_criterion(
         request: Request,
         data_in: schemas.get_uids,
-        ckdb: CKDrive = Depends(get_ck_db)
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取标准版简历"""
     # sql = f"""
@@ -711,7 +729,8 @@ async def find_criterion(
 async def resume_affix(
         request: Request,
         data_in: schemas.Post_uid,
-        ckdb: CKDrive = Depends(get_ck_db)
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """保存附件id"""
     sql = f"""
@@ -726,7 +745,8 @@ async def updata_entry(
         request: Request,
         data_in: schemas.Entry,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """修改入职状态"""
     sql = f"""
@@ -744,7 +764,8 @@ async def updata_entry(
 async def get_str_mail(
         request: Request,
         data_in: schemas.Email_str,
-        ckdb: CKDrive = Depends(get_ck_db)
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取邮件文本"""
     data = f"""
@@ -758,7 +779,8 @@ async def get_str_mail(
 async def send_mail(
         request: Request,
         data_in: schemas.send_str_mail,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """发送邮件"""
     try:
@@ -782,7 +804,8 @@ async def send_mail(
 async def email_record(
         request: Request,
         data_in: schemas.get_email_record,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取发送邮件的记录"""
     data = await crud.email_record.all_record(db, data_in)
@@ -793,7 +816,8 @@ async def email_record(
 async def operate_log(
         request: Request,
         data_in: schemas.operate_log,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """同步插入操作记录"""
     await crud.operate_log.create(db, data_in)
@@ -804,7 +828,8 @@ async def operate_log(
 async def get_operate_log(
         request: Request,
         data_in: schemas.get_operate_log,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取操作记录"""
     data = await crud.operate_log.all_log(db, data_in)
@@ -827,7 +852,8 @@ async def get_dding_user(
 async def teacher_state(
         request: Request,
         data_in: schemas.user_id,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """催促面试官反馈功能"""
     # 同步钉钉发消息给面试官
@@ -845,6 +871,7 @@ async def add_mode(
         request: Request,
         data_in: schemas.InsertModes,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """添加hr分组"""
     await crud.api_interview_modes.insert_modes(db, data_in)
@@ -856,6 +883,7 @@ async def add_table(
         request: Request,
         data_in: schemas.InsertTables,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """添加分组报表"""
     await crud.api_interview_tables.insert_tables(db, data_in)
@@ -875,6 +903,7 @@ async def update_table(
         request: Request,
         data_in: schemas.UpdateTables,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """修改表报"""
     await crud.api_interview_tables.update_tables(db, data_in)
@@ -886,6 +915,7 @@ async def find_mode(
         request: Request,
         data_in: schemas.FindModes,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """查找hr所有分组"""
     data = await crud.api_interview_modes.get_interview_modes(db, data_in)
@@ -897,6 +927,7 @@ async def find_mode_tables(
         request: Request,
         data_in: schemas.UpdateModes,
         db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取一条hr分组包含的报表"""
     mode_data = await crud.api_interview_modes.get_one_mode(db, data_in.mode_id)
@@ -983,11 +1014,12 @@ async def delete(
 
 
 @router.post("/get_dashboard")
-async def get_dashboard(request: Request,
-                        data_in: schemas.ReadDashboard,
-                        db: AsyncIOMotorDatabase = Depends(get_database),
-                        current_user: schemas.UserDB = Depends(deps.get_current_user)
-                        ):
+async def get_dashboard(
+        request: Request,
+        data_in: schemas.ReadDashboard,
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
+):
     """获取一个看板"""
     res = await crud.dashboard.get(db, id=data_in.id)
     reports = {item['report_id']: item for item in res['reports']}
@@ -1014,7 +1046,8 @@ async def notarize(
         request: Request,
         data_in: schemas.Notarize,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """简历确认进入面试阶段"""
     sql = f"""
@@ -1054,7 +1087,8 @@ async def alter(
         request: Request,
         data_in: schemas.Stage,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """更改简历阶段"""
     if data_in.stage == 1:  # 初筛
@@ -1074,7 +1108,8 @@ async def get_report(
         request: Request,
         data_in: schemas.Findreport,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获得已添加的报表"""
     # 获取自己的报表
@@ -1087,7 +1122,8 @@ async def create_report(
         request: Request,
         data_in: schemas.ReportCreate,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """新建报表"""
     try:
@@ -1101,7 +1137,8 @@ async def create_report(
 async def up_report(
         request: Request,
         data_in: schemas.ReportEdit,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """编辑报表"""
     await crud.report.update_one(db, {'_id': data_in.report_id},
@@ -1117,10 +1154,12 @@ async def up_report(
 
 
 @router.post("/edit_report")
-async def edit_report(request: Request,
-                      data_in: schemas.EditReport,
-                      db: AsyncIOMotorDatabase = Depends(get_database)
-                      ) -> schemas.Msg:
+async def edit_report(
+        request: Request,
+        data_in: schemas.EditReport,
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
+) -> schemas.Msg:
     """看板样式设置"""
     await crud.dashboard.update_one(db, {'_id': data_in.id, 'reports.report_id': data_in.report.report_id},
                                     {'$set': {f'reports.$.{k}': v for k, v in
@@ -1132,7 +1171,8 @@ async def edit_report(request: Request,
 async def head(
         request: Request,
         data_in: schemas.Post_head,
-        ckdb: CKDrive = Depends(get_ck_db)
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """保存头像链接"""
     sql = f"""
@@ -1146,7 +1186,8 @@ async def head(
 async def hint(
         request: Request,
         data_in: schemas.Get_hr,
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """重要事项-面试相关"""
     # 邮件，反馈
@@ -1164,7 +1205,8 @@ async def up_hint(
         request: Request,
         data_in: schemas.Up_hint,
         ckdb: CKDrive = Depends(get_ck_db),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """批量修改已读状态"""
     # 面试记录
@@ -1180,7 +1222,7 @@ async def up_hint(
 async def login(
         #data_in: schemas.Login,
         data: ExtendendOAuth = Depends(),
-        db: AsyncIOMotorDatabase = Depends(get_database)
+        db: AsyncIOMotorDatabase = Depends(get_database),
 ) -> Any:
     """
     OAuth2兼容令牌登录，获取将来令牌的访问令牌
@@ -1232,7 +1274,7 @@ async def login(
 async def reset_password(request: Request,
                          data_in: schemas.UserRestPassword,
                          db: AsyncIOMotorDatabase = Depends(get_database),
-                         # current_user: schemas.User = Depends(deps.get_current_user)
+                         current_user: schemas.User = Depends(deps.get_current_user)
                          ) -> Any:
     """
     修改其他人密码
@@ -1263,7 +1305,7 @@ async def add_account(
         request: Request,
         data_in: schemas.Createuser,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.User = Depends(deps.get_current_user)
+        current_user: schemas.User = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     创建新账号
@@ -1287,7 +1329,7 @@ async def add_account(
 async def forbid_login(request: Request,
                        data_in: schemas.Get_userid,
                        db: AsyncIOMotorDatabase = Depends(get_database),
-                       # current_user: schemas.User = Depends(deps.get_current_user)
+                       current_user: schemas.User = Depends(deps.get_current_user)
                        ) -> schemas.Msg:
     """
     禁止/解禁用户登录功能
@@ -1304,7 +1346,7 @@ async def owner_list(
         request: Request,
         data_in: schemas.Getdate,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取基本信息列表"""
     try:
@@ -1320,7 +1362,7 @@ async def owner_edit(
         request: Request,
         data_in: schemas.Ownername,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """新增，删除基本信息"""
     await crud.basic_data.update(db, data_in)
@@ -1331,7 +1373,7 @@ async def owner_edit(
 async def get_section(
         request: Request,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """获取对应部门及职位"""
     res = await crud.section.get_all(db)
@@ -1343,7 +1385,7 @@ async def get_section(
         request: Request,
         data_in: schemas.Ins_section,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """新增对应部门及职位"""
     await crud.section.ins_section(db, data_in)
@@ -1354,7 +1396,7 @@ async def get_section(
 async def api_list(
         request: Request,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     所有的api
@@ -1373,7 +1415,7 @@ async def add_policy(
         request: Request,
         data_in: schemas.Add_module,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ):
     """修改api_module权限状态"""
     res = await crud.api_module.get_one_module(db, data_in)
@@ -1391,7 +1433,7 @@ async def add_policy(
 async def domain_list(
         request: Request,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     角色管理创建角色时显示的各个模块
@@ -1412,7 +1454,7 @@ async def add_api(
         request: Request,
         data_in: schemas.AddApi,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     '在api_module'添加api
@@ -1453,7 +1495,7 @@ async def add_policy(
         request: Request,
         data_in: schemas.Datalist,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ):
     """
     向当前权限添加新路由
@@ -1471,7 +1513,7 @@ async def remove_policy(
         request: Request,
         data_in: schemas.Del_role,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ):
     """
     修改角色api权限
@@ -1488,7 +1530,7 @@ async def remove_policy(
 async def roles(
         request: Request,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     获取所有的管理员用户
@@ -1530,7 +1572,7 @@ async def add_roles(
         request: Request,
         data_in: schemas.Add_role,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        # current_user: schemas.UserDB = Depends(deps.get_current_user)
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """
     创建不同权限角色
@@ -1595,7 +1637,8 @@ async def add_role_domain(
         request: Request,
         data_in: schemas.AddRoleForUsersInDomain,
         db: AsyncIOMotorDatabase = Depends(get_database),
-        current_user: schemas.UserDB = Depends(deps.get_current_user)):
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
+):
     """
     在当前项目为角色添加相应权限
     """