Utilize Client.interaction_tree

This commit is contained in:
Zeva Rose 2022-09-10 17:40:02 -06:00
parent c70c7e20be
commit d3fe0a0744

View file

@ -58,35 +58,38 @@ class EventMixin(MemberEventMixin, MessageEventMixin, ComponentEventMixin):
"{}&permissions=8&scope=bot%20applications.commands".format(self.user.id) "{}&permissions=8&scope=bot%20applications.commands".format(self.user.id)
) )
global_base_commands = {} global_base_commands = 0
guild_base_commands = {} guild_base_commands = 0
global_context_menus = [] global_context_menus = 0
guild_context_menus = [] guild_context_menus = 0
for cid in self.interactions: for cid in self.interaction_tree:
commands = self.interactions[cid] if cid == 0:
to_update = global_base_commands if cid == 0 else guild_base_commands global_base_commands = sum(
for command in commands: 1
if isinstance(commands[command], ContextMenu): for _ in self.interaction_tree[cid]
if cid == 0: if not isinstance(self.interaction_tree[cid][_], ContextMenu)
global_context_menus.append(command) )
else: global_context_menus = sum(
guild_context_menus.append(command) 1
continue for _ in self.interaction_tree[cid]
full = command.split(" ") if isinstance(self.interaction_tree[cid][_], ContextMenu)
base = full[0] )
if base not in to_update: else:
to_update[base] = {} guild_base_commands += sum(
if len(full) == 3: 1
to_update[base][full[1]] = full[2] for _ in self.interaction_tree[cid]
elif len(full) == 2: if not isinstance(self.interaction_tree[cid][_], ContextMenu)
to_update[base][full[1]] = None )
guild_context_menus += sum(
1
for _ in self.interaction_tree[cid]
if isinstance(self.interaction_tree[cid][_], ContextMenu)
)
self.logger.info( self.logger.info("Loaded {:>3} global base slash commands".format(global_base_commands))
"Loaded {:>2} global base slash commands".format(len(global_base_commands)) self.logger.info("Loaded {:>3} global context menus".format(global_context_menus))
) self.logger.info("Loaded {:>3} guild base slash commands".format(guild_base_commands))
self.logger.info("Loaded {:>2} global context menus".format(len(global_context_menus))) self.logger.info("Loaded {:>3} guild context menus".format(guild_context_menus))
self.logger.info("Loaded {:>2} guild base slash commands".format(len(guild_base_commands)))
self.logger.info("Loaded {:>2} guild context menus".format(len(guild_context_menus)))
self.logger.debug("Hitting Reminders for faster loads") self.logger.debug("Hitting Reminders for faster loads")
_ = await Reminder.find().to_list(None) _ = await Reminder.find().to_list(None)