build.cogs.leaderboardcog

Complexity

Total Complexity

20

Size/Duplication

Total Lines	112
Duplicated Lines	0 %

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	LeaderboardsCog.__init__()	0	3	1
A	LeaderboardsCog.update_member_roles()	0	7	2
A	LeaderboardsCog.update_leaderboards()	0	9	1
A	LeaderboardsCog.update_role()	0	11	5
A	LeaderboardsCog.update_role_ext()	0	10	4
A	LeaderboardsCog.update_leaderboards_loop()	0	3	1
A	LeaderboardsCog.cog_unload()	0	2	1
A	LeaderboardsCog.before_update_leaderboards_loop()	0	5	1
A	LeaderboardsCog.reload_leaderboards()	0	6	1
A	LeaderboardsCog.update_leaderboard()	0	25	2

import discord
import pandas as pd
from discord.ext import commands, tasks
from discord.utils import get
from discord_slash import cog_ext

import cogs.cogbase as cogbase
from cogs.databasecog import DatabaseCog
from modules.utils import get_dominant_color

legend_multiplier = 5


class LeaderboardsCog(cogbase.BaseCog):
    def __init__(self, base):
        super().__init__(base)
        self.update_leaderboards_loop.start()

    def cog_unload(self):
        self.update_leaderboards_loop.cancel()

    # Send leaderboards to specified channel
    async def update_leaderboard(self, channel: int, ch_type: str):
        top_ch = self.bot.get_channel(channel)
        spots_df = await DatabaseCog.db_get_spots_df()
        spots_df = pd.DataFrame(spots_df)
        spots_df["total"] = spots_df["legendary"] * legend_multiplier + spots_df["rare"]
        spots_df_top = spots_df.sort_values(ch_type, ascending=False).head(15)
        spots_df_top = spots_df_top.reset_index(drop=True)
        await top_ch.purge()

        top_print = []
        for index, row in spots_df_top.iterrows():
            member_stats = [f"**[{index + 1}]**  {row['display_name']} - {row[ch_type]}"]
            top_print.append(member_stats)
        top_print = ['\n'.join([elem for elem in sublist]) for sublist in top_print]
        top_print = "\n".join(top_print)
        ch_type = ''.join([i for i in ch_type if not i.isdigit()])

        top_user_id = int(spots_df_top.at[0, 'member_id'])
        top_user = get(self.bot.get_all_members(), id=top_user_id)
        top_user_color = get_dominant_color(top_user.avatar_url)
        embed_command = discord.Embed(title=f"TOP 15 {ch_type.upper()}", description=top_print,
                                      color=top_user_color)
        member = self.bot.get_user(spots_df_top['member_id'].iloc[0])
        embed_command.set_thumbnail(url=f'{member.avatar_url}')
        await top_ch.send(embed=embed_command)

    # Update member spotting role(total/common)
    async def update_role(self, guild, guild_member, spot_roles, common: bool):
        roles_type = "common" if common else "total"
        try:
            spots_df = await DatabaseCog.db_get_member_stats(guild_member.id)
            if not common:
                spots_df["total"] = spots_df["legendary"] * legend_multiplier + spots_df["rare"]
            roles_list = [key for (key, value) in spot_roles.items() if spots_df.loc[0, roles_type] >= value]
            if roles_list:
                await self.update_role_ext(guild, roles_list, guild_member)
        except KeyError as e:
            print(e)

    async def update_role_ext(self, guild, roles_list, guild_member):
        await self.create_role(guild, roles_list[-1])
        role_new = get(guild.roles, name=roles_list[-1])
        if role_new not in guild_member.roles:
            await guild_member.add_roles(role_new)
        if len(roles_list) > 1:
            await self.create_role(guild, roles_list[-2])
            role_old = get(guild.roles, name=roles_list[-2])
            if role_old in guild_member.roles:
                await guild_member.remove_roles(role_old)

    # Update members' spotting roles
    async def update_member_roles(self):
        guild = self.bot.get_guild(self.bot.guild[0])
        spot_roles_total = self.bot.config["total_milestones"][0]
        spot_roles_common = self.bot.config["common_milestones"][0]
        for guild_member in guild.members:
            await self.update_role(guild, guild_member, spot_roles_total, False)
            await self.update_role(guild, guild_member, spot_roles_common, True)

    async def update_leaderboards(self):
        await self.update_leaderboard(self.bot.ch_leaderboards, "total")
        await self.update_leaderboard(self.bot.ch_leaderboards_common, "common")
        await self.update_leaderboard(self.bot.ch_leaderboards_event, "event1")
        dt_string = self.bot.get_current_time()
        print(f'({dt_string})\t[{self.__class__.__name__}]: Leaderboards updated')
        await self.update_member_roles()
        dt_string = self.bot.get_current_time()
        print(f"({dt_string})\t[{self.__class__.__name__}]: Members' roles updated")

    @tasks.loop(minutes=15)
    async def update_leaderboards_loop(self):
        await self.update_leaderboards()

    @update_leaderboards_loop.before_loop
    async def before_update_leaderboards_loop(self):
        dt_string = self.bot.get_current_time()
        print(f'({dt_string})\t[{self.__class__.__name__}]: Waiting until Bot is ready')
        await self.bot.wait_until_ready()

    @cog_ext.cog_slash(name="reloadLeaderboards", guild_ids=cogbase.GUILD_IDS,
                       description=" ",
                       default_permission=True)
    async def reload_leaderboards(self, ctx):
        await self.update_leaderboards()
        await ctx.send(f"Leaderboards reloaded", hidden=True)


def setup(bot: commands.Bot):
    bot.add_cog(LeaderboardsCog(bot))