fix poll-voting

master
Aevann1 2022-06-26 22:53:12 +02:00
parent 8185340e15
commit e7ce39bd80
3 changed files with 6 additions and 5 deletions

View File

@ -135,8 +135,8 @@ class Comment(Base):
@lazy @lazy
def total_choice_voted(self, v): def total_choice_voted(self, v):
if v: if v:
return g.db.query(CommentVote.comment_id).filter(CommentVote.user_id == v.id, CommentVote.comment_id.in_([x.id for x in self.choices])).first() return g.db.query(CommentVote).filter(CommentVote.user_id == v.id, CommentVote.comment_id.in_([x.id for x in self.choices])).first()
return False return None
@property @property
@lazy @lazy

View File

@ -128,8 +128,8 @@ class Submission(Base):
@lazy @lazy
def total_choice_voted(self, v): def total_choice_voted(self, v):
if v and self.choices: if v and self.choices:
return g.db.query(CommentVote.comment_id).filter(CommentVote.user_id == v.id, CommentVote.comment_id.in_([x.id for x in self.choices])).first() return g.db.query(CommentVote).filter(CommentVote.user_id == v.id, CommentVote.comment_id.in_([x.id for x in self.choices])).first()
return False return None
@lazy @lazy
def total_bet_voted(self, v): def total_bet_voted(self, v):

View File

@ -278,7 +278,8 @@ def api_vote_choice(comment_id, v):
if comment.parent_comment: parent = comment.parent_comment if comment.parent_comment: parent = comment.parent_comment
else: parent = comment.post else: parent = comment.post
for vote in parent.total_choice_voted(v): vote = parent.total_choice_voted(v)
if vote:
vote.comment.upvotes = g.db.query(CommentVote).filter_by(comment_id=vote.comment.id, vote_type=1).count() - 1 vote.comment.upvotes = g.db.query(CommentVote).filter_by(comment_id=vote.comment.id, vote_type=1).count() - 1
g.db.add(vote.comment) g.db.add(vote.comment)
g.db.delete(vote) g.db.delete(vote)