From 941b5fa82dacf9e5cfcfa9cd5e9e2f925a10becd Mon Sep 17 00:00:00 2001 From: justcool393 Date: Tue, 11 Oct 2022 23:47:47 -0700 Subject: [PATCH] vote realness is in user model --- files/classes/user.py | 8 ++++++++ files/routes/votes.py | 4 +--- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/files/classes/user.py b/files/classes/user.py index 0547c4fe6..e3f19d4f3 100644 --- a/files/classes/user.py +++ b/files/classes/user.py @@ -289,6 +289,14 @@ class User(Base): if self.bite: return "565656" return self.namecolor + @property + @lazy + def is_votes_real(self): + if self.is_suspended_permanently or self.shadowbanned: return False + if self.agendaposter: return False + if self.profile_url.startswith('/e/') and not self.customtitle and self.namecolor == DEFAULT_COLOR: return False + return True + @lazy def mods(self, sub): if self.is_suspended_permanently or self.shadowbanned: return False diff --git a/files/routes/votes.py b/files/routes/votes.py index 18fc7ec85..76a3df77d 100644 --- a/files/routes/votes.py +++ b/files/routes/votes.py @@ -98,9 +98,7 @@ def vote_post_comment(target_id, new, v, cls, vote_cls): target.author.truecoins += coin_delta g.db.add(target.author) - if new == 1 and (v.agendaposter or v.shadowbanned or (v.is_banned and not v.unban_utc) or (v.profile_url.startswith('/e/') and not v.customtitle and v.namecolor == DEFAULT_COLOR)): real = False - else: real = True - + real = new != 1 or v.is_votes_real vote = None if vote_cls == Vote: vote = Vote(user_id=v.id,