fix local filter for suggested users

This commit is contained in:
Hugh Rundle 2021-10-16 08:20:12 +11:00
parent 864d8cea2b
commit e261bfdb6a

View file

@ -81,7 +81,7 @@ class SuggestedUsers(RedisStore):
"""take a user out of someone's suggestions""" """take a user out of someone's suggestions"""
self.bulk_remove_objects_from_store([suggested_user], self.store_id(user)) self.bulk_remove_objects_from_store([suggested_user], self.store_id(user))
def get_suggestions(self, user, **filters): def get_suggestions(self, user, local=False):
"""get suggestions""" """get suggestions"""
values = self.get_store(self.store_id(user), withscores=True) values = self.get_store(self.store_id(user), withscores=True)
results = [] results = []
@ -97,10 +97,7 @@ class SuggestedUsers(RedisStore):
logger.exception(err) logger.exception(err)
continue continue
user.mutuals = counts["mutuals"] user.mutuals = counts["mutuals"]
if filters.local: if (local and user.local) or not local:
if user.local:
results.append(user)
else:
results.append(user) results.append(user)
if len(results) >= 5: if len(results) >= 5:
break break