码迷,mamicode.com
首页 > 数据库 > 详细

python-mongodb简单封装

时间:2020-01-01 16:46:49      阅读:170      评论:0      收藏:0      [点我收藏+]

标签:size   批量更新   bin   mongodb   address   lag   odi   one   har   

#!/usr/bin/python
# -*- coding: UTF-8 -*-
‘‘‘@auther :mr.qin
@IDE:pycharm‘‘‘


import pymongo
from tool.Common import *
import sys

class Operation_Mongo(object):
    def __init__(self,db=creeper_test):
        ‘‘‘初始化连接‘‘‘
        self.connect_client=pymongo.MongoClient("mongodb://{}:{}@{}:27017/".format(xx,xx,tencent_cloud_host))
        self.mydb = self.connect_client[db]#连接指定数据库

    def insert_collection(self,collection_name,value):#单个插入
        mycol=self.mydb[collection_name]
        mycol_id=mycol.insert_one(value)
        return mycol_id.inserted_id #返回insert_id,即插入文档的id值

    def insert_batch_collection(self,collection_name,value_list):#批量插入
        mycol=self.mydb[collection_name]
        mycol_id=mycol.insert_many(value_list)
        return mycol_id.inserted_ids #返回insert_id集合,即插入文档的id值


    def select_one_collection(self,collection_name,search_col=None):#获取一条数据
        ‘‘‘search_col:只能是dict类型,key大于等于一个即可,也可为空
        可使用修饰符查询:{"name": {"$gt": "H"}}#读取 name 字段中第一个字母 ASCII 值大于 "H" 的数据
        使用正则表达式查询:{"$regex": "^R"}#读取 name 字段中第一个字母为 "R" 的数据‘‘‘
        my_col=self.mydb[collection_name]
        try:
            result = my_col.find_one(search_col)  # 这里只会返回一个对象,数据需要自己取
            return result
        except TypeError as e:
            print(查询条件只能是dict类型)
            return None

    def select_all_collection(self,collection_name,search_col=None,limit_num=sys.maxsize,sort_col=None_sort,sort=asc):
        ‘‘‘search_col:只能是dict类型,key大于等于一个即可,也可为空
        可使用修饰符查询:{"name": {"$gt": "H"}}#读取 name 字段中第一个字母 ASCII 值大于 "H" 的数据
        使用正则表达式查询:{"$regex": "^R"}#读取 name 字段中第一个字母为 "R" 的数据
        limit_num:返回指定条数记录,该方法只接受一个数字参数(sys.maxsize:返回一个最大的整数值)‘‘‘
        my_col=self.mydb[collection_name]
        try:
            if sort_col==False or sort_col==None_sort:
                results=my_col.find(search_col).limit(limit_num)#这里只会返回一个对象,数据需要自己取
            else:
                sort_flag = 1
                if sort == desc:
                    sort_flag = -1
                results = my_col.find(search_col).sort(sort_col,sort_flag).limit(limit_num)  # 这里只会返回一个对象,数据需要自己取
            result_all=[i for i in results]#将获取到的数据添加至list
            return result_all
        except TypeError as e:
            print(查询条件只能是dict类型)
            return None

    def update_one_collecton(self,collection_name,search_col,update_col):
        ‘‘‘该方法第一个参数为查询的条件,第二个参数为要修改的字段。
            如果查找到的匹配数据多余一条,则只会修改第一条。
            修改后字段的定义格式: { "$set": { "alexa": "12345" } }‘‘‘
        my_col=self.mydb[collection_name]
        try:
            relust=my_col.update_one(search_col,update_col)
            return relust
        except TypeError as e:
            print(查询条件与需要修改的字段只能是dict类型)
            return None

    def update_batch_collecton(self,collection_name,search_col,update_col):
        ‘‘‘批量更新数据‘‘‘
        my_col=self.mydb[collection_name]
        try:
            relust=my_col.update_many(search_col,update_col)
            return relust
        except TypeError as e:
            print(查询条件与需要修改的字段只能是dict类型)
            return None

    def delete_one_collection(self,collection_name,search_col):#删除集合中的文档
        my_col = self.mydb[collection_name]
        try:
            relust=my_col.delete_one(search_col)
            return relust
        except TypeError as e:
            print(查询条件与需要修改的字段只能是dict类型)
            return None


    def delete_batch_collection(self,collection_name,search_col):#删除集合中的多个文档
        ‘‘‘删除所有 name 字段中以 F 开头的文档:{ "name": {"$regex": "^F"} }
        删除所有文档:{}‘‘‘
        my_col = self.mydb[collection_name]
        try:
            relust=my_col.delete_many(search_col)
            return relust
        except TypeError as e:
            print(查询条件与需要修改的字段只能是dict类型)
            return None


    def drop_collection(self,collection_name):
        ‘‘‘删除集合,如果删除成功 drop() 返回 true,如果删除失败(集合不存在)则返回 false‘‘‘
        my_col = self.mydb[collection_name]
        result=my_col.drop()
        return result

    def get_connections(self):#获取所有的connections
        return self.mydb.list_collection_names()

    def close_connect(self):
        self.connect_client.close()
        return mongo连接已关闭



if __name__=="__main__":
    om=Operation_Mongo()
    dict={ "name": "RUNOOB", "alexa": "10000", "url": "https://www.runoob.com" }
    my_dict_list=[
  { "name": "Taobao", "alexa": "100", "url": "https://www.taobao.com" },
  { "name": "QQ", "alexa": "101", "url": "https://www.qq.com" },
  { "name": "Facebook", "alexa": "10", "url": "https://www.facebook.com" },
  { "name": "知乎", "alexa": "103", "url": "https://www.zhihu.com" },
  { "name": "Github", "alexa": "109", "url": "https://www.github.com" }
]#不指定id
    my_dict_list_assign_id = [
  { "_id": 1, "name": "RUNOOB", "cn_name": "菜鸟教程"},
  { "_id": 2, "name": "Google", "address": "Google 搜索"},
  { "_id": 3, "name": "Facebook", "address": "脸书"},
  { "_id": 4, "name": "Taobao", "address": "淘宝"},
  { "_id": 5, "name": "Zhihu", "address": "知乎"}
]#指定id
    # batch_in=om.insert_batch_collection(‘test3‘,my_dict_list_assign_id)
    # print(batch_in)
    # a=om.create_colltion(‘test2‘,dict)
    # myquery = {"name": {"$gt": "H"}}#可使用修饰符查询
    # myquery_2={ "name": { "$regex": "^\W" } }
    # data=om.select_all_collection(‘test3‘,sort_col=‘name‘,sort=‘desc‘)
    # data=om.select_all_collection(‘test3‘)
    # om.delete_one_collection(‘test3‘,search_col={ "_id": 1, "name": "RUNOOB", "cn_name": "菜鸟教程"})
    # om.delete_batch_collection(‘test3‘,search_col={})
    # data = om.select_all_collection(‘test3‘)
    # datas=om.select_all_collection(‘test3‘,myquery_2)
    # newvalues = {"$set": {"name": "{}_1"}}
    # up_data=om.update_one_collecton(‘test3‘,myquery_2,newvalues)
    # bup_data = om.update_batch_collecton(‘test3‘, myquery_2, newvalues)
    # data = om.select_all_collection(‘test3‘, myquery_2)
    print(om.get_connections())
    # print(om.drop_collection(‘test2‘))
    om.close_connect()
    print(om.get_connections())

python-mongodb简单封装

标签:size   批量更新   bin   mongodb   address   lag   odi   one   har   

原文地址:https://www.cnblogs.com/qtclm/p/12128946.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!