def getDB(name):
if name=='mongo':
conn=MongoReplicaSetClient('dbserver-1:27017,dbserver-2:27017,dbserver-3:27017', replicaSet='ats_rs')
conn.read_preference = ReadPreference.SECONDARY_PREFERRED
return conn.brauth
elif name=='redis':
return redis.Redis(connection_pool = redis.ConnectionPool(host='192.168.7.210', port=6379, db=6))
else:
return None
def getDB(name):
if name == 'mongo':
conn = MongoReplicaSetClient(
'dbserver-1:27017,dbserver-2:27017,dbserver-3:27017',
replicaSet='ats_rs')
conn.read_preference = ReadPreference.SECONDARY_PREFERRED
return conn.brauth
elif name == 'redis':
return redis.Redis(connection_pool=redis.ConnectionPool(
host='192.168.7.210', port=6379, db=6))
else:
return None
def init_mongodb_client(conf_item):
section = conf_item
servers = section.get('servers')
def build_node_desc(node_conf):
return '%s:%s' % (node_conf['host'], node_conf['port'])
if section.get('replica', False):
from pymongo import MongoReplicaSetClient
from pymongo import ReadPreference
client = MongoReplicaSetClient(','.join(map(build_node_desc, servers)), replicaSet=section.get('replName'))
pref = section.get('readPref', 'PRIMARY')
client.read_preference = getattr(ReadPreference, pref)
else:
from pymongo import MongoClient
s = servers[0]
client = MongoClient(s['host'], s['port'])
setattr(client, 'auth_list', set([]))
return client
def init_mongodb_client(conf_item):
section = conf_item
servers = section.get('servers')
def build_node_desc(node_conf):
return '%s:%s' % (node_conf['host'], node_conf['port'])
if section.get('replica', False):
from pymongo import MongoReplicaSetClient
from pymongo import ReadPreference
client = MongoReplicaSetClient(','.join(map(build_node_desc, servers)),
replicaSet=section.get('replName'))
pref = section.get('readPref', 'PRIMARY')
client.read_preference = getattr(ReadPreference, pref)
else:
from pymongo import MongoClient
s = servers[0]
client = MongoClient(s['host'], s['port'])
setattr(client, 'auth_list', set([]))
return client
def getMongoDB(self):
conn=MongoReplicaSetClient(Config['ReplicaSetServer'], replicaSet=Config['ReplicaSetName'])
conn.read_preference = ReadPreference.SECONDARY_PREFERRED
return conn.bugreporter
def __new__(mcs, name, bases, attrs):
new_class = super(ModelBase,
mcs).__new__(mcs, name, bases, attrs)
parents = [b for b in bases if isinstance(b, ModelBase)]
if not parents:
# If this isn't a subclass of Model, don't do anything special.
return new_class
# Processing Model metadata.
try:
meta = getattr(new_class, 'Meta')
except AttributeError:
meta = None
else:
# Won't need the original metadata container anymore.
delattr(new_class, 'Meta')
options = _Options(meta)
options.collection = options.collection or to_underscore(name)
if options.interface:
new_class._meta = None
new_class.database = None
new_class.collection = DummyCollection
return new_class
if not (options.host and options.port and options.database):
raise Exception(
'Model %r improperly configured: %s %s %s' % (
name, options.host, options.port, options.database))
# Checking connection / client pool for an existing connection / client.
pool_key = options.host,options.port
if options.replica_set_name:
logging.debug("Using replica_set_name=%s as database pool key." % options.replica_set_name)
pool_key = options.replica_set_name
if pool_key in mcs._connections:
client = mcs._connections[pool_key]
logging.debug("Got database client from pool for pool_key=%s" % (pool_key,))
else:
logging.debug("Creating new database client for pool_key=%s" % (pool_key,))
if options.replica_set_name:
logging.debug("Setting up a replica set client...")
client = MongoReplicaSetClient(options.replica_set_uri, replicaSet=options.replica_set_name)
client.read_preference = ReadPreference.SECONDARY_PREFERRED
else:
logging.debug("Setting up a normal client...")
client = MongoClient(options.host, options.port)
mcs._connections[pool_key] = client
new_class._meta = options
new_class.connection = client
new_class.database = client[options.database]
if options.username and options.password:
new_class.database.authenticate(options.username, options.password)
new_class.collection = options.collection_class(
new_class.database, options.collection, document_class=new_class)
if options.auto_index:
new_class.auto_index() # Generating required indices.
return new_class
def getMongoDB(self):
conn = MongoReplicaSetClient(Config['ReplicaSetServer'],
replicaSet=Config['ReplicaSetName'])
conn.read_preference = ReadPreference.SECONDARY_PREFERRED
return conn.bugreporter
