From 04a878a8c40daf4d30caa1256bf1f66f569b3fb1 Mon Sep 17 00:00:00 2001
From: TLSM <duolsm@outlook.com>
Date: Mon, 30 May 2022 08:13:11 -0400
Subject: [PATCH] Make betting polls respect can_gamble.

---
 files/classes/submission.py           | 2 +-
 files/routes/votes.py                 | 1 +
 files/templates/settings_filters.html | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/files/classes/submission.py b/files/classes/submission.py
index 87f73b8f3..9ea0b3edf 100644
--- a/files/classes/submission.py
+++ b/files/classes/submission.py
@@ -414,7 +414,7 @@ class Submission(Base):
 		for c in self.bet_options:
 			body += f'''<div class="custom-control mt-3"><input autocomplete="off" class="custom-control-input bet" type="radio" id="{c.id}" onchange="bet_vote('{c.id}')"'''
 			if c.poll_voted(v): body += " checked "
-			if not (v and v.coins > 200) or self.total_bet_voted(v): body += " disabled "
+			if not (v and v.coins > 200) or self.total_bet_voted(v) or not v.can_gamble: body += " disabled "
 			body += f'''><label class="custom-control-label" for="{c.id}">{c.body_html} - <a href="/votes?link=t3_{c.id}"><span id="bet-{c.id}">{c.upvotes}</span> bets</a>'''
 			if not self.total_bet_voted(v):
 				body += '''<span class="cost"> (cost of entry: 200 coins)</span>'''
diff --git a/files/routes/votes.py b/files/routes/votes.py
index 02a5da205..bf777981b 100644
--- a/files/routes/votes.py
+++ b/files/routes/votes.py
@@ -235,6 +235,7 @@ def api_vote_poll(comment_id, v):
 @limiter.limit("1/second;30/minute;200/hour;1000/day", key_func=lambda:f'{request.host}-{session.get("lo_user")}')
 @is_not_permabanned
 def bet(comment_id, v):
+	if not v.can_gamble: return {"error": "You have gambling disabled!"}
 	
 	if v.coins < 200: return {"error": "You don't have 200 coins!"}
 
diff --git a/files/templates/settings_filters.html b/files/templates/settings_filters.html
index ed850022e..546cfb3f2 100644
--- a/files/templates/settings_filters.html
+++ b/files/templates/settings_filters.html
@@ -156,7 +156,7 @@
 							<input autocomplete="off" type="checkbox" class="custom-control-input" id="can_gamble" name="can_gamble"{% if v.can_gamble %} checked{% endif %} onchange="post_toast(this,'/settings/profile?can_gamble='+document.getElementById('can_gamble').checked);">
 							<label class="custom-control-label" for="can_gamble"></label>
 						</div>
-						<span class="text-small-extra text-muted">Disable to prevent use of blackjack, slots, treasure chests, and the lottery.</span>
+						<span class="text-small-extra text-muted">Disable to prevent use of blackjack, slots, treasure chests, the lottery, and betting polls.</span>
 					</div>
 				</div>
 			</div>