import time
from urllib import parse

import pymongo

from setting import DB_USER, DB_PASS, DB_HOST, DB_PORT

USER = parse.quote_plus(DB_USER)
PASSWORD = parse.quote_plus(DB_PASS)

client = pymongo.MongoClient('mongodb://%s:%s@%s:%s/' % (USER, PASSWORD, DB_HOST, DB_PORT))


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


def judege_api_code(code):
    if code == 300001:
        return '请求失败'
    elif code == 300002:
        return '账号失效'
    elif code == 300003:
        return '账号过期'
    elif code == 300006:
        return '余额不足'
    elif code == 300007:
        return '剩余次数不足'
    elif code == 300009:
        return '账号信息有误'
    elif code == 300000:
        return '请求成功'


if __name__ == '__main__':
    res = judege_api_code(300000)
    print(res)