def assign_trial(account, ip, slash16): from r2.models import Jury, Subreddit, Trial from r2.lib.db import queries defendants_voted_upon = [] defendants_assigned_to = [] for jury in Jury.by_account(account): defendants_assigned_to.append(jury._thing2_id) if jury._name != '0': defendants_voted_upon.append(jury._thing2_id) subscribed_sr_ids = Subreddit.user_subreddits(account, ids=True, limit=None) # Pull defendants, except ones which already have lots of juryvotes defs = Trial.all_defendants(quench=True) # Filter out defendants outside this user's subscribed SRs defs = filter(lambda d: d.sr_id in subscribed_sr_ids, defs) # Dictionary of sr_id => SR for all defendants' SRs srs = Subreddit._byID(set([d.sr_id for d in defs])) # Dictionary of sr_id => eligibility bool submit_srs = {} for sr_id, sr in srs.iteritems(): submit_srs[sr_id] = sr.can_submit(account) and not sr._spam # Filter out defendants with ineligible SRs defs = filter(lambda d: submit_srs.get(d.sr_id), defs) likes = queries.get_likes(account, defs) if not g.debug: # Filter out things that the user has upvoted or downvoted defs = filter(lambda d: likes.get((account, d)) is None, defs) # Prefer oldest trials defs.sort(key=lambda x: x._date) for defendant in defs: sr = srs[defendant.sr_id] if voir_dire(account, ip, slash16, defendants_voted_upon, defendant, sr): if defendant._id not in defendants_assigned_to: j = Jury._new(account, defendant) return defendant return None
def assign_trial(account, ip, slash16): from r2.models import Jury, Subreddit, Trial from r2.lib.db import queries defendants_voted_upon = [] defendants_assigned_to = [] for jury in Jury.by_account(account): defendants_assigned_to.append(jury._thing2_id) if jury._name != '0': defendants_voted_upon.append(jury._thing2_id) subscribed_sr_ids = Subreddit.user_subreddits(account, ids=True, limit=None) # Pull defendants, except ones which already have lots of juryvotes defs = Trial.all_defendants(quench=True) # Filter out defendants outside this user's subscribed SRs defs = filter (lambda d: d.sr_id in subscribed_sr_ids, defs) # Dictionary of sr_id => SR for all defendants' SRs srs = Subreddit._byID(set([ d.sr_id for d in defs ])) # Dictionary of sr_id => eligibility bool submit_srs = {} for sr_id, sr in srs.iteritems(): submit_srs[sr_id] = sr.can_submit(account) and not sr._spam # Filter out defendants with ineligible SRs defs = filter (lambda d: submit_srs.get(d.sr_id), defs) likes = queries.get_likes(account, defs) if not g.debug: # Filter out things that the user has upvoted or downvoted defs = filter (lambda d: likes.get((account, d)) is None, defs) # Prefer oldest trials defs.sort(key=lambda x: x._date) for defendant in defs: sr = srs[defendant.sr_id] if voir_dire(account, ip, slash16, defendants_voted_upon, defendant, sr): if defendant._id not in defendants_assigned_to: j = Jury._new(account, defendant) return defendant return None
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import add_attr user = c.user if c.user_is_loggedin else None #get authors #TODO pull the author stuff into add_props for links and #comments and messages? aids = set(l.author_id for l in items if hasattr(l, 'author_id') and l.author_id is not None) authors = {} cakes = {} friend_rels = None if aids: authors = Account._byID(aids, data=True, stale=self.stale) if aids else {} now = datetime.datetime.now(g.tz) cakes = { a._id for a in authors.itervalues() if a.cake_expiration and a.cake_expiration >= now } if user and user.gold: friend_rels = user.friend_rels() subreddits = Subreddit.load_subreddits(items, stale=self.stale) can_ban_set = set() can_flair_set = set() can_own_flair_set = set() if user: for sr_id, sr in subreddits.iteritems(): if sr.can_ban(user): can_ban_set.add(sr_id) if sr.is_moderator_with_perms(user, 'flair'): can_flair_set.add(sr_id) if sr.link_flair_self_assign_enabled: can_own_flair_set.add(sr_id) #get likes/dislikes try: likes = queries.get_likes(user, items) except tdb_cassandra.TRANSIENT_EXCEPTIONS as e: g.log.warning("Cassandra vote lookup failed: %r", e) likes = {} uid = user._id if user else None types = {} wrapped = [] modlink = {} modlabel = {} for s in subreddits.values(): modlink[s._id] = '/r/%s/about/moderators' % s.name modlabel[s._id] = ( _('moderator of /r/%(reddit)s, speaking officially') % dict(reddit=s.name)) for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) #TODO pull the author stuff into add_props for links and #comments and messages? w.author = None w.friend = False # List of tuples (see add_attr() for details) w.attribs = [] w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == 'yes': w.distinguished = 'moderator' elif item.distinguished in ('admin', 'special', 'gold', 'gold-auto'): w.distinguished = item.distinguished try: w.author = authors.get(item.author_id) if user and item.author_id in user.friends: # deprecated old way: w.friend = True # new way: label = None if friend_rels: rel = friend_rels[item.author_id] note = getattr(rel, "note", None) if note: label = u"%s (%s)" % (_("friend"), _force_unicode(note)) add_attr(w.attribs, 'F', label) except AttributeError: pass if (w.distinguished == 'admin' and w.author): add_attr(w.attribs, 'A') if w.distinguished == 'moderator': add_attr(w.attribs, 'M', label=modlabel[item.sr_id], link=modlink[item.sr_id]) if w.distinguished == 'special': args = w.author.special_distinguish() args.pop('name') if not args.get('kind'): args['kind'] = 'special' add_attr(w.attribs, **args) if w.author and w.author._id in cakes and not c.profilepage: add_attr( w.attribs, kind="cake", label=(_("%(user)s just celebrated a reddit birthday!") % { "user": w.author.name }), link="/user/%s" % w.author.name, ) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] w.likes = likes.get((user, item)) # update vote tallies compute_votes(w, item) w.score = w.upvotes - w.downvotes if w.likes: base_score = w.score - 1 elif w.likes is None: base_score = w.score else: base_score = w.score + 1 # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) if c.user_is_admin and getattr(item, 'ip', None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.can_flair = False w.use_big_modbuttons = self.spam_listing if (c.user_is_admin or (user and hasattr(item, 'sr_id') and item.sr_id in can_ban_set)): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, 'ban_info', {}) w.unbanner = ban_info.get('unbanner') if item._spam: w.show_spam = True w.moderator_banned = ban_info.get('moderator_banned', False) w.autobanned = ban_info.get('auto', False) w.banner = ban_info.get('banner') if ban_info.get('note', None) and w.banner: w.banner += ' (%s)' % ban_info['note'] w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" if c.user == w.author and c.user._spam: w.show_spam = False w._spam = False w.use_big_modbuttons = False elif (getattr(item, 'reported', 0) > 0 and (not getattr(item, 'ignore_reports', False) or c.user_is_admin)): w.show_reports = True w.use_big_modbuttons = True if (c.user_is_admin or (user and hasattr(item, 'sr_id') and (item.sr_id in can_flair_set or (w.author and w.author._id == user._id and item.sr_id in can_own_flair_set)))): w.can_flair = True w.approval_checkmark = None if w.can_ban: verdict = getattr(w, "verdict", None) if verdict in ('admin-approved', 'mod-approved'): approver = None approval_time = None baninfo = getattr(w, "ban_info", None) if baninfo: approver = baninfo.get("unbanner", None) approval_time = baninfo.get("unbanned_at", None) approver = approver or _("a moderator") if approval_time: text = _("approved by %(who)s %(when)s ago") % { "who": approver, "when": timesince(approval_time) } else: text = _("approved by %s") % approver w.approval_checkmark = text # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import add_attr user = c.user if c.user_is_loggedin else None aids = set(l.author_id for l in items if hasattr(l, 'author_id') and l.author_id is not None) authors = Account._byID(aids, data=True, stale=self.stale) now = datetime.datetime.now(g.tz) cakes = {a._id for a in authors.itervalues() if a.cake_expiration and a.cake_expiration >= now} friend_rels = user.friend_rels() if user and user.gold else {} subreddits = Subreddit.load_subreddits(items, stale=self.stale) can_ban_set = set() can_flair_set = set() can_own_flair_set = set() if user: for sr_id, sr in subreddits.iteritems(): if sr.can_ban(user): can_ban_set.add(sr_id) if sr.is_moderator_with_perms(user, 'flair'): can_flair_set.add(sr_id) if sr.link_flair_self_assign_enabled: can_own_flair_set.add(sr_id) #get likes/dislikes try: likes = queries.get_likes(user, items) except tdb_cassandra.TRANSIENT_EXCEPTIONS as e: g.log.warning("Cassandra vote lookup failed: %r", e) likes = {} types = {} wrapped = [] modlink = {} modlabel = {} for s in subreddits.values(): modlink[s._id] = '/r/%s/about/moderators' % s.name modlabel[s._id] = (_('moderator of /r/%(reddit)s, ' 'speaking officially') % {'reddit': s.name}) for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) w.author = None w.friend = False # List of tuples (see add_attr() for details) w.attribs = [] w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == 'yes': w.distinguished = 'moderator' elif item.distinguished in ('admin', 'special', 'gold', 'gold-auto'): w.distinguished = item.distinguished try: w.author = authors.get(item.author_id) if user and item.author_id in user.friends: # deprecated old way: w.friend = True # new way: label = None if friend_rels: rel = friend_rels[item.author_id] note = getattr(rel, "note", None) if note: label = u"%s (%s)" % (_("friend"), _force_unicode(note)) add_attr(w.attribs, 'F', label) except AttributeError: pass if (w.distinguished == 'admin' and w.author): add_attr(w.attribs, 'A') if w.distinguished == 'moderator': add_attr(w.attribs, 'M', label=modlabel[item.sr_id], link=modlink[item.sr_id]) if w.distinguished == 'special': args = w.author.special_distinguish() args.pop('name') if not args.get('kind'): args['kind'] = 'special' add_attr(w.attribs, **args) if w.author and w.author._id in cakes and not c.profilepage: add_attr( w.attribs, kind="cake", label=(_("%(user)s just celebrated a reddit birthday!") % {"user": w.author.name}), link="/user/%s" % w.author.name, ) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] w.likes = likes.get((user, item)) # update vote tallies compute_votes(w, item) w.score = w.upvotes - w.downvotes if w.likes: base_score = w.score - 1 elif w.likes is None: base_score = w.score else: base_score = w.score + 1 # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) if c.user_is_admin and getattr(item, 'ip', None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.can_flair = False w.use_big_modbuttons = self.spam_listing if (c.user_is_admin or (user and hasattr(item,'sr_id') and item.sr_id in can_ban_set)): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, 'ban_info', {}) w.unbanner = ban_info.get('unbanner') if item._spam: w.show_spam = True w.moderator_banned = ban_info.get('moderator_banned', False) w.autobanned = ban_info.get('auto', False) w.banner = ban_info.get('banner') if ban_info.get('note', None) and w.banner: w.banner += ' (%s)' % ban_info['note'] w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" if c.user == w.author and c.user._spam: w.show_spam = False w._spam = False w.use_big_modbuttons = False elif (getattr(item, 'reported', 0) > 0 and (not getattr(item, 'ignore_reports', False) or c.user_is_admin)): w.show_reports = True w.use_big_modbuttons = True if (c.user_is_admin or (user and hasattr(item, 'sr_id') and (item.sr_id in can_flair_set or (w.author and w.author._id == user._id and item.sr_id in can_own_flair_set)))): w.can_flair = True w.approval_checkmark = None if w.can_ban: verdict = getattr(w, "verdict", None) if verdict in ('admin-approved', 'mod-approved'): approver = None approval_time = None baninfo = getattr(w, "ban_info", None) if baninfo: approver = baninfo.get("unbanner", None) approval_time = baninfo.get("unbanned_at", None) approver = approver or _("a moderator") if approval_time: text = _("approved by %(who)s %(when)s ago") % { "who": approver, "when": timesince(approval_time)} else: text = _("approved by %s") % approver w.approval_checkmark = text # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import add_attr user = c.user if c.user_is_loggedin else None #get authors #TODO pull the author stuff into add_props for links and #comments and messages? aids = set(l.author_id for l in items if hasattr(l, 'author_id') and l.author_id is not None) authors = {} cup_infos = {} email_attrses = {} friend_rels = None if aids: authors = Account._byID(aids, data=True, stale=self.stale) if aids else {} cup_infos = Account.cup_info_multi(aids) if c.user_is_admin: email_attrses = admintools.email_attrs(aids, return_dict=True) if user and user.gold: friend_rels = user.friend_rels() subreddits = Subreddit.load_subreddits(items, stale=self.stale) if not user: can_ban_set = set() else: can_ban_set = set(id for (id,sr) in subreddits.iteritems() if sr.can_ban(user)) #get likes/dislikes try: likes = queries.get_likes(user, items) except tdb_cassandra.TRANSIENT_EXCEPTIONS as e: g.log.warning("Cassandra vote lookup failed: %r", e) likes = {} uid = user._id if user else None types = {} wrapped = [] count = 0 modlink = {} modlabel = {} for s in subreddits.values(): modlink[s._id] = '/r/%s/about/moderators' % s.name modlabel[s._id] = (_('moderator of /r/%(reddit)s, speaking officially') % dict(reddit = s.name) ) for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) #TODO pull the author stuff into add_props for links and #comments and messages? w.author = None w.friend = False # List of tuples (see add_attr() for details) w.attribs = [] w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == 'yes': w.distinguished = 'moderator' elif item.distinguished in ('admin', 'special'): w.distinguished = item.distinguished try: w.author = authors.get(item.author_id) if user and item.author_id in user.friends: # deprecated old way: w.friend = True # new way: label = None if friend_rels: rel = friend_rels[item.author_id] note = getattr(rel, "note", None) if note: label = u"%s (%s)" % (_("friend"), _force_unicode(note)) add_attr(w.attribs, 'F', label) except AttributeError: pass if (w.distinguished == 'admin' and w.author): add_attr(w.attribs, 'A') if w.distinguished == 'moderator': add_attr(w.attribs, 'M', label=modlabel[item.sr_id], link=modlink[item.sr_id]) if w.distinguished == 'special': args = w.author.special_distinguish() args.pop('name') if not args.get('kind'): args['kind'] = 'special' add_attr(w.attribs, **args) if False and w.author and c.user_is_admin: for attr in email_attrses[w.author._id]: add_attr(w.attribs, attr[2], label=attr[1]) if w.author and w.author._id in cup_infos and not c.profilepage: cup_info = cup_infos[w.author._id] label = _(cup_info["label_template"]) % \ {'user':w.author.name} add_attr(w.attribs, 'trophy:' + cup_info["img_url"], label=label, link = "/user/%s" % w.author.name) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] w.likes = likes.get((user, item)) # update vote tallies compute_votes(w, item) w.score = w.upvotes - w.downvotes if w.likes: base_score = w.score - 1 elif w.likes is None: base_score = w.score else: base_score = w.score + 1 # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) w.rowstyle = getattr(w, 'rowstyle', "") w.rowstyle += ' ' + ('even' if (count % 2) else 'odd') count += 1 if c.user_is_admin and getattr(item, 'ip', None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.reveal_trial_info = False w.use_big_modbuttons = False if (c.user_is_admin or (user and hasattr(item,'sr_id') and item.sr_id in can_ban_set)): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, 'ban_info', {}) w.unbanner = ban_info.get('unbanner') if item._spam: w.show_spam = True w.moderator_banned = ban_info.get('moderator_banned', False) w.autobanned = ban_info.get('auto', False) w.banner = ban_info.get('banner') if ban_info.get('note', None) and w.banner: w.banner += ' (%s)' % ban_info['note'] w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" elif getattr(item, 'reported', 0) > 0: w.show_reports = True w.use_big_modbuttons = True # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import add_attr user = c.user if c.user_is_loggedin else None #get authors #TODO pull the author stuff into add_props for links and #comments and messages? aids = set(l.author_id for l in items if hasattr(l, 'author_id') and l.author_id is not None) authors = {} cup_infos = {} email_attrses = {} friend_rels = None if aids: authors = Account._byID(aids, True) if aids else {} cup_infos = Account.cup_info_multi(aids) if c.user_is_admin: email_attrses = admintools.email_attrs(aids, return_dict=True) if c.user.gold: friend_rels = c.user.friend_rels() subreddits = Subreddit.load_subreddits(items) if not user: can_ban_set = set() else: can_ban_set = set(id for (id, sr) in subreddits.iteritems() if sr.can_ban(user)) #get likes/dislikes likes = queries.get_likes(user, items) uid = user._id if user else None types = {} wrapped = [] count = 0 modlink = {} modlabel = {} for s in subreddits.values(): modlink[s._id] = '/r/%s/about/moderators' % s.name modlabel[s._id] = ( _('moderator of /r/%(reddit)s, speaking officially') % dict(reddit=s.name)) for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) #TODO pull the author stuff into add_props for links and #comments and messages? w.author = None w.friend = False # List of tuples (see add_attr() for details) w.attribs = [] w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == 'yes': w.distinguished = 'moderator' elif item.distinguished == 'admin': w.distinguished = 'admin' try: w.author = authors.get(item.author_id) if user and item.author_id in user.friends: # deprecated old way: w.friend = True # new way: label = None if friend_rels: rel = friend_rels[item.author_id] note = getattr(rel, "note", None) if note: label = "%s (%s)" % (_("friend"), note) add_attr(w.attribs, 'F', label) except AttributeError: pass if (w.distinguished == 'admin' and w.author and w.author.name in g.admins): add_attr(w.attribs, 'A') if w.distinguished == 'moderator': add_attr(w.attribs, 'M', label=modlabel[item.sr_id], link=modlink[item.sr_id]) if False and w.author and c.user_is_admin: for attr in email_attrses[w.author._id]: add_attr(w.attribs, attr[2], label=attr[1]) if w.author and w.author._id in cup_infos and not c.profilepage: cup_info = cup_infos[w.author._id] label = _(cup_info["label_template"]) % \ {'user':w.author.name} add_attr(w.attribs, 'trophy:' + cup_info["img_url"], label=label, link="/user/%s" % w.author.name) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] w.likes = likes.get((user, item)) # update vote tallies compute_votes(w, item) w.score = w.upvotes - w.downvotes if w.likes: base_score = w.score - 1 elif w.likes is None: base_score = w.score else: base_score = w.score + 1 # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) w.rowstyle = getattr(w, 'rowstyle', "") w.rowstyle += ' ' + ('even' if (count % 2) else 'odd') count += 1 if c.user_is_admin and getattr(item, 'ip', None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.reveal_trial_info = False w.use_big_modbuttons = False if (c.user_is_admin or (user and hasattr(item, 'sr_id') and item.sr_id in can_ban_set)): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, 'ban_info', {}) w.unbanner = ban_info.get('unbanner') if item._spam: w.show_spam = True w.moderator_banned = ban_info.get('moderator_banned', False) w.autobanned = ban_info.get('auto', False) w.banner = ban_info.get('banner') w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" elif getattr(item, 'reported', 0) > 0: w.show_reports = True w.use_big_modbuttons = True # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import add_attr user = c.user if c.user_is_loggedin else None # get authors # TODO pull the author stuff into add_props for links and # comments and messages? aids = set(l.author_id for l in items if hasattr(l, "author_id") and l.author_id is not None) authors = {} cup_infos = {} email_attrses = {} friend_rels = None if aids: authors = Account._byID(aids, True) if aids else {} cup_infos = Account.cup_info_multi(aids) if c.user_is_admin: email_attrses = admintools.email_attrs(aids, return_dict=True) if c.user.gold: friend_rels = c.user.friend_rels() subreddits = Subreddit.load_subreddits(items) if not user: can_ban_set = set() else: can_ban_set = set(id for (id, sr) in subreddits.iteritems() if sr.can_ban(user)) # get likes/dislikes likes = queries.get_likes(user, items) uid = user._id if user else None types = {} wrapped = [] count = 0 modlink = {} modlabel = {} for s in subreddits.values(): modlink[s._id] = "/r/%s/about/moderators" % s.name modlabel[s._id] = _("moderator of /r/%(reddit)s, speaking officially") % dict(reddit=s.name) for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) # TODO pull the author stuff into add_props for links and # comments and messages? w.author = None w.friend = False # List of tuples (see add_attr() for details) w.attribs = [] w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == "yes": w.distinguished = "moderator" elif item.distinguished == "admin": w.distinguished = "admin" try: w.author = authors.get(item.author_id) if user and item.author_id in user.friends: # deprecated old way: w.friend = True # new way: label = None if friend_rels: rel = friend_rels[item.author_id] note = getattr(rel, "note", None) if note: label = "%s (%s)" % (_("friend"), note) add_attr(w.attribs, "F", label) except AttributeError: pass if w.distinguished == "admin" and w.author and w.author.name in g.admins: add_attr(w.attribs, "A") if w.distinguished == "moderator": add_attr(w.attribs, "M", label=modlabel[item.sr_id], link=modlink[item.sr_id]) if False and w.author and c.user_is_admin: for attr in email_attrses[w.author._id]: add_attr(w.attribs, attr[2], label=attr[1]) if w.author and w.author._id in cup_infos and not c.profilepage: cup_info = cup_infos[w.author._id] label = _(cup_info["label_template"]) % {"user": w.author.name} add_attr(w.attribs, "trophy:" + cup_info["img_url"], label=label, link="/user/%s" % w.author.name) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] w.likes = likes.get((user, item)) # update vote tallies compute_votes(w, item) w.score = w.upvotes - w.downvotes if w.likes: base_score = w.score - 1 elif w.likes is None: base_score = w.score else: base_score = w.score + 1 # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) w.rowstyle = getattr(w, "rowstyle", "") w.rowstyle += " " + ("even" if (count % 2) else "odd") count += 1 if c.user_is_admin and getattr(item, "ip", None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.reveal_trial_info = False w.use_big_modbuttons = False if c.user_is_admin or (user and hasattr(item, "sr_id") and item.sr_id in can_ban_set): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, "ban_info", {}) w.unbanner = ban_info.get("unbanner") if item._spam: w.show_spam = True w.moderator_banned = ban_info.get("moderator_banned", False) w.autobanned = ban_info.get("auto", False) w.banner = ban_info.get("banner") w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" elif getattr(item, "reported", 0) > 0: w.show_reports = True w.use_big_modbuttons = True # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped
def wrap_items(self, items): from r2.lib.db import queries from r2.lib.template_helpers import ( add_friend_distinguish, add_admin_distinguish, add_moderator_distinguish, add_cakeday_distinguish, add_special_distinguish, ) user = c.user if c.user_is_loggedin else None aids = set(l.author_id for l in items if hasattr(l, "author_id") and l.author_id is not None) authors = Account._byID(aids, data=True, stale=self.stale) now = datetime.datetime.now(g.tz) cakes = {a._id for a in authors.itervalues() if a.cake_expiration and a.cake_expiration >= now} friend_rels = user.friend_rels() if user and user.gold else {} subreddits = Subreddit.load_subreddits(items, stale=self.stale) can_ban_set = set() if user: for sr_id, sr in subreddits.iteritems(): if sr.can_ban(user): can_ban_set.add(sr_id) # get likes/dislikes try: likes = queries.get_likes(user, items) except tdb_cassandra.TRANSIENT_EXCEPTIONS as e: g.log.warning("Cassandra vote lookup failed: %r", e) likes = {} types = {} wrapped = [] for item in items: w = self.wrap(item) wrapped.append(w) # add for caching (plus it should be bad form to use _ # variables in templates) w.fullname = item._fullname types.setdefault(w.render_class, []).append(w) w.author = None w.friend = False w.distinguished = None if hasattr(item, "distinguished"): if item.distinguished == "yes": w.distinguished = "moderator" elif item.distinguished in ("admin", "special", "gold", "gold-auto"): w.distinguished = item.distinguished if getattr(item, "author_id", None): w.author = authors.get(item.author_id) if hasattr(item, "sr_id") and item.sr_id is not None: w.subreddit = subreddits[item.sr_id] distinguish_attribs_list = [] # if display_author exists, then w.author is unknown to the # receiver, so we can't check for friend or cakeday author_is_hidden = hasattr(item, "display_author") if not author_is_hidden and user and w.author and w.author._id in user.friends: w.friend = True # TODO: deprecated? if item.author_id in friend_rels: note = getattr(friend_rels[w.author._id], "note", None) else: note = None add_friend_distinguish(distinguish_attribs_list, note) if w.distinguished == "admin" and w.author: add_admin_distinguish(distinguish_attribs_list) if w.distinguished == "moderator": add_moderator_distinguish(distinguish_attribs_list, w.subreddit) if w.distinguished == "special": add_special_distinguish(distinguish_attribs_list, w.author) if not author_is_hidden and w.author and w.author._id in cakes and not c.profilepage: add_cakeday_distinguish(distinguish_attribs_list, w.author) w.attribs = distinguish_attribs_list user_vote_dir = likes.get((user, item)) if user_vote_dir == Vote.DIRECTIONS.up: w.likes = True elif user_vote_dir == Vote.DIRECTIONS.down: w.likes = False else: w.likes = None w.upvotes = item._ups w.downvotes = item._downs total_votes = max(item.num_votes, 1) w.upvote_ratio = float(item._ups) / total_votes w.is_controversial = self._is_controversial(w) w.score = w.upvotes - w.downvotes if user_vote_dir == Vote.DIRECTIONS.up: base_score = w.score - 1 elif user_vote_dir == Vote.DIRECTIONS.down: base_score = w.score + 1 else: base_score = w.score # store the set of available scores based on the vote # for ease of i18n when there is a label w.voting_score = [(base_score + x - 1) for x in range(3)] w.deleted = item._deleted w.link_notes = [] if c.user_is_admin: if item._deleted: w.link_notes.append("deleted link") if getattr(item, "verdict", None): if not item.verdict.endswith("-approved"): w.link_notes.append(w.verdict) if c.user_is_admin and getattr(item, "ip", None): w.ip_span = ip_span(item.ip) else: w.ip_span = "" # if the user can ban things on a given subreddit, or an # admin, then allow them to see that the item is spam, and # add the other spam-related display attributes w.show_reports = False w.show_spam = False w.can_ban = False w.use_big_modbuttons = self.spam_listing if c.user_is_admin or (user and hasattr(item, "sr_id") and item.sr_id in can_ban_set): if getattr(item, "promoted", None) is None: w.can_ban = True ban_info = getattr(item, "ban_info", {}) w.unbanner = ban_info.get("unbanner") if item._spam: w.show_spam = True w.moderator_banned = ban_info.get("moderator_banned", False) w.autobanned = ban_info.get("auto", False) w.banner = ban_info.get("banner") w.banned_at = ban_info.get("banned_at", None) if ban_info.get("note", None) and w.banner: w.banner += " (%s)" % ban_info["note"] w.use_big_modbuttons = True if getattr(w, "author", None) and w.author._spam: w.show_spam = "author" if c.user == w.author and c.user._spam: w.show_spam = False w._spam = False w.use_big_modbuttons = False elif getattr(item, "reported", 0) > 0 and ( not getattr(item, "ignore_reports", False) or c.user_is_admin ): w.show_reports = True w.use_big_modbuttons = True # report_count isn't used in any template, but add it to # the Wrapped so it's pulled into the render cache key in # instances when reported will be used in the template w.report_count = item.reported w.approval_checkmark = None if w.can_ban: verdict = getattr(w, "verdict", None) if verdict in ("admin-approved", "mod-approved"): approver = None approval_time = None baninfo = getattr(w, "ban_info", None) if baninfo: approver = baninfo.get("unbanner", None) approval_time = baninfo.get("unbanned_at", None) approver = approver or _("a moderator") if approval_time: text = _("approved by %(who)s %(when)s ago") % { "who": approver, "when": timesince(approval_time), } else: text = _("approved by %s") % approver w.approval_checkmark = text hooks.get_hook("builder.wrap_items").call(item=item, wrapped=w) # recache the user object: it may be None if user is not logged in, # whereas now we are happy to have the UnloggedUser object user = c.user for cls in types.keys(): cls.add_props(user, types[cls]) return wrapped