Apply suggested changes

redbrick · Feb 1, 2024 · ae7cea8 · ae7cea8
1 parent cb1d77f
commit ae7cea8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/extensions/boosts.py b/src/extensions/boosts.py
@@ -30,7 +30,7 @@ def build_boost_message(
     message = base_message + multiple_boosts_message + "!"
 
     if (message_type in BOOST_TIERS):
-        count = BOOST_TIERS.index(message_type)
+        count = BOOST_TIERS.index(message_type) + 1
         message += f"\n{guild.name} has reached **Level {count}!**"
 
     return message

diff --git a/src/utils.py b/src/utils.py
@@ -4,4 +4,5 @@
 async def get_guild(client: GatewayClient, event: hikari.GuildMessageCreateEvent):
     return event.get_guild() or await client.rest.fetch_guild(event.guild_id)
 
-def role_mention(role: hikari.Snowflake | str): return f"<@&{role}>"
+def role_mention(role_id: hikari.Snowflake | int | str):
+    return f"<@&{role_id}>"