MongoUtils mongo utils python

 

 

 

 如果是 pymongo==4.x   pls   

pip3 install pymongo==4.5.0  

 client = pymongo.MongoClient(host="192.168.12.153", port=27017, username="root", password="root", authSource="admin")

 client = pymongo.MongoClient(host="mongodb://admin:123456@localhost:27017/?authSource=db_sea")

 

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @mail    : lshan523@163.com
# @Time    : 2023/9/11 11:06
# @Author  : Sea
# @File    : MongoUtils.py
# @history:
# ****************************

import pymongo
from urllib import parse

class MongoUtils:
    def __init__(self, host="127.0.0.1", port=27017, dbname="admin", mongo_user=None, mongo_password=None):
        if mongo_user and mongo_password:
            mongo_password = parse.quote_plus(mongo_password)
            # uri = "mongodb://" + mongo_user + ":" + mongo_password + "@" + host + ":" + str(port) + "/?authSource=" + dbname
            uri = "mongodb://{0}:{1}@{2}:{3}/?authSource={4}"
            uri = uri.format(mongo_user, mongo_password, host, port, dbname)
        else:
            uri = "mongodb://" + host + ":" + str(port) + "/?directConnection=true"
            # uri = "mongodb://192.168.16.122:27017/?directConnection=true"
            print(uri)
        self.client = pymongo.MongoClient(host=uri)
        # 指向指定的数据库
        self.mdb = self.client[dbname]

    def insert_json_str(self, table, str_json_data):
        post = self.mdb[str(table)]
        post.insert_one(dict(str_json_data))

    def insert_dict(self, table, dicts):
        post = self.mdb[str(table)]
        post.insert_many(dicts)

    def saveOrUpdate(self, table, data={}):
        tb = self.mdb[str(table)]
        print(str(data))
        if not data.get("_id", None)==None:
            # maybe need do_update "
            query = {'_id': data["_id"]}
            del data["_id"]
            result = tb.update_one(filter=query, update={'$set': data}, upsert=True)
            print("update data  success")
            return result
        result = tb.insert_one(data)
        print("insert data success")
        return result

    def drop_table(self, table):
        post = self.mdb[str(table)]
        post.drop()

    def delById(self, table, id):
        print("&&&&  delById &&&&&&" + str(id))
        tb = self.mdb[str(table)]
        tb.delete_one({"_id": id})

    def delByQuery(self, table, query={}):
        """{"_id": id}"""
        print("&&&&& delByQuery &&&&&&&" + str(query))
        tb = self.mdb[str(table)]
        tb.delete_one(query)

    def find(self, table, query={}, limit=100):
        """ query = {'name':'sea'}  return list """
        tb = self.mdb[str(table)]
        return tb.find(query).limit(limit)

    def find_one(self, table, query={}):
        """ query = {'name':'sea'}  return list """
        tb = self.mdb[str(table)]
        return tb.find_one(query)

    def find_one_by(self, table, query={}, projection=None, sortBy="ts"):
        """ query = {'name':'sea'} projection={"_id": -1}  return list """
        tb = self.mdb[str(table)]
        return tb.find_one(filter=query, projection=projection, sort=[(sortBy, pymongo.ASCENDING)])


    def update_one(self, table, query=None, update={}):
        """ query = {'name':'sea'}  update = {"no":1}  """
        tb = self.mdb[str(table)]
        if filter:
            return tb.update_one(filter=query, update={'$set': update}, upsert=True)
        return 0

    def count(self, table, query=None):
        """ query = {'name':'sea'}   """
        tb = self.mdb[str(table)]
        if filter:
            return tb.count_documents(filter=query)


if __name__ == '__main__':
    utils = MongoUtils(host="192.168.182.122", port=27017, dbname="bx_blog_script")
    # utils = MongoUtils(host="192.168.123.52", port=27017, dbname="sea", mongo_user="root",mongo_password="root")
    utils.saveOrUpdate(table="sea_1231", data={"_id": 1, "page": 0, "data": "123213"})
    # # # data = utils.findoneBy(table="sea_blacklist",projection={"page": 1},sort=[{"page":1}])
    # # tb = utils.mdb[str("sea_blacklist")]
    # # # data = tb.find(filter={}).sort("page ",pymongo.DESCENDING)
    # # data = tb.find_one(filter={}, projection={"page":1, "_id": -1}, sort=[('page', pymongo.DESCENDING )])
    # # print(data)
    # by = utils.findoneBy(table="sea_blacklist")
    # print(by["page"])
    #
    # alluser = utils.find(table="user", query={})
    # i = 0
    # for user in alluser:
    #     i = i + 1
    #     print(i)
    #     print(user)

 

 

 

 

pip3 install pymongo==3.13.0  

import pymongo
class MongoUtils:

    def __init__(self, host="127.0.0.1", port=27017, dbname="admin", mongo_user=None, mongo_password=None):
        # myclient = pymongo.MongoClient(host="mongodb://admin:123456@localhost:27017/?authSource=db_sea")
        client = pymongo.MongoClient(host=host, port=port)
        # 指向指定的数据库
        self.mdb = client[dbname]
        if mongo_user:
            self.mdb.authenticate(mongo_user, mongo_password)
        # 获取数据库里存放数据的表名
        # self.post = mdb["categoryname"]

    def insert_json_str(self, table, str_json_data):
        post = self.mdb[str(table)]
        post.insert(dict(str_json_data))

    def insert_dict(self, table, dicts):
        post = self.mdb[str(table)]
        post.insert(dicts)

    def saveOrUpdate(self, table, data={}):
        post = self.mdb[str(table)]
        return post.save(data)

    def drop_table(self, table):
        post = self.mdb[str(table)]
        post.drop()

    def delById(self, table, id):
        tb = self.mdb[str(table)]
        tb.delete_one({"_id": id})

    def find(self, table, query={}):
        """ query = {'name':'sea'}  return list """
        tb = self.mdb[str(table)]
        return tb.find(query)

    def find_one(self, table, query={}):
        """ query = {'name':'sea'}  return list """
        tb = self.mdb[str(table)]
        return tb.find_one(query)

    def find_one_by(self, table, query={}, projection=None, sortBy="ts"):
        """ query = {'name':'sea'} projection={"_id": -1}  return list """
        tb = self.mdb[str(table)]
        return tb.find_one(filter=query, projection=projection, sort=[(sortBy, pymongo.ASCENDING)])

    def save(self, table, data={}):
        """ query = {'name':'sea'}  return list """
        tb = self.mdb[str(table)]
        return tb.save(data)

    def update_one(self, table, query=None, update={}):
        """ query = {'name':'sea'}  update = {"no":1}  """
        tb = self.mdb[str(table)]
        if filter:
            return tb.update_one(filter=query, update={'$set': update}, upsert=True)
        return 0

    def count(self, table, query=None):
        """ query = {'name':'sea'}   """
        tb = self.mdb[str(table)]
        if filter:
            return tb.count(filter=query)

    def __del__(self):
        self.mdb.logout()


if __name__ == '__main__':
    utils = MongoUtils(host="192.168.186.122", port=27017, dbname="bx_blog_script")
    # utils.insert_dict(table="sea_blacklist",dicts=[{"_id":0,"page":0,"data":""}])
    # # # data = utils.findoneBy(table="sea_blacklist",projection={"page": 1},sort=[{"page":1}])
    # # tb = utils.mdb[str("sea_blacklist")]
    # # # data = tb.find(filter={}).sort("page ",pymongo.DESCENDING)
    # # data = tb.find_one(filter={}, projection={"page":1, "_id": -1}, sort=[('page', pymongo.DESCENDING )])
    # # print(data)
    # by = utils.findoneBy(table="sea_blacklist")
    # print(by["page"])

    alluser = utils.find(table="user", query={})
    print(type(alluser))
    i = 0
    for user in alluser:
        i = i + 1
        print(i)
        user = dict(user)
        print(type(user))
        user["nickName"] = user.get("nickname", "")
        if "nickname" in user.keys():
            del user["nickname"]
        user["avatar"] = user.get("headImgUrl", "")
        if "headImgUrl" in user.keys():
            del user["headImgUrl"]
        user["username"] = user.get("userName", "")
        if "username" in user.keys():
            del user["username"]
        # utils.insert_dict(table="user",dict=user)
        utils.save(table="user", data=user)
        print(user)