watcher/commands/info.py

import hikari
import lightbulb
from miru.ext import nav

import config as c
from tools import components


plugin = lightbulb.Plugin("info", default_enabled_guilds=c.config["guild"])


# Get list of members not in activity database
@plugin.command
@lightbulb.add_checks(lightbulb.has_guild_permissions(hikari.Permissions.MANAGE_GUILD))
@lightbulb.command("limbo", "List members not in activity database", ephemeral=True)
@lightbulb.implements(lightbulb.SlashCommand)
async def limbo(context):
    # Embed builder
    def build(index, content):
        return hikari.Embed(
            title="Limbo", description=content, color=context.get_guild().get_my_member().get_top_role().color
        ).set_footer(f"{len(limbo)} members")

    # Get, sort, and filter list of members if not a bot, without excluded role, and not in db
    limbo = {
        snowflake: member
        for snowflake, member in sorted(
            context.get_guild().get_members().items(), key=lambda item: item[1].display_name
        )
        if not member.is_bot
        and c.config["exclude"] not in [role.id for role in member.get_roles()]
        and snowflake not in c.db
    }

    # Build paginator
    paginator = lightbulb.utils.EmbedPaginator()
    paginator.set_embed_factory(build)
    for snowflake, member in limbo.items():
        paginator.add_line(f"{member.mention} {snowflake}")
    pages = [page for page in paginator.build_pages()]

    # Send paginator
    if len(pages) > 1:
        navigator = nav.NavigatorView(
            pages=pages,
            buttons=[components.Back(), components.Forward()],
            timeout=600,
        )
        await navigator.send(context.interaction, ephemeral=True)
    elif pages:
        await context.respond(pages[0])
    else:
        await context.respond("***All members accounted for***")


def load(bot):
    bot.add_plugin(plugin)


def unload(bot):
    bot.remove_plugin(plugin)