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)