37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
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__':
|
||
|
||
json_file_relative_path = "/static/data/ESG评价数据/填报数据.json"
|
||
data_position_in_mongo_db = {"DB": "ESG评价数据", "Collection": "填报数据"}
|
||
|
||
Initializer(json_file_relative_path, data_position_in_mongo_db)
|