python操作mongodb

#coding=utf-8
from pymongo import *

IP = "localhost"
PORT = 27017




class MyMongoDB:

    '''成员变量'''
    client = None  ##连接数据库的对象
    db = None  ## 指定的的数据库
    collection = None ## 制定的文档

    '''构造函数'''
    def __init__(self):
        self.client = MongoClient(IP,PORT)

    '''构造函数'''
    def __init__(self, db, collection):
        self.client = MongoClient(IP,PORT)
        self.db = self.client[db]
        self.collection = self.db[collection]

    '''插入一条记录'''
    def insert_one(self, dict_record):
        try:
            self.collection.insert(dict_record)
            return 1
        except:
            return 0

    '''插入多条记录'''
    def insert_many(self, dict_list):
        try:
            self.collection.insert_many(dict_list)
            return 1
        except:
            return 0

    '''查询记录'''
    def query(self, query_regex={}):
        result_list = []
        for i in self.collection.find(query_regex):
            result_list.append(i)
        return result_list

    '''修改记录'''
    def update(self, criteria,ObjNew,upset=False, Multi=False):
        self.collection.update(criteria, ObjNew, upset,Multi)
        return 1

    '''删除记录'''
    def remove(self, remove_regex):
        self.collection.remove(remove_regex)


def main():
    ### 初始化
    db = 'quincy'
    collection = 'quincyDB'
    mongodb = MyMongoDB(db,collection)

    # ### 插入一条记录
    # dict_record = {'name':'quincy2','age':18}
    # print mongodb.insert_one(dict_record)

    # ### 修改记录
    # criteria = {'age':22}
    # objNew = {'$set':{'name':'quincy3'}}
    # mongodb.update(criteria,objNew)

    ### 删除记录
    remove_regex = {'age':22}
    mongodb.remove(remove_regex)

    ### 查询记录
    query_list = mongodb.query()
    for i in query_list:
        print i

if __name__ == '__main__':
    main()