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 conserve_data_by_hold(db, h_id, h_name, key, value):
    """
    保存天眼查api接口数据
    传入数据库名称，企业名称，集合名称，需要保存的数据
    """
    insert_data = dict()
    insert_data['人员ID'] = h_id
    insert_data['姓名'] = h_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({"人员ID": h_id}, {"$set": insert_data}, upsert=True)


def find_db_data(**kwargs):
    """
    查询数据库记录
    Parameters:
        kwargs: 查询条件
    Returns:
        data: tfse数据
    """
    collection = client[kwargs['db']][kwargs['col']]
    result = collection.find(kwargs['query'], {'_id': False})
    data = list(result)
    return data