tfse-admin-api-v0.2/Manage/industry/industry_db.py

36 lines
832 B
Python

import pymongo
MG_USER = "root"
MG_PWD = "sromitdTW569kC#M"
MG_HOST = "116.63.130.34"
MG_PORT = "27018"
CLIENT = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(MG_USER, MG_PWD, MG_HOST, MG_PORT))
def INSERT_DATA(param1, param2):
"""
插入一条数据
Parameters:
param1: str 数据集
param2: obj 插入数据
Returns:
True 执行成功
"""
collection = CLIENT['宏观'][param1]
collection.insert_one(param2)
return True
def UPDATE_DATA(param1, param2, param3):
"""
新建/更新一条数据
Parameters:
param1: str 数据集
param2: obj 搜索条件
param3: obj 插入数据
Returns:
True 执行成功
"""
collection = CLIENT['宏观'][param1]
collection.update_one(param2, {"$set": param3}, upsert=True)