Fixed issue cheran-senthil#456

ToxicPie · May 3, 2021 · 20d3d73 · 20d3d73
1 parent a64faa1
commit 20d3d73
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tle/util/db/user_db_conn.py b/tle/util/db/user_db_conn.py
@@ -475,11 +475,11 @@ def add_starboard_message(self, original_msg_id, starboard_msg_id, guild_id):
         self.conn.commit()
 
     def check_exists_starboard_message(self, original_msg_id):
-        query = ('SELECT 1 '
+        query = ('SELECT COUNT(*) AS cnt '
                  'FROM starboard_message '
                  'WHERE original_msg_id = ?')
         res = self.conn.execute(query, (original_msg_id,)).fetchone()
-        return res is not None
+        return res.cnt != 0
 
     def remove_starboard_message(self, *, original_msg_id=None, starboard_msg_id=None):
         assert (original_msg_id is None) ^ (starboard_msg_id is None)