def __init__(
self,
mongogrant_spec: str,
collection_name: str,
mgclient_config_path: Optional[str] = None,
**kwargs,
):
"""
Args:
mongogrant_spec: of the form `<role>`:`<host>`/`<db>`, where
role is one of {"read", "readWrite"} or aliases {"ro", "rw"};
host is a db host (w/ optional port) or alias; and db is a db
on that host, or alias. See mongogrant documentation.
collection_name: name of mongo collection
mgclient_config_path: Path to mongogrant client config file,
or None if default path (`mongogrant.client.path`).
"""
self.mongogrant_spec = mongogrant_spec
self.collection_name = collection_name
self.mgclient_config_path = mgclient_config_path
self._collection = None
if self.mgclient_config_path:
config = Config(check=check, path=self.mgclient_config_path)
client = Client(config)
else:
client = Client()
if set(("username", "password", "database", "host")) & set(kwargs):
raise StoreError("MongograntStore does not accept "
"username, password, database, or host "
"arguments. Use `mongogrant_spec`.")
self.kwargs = kwargs
_auth_info = client.get_db_auth_from_spec(self.mongogrant_spec)
super(MongograntStore, self).__init__(
host=_auth_info["host"],
database=_auth_info["authSource"],
username=_auth_info["username"],
password=_auth_info["password"],
collection_name=self.collection_name,
**kwargs,
)
def collection(self):
""" Property referring to underlying pymongo collection """
if self._collection is None:
raise StoreError(
"Must connect Mongo-like store before attemping to use it")
return self._collection