import time

from Utils.DataBase.MongoHelperUtils import get_tyc_mongodb

token = "32737167-cb63-4ce9-9397-d66169488f51"
headers = {'Authorization': token}

# 数据库参数
DB_GSBJ = '公司背景'
DB_SFFX = '司法风险'
DB_JYFX = '经营风险'
DB_TYFX = '天眼风险'
DB_JYZK = '经营状况'
DB_ZSCQ = '知识产权'
DB_GSFZ = '公司发展'


def judege_api_code(code):
    """
    判断状态码
    Parameters:
        code: 传入状态码，返回状态码对应状态说明
    Returns:
        状态码说明信息
    """
    if code == 0:
        return '请求成功'
    elif code == 300000:
        return '无数据'
    elif code == 300001:
        return '请求失败'
    elif code == 300002:
        return '账号失效'
    elif code == 300003:
        return '账号过期'
    elif code == 300004:
        return '访问频率过快'
    elif code == 300005:
        return '无权限访问此api'
    elif code == 300006:
        return '余额不足'
    elif code == 300007:
        return '剩余次数不足'
    elif code == 300008:
        return '缺少必要参数'
    elif code == 300009:
        return '账号信息有误'
    elif code == 300010:
        return 'URL不存在'
    elif code == 300011:
        return '此IP无权限访问此api'
    elif code == 300012:
        return '报告生成中'


def conserve_data(db, name, key, value):
    """
    保存天眼查api接口数据
    传入数据库名称，企业名称，集合名称，需要保存的数据
    """
    tyc_mongodb = get_tyc_mongodb()
    insert_data = dict()
    insert_data['企业名称'] = name
    insert_data['录入时间'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
    insert_data[key] = value
    # 数据库操作
    tyc_mongodb.upsert_data(
        db,
        key,
        {"企业名称": name},
        insert_data
    )