def wrap_items(self, items): categories = LinkSavesByAccount.fast_query(c.user, items).items() categories += CommentSavesByAccount.fast_query(c.user, items).items() categories = {item[1]._id: category for item, category in categories if category} wrapped = QueryBuilder.wrap_items(self, items) for w in wrapped: category = categories.get(w._id, "") w.savedcategory = category return wrapped
def wrap_items(self, items): from r2.lib.template_helpers import add_attr categories = LinkSavesByAccount.fast_query(c.user, items).items() categories += CommentSavesByAccount.fast_query(c.user, items).items() categories = {item[1]._id: category for item, category in categories if category} wrapped = QueryBuilder.wrap_items(self, items) for w in wrapped: category = categories.get(w._id, '') w.savedcategory = category return wrapped
def wrap_items(self, items): from r2.lib.template_helpers import add_attr categories = LinkSavesByAccount.fast_query(c.user, items).items() categories += CommentSavesByAccount.fast_query(c.user, items).items() categories = {item[1]._id: category for item, category in categories if category} wrapped = QueryBuilder.wrap_items(self, items) for w in wrapped: category = categories.get(w._id, '') w.savedcategory = category return wrapped