RUN__IT # Python操作MongoDB的方法-增删改查

Python操作MongoDB的方法

# -*- coding: utf-8 -*-
# @Time : 2020/12/16 13:40
# @Author : admin

from pymongo import MongoClient
from bson.objectid import ObjectId


class Mongo_DB:
    def __init__(self):
        conn = MongoClient('xxx.xx.241.245', 27017)
        db = conn.test  # 连接test数据库，没有则自动创建
        db.authenticate("", "mongo")
        self.my_set = db.stu  # 使用test_set集合，没有则自动创建

    def insert_data(self, data):
        if isinstance(data, dict):
            # 单个插入
            result = self.my_set.insert_one(data)
            print(result.inserted_id)
        elif isinstance(data, list):
            # 多个插入
            result = self.my_set.insert_many(data)
            print(result.inserted_ids)
        else:
            print("参数类型error")

    def delete_data(self, data):
        if data is not None:
            if isinstance(data, dict):
                if "_id" in data.keys():
                    data["_id"] = ObjectId(data["_id"])
                # 指定条件del
                results = self.my_set.delete_many(data)
                print(results.deleted_count)
            else:
                print("参数类型error")
        pass

    def update_data(self, query, data):
        if data is not None:
            if isinstance(data, dict) and isinstance(query, dict):
                if "_id" in data.keys():
                    data["_id"] = ObjectId(data["_id"])
                # 指定条件update
                results = self.my_set.update_many(query, {"$set": data})
                print(results.modified_count)
            else:
                print("参数类型error")
        pass

    def query(self, data=None):
        ObjectId("5fdd97fbdef9c0f43bc9e7e9")
        if data is not None:
            if isinstance(data, dict):
                if "_id" in data.keys():
                    data["_id"] = ObjectId(data["_id"])
                # 指定条件查询
                results = self.my_set.find(data)
                print(list(results))
            else:
                print("参数类型error")
        elif data is None:
            # 查所有
            results = self.my_set.find()
            # print(list(results))
            for result in results:
                print(result)


if __name__ == '__main__':
    MD = Mongo_DB()
    
    # data = {"name": '华子', "hometown": '河南', "age": 20, "gender": "true"}
    # data = [{"name": '华子', "hometown": '河南', "age": 20, "gender": "true"}]
    # MD.insert_data(data)

    # myquery = {"hometown": "英国", "name": "Lily"}
    # myquery = {"_id": "5fdd97fbdef9c0f43bc9e7e9", "name": "华子"} 
    # mydata = {"age": 27}
    # MD.update_data(myquery, mydata)

    myquery = {"_id": "5fdd97fbdef9c0f43bc9e7e9"}
    # myquery = {"hometown": {"$in": ['蒙古']}}
    MD.delete_data(myquery)
    
    # 查
    MD.query(myquery)

# 创建root用户
# db.createUser({user: "root",pwd: "mongo", roles: [ { role: "root", db: "admin" } ]})
# 开启配置文件auth = ture