26 lines
760 B
Python
26 lines
760 B
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)
|