Add command /selftimeout

vutfitdiscord · May 27, 2023 · c90a51b · c90a51b
1 parent 744aeee
commit c90a51b
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/cogs/README.md b/cogs/README.md
@@ -362,6 +362,7 @@ Containing timeout commands and manipulating with timeout.\
     - /timeout user
     - /timeout remove
     - /timeout list
+    - /selftimeout
 **Tasks:**
 
     - refresh_timeout

diff --git a/cogs/timeout.py b/cogs/timeout.py
@@ -11,6 +11,7 @@
 
 import utils
 from cogs.base import Base
+from config import cooldowns
 from config.app_config import config
 from config.messages import Messages
 from permissions import permission_check
@@ -341,6 +342,23 @@ async def on_audit_log_entry_create(self, entry):
 
                 await self.submod_helper_room.send(embed=embed)
 
+    @cooldowns.default_cooldown
+    @commands.slash_command(name="selftimeout", description=Messages.self_timeout)
+    async def self_timeout(self, inter: disnake.ApplicationCommandInteraction, endtime: str = commands.Param(
+            autocomplete=autocomplete_times,
+            max_length=20, description=Messages.timeout_time
+            )):
+        await inter.response.defer(ephemeral=True)
+        endtime = await self.timeout_parse(inter, inter.user, endtime, Messages.self_timeout_reason)
+        if endtime is None:
+            raise commands.BadArgument
+        starttime = inter.created_at.astimezone(tz=utils.get_local_zone()).replace(tzinfo=None)
+        # convert to local time and remove timezone info
+        self.timeout_repo.add_timeout(inter.user.id, inter.author.id, starttime,
+                                      endtime, Messages.self_timeout_reason)
+        await inter.send(content=Messages.self_timeout_success, ephemeral=True)
+
+    @self_timeout.error
     @timeout_user.error
     async def timeout_error(self, inter: disnake.ApplicationCommandInteraction, error):
         if isinstance(error, commands.BadArgument):

diff --git a/config/messages.py b/config/messages.py
@@ -488,6 +488,12 @@ class Messages:
     timeout_list_none = "Nenalezeny žádné umlčení."
     timeout_member_not_found = "{member} Nikoho takového jsem na serveru nenašel. Ujisti se, že jsi uživatele zadal @tagem."
 
+    # SELFTIMEOUT COG
+    self_timeout = "Dočasně si zakážeš interakce na serveru"
+    self_timeout_reason = "Sebeumlčení"
+    self_timeout_something_went_wrong = "Něco se pokazilo"
+    self_timeout_success = "Sebeumlčení proběhlo úspěšně"
+
     # TIMEOUT WARS
     timeout_wars_user = "Uživatel {user} byl umlčen na {time:.0f} minut."
     timeout_wars_message_delete = "Uživatel {user} byl přistižen při mazání zpráv. Byl proto umlčen na {time:.0f} minut."