tfse-etl-api-v0.2/DBInitial/DBInitializer.py

37 lines
1.2 KiB
Python
Raw Normal View History

2022-03-15 17:09:17 +08:00
import json
import os
import pymongo
HOST = "116.63.130.34"
PASS = "UTlC9cCoglD1cI1*"
USER = "root"
PORT = "27021"
mongo_client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(USER, PASS, HOST, PORT))
def Initializer(relative_file_path, data_position):
"""
Parameters:
relative_file_path: str 以该package下static起始的相对路径 "/static/app/企业用户.json"
data_position: dict 数据存储位置DB是存储数据库Collection是存储数据集
Returns:
-
"""
absolute_folder_path = os.path.abspath(os.path.dirname(__file__))
absolute_file_path = os.path.abspath(absolute_folder_path + relative_file_path)
with open(absolute_file_path, encoding="utf8") as f:
data = json.load(f)
collection = mongo_client[data_position['DB']][data_position['Collection']]
collection.insert_one(data)
print("完成数据初始化:【{}】-【{}".format(data_position['DB'], data_position['Collection']))
if __name__ == '__main__':
2022-03-17 15:31:18 +08:00
json_file_relative_path = "/static/data/ESG评价数据/填报数据.json"
data_position_in_mongo_db = {"DB": "ESG评价数据", "Collection": "填报数据"}
2022-03-15 17:09:17 +08:00
Initializer(json_file_relative_path, data_position_in_mongo_db)