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