diff --git a/r2/r2/models/query_cache.py b/r2/r2/models/query_cache.py index 9f0a1f113..4e087da84 100644 --- a/r2/r2/models/query_cache.py +++ b/r2/r2/models/query_cache.py @@ -196,14 +196,10 @@ class CachedQueryMutator(object): self.to_prune = set() def __enter__(self): - self.mutator.__enter__() return self def __exit__(self, type, value, traceback): - self.mutator.__exit__(type, value, traceback) - - if self.to_prune: - CachedQuery._prune_multi(self.to_prune) + self.send() def insert(self, query, things): if not things: @@ -220,6 +216,12 @@ class CachedQueryMutator(object): query._delete(self.mutator, things) + def send(self): + self.mutator.send() + + if self.to_prune: + CachedQuery._prune_multi(self.to_prune) + def filter_identity(x): return x