MongoDB 去重

# @Time : 2019/11/30 18:06
# @Author : GKL
# FileName : mongodb去重.py
# Software : PyCharm

import pymongo
# 去重异常捕获
from pymongo.errors import DuplicateKeyError


class MongoDb(object):

    def __init__(self):
        self.client = pymongo.MongoClient()
        self.db = self.client['test']
        self.collection = self.db['test']
        # 去重操作（通过name去重）
        self.collection.ensure_index('name', unique=True)

    def insert(self, data):
        try:
            self.collection.insert_one(data)
        except DuplicateKeyError as e:
            print(e)

if __name__ == '__main__':
    data = {
        'name': "python",
        'age': 18
    }
    s = MongoDb()
    s.insert(data)