From 2464775210e526d613805e4d06cc113655efffe5 Mon Sep 17 00:00:00 2001
From: Aevann1 <randomname42029@gmail.com>
Date: Mon, 7 Feb 2022 14:38:45 +0200
Subject: [PATCH] cxv

---
 files/classes/comment.py    | 5 +++--
 files/classes/submission.py | 5 +++--
 files/classes/votes.py      | 8 ++++----
 files/routes/votes.py       | 2 ++
 4 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/files/classes/comment.py b/files/classes/comment.py
index 5a0554ee5..3a6fbf2a8 100644
--- a/files/classes/comment.py
+++ b/files/classes/comment.py
@@ -381,8 +381,9 @@ class Comment(Base):
 			body += f'"> - <a href="/votes?link=t3_{c.id}"><span id="poll-{c.id}">{c.upvotes}</span> votes</a></span></label></div>'
 
 		curr = self.total_choice_voted(v)
-		if curr:
-			body += f'<input class="d-none" id="current-{self.id}" value={curr[0].comment_id}>'
+		if curr: curr = " value=" + str(curr[0].comment_id)
+		else: curr = ''
+		body += f'<input class="d-none" id="current-{self.id}"{curr}>'
 
 		for c in self.choices:
 			body += f'''<div class="custom-control"><input name="choice" autocomplete="off" class="custom-control-input" type="radio" id="{c.id}" onchange="choice_vote('{c.id}','{self.id}')"'''
diff --git a/files/classes/submission.py b/files/classes/submission.py
index 232667605..8cc64cf05 100644
--- a/files/classes/submission.py
+++ b/files/classes/submission.py
@@ -394,8 +394,9 @@ class Submission(Base):
 			body += f'"> - <a href="/votes?link=t3_{c.id}"><span id="poll-{c.id}">{c.upvotes}</span> votes</a></span></label></div>'
 
 		curr = self.total_choice_voted(v)
-		if curr:
-			body += f'<input class="d-none" id="current-{self.id}" value={curr[0].comment_id}>'
+		if curr: curr = " value=" + str(curr[0].comment_id)
+		else: curr = ''
+		body += f'<input class="d-none" id="current-{self.id}"{curr}>'
 
 		for c in self.choices:
 			body += f'''<div class="custom-control"><input name="choice" autocomplete="off" class="custom-control-input" type="radio" id="{c.id}" onchange="choice_vote('{c.id}','{self.id}')"'''
diff --git a/files/classes/votes.py b/files/classes/votes.py
index 22f431c64..e2cbdbc07 100644
--- a/files/classes/votes.py
+++ b/files/classes/votes.py
@@ -15,8 +15,8 @@ class Vote(Base):
 	app_id = Column(Integer, ForeignKey("oauth_apps.id"))
 	real = Column(Boolean, default=True)
 
-	user = relationship("User", lazy="subquery", viewonly=True)
-	post = relationship("Submission", lazy="subquery", viewonly=True)
+	user = relationship("User", viewonly=True)
+	post = relationship("Submission", viewonly=True)
 
 	def __init__(self, *args, **kwargs):
 
@@ -56,8 +56,8 @@ class CommentVote(Base):
 	app_id = Column(Integer, ForeignKey("oauth_apps.id"))
 	real = Column(Boolean, default=True)
 
-	user = relationship("User", lazy="subquery")
-	comment = relationship("Comment", lazy="subquery", viewonly=True)
+	user = relationship("User")
+	comment = relationship("Comment")
 
 	def __init__(self, *args, **kwargs):
 
diff --git a/files/routes/votes.py b/files/routes/votes.py
index 8ad1e63e1..a4fd18167 100644
--- a/files/routes/votes.py
+++ b/files/routes/votes.py
@@ -259,6 +259,8 @@ def api_vote_choice(comment_id, v):
 	else: parent = comment.post
 
 	for vote in parent.total_choice_voted(v):
+		vote.comment.upvotes = g.db.query(CommentVote.id).filter_by(comment_id=vote.comment.id, vote_type=1).count() - 1
+		g.db.add(vote.comment)
 		g.db.delete(vote)
 
 	try: