game/clear.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+# Copyright 2022 - c0de <c0de@c0de.dev>
+# Licensed under the MIT License (https://opensource.org/licenses/MIT)
+
+# pylint: disable=missing-module-docstring,too-few-public-methods
+
+from database.models import PlayerModel as Player
+from game.base import BaseGameManager
+
+
+class ClearManager(BaseGameManager):
+    """Commands that run when a player clears the session leaderboard"""
+
+    def __init__(self):
+        super().__init__()
+        self.commands.append(("clear", self.clear))
+
+    async def clear(self):
+        """Clear command - Clears the session scoreboard"""
+
+        players = Player.select(Player.total_points)
+
+        for player in players:
+            player.total_points = 0
+
+        Player.bulk_update(players, Player.total_points)
+
+        clear_message = "The score has been cleared!"
+
+        recipient = await self.discord.fetch_user(self.message.author.id)
+        await recipient.send(clear_message)