mongodb帮助类

import pymongo


class MongoDBHelp:

    def __init__(self, database, collection):
        """
        初始化链接数据库集合
        :param database:数据库名
        :param collection: 集合名
        """
        self.connect = pymongo.MongoClient()
        self.connection = self.connect.get_database(database).get_collection(collection)
        self.insert_max_count = 48
        self.insert_many_list = []

    def insert_one(self, data: dict):
        """
        插入一条数据
        :param data:数据内容为字典格式
        """
        self.connection.insert_one(data)

    def insert_many(self, data: dict):
        """
        插入多条数据
        :param data:数据内容为字典格式
        """
        if len(self.insert_many_list) >= self.insert_max_count:
            self.connection.insert_many(self.insert_many_list)
            self.insert_many_list.clear()
        else:
            self.insert_many_list.append(data)

    def close(self):
        """
        先将列表中剩余元素插入数据库再关闭数据库。
        """
        self.connection.insert_many(self.insert_many_list)
        self.connect.close()