1.添加token验证

This commit is contained in:
李伟 2022-08-31 16:35:31 +08:00
parent 3e1195febb
commit 1e43b15cfb

View File

@ -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,7 +1014,8 @@ async def delete(
@router.post("/get_dashboard")
async def get_dashboard(request: Request,
async def get_dashboard(
request: Request,
data_in: schemas.ReadDashboard,
db: AsyncIOMotorDatabase = Depends(get_database),
current_user: schemas.UserDB = Depends(deps.get_current_user)
@ -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,9 +1154,11 @@ async def up_report(
@router.post("/edit_report")
async def edit_report(request: Request,
async def edit_report(
request: Request,
data_in: schemas.EditReport,
db: AsyncIOMotorDatabase = Depends(get_database)
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},
@ -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)
):
"""
在当前项目为角色添加相应权限
"""