prs_server/api/api_v1/endpoints/forms.py
2022-07-25 18:13:17 +08:00

650 lines
21 KiB
Python

import operator
import os
import re
import pandas as pd
from copy import deepcopy
from fastapi import APIRouter, Depends, Request, File, UploadFile
from motor.motor_asyncio import AsyncIOMotorDatabase
from utils.dingding import get_redis_alluid, send_dates
from utils.jianli import get_resume
from utils.func import get_every_days
import crud, schemas
from datetime import datetime
from core.configuration import *
from db import get_database
from db.ckdb import get_ck_db, CKDrive
from models.interview_zsgc import InterviewDo
from utils import get_time, qujian_time, Download_xlsx, send_str_mail
router = APIRouter()
# 候选人质量报表
@router.post("/man_mass_form")
async def man_mass_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
""" 候选人质量报表 """
await interview.init()
res = interview.get_man_mass_form_sql()
sql = res['sql']
data = await db.execute(sql)
school_985 = ['清华大学', '北京大学', '中国人民大学', '北京理工大学', '北京航空航天大学', '中央民族大学', '北京师范大学',
'中国农业大学', '天津大学', '南开大学', '复旦大学', '上海交通大学', '同济大学', '华东师范大学', '重庆大学', '四川大学',
'电子科技大学', '湖南大学', '国防科技大学', '中南大学', '厦门大学']
school_211 = ['上海外国语大学', '东华大学', '上海财经大学', '华东理工大学', '上海大学', '天津医科大学', '吉林大学', '东北师范大学',
'延边大学', '哈尔滨工业大学', '哈尔滨工程大学', '东北农业大学', '东北林业大学', '南京大学', '东南大学', '苏州大学',
'中国矿业大学', '中国药科大学', '河海大学', '南京航空航天大学', '江南大学', '南京农业大学', '南京理工大学', '浙江大学',
'中国科技大学', '安徽大学', '合肥工业大学', '福州大学', '南昌大学', '山东大学', '中国海洋大学', '石油大学',
'湖南师范大学', '广西大学', '中山大学', '暨南大学', '华南理工大学', '华南师范大学', '广州中医药大学', '武汉大学',
'华中科技大学', '中国地质大学', '武汉理工大学', '华中师范大学', '华中农业大学', '中南财经政法大学']
# 性别
gander = {
'': [],
'': []
}
# 年龄
age = {
'20-29': [],
'30-39': [],
'40-49': [],
'50-59': [],
'60-100': [],
}
# 学历
education = {
'大专': [],
'本科': [],
'研究生': [],
'博士': [],
'硕士': []
}
# 工作经验
work_exp = {
'0-2': [],
'3-6': [],
'6-10': [],
'10-20': [],
'20-40': []
}
# 所在地
account = {
}
# 毕业院校
school = {
}
# 就职公司
work_for = {
}
# 渠道
owner_name = {
'前程无忧': [],
'人才库': [],
'智联招聘': [],
'Boss直聘': [],
'58同城': []
}
# 职位
job_name = {
}
# 学校类型
school_type = {
'985': [],
'211': [],
'其他': [],
}
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
# 处理数据
for interview_data in data.values():
i_name = interview_data.get('name', '默认名字')
i_gander = interview_data.get('gander', '')
if i_gander == '':
gander[''].append(i_name)
else:
gander[''].append(i_name)
i_age = interview_data.get('age', 20)
if i_age <= 29:
age['20-29'].append(i_name)
elif 30 <= i_age <= 39:
age['30-39'].append(i_name)
elif 40 <= i_age <= 49:
age['40-49'].append(i_name)
elif 50 <= i_age <= 59:
age['50-59'].append(i_name)
else:
age['60-100'].append(i_name)
i_education = interview_data.get('education', 1)
if i_education == 1:
education['大专'].append(i_name)
elif i_education == 2:
education['本科'].append(i_name)
elif i_education == 3:
education['研究生'].append(i_name)
elif i_education == 4:
education['博士'].append(i_name)
else:
education['硕士'].append(i_name)
i_owner_name = interview_data.get('owner_name', 2)
if i_owner_name == 1:
owner_name['前程无忧'].append(i_name)
elif i_owner_name == 2:
owner_name['人才库'].append(i_name)
elif i_owner_name == 3:
owner_name['智联招聘'].append(i_name)
elif i_owner_name == 4:
owner_name['Boss直聘'].append(i_name)
else:
owner_name['58同城'].append(i_name)
i_school = interview_data.get('school', '')
if i_school:
if i_school in school:
school[i_school].append(i_name)
else:
school[i_school] = [i_name]
if i_school in school_985:
school_type['985'].append(i_name)
school_type['211'].append(i_name)
elif i_school in school_211:
school_type['211'].append(i_name)
else:
school_type['其他'].append(i_name)
i_work_exp = interview_data.get('work_exp', 1)
if i_work_exp <= 2:
work_exp['0-2'].append(i_name)
elif 3 <= i_work_exp < 6:
work_exp['3-6'].append(i_name)
elif 6 <= i_work_exp < 10:
work_exp['6-10'].append(i_name)
elif 10 <= i_work_exp < 20:
work_exp['10-20'].append(i_name)
else:
work_exp['20-40'].append(i_name)
i_job_name = interview_data.get('job_names', '')
if i_job_name:
if i_job_name in account:
job_name[i_job_name].append(i_name)
else:
job_name[i_job_name] = [i_name]
i_account = interview_data.get('account', '')
if i_account:
if i_account in account:
account[i_account].append(i_name)
else:
account[i_account] = [i_name]
i_work_list = interview_data.get('work_list', '')
if i_work_list:
data_work = eval(i_work_list[0])
i_work_for = data_work.get('name', '')
if i_work_for:
if i_work_for in work_for:
work_for[i_work_for].append(i_name)
else:
work_for[i_work_for] = [i_name]
res_msg = {
'gander': gander,
'age': age,
'education': education,
'work_exp': work_exp,
'account': account,
'school': school,
'work_for': work_for,
'owner_name': owner_name,
'job_name': job_name,
'school_type': school_type
}
return schemas.Msg(code=200, msg='ok', data=res_msg)
# 候选人明细报表
@router.post("/man_info_form")
async def man_info_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
""" 候选人明细报表 """
await interview.init()
res = interview.get_man_info_form_sql()
sql = res['sql']
data = await db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
for key, interview_data in data.items():
i_work_list = interview_data.get('work_list', '')
if i_work_list:
data_work = eval(i_work_list[0])
i_work_for = data_work.get('name', '')
if i_work_for:
data[key]['work_for'] = i_work_for
continue
data[key]['work_for'] = ''
return schemas.Msg(code=200, msg='ok', data=data)
# 职位阶段数据报表
@router.post("/every_stage_form")
async def every_stage_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
ck_db: CKDrive = Depends(get_ck_db),
db: AsyncIOMotorDatabase = Depends(get_database)
) -> schemas.Msg:
""" 职位阶段数据报表 """
all_jobs = await crud.jobs.all_some_field(db)
res_msg = {}
job_id_to_name = {}
job_ids = []
for job_data in all_jobs:
job_id = job_data['job_id']
job_ids.append(job_id)
job_name = job_data['job_name']
job_id_to_name[job_id] = job_name
res_job_data = deepcopy({k: v for k, v in job_data.items() if k != 'job_id'})
res_job_data.update({
'初筛': 0,
'复筛': 0,
'面试': 0,
'offer': 0,
'待入职': 0,
'已入职': job_data['now_job_num'],
})
res_msg[job_name] = res_job_data
await interview.init()
res = interview.get_every_stage_form_sql(job_ids)
sql = res['sql']
data = await ck_db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
for ck_data in data.values():
# 职位id不存在跳过
if ck_data['job_id'] not in job_id_to_name:
continue
ck_job_name = job_id_to_name[ck_data['job_id']]
stage = ck_data['interview_stage']
if stage == 1:
res_msg[ck_job_name]['初筛'] += 1
elif stage == 2:
res_msg[ck_job_name]['复筛'] += 1
elif stage == 3:
res_msg[ck_job_name]['面试'] += 1
elif stage == 4:
res_msg[ck_job_name]['offer'] += 1
elif stage == 5:
res_msg[ck_job_name]['待入职'] += 1
return schemas.Msg(code=200, msg='ok', data=res_msg)
# hr工作量报表
@router.post("/hr_works_form")
async def hr_works_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
ck_db: CKDrive = Depends(get_ck_db),
db: AsyncIOMotorDatabase = Depends(get_database)
) -> schemas.Msg:
""" hr工作量报表 """
await interview.init()
res = interview.get_hr_works_form_sql()
sql = res['sql']
data = await ck_db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
res_msg = {}
job_name_sector = {}
hr_names = [] # 查询面试数据
job_ids = []
for ck_data in data.values():
ck_job_id = ck_data['job_id']
hr_name = ck_data['hr_name']
if hr_name not in hr_names:
hr_names.append(hr_name)
job_data = await crud.jobs.find_job_name(db, ck_job_id)
job_name = job_data['job_name'] # 职位
job_sector = job_data['job_sector'] # 部门
job_name_sector.update({
job_name: job_sector
})
# 未录入职位
if ck_job_id not in job_ids:
job_ids.append(ck_job_id)
res_msg[job_name] = {hr_name: {
'初筛': 0,
'复筛': 0,
'面试': 0,
'创建面试': 0,
'面试签到': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
}}
# 已录入职位
else:
if hr_name not in res_msg[job_name]:
res_msg[job_name][hr_name] = {
'初筛': 0,
'复筛': 0,
'面试': 0,
'创建面试': 0,
'面试签到': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
}
stage = ck_data['interview_stage']
if stage >= 1:
res_msg[job_name][hr_name]['初筛'] += 1
if stage >= 2:
res_msg[job_name][hr_name]['复筛'] += 1
if stage >= 3:
res_msg[job_name][hr_name]['面试'] += 1
if stage >= 4:
res_msg[job_name][hr_name]['offer'] += 1
if stage >= 5:
res_msg[job_name][hr_name]['待入职'] += 1
if stage >= 7:
res_msg[job_name][hr_name]['已入职'] += 1
interview_records = await crud.interview_record.find_job_some(db, hr_names)
for record in interview_records:
record_job_name = record['job_names']
record_hr_name = record['hr_name']
record_interview_sign = record['interview_sign']
if record_job_name in res_msg:
if record_hr_name not in res_msg[record_job_name]:
res_msg[record_job_name][record_hr_name] = {
'初筛': 0,
'复筛': 0,
'面试': 0,
'创建面试': 0,
'面试签到': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
}
res_msg[record_job_name][record_hr_name]['创建面试'] += 1
if record_interview_sign:
res_msg[record_job_name][record_hr_name]['面试签到'] += 1
for job_name1, msg_data in res_msg.items():
count_data = {
'初筛': 0,
'复筛': 0,
'面试': 0,
'创建面试': 0,
'面试签到': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
}
for key, value_data in msg_data.items():
for true_key, num in value_data.items():
count_data[true_key] += num
res_msg[job_name1].update({'总计': count_data})
res_data = {
'msg': res_msg,
'job_sector': job_name_sector
}
return schemas.Msg(code=200, msg='ok', data=res_data)
# 职位阶段通过率报表
@router.post("/stage_success_form")
async def stage_success_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
ck_db: CKDrive = Depends(get_ck_db),
db: AsyncIOMotorDatabase = Depends(get_database)
) -> schemas.Msg:
""" 职位阶段通过率报表 """
all_jobs = await crud.jobs.all_some_field(db)
res_msg = {}
job_id_to_name = {}
job_ids = []
for job_data in all_jobs:
job_id = job_data['job_id']
job_ids.append(job_id)
job_name = job_data['job_name']
job_id_to_name[job_id] = job_name
res_job_data = deepcopy({k: v for k, v in job_data.items() if k not in ['job_id', 'now_job_num']})
res_job_data.update({
'初筛': 0,
'复筛': 0,
'面试': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
})
res_msg[job_name] = res_job_data
await interview.init()
res = interview.get_every_stage_form_sql(job_ids)
sql = res['sql']
data = await ck_db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
for ck_data in data.values():
# 职位id不存在跳过
if ck_data['job_id'] not in job_id_to_name:
continue
ck_job_name = job_id_to_name[ck_data['job_id']]
stage = ck_data['interview_stage']
if stage >= 1:
res_msg[ck_job_name]['初筛'] += 1
if stage >= 2:
res_msg[ck_job_name]['复筛'] += 1
if stage >= 3:
res_msg[ck_job_name]['面试'] += 1
if stage >= 4:
res_msg[ck_job_name]['offer'] += 1
if stage >= 5:
res_msg[ck_job_name]['待入职'] += 1
if stage >= 7:
res_msg[ck_job_name]['已入职'] += 1
for res_name, res_data in res_msg.items():
chk_num1 = res_data['初筛']
chk_num2 = res_data['复筛']
chk_num3 = res_data['面试']
chk_num4 = res_data['offer']
chk_num5 = res_data['待入职']
chk_num7 = res_data['已入职']
if chk_num1 and chk_num2:
chance_1 = round(chk_num2 / chk_num1, 2)
else:
chance_1 = 0
if chk_num2 and chk_num3:
chance_2 = round(chk_num3 / chk_num2, 2)
else:
chance_2 = 0
if chk_num3 and chk_num4:
chance_3 = round(chk_num4 / chk_num3, 2)
else:
chance_3 = 0
if chk_num4 and chk_num5:
chance_4 = round(chk_num5 / chk_num4, 2)
else:
chance_4 = 0
if chk_num5 and chk_num7:
chance_5 = round(chk_num7 / chk_num5, 2)
else:
chance_5 = 0
res_msg[res_name].update({
'初筛通过率': chance_1,
'复筛通过率': chance_2,
'面试通过率': chance_3,
'offer通过率': chance_4,
'待入职通过率': chance_5
})
return schemas.Msg(code=200, msg='ok', data=res_msg)
# 招聘漏斗报表
@router.post("/interview_funnel_form")
async def interview_funnel_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
ck_db: CKDrive = Depends(get_ck_db)
) -> schemas.Msg:
""" 招聘漏斗报表 """
await interview.init()
res = interview.get_hr_works_form_sql()
sql = res['sql']
data = await ck_db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
res_msg = {
'初筛': 0,
'复筛': 0,
'面试': 0,
'offer': 0,
'待入职': 0,
'已入职': 0
}
for i_data in data.values():
stage = i_data['interview_stage']
if stage >= 1:
res_msg['初筛'] += 1
if stage >= 2:
res_msg['复筛'] += 1
if stage >= 3:
res_msg['面试'] += 1
if stage >= 4:
res_msg['offer'] += 1
if stage >= 5:
res_msg['待入职'] += 1
if stage >= 7:
res_msg['已入职'] += 1
return schemas.Msg(code=200, msg='ok', data=res_msg)
# 渠道质量报表
@router.post("/owner_form")
async def owner_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
""" 渠道质量报表 """
await interview.init()
res = interview.get_owner_form_sql()
sql = res['sql']
data = await db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
# 简历渠道
owner_dict = {1: '前程无忧', 2: '人才库', 3: '智联招聘', 4: 'Boss直聘', 5: '58同城'}
chk_data = {
'前程无忧': [],
'人才库': [],
'智联招聘': [],
'Boss直聘': [],
'58同城': [],
}
for i in data.values():
owner_name = owner_dict[i['owner_name']]
i_data = deepcopy(i)
chk_data[owner_name].append(i_data)
res_msg = {
'前程无忧': {
'count': 0
},
'人才库': {
'count': 0
},
'智联招聘': {
'count': 0
},
'Boss直聘': {
'count': 0
},
'58同城': {
'count': 0
},
}
for key, data_list in chk_data.items():
count_num = len(data_list)
offer_num = len([1 for i in data_list if i['interview_stage'] >= 5])
work_num = len([1 for i in data_list if i['interview_stage'] >= 7])
if offer_num:
offer_chance = round(offer_num / count_num, 2)
else:
offer_chance = 0
if work_num:
work_chance = round(work_num / count_num, 2)
else:
work_chance = 0
res_msg[key]['count'] = count_num
res_msg[key]['offer_num'] = offer_num
res_msg[key]['work_num'] = work_num
res_msg[key]['offer_chance'] = offer_chance
res_msg[key]['work_chance'] = work_chance
return schemas.Msg(code=200, msg='ok', data=res_msg)
# 招聘趋势分析报表
@router.post("/interview_trend_form")
async def interview_trend_form(
request: Request,
interview: InterviewDo = Depends(InterviewDo),
ck_db: CKDrive = Depends(get_ck_db)
) -> schemas.Msg:
""" 招聘趋势分析报表 """
await interview.init()
res = interview.get_trend_form_sql()
sql = res['sql']
data = await ck_db.execute(sql)
if not data:
return schemas.Msg(code=-9, msg='无数据', data=None)
sdate = res['sdate']
edate = res['edate']
days = get_every_days(sdate, edate)
res_msg = {}
for i in days:
res_msg[str(i)] = {
'初筛简历数': 0,
'初筛': 0,
'创建面试的申请数': 0
}
for chk_data in data.values():
chk_date = datetime.strftime(chk_data['date'], '%Y-%m-%d')
if chk_date not in res_msg:
continue
chk_stage = chk_data['interview_stage']
count_num = chk_data['value']
if chk_stage >= 1:
res_msg[chk_date]['初筛简历数'] += 1 * count_num
if chk_stage >= 2:
res_msg[chk_date]['初筛'] += 1 * count_num
if chk_stage >= 3:
res_msg[chk_date]['创建面试的申请数'] += 1 * count_num
return schemas.Msg(code=200, msg='ok', data=res_msg)