55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
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
|