PYTHON-1019 - GridIn must use ReadPreference.PRIMARY

Author: behackett

gridfs/grid_file.py

@@ -29,6 +29,7 @@
 from pymongo.errors import (ConfigurationError,
  DuplicateKeyError,
  OperationFailure)
+from pymongo.read_preferences import ReadPreference
 
 try:
  _SEEK_SET = os.SEEK_SET
@@ -155,12 +156,15 @@ def __init__(self, root_collection, **kwargs):
  if "chunk_size" in kwargs:
  kwargs["chunkSize"] = kwargs.pop("chunk_size")
 
+ coll = root_collection.with_options(
+ read_preference=ReadPreference.PRIMARY)
+
  kwargs['md5'] = md5()
  # Defaults
  kwargs["_id"] = kwargs.get("_id", ObjectId())
  kwargs["chunkSize"] = kwargs.get("chunkSize", DEFAULT_CHUNK_SIZE)
- object.__setattr__(self, "_coll", root_collection)
- object.__setattr__(self, "_chunks", root_collection.chunks)
+ object.__setattr__(self, "_coll", coll)
+ object.__setattr__(self, "_chunks", coll.chunks)
  object.__setattr__(self, "_file", kwargs)
  object.__setattr__(self, "_buffer", StringIO())
  object.__setattr__(self, "_position", 0)

test/test_gridfs.py

@@ -468,6 +468,10 @@ def test_gridfs_replica_set(self):
  read_preference=ReadPreference.SECONDARY)
 
  fs = gridfs.GridFS(rsc.pymongo_test)
+
+ gin = fs.new_file()
+ self.assertEqual(gin._coll.read_preference, ReadPreference.PRIMARY)
+
  oid = fs.put(b'foo')
  content = fs.get(oid).read()
  self.assertEqual(b'foo', content)