import datetime
import mimetypes
from collections import defaultdict
import time
import os
from urllib.parse import quote
import re
from clickhouse_driver import Client
import pandas as pd
import numpy as np
from fastapi import APIRouter, Depends, Request, File, UploadFile
from motor.motor_asyncio import AsyncIOMotorDatabase
from pandas import DataFrame
from starlette.responses import StreamingResponse
from  utils.jianli import get_resume
import crud, schemas
from common import *

from api import deps
from db import get_database
from db.ckdb import get_ck_db, CKDrive, ckdb
from db.redisdb import get_redis_pool, RedisDrive

from models.behavior_analysis import BehaviorAnalysis
from models.user_analysis import UserAnalysis
from models.interview_zsgc import InterviewDo
from utils import DfToStream, get_bijiao

router = APIRouter()


# 面试查询
@router.post("/interview_find")
async def interview_find(
        request: Request,
        interview: InterviewDo = Depends(InterviewDo),
        db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
    """ interview面试数据查询 """
    await interview.init()
    res = interview.find_interview_sql()
    sql = res['sql']
    data = await db.execute(sql)
    if not data:
        return schemas.Msg(code=-9, msg='无数据', data=None)
    return schemas.Msg(code=200, msg='ok', data=data)


# 面试修改
@router.post("/interview_update")
async def interview_update(
        request: Request,
        interview: InterviewDo = Depends(InterviewDo),
        db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
    """ interview面试数据更新 """
    await interview.init()
    res = interview.update_interview_sql()
    sql = res['sql']
    try:
        data = await db.execute(sql)
    except:
        return schemas.Msg(code=-9, msg='数据有误', data=None)
    return schemas.Msg(code=200, msg='ok', data=data)


# 写入面试数据
@router.post("/interview_insert")
async def interview_insert(
        request: Request,
        interview: InterviewDo = Depends(InterviewDo),
        db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
    """ interview面试数据写入 """
    await interview.init()
    res = interview.insert_interview_sql()
    sql = res['sql']
    insert_data = res['insert_data']
    data = await db.execute_dict(sql, insert_data)
    return schemas.Msg(code=200, msg='ok', data=data)


# 写入面试数据
@router.post("/interview_file_insert")
async def interview_insert(
        request: Request,
        file: UploadFile = File(...),
        db: CKDrive = Depends(get_ck_db),
) -> schemas.Msg:
    """ interview面试数据导入 """
    path_data = os.getcwd() + '/jianli/'  # 当前文件所在的目录
    contents = await file.read()
    filename = file.filename
    try:
        with open(path_data + filename, "wb") as f:
            #  将获取的file文件内容，写入到新文件中
            f.write(contents)
            f.close()
    except:
        return schemas.Msg(code=400, msg='上传文件有误', data=None)
    data = get_resume(filename, path_data)
    education = data['education']
    # 学历int转化
    education_int = {
        '大专': 1,
        '本科': 2,
        '研究生': 3,
        '博士': 4,
        '硕士': 5,
    }
    if education and isinstance(education, str):
        data['education'] = education_int.get(education, 1)
    age = data['age']
    # 年龄int转化
    if age and isinstance(age, str):
        true_age = re.search(r"\d+\.?\d*", age)
        if len(true_age.group()) > 2:
            data['age'] = 20
        else:
            data['age'] = int(true_age.group())
    work_exp = data['work_exp']
    # 工作经验float转化
    if work_exp and isinstance(work_exp, str):
        true_work_exp = re.search(r"\d+\.?\d*", work_exp)
        if len(true_work_exp.group()) > 3:
            data['work_exp'] = 1.0
        else:
            data['work_exp'] = float(true_work_exp.group())
    print(data)
    return schemas.Msg(code=200, msg='ok', data=data)


# @router.post("/interview_insert")
# async def interview_insert(
#         request: Request,
#         data_in: schemas.Interview,
#         ckdb: CKDrive = Depends(get_ck_db),
# ) -> schemas.Msg:
#     """ 面试情况 """
#     await interview.init()
#     res = interview.insert_interview_sql()
#     sql = res['sql']
#     insert_data = res['insert_data']
#     data = await db.execute_dict(sql, insert_data)
#     return schemas.Msg(code=200, msg='ok', data=data)

@router.post("/add_job")
async def event_edit(
        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)
    return schemas.Msg(code=200, msg='ok', data='')