leafsummer
9/9/2015 - 8:05 AM

A class of custom mongo client

A class of custom mongo client

import pymongo
import conf as ncconf

class MongoDB(object):

    def __init__(self):
        # self.conn = pymongo.Connection(ncconf.MONGO_SERVER, ncconf.MONGO_PORT)
        # self.mongo_db = pymongo.database.Database(self.conn, ncconf.MONGO_DB)
        self.conn = pymongo.MongoClient(host=ncconf.MONGO_SERVER, port=ncconf.MONGO_PORT, max_pool_size=50, socketTimeoutMS=60000)
        self.mongo_db = self.conn[ncconf.MONGO_DB]

    def __enter__(self):
        return self.mongo_db

    def __exit__(self):
        self.conn.disconnect()

mongodb = MongoDB().mongo_db