Update command contexts and ensure data directory

* ✨ Make /rtfm user installable

* 🙈 Ignore `.idea/` (JetBrains)

* 🩹 Create `data` directory if it does not exist

* 🐛 Use `discord.InteractionContextType.guild` instead of `guild_only=True`.

* Format

---------

Co-authored-by: Dorukyum <[email protected]>

Author: Paillat-dev

.gitignore

@@ -3,3 +3,4 @@ __pycache__
 *.log
 data/
 venv/
+.idea/

cogs/developer/__init__.py

@@ -250,7 +250,10 @@ async def fetch_code_snippet(
             view=Delete(ctx.author),
         )
 
-    @discord.message_command(name="Link GitHub Issues", guild_only=True)
+    @discord.message_command(
+        name="Link GitHub Issues",
+        contexts={discord.InteractionContextType.guild},
+    )
     async def link_github_issues(
         self, ctx: discord.ApplicationContext, message: discord.Message
     ):
@@ -278,8 +281,7 @@ async def link_github_issues(
             links[0] if len(links) == 1 else "\n".join(f"<{link}>" for link in links)
         )
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_guild=True)
     @discord.option(
         "repo",
@@ -330,7 +332,17 @@ async def get_rtfm_results(self, target: str, query: str) -> list:
         )
         return results
 
-    @discord.command()
+    @discord.command(
+        integration_types={
+            discord.IntegrationType.guild_install,
+            discord.IntegrationType.user_install,
+        },
+        contexts={
+            discord.InteractionContextType.bot_dm,
+            discord.InteractionContextType.guild,
+            discord.InteractionContextType.private_channel,
+        },
+    )
     @discord.option(
         "documentation",
         description="The documentation to search through.",

cogs/logs.py

@@ -268,7 +268,7 @@ async def on_guild_role_update(self, before: discord.Role, after: discord.Role):
     logs = discord.SlashCommandGroup(
         "logs",
         "Commands related to logs.",
-        guild_only=True,
+        contexts={discord.InteractionContextType.guild},
         default_member_permissions=discord.Permissions(manage_guild=True),
     )
 

cogs/misc.py

@@ -12,8 +12,7 @@
 class Miscellaneous(Cog):
     """Miscellaneous commands."""
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     async def serverinfo(self, ctx: Context):
         """View information/statistics about the server."""
         guild = ctx.guild
@@ -218,8 +217,7 @@ async def magic_8ball_yes_or_no(self, ctx: Context, question: str):
         await sleep(2)
         await ctx.respond(f"> {question}\n{answer}")
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.option("text", description="The text to check for in display names.")
     async def how_many(self, ctx: Context, *, text: str):
         """Shows the amount of members that have the provided text in their display name."""

cogs/moderation.py

@@ -7,8 +7,7 @@
 class Moderation(Cog):
     """Commands related to moderation."""
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(ban_members=True)
     @discord.option(
         "members",
@@ -42,8 +41,7 @@ async def massban(
             )
         await ctx.respond(f"Banned **{count}** member{s(converted_members)}.")
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_messages=True)
     @discord.option(
         "seconds",
@@ -62,8 +60,7 @@ async def slowmode(self, ctx: Context, seconds: int):
             else "Slowmode is now disabled."
         )
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_roles=True)
     @discord.option(
         "permission",
@@ -105,8 +102,7 @@ async def lock(self, ctx: Context, permission: str, role: discord.Role | None, *
             f"The {permission} permission is now denied for {role.mention} in this channel."
         )
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_roles=True)
     @discord.option(
         "permission",
@@ -165,7 +161,7 @@ async def purge_channel(self, ctx: Context, **kwargs):
     purge = discord.SlashCommandGroup(
         "purge",
         "Commands to purge messages.",
-        guild_only=True,
+        contexts={discord.InteractionContextType.guild},
         default_member_permissions=discord.Permissions(manage_messages=True),
     )
 

cogs/server.py

@@ -12,7 +12,7 @@ class Server(Cog):
     emoji = discord.SlashCommandGroup(
         "emoji",
         "Commands related to emojis.",
-        guild_only=True,
+        contexts={discord.InteractionContextType.guild},
         default_member_permissions=discord.Permissions(manage_guild=True),
     )
 
@@ -59,7 +59,7 @@ async def emoji_delete(
     suggestions = discord.SlashCommandGroup(
         "suggestions",
         "Commands related to member suggestions.",
-        guild_only=True,
+        contexts={discord.InteractionContextType.guild},
         default_member_permissions=discord.Permissions(manage_guild=True),
     )
 
@@ -89,8 +89,7 @@ async def suggestions_disable(self, ctx: Context):
             )
         await ctx.respond("Member suggestions are already disabled for this server.")
 
-    @discord.slash_command()
-    @discord.guild_only()
+    @discord.slash_command(contexts={discord.InteractionContextType.guild})
     @discord.option("suggestion", description="The suggestion.")
     async def suggest(self, ctx: Context, *, suggestion: str):
         """Make a suggestion for the server. This will be sent to the channel set by the server managers."""

cogs/tags.py

@@ -9,7 +9,11 @@
 class Tags(Cog):
     """Commands related to tags."""
 
-    tag = discord.SlashCommandGroup("tag", "Commands related to tags.", guild_only=True)
+    tag = discord.SlashCommandGroup(
+        "tag",
+        "Commands related to tags.",
+        contexts={discord.InteractionContextType.guild},
+    )
 
     async def get_tag_names(self, ctx: discord.AutocompleteContext) -> List[str]:
         assert ctx.interaction.guild_id

cogs/warnings.py

@@ -75,24 +75,25 @@ def format_warning(self, warn: WarnModel) -> str:
             f"By **{f'{mod.name} ({mod.mention})' if mod else f'Unknown User (<@{warn.mod_id}>)'}**"
         )
 
-    @discord.slash_command(name="warn")
-    @discord.guild_only()
+    @discord.slash_command(name="warn", contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_messages=True)
     @discord.option("member", description="The member to warn.")
     @discord.option("reason", description="The reason of the warning.")
     async def warn_slash(self, ctx: Context, member: discord.Member, *, reason: str):
         """Warn a member."""
         await warn(ctx.interaction, member, reason)
 
-    @discord.user_command(name="Warn Member")
-    @discord.guild_only()
+    @discord.user_command(name="Warn Member", contexts={discord.InteractionContextType.guild})
     @discord.default_permissions(manage_messages=True)
     async def warn_member(self, ctx: Context, member: discord.Member):
         """Warn a member."""
         await ctx.send_modal(WarnModal(member, self.bot.get_cog("Logs")))
 
     warning = discord.SlashCommandGroup(
-        "warning", "Commands related to warnings.", guild_only=True, default_member_permissions=discord.Permissions(manage_messages=True)
+        "warning",
+        "Commands related to warnings.",
+        contexts={discord.InteractionContextType.guild},
+        default_member_permissions=discord.Permissions(manage_messages=True),
     )
 
     @warning.command()

core/bot.py

@@ -1,4 +1,4 @@
-from os import environ, getenv
+from os import environ, getenv, makedirs
 from traceback import format_exception
 
 import discord
@@ -29,6 +29,7 @@ def __init__(self) -> None:
         self.cache: dict[str, dict] = {"example_list": {}}
 
     async def setup_tortoise(self) -> None:
+        makedirs("data", exist_ok=True)
         await Tortoise.init(
             db_url="sqlite://data/database.db", modules={"models": ["core.models"]}
         )