From 81acc136f4f21206a048d08b3e3a2826ac0c04bf Mon Sep 17 00:00:00 2001 From: Logan Hanks Date: Wed, 20 Feb 2013 17:39:05 -0800 Subject: [PATCH] Fix memoization of X_ids and reverse_X_ids. --- r2/r2/lib/db/userrel.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/r2/r2/lib/db/userrel.py b/r2/r2/lib/db/userrel.py index 714160668..dfb74f5e8 100644 --- a/r2/r2/lib/db/userrel.py +++ b/r2/r2/lib/db/userrel.py @@ -105,21 +105,21 @@ class MemoizedUserRelManager(UserRelManager): self.add = self._update_caches_on_success(sup.add) self.remove = self._update_caches_on_success(sup.remove) - def _update_caches(self, thing): + def _update_caches(self, thing, user): if not self.disable_ids_fn: self.ids(thing, _update=True) if not self.disable_reverse_ids_fn: - self.reverse_ids(thing, _update=True) + self.reverse_ids(user, _update=True) def _update_caches_on_success(self, method): @functools.wraps(method) - def wrapper(thing, *args, **kwargs): + def wrapper(thing, user, *args, **kwargs): try: - result = method(thing, *args, **kwargs) + result = method(thing, user, *args, **kwargs) except: raise else: - self._update_caches(thing) + self._update_caches(thing, user) return result return wrapper