pymongo 的增删改查封装

import pymongo, math


mongo_url = 'mongodb://127.0.0.1'
# 链接数据库
client = pymongo.MongoClient(mongo_url)


class mongo(object):
    def __init__(self, dn_name):
        # 需要打开的数据库
        self.db = client[dn_name]
        # 查询数据库的所有集合
        self.collist = self.db.list_collection_names()

    # 插入文档
    def insert_one(self, col_name, data):
        # 打开集合
        col = self.db[col_name]
        return col.insert_one(data)

    # 插入多个文档
    def insert_many(self, col_name, list):
        # 打开集合
        col = self.db[col_name]
        return col.insert_many(list)

    # 查询一条数据
    def find_one(self, col_name, where={}):
        # 打开集合
        col = self.db[col_name]
        return col.find_one(where)

    # 查询多条数据
    def find(self, col_name, where={}, sort=[(date, -1)], select={_id: 0, __v: 0}):
        # 打开集合
        col = self.db[col_name]
        return col.find(where, select).sort(sort)

    # 查询集合的条数
    def count(self, col_name, where={}):
        # 打开集合
        col = self.db[col_name]
        return col.find(where).count()

    # 分页查询
    def find_page(self, col_name, where={}, page=1, sort=[(date, -1)], page_num=20, select={_id: 0, __v: 0}):
        # 打开集合
        col = self.db[col_name]
        # 数据总条数
        counts = col.find(where).count()
        # 总页数
        all_page = math.abs(counts / page_num)
        # 数据偏移量
        if page < all_page:
            skip = (page - 1) * page_num
        else:
            skip = all_page * page_num
        # 查询总条数
        list = col.find(where, select).sort(sort).limit(page_num).skip(skip)

        return {"cur_page": page, "counts": counts, "list": list, "all_page": all_page}

    # 更新一条数据
    def update_one(self,col_name,where,update):
        # 打开集合
        col = self.db[col_name]
        return col.update_one(where,update)

    # 更新多条数据
    def update_many(self,col_name,where,update):
        # 打开集合
        col = self.db[col_name]
        return col.update_many(where, update)

    # 删除一条数据
    def delete_one(self,col_name,where):
        # 打开集合
        col = self.db[col_name]
        return col.delete_one(where)
    # 删除多条数据
    def delete_many(self,col_name,where):
        # 打开集合
        col = self.db[col_name]
        return col.delete_many(where)