def build_cluster_from_uri(uri):
uri_wrapper = parse_mongo_uri(uri)
database = uri_wrapper.database or "admin"
username = uri_wrapper.username
password = uri_wrapper.password
nodes = uri_wrapper.node_list
cluster_doc = {
"_id": mask_mongo_uri(uri)
}
member_doc_list = []
for node in nodes:
host = node[0]
port = node[1]
member_doc = {
"host": "%s:%s" % (host, port)
}
member_doc_list.append(member_doc)
cluster_doc["members"] = member_doc_list
cluster = new_cluster(cluster_doc)
# set login user if specified
if username:
for member in cluster.get_members():
member.get_server().set_login_user(database, username, password)
return cluster
def build_cluster_from_uri(uri):
uri_wrapper = parse_mongo_uri(uri)
database = uri_wrapper.database or "admin"
username = uri_wrapper.username
password = uri_wrapper.password
nodes = uri_wrapper.node_list
cluster_doc = {"_id": mask_mongo_uri(uri)}
member_doc_list = []
for node in nodes:
host = node[0]
port = node[1]
member_doc = {"host": "%s:%s" % (host, port)}
member_doc_list.append(member_doc)
cluster_doc["members"] = member_doc_list
cluster = new_cluster(cluster_doc)
# set login user if specified
if username:
for member in cluster.get_members():
member.get_server().set_login_user(database, username, password)
return cluster
def mongo_connect(uri, conn_timeout=None, **kwargs):
conn_timeout_mills = (conn_timeout or CONN_TIMEOUT) * 1000
kwargs = kwargs or {}
kwargs["connectTimeoutMS"] = conn_timeout_mills
kwargs["socketTimeoutMS"] = SOCKET_TIMEOUT * 1000
# default connection timeout and convert to mills
uri_wrapper = parse_mongo_uri(uri)
try:
dbname = uri_wrapper.database
if not dbname:
if uri.endswith("/"):
uri += "admin"
else:
uri += "/admin"
# add serverSelectionTimeoutMS for pymongo 3.2
if pymongo.get_version_string().startswith("3.2"):
kwargs["serverSelectionTimeoutMS"] = conn_timeout_mills
kwargs["maxPoolSize"] = 1
mongo_client = _mongo_client(uri, **kwargs)
return mongo_client
except Exception, e:
if is_connection_exception(e):
raise ConnectionError(uri_wrapper.masked_uri, cause=e)
elif "authentication failed" in safe_stringify(e):
raise AuthenticationFailedError(uri_wrapper.masked_uri, cause=e)
else:
raise
def mongo_connect(uri, conn_timeout=None, **kwargs):
conn_timeout_mills = (conn_timeout or CONN_TIMEOUT) * 1000
kwargs = kwargs or {}
kwargs["connectTimeoutMS"] = conn_timeout_mills
kwargs["socketTimeoutMS"] = SOCKET_TIMEOUT * 1000
# default connection timeout and convert to mills
uri_wrapper = parse_mongo_uri(uri)
try:
dbname = uri_wrapper.database
if not dbname:
if uri.endswith("/"):
uri += "admin"
else:
uri += "/admin"
# add serverSelectionTimeoutMS for pymongo 3.2
if pymongo.get_version_string().startswith("3.2"):
kwargs["serverSelectionTimeoutMS"] = conn_timeout_mills
kwargs["maxPoolSize"] = 1
mongo_client = _mongo_client(uri, **kwargs)
return mongo_client
except Exception, e:
if is_connection_exception(e):
raise ConnectionError(uri_wrapper.masked_uri, cause=e)
elif "authentication failed" in safe_stringify(e):
raise AuthenticationFailedError(uri_wrapper.masked_uri, cause=e)
else:
raise
def __init__(self,
uri,
connector_id=None,
display_name=None,
conn_timeout=None):
self._uri_wrapper = parse_mongo_uri(uri)
self._connector_id = connector_id
self._conn_timeout = conn_timeout or CONN_TIMEOUT
self._connection_id = None
self._display_name = display_name
def generate_tag_value(self, backup):
try:
uri_wrapper = parse_mongo_uri(backup.source.uri)
host = uri_wrapper.host
ips = get_host_ips(host)
if ips:
return ips[0][0]
except Exception, e:
logger.error("SourceIPTag: Error while generating tag value: %s" %
e)
def generate_tag_value(self, plan):
try:
uri_wrapper = parse_mongo_uri(plan.source.uri)
host = uri_wrapper.host
ips = get_host_ips(host)
if ips:
return ips[0][0]
except Exception, e:
logger.error("SourceIPTag: Error while generating tag value: %s" %
e)
def build_server_from_uri(uri):
uri_wrapper = parse_mongo_uri(uri)
node = uri_wrapper.node_list[0]
host = node[0]
port = node[1]
database = uri_wrapper.database or "admin"
username = uri_wrapper.username
password = uri_wrapper.password
address = "%s:%s" % (host, port)
server = build_server_from_address(address)
# set login user if specified
if username:
server.set_login_user(database, username, password)
return server
def build_server_from_uri(uri):
uri_wrapper = parse_mongo_uri(uri)
node = uri_wrapper.node_list[0]
host = node[0]
port = node[1]
database = uri_wrapper.database or "admin"
username = uri_wrapper.username
password = uri_wrapper.password
address = "%s:%s" % (host, port)
server = build_server_from_address(address)
# set login user if specified
if username:
server.set_login_user(database, username, password)
return server
def mongo_connect(uri):
uri_wrapper = parse_mongo_uri(uri)
try:
dbname = uri_wrapper.database
if not dbname:
dbname = "admin"
if uri.endswith("/"):
uri += "admin"
else:
uri += "/admin"
conn = pymongo.Connection(uri, socketTimeoutMS=CONN_TIMEOUT,
connectTimeoutMS=CONN_TIMEOUT)
return conn[dbname]
except Exception, e:
if is_connection_exception(e):
raise ConnectionError(uri_wrapper.masked_uri, cause=e)
elif "authentication failed" in str(e):
raise AuthenticationFailedError(uri_wrapper.masked_uri, cause=e)
else:
raise
def mongo_connect(uri):
uri_wrapper = parse_mongo_uri(uri)
try:
dbname = uri_wrapper.database
if not dbname:
dbname = "admin"
if uri.endswith("/"):
uri += "admin"
else:
uri += "/admin"
conn = pymongo.Connection(uri,
socketTimeoutMS=CONN_TIMEOUT,
connectTimeoutMS=CONN_TIMEOUT)
return conn[dbname]
except Exception, e:
if is_connection_exception(e):
raise ConnectionError(uri_wrapper.masked_uri, cause=e)
elif "authentication failed" in str(e):
raise AuthenticationFailedError(uri_wrapper.masked_uri, cause=e)
else:
raise
def __init__(self, uri):
self._uri_wrapper = parse_mongo_uri(uri)
def _db_repo_connect():
db_conf = config.get_database_repository_conf()
uri = db_conf["databaseURI"]
conn = pymongo.Connection(uri)
dbname = parse_mongo_uri(uri).database
return conn, dbname
def _db_repo_connect():
db_conf = config.get_database_repository_conf()
uri = db_conf["databaseURI"]
conn = pymongo.Connection(uri)
dbname = parse_mongo_uri(uri).database
return conn, dbname
def __init__(self, uri):
self._uri_wrapper = parse_mongo_uri(uri)
def __init__(self, uri, connector_id=None,display_name=None, conn_timeout=None):
self._uri_wrapper = parse_mongo_uri(uri)
self._connector_id = connector_id
self._conn_timeout = conn_timeout or CONN_TIMEOUT
self._connection_id = None
self._display_name = display_name
