openstack/deb-python-pysaml2
Code Issues Proposed changes

Merge branch 'master' of github.com:rohe/pysaml2

Browse Source
This commit is contained in:
Roland Hedberg
2013-06-13 15:03:14 +02:00
parent 30ec6e6a93 3a79a932fb
commit 0841c814d8
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/saml2/mongo_store.py
View File

@@ -266,25 +266,28 @@ def _mdb_get_database(uri, **kwargs):
:params database: name as string or (uri, name) :params database: name as string or (uri, name)
:returns: pymongo database object :returns: pymongo database object
""" """
if not "tz_aware" in kwargs:
# default, but not forced
kwargs["tz_aware"] = True
connection_factory = MongoClient connection_factory = MongoClient
_parsed_uri = {} _parsed_uri = {}
db_name = None db_name = None
_conn = None
try: try:
_parsed_uri = pymongo.uri_parser.parse_uri(uri) _parsed_uri = pymongo.uri_parser.parse_uri(uri)
except pymongo.errors.InvalidURI: except pymongo.errors.InvalidURI:
# assume URI to be just the database name # assume URI to be just the database name
db_name = uri db_name = uri
_conn = MongoClient()
pass pass
else: else:
if "replicaset" in _parsed_uri["options"]: if "replicaset" in _parsed_uri["options"]:
connection_factory = MongoReplicaSetClient connection_factory = MongoReplicaSetClient
db_name = _parsed_uri.get("database", "pysaml2") db_name = _parsed_uri.get("database", "pysaml2")
if not "tz_aware" in kwargs:
# default, but not forced
kwargs["tz_aware"] = True
_conn = connection_factory(uri, **kwargs) _conn = connection_factory(uri, **kwargs)
_db = _conn[db_name] _db = _conn[db_name]
if "username" in _parsed_uri: if "username" in _parsed_uri: