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