Allow users to turn off their own level up notifications. Also allow guild owners to set channels to ignore level up notifications.

Author: brandons209

levelup/commands/admin.py

@@ -354,6 +354,15 @@ async def view_settings(self, ctx: commands.Context):
                 [f"<#{channel_id}>" for channel_id in conf.ignoredchannels if ctx.guild.get_channel(channel_id)]
             )
             embed.add_field(name=_("Ignored Channels"), value=joined, inline=False)
+        if conf.ignore_notification_channels:
+            joined = ", ".join(
+                [
+                    f"<#{channel_id}>"
+                    for channel_id in conf.ignore_notification_channels
+                    if ctx.guild.get_channel(channel_id)
+                ]
+            )
+            embed.add_field(name=_("Notification Ignored Channels"), value=joined, inline=False)
         if conf.ignoredusers:
             joined = ", ".join([f"<@{user_id}>" for user_id in conf.ignoredusers if ctx.guild.get_member(user_id)])
             embed.add_field(name=_("Ignored Users"), value=joined, inline=False)
@@ -855,6 +864,29 @@ async def ignore(self, ctx: commands.Context):
         """Base command for all ignore lists"""
         pass
 
+    @ignore.command(name="notify")
+    async def ignore_notify(
+        self,
+        ctx: commands.Context,
+        *,
+        channel: t.Union[discord.TextChannel, discord.VoiceChannel, discord.CategoryChannel, discord.ForumChannel],
+    ):
+        """
+        Add/Remove a channel in the notify ignore list
+        Channels in the notify ignore list won't have level up notifications sent there
+
+        Use the command with a channel already in the notify ignore list to remove it
+        """
+        conf = self.db.get_conf(ctx.guild)
+        if channel.id in conf.ignore_notification_channels:
+            conf.ignore_notification_channels.remove(channel.id)
+            txt = _("Channel {} has been removed from the notify ignore list").format(channel.mention)
+        else:
+            conf.ignore_notification_channels.append(channel.id)
+            txt = _("Channel {} has been added to the notify ignore list").format(channel.mention)
+        self.save()
+        await ctx.send(txt)
+
     @ignore.command(name="channel")
     async def ignore_channel(
         self,

levelup/commands/user.py

@@ -136,6 +136,33 @@ async def profile_data(self, ctx: commands.Context, user_id: int):
         )
         await ctx.send(txt)
 
+    @commands.hybrid_command(name="lvlupnotify", aliases=["notify"])
+    @commands.guild_only()
+    @commands.cooldown(3, 10, commands.BucketType.user)
+    async def level_up_notify(self, ctx: commands.Context):
+        """
+        If enabled, level up notifications won't be sent for you in the server
+        """
+        conf = self.db.get_conf(ctx.guild)
+        profile = conf.get_profile(ctx.author.id)
+
+        profile.ignore_level_up_notification = not profile.ignore_level_up_notification
+
+        if profile.ignore_level_up_notification:
+            await ctx.reply(
+                "Level up notifications are now **disabled** for you.",
+                mention_author=False,
+                delete_after=30,
+            )
+        else:
+            await ctx.reply(
+                "Level up notifications are now **enabled** for you.",
+                mention_author=False,
+                delete_after=30,
+            )
+
+        self.save()
+
     @commands.hybrid_command(name="profile", aliases=["pf"])
     @commands.guild_only()
     @commands.cooldown(3, 10, commands.BucketType.user)
@@ -226,13 +253,17 @@ async def prestige(self, ctx: commands.Context):
         if not role:
             return await ctx.send(_("The prestige role for this level no longer exists, please contact an admin!"))
 
-        current_xp = int(profile.xp)
-        xp_at_prestige = conf.algorithm.get_xp(conf.prestigelevel)
-        leftover_xp = current_xp - xp_at_prestige if current_xp > xp_at_prestige else 0
-        newlevel = conf.algorithm.get_level(leftover_xp)
-
-        profile.level = newlevel
-        profile.xp = leftover_xp
+        # current_xp = int(profile.xp)
+        # xp_at_prestige = conf.algorithm.get_xp(conf.prestigelevel)
+        # leftover_xp = current_xp - xp_at_prestige if current_xp > xp_at_prestige else 0
+        # newlevel = conf.algorithm.get_level(leftover_xp)
+        # change prestige to work by levels instead of exp
+        current_level = profile.level
+        new_level = max(1, current_level - conf.prestigelevel)
+        new_xp = conf.algorithm.get_xp(new_level)
+
+        profile.level = new_level
+        profile.xp = new_xp
         profile.prestige = next_prestige
         self.save()
 

levelup/common/models.py

@@ -124,6 +124,7 @@ class Profile(Base):
         description="Last time the user was active in the guild (message sent or voice activity)",
     )
     show_tutorial: bool = True  # Init with True, show tutorial on first command usage
+    ignore_level_up_notification: bool = False  # if true level up notifications will not be announced for this user
 
     # Profile customization
     style: str = "default"  # Can be default, runescape... (WIP)
@@ -308,6 +309,7 @@ class GuildSettings(Base):
     ignoredchannels: t.List[int] = []  # Channels that dont gain XP
     ignoredroles: t.List[int] = []  # Roles that dont gain XP
     ignoredusers: t.List[int] = []  # Ignored users won't gain XP
+    ignore_notification_channels: t.List[int] = []  # channels where level up announcements won't be sent if configured
 
     # Prestige
     prestigelevel: int = 0  # Level required to prestige, 0 is disabled

levelup/shared/levelups.py

@@ -115,18 +115,23 @@ async def check_levelups(
             )
             if current_channel and conf.notify:
                 with suppress(discord.HTTPException):
-                    if conf.notifymention:
-                        await current_channel.send(member.mention, embed=embed)
-                    else:
-                        await current_channel.send(embed=embed)
+                    if (
+                        not profile.ignore_level_up_notification
+                        and current_channel.id not in conf.ignore_notification_channels
+                    ):
+                        if conf.notifymention:
+                            await current_channel.send(member.mention, embed=embed)
+                        else:
+                            await current_channel.send(embed=embed)
 
             current_channel_id = current_channel.id if current_channel else 0
             if log_channel and log_channel.id != current_channel_id:
                 with suppress(discord.HTTPException):
-                    if conf.notifymention:
-                        await log_channel.send(member.mention, embed=embed)
-                    else:
-                        await log_channel.send(embed=embed)
+                    if not profile.ignore_level_up_notification:
+                        if conf.notifymention:
+                            await log_channel.send(member.mention, embed=embed)
+                        else:
+                            await log_channel.send(embed=embed)
 
         else:
             fonts = list(self.fonts.glob("*.ttf")) + list(self.custom_fonts.iterdir())
@@ -205,16 +210,21 @@ def _run() -> t.Tuple[bytes, bool]:
             if current_channel and conf.notify:
                 file = discord.File(BytesIO(img_bytes), filename=f"levelup.{ext}")
                 with suppress(discord.HTTPException):
-                    if conf.notifymention and message is not None:
-                        await message.reply(msg_txt, file=file, mention_author=True)
-                    else:
-                        await current_channel.send(msg_txt, file=file)
+                    if (
+                        not profile.ignore_level_up_notification
+                        and current_channel.id not in conf.ignore_notification_channels
+                    ):
+                        if conf.notifymention and message is not None:
+                            await message.reply(msg_txt, file=file, mention_author=True)
+                        else:
+                            await current_channel.send(msg_txt, file=file)
 
             current_channel_id = current_channel.id if current_channel else 0
             if log_channel and log_channel.id != current_channel_id:
                 file = discord.File(BytesIO(img_bytes), filename=f"levelup.{ext}")
                 with suppress(discord.HTTPException):
-                    await log_channel.send(msg_txt, file=file)
+                    if not profile.ignore_level_up_notification:
+                        await log_channel.send(msg_txt, file=file)
 
         payload = {
             "guild": guild,  # discord.Guild