pycharm連接MongoDB

 

import pymongo


class MyMonDb(object):
    def __init__(self, db_name, col_name):
        self.conn = pymongo.MongoClient()
        self.db_name = self.conn[db_name]
        self.col_name = self.db_name[col_name]

    def insert(self, data):
        if isinstance(data, list):
            self.col_name.insert_many(data)
        elif isinstance(data, dict):
            self.col_name.insert_one(data)

    def delete(self, query, _all=False):
        if _all:
            self.col_name.delete_many(query)
        else:
            self.col_name.delete_one(query)

    def update(self, query, data):
        self.col_name.update_many(query, data)

    def find(self, query, _all=False):
        if _all:
            return self.col_name.find(query)
        else:
            return self.col_name.find_one(query)


if __name__ == '__main__':
    m = MyMonDb("gkl", "students")
    m.insert([{"_id": 6, "name": "gkl"}, {"_id": 7, "name": "rfy"}])
    m.update({"name":"rfy"}, {"$set": {"name": "郭康倫"}})
    for i in m.find({}, _all=True):
        print(i)