Merge pull request pandas-dev#167 from manahl/chunkstore-write-fix

Chunkstore fix for removing SHAs Issue pandas-dev#166

Author: bmoscon

CHANGES.md

@@ -6,6 +6,7 @@
   * Bugfix: TickStore.read now respects `allow_secondary=True`
   * Bugfix: #147 Add get_info method to ChunkStore
   * Bugfix: Periodically re-cache the library.quota to pick up any changes
+  * Bugfix: #166 Add index on SHA for ChunkStore
 
 ### 1.25 (2016-05-23)
 

arctic/chunkstore/chunkstore.py

@@ -219,7 +219,7 @@ def write(self, symbol, item, chunk_size):
         doc['append_count'] = 0
 
         if previous_shas:
-            mongo_retry(self._collection.delete_many)({'sha': {'$in': list(previous_shas)}})
+            mongo_retry(self._collection.delete_many)({'symbol': symbol, 'sha': {'$in': list(previous_shas)}})
 
         mongo_retry(self._symbols.update_one)({'symbol': symbol},
                                               {'$set': doc},