pincer.commands.chat_command_handler.ChatCommandHandler.__build_local_commands() - Code Metrics - Inspection of ":sparkles: Rewrote command registration and added..." - Pincer-org/Pincer - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — main (#389)

unknown

created 2022-01-17 02:00 UTC

ChatCommandHandler.__build_local_commands() C

↳ Parent: pincer.commands.chat_command_handler

Complexity

Conditions

Size

Total Lines	125
Code Lines	62

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	62
dl	0
loc	125
rs	5.0236
c	0
b	0
f	0
cc	11
nop	0

How to fix Long Method Complexity

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

If many parameters/temporary variables are present:
- Replace temporary variables with Query
- Introduce parameter object; often combined with preserve whole object
- If the above two are insufficient: Replace method with method object
If you have long conditionals: Decompose Conditional
Otherwise: Extract method

Complexity

Complex classes like pincer.commands.chat_command_handler.ChatCommandHandler.__build_local_commands() often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.

Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.

# Copyright Pincer 2021-Present

# Full MIT License can be found in `LICENSE` at the project root.

from __future__ import annotations

from typing import TYPE_CHECKING

import logging
from asyncio import gather

from ..utils.types import MISSING, Singleton

from ..exceptions import ForbiddenError
from ..objects.guild.guild import Guild
from ..objects.app.command import AppCommand, AppCommandOption
from ..objects.app.command_types import AppCommandOptionType, AppCommandType

if TYPE_CHECKING:
    from typing import List, Dict, Optional, ValuesView, Union

    from .interactable import Interactable
    from ..client import Client
    from ..utils.snowflake import Snowflake
    from ..objects.app.command import InteractableStructure

_log = logging.getLogger(__name__)


class ChatCommandHandler(metaclass=Singleton):
    """Singleton containing methods used to handle various commands

    The register and built_register
    -------------------------------
    I found the way Discord expects commands to be registered to be very different than
    how you want to think about command registration. i.e. Discord wants nesting but we
    don't want any nesting. Nesting makes it hard to think about commands and also will
    increase lookup time.
    The way this problem is avoided is by storing a version of the commands that we can
    deal with as library developers and a version of the command that Discord thinks we
    should provide. That is where the register and the built_register help simplify the
    design of the library.
    The register is simply where the "Pincer version" of commands gets saved to memory.
    The built_register is where the version of commands that Discord requires is saved.
    The register allows for O(1) lookups by storing commands in a Python dictionary. It
    does cost some memory to save two copies in the current iteration of the system but
    we should be able to drop the built_register in runtime if we want to. I don't feel
    that lost maintainability from this is optimal. We can index by in O(1) by checking
    the register but can still use the built_register if we need to do a nested lookup.

    Attributes
    ----------
    client: :class:`Client`
        The client object
    managers: Dict[:class:`str`, :class:`~typing.Any`]
        Dictionary of managers
    register: Dict[:class:`str`, :class:`~pincer.objects.app.command.InteractableStructure`]
        Dictionary of ``InteractableStructure``
    built_register: Dict[:class:`str`, :class:`~pincer.objects.app.command.InteractableStructure`]
        Dictionary of ``InteractableStructure`` where the commands are converted to
        the format that Discord expects for sub commands and sub command groups.
    """  # noqa: E501

    has_been_initialized = False
    managers: List[Interactable] = []
    register: Dict[str, InteractableStructure] = {}
    built_register: Dict[str, AppCommand] = {}

    # Endpoints:
    __get = "/commands"
    __delete = "/commands/{command.id}"
    __update = "/commands/{command.id}"
    __add = "/commands"
    __add_guild = "/guilds/{command.guild_id}/commands"
    __get_guild = "/guilds/{guild_id}/commands"
    __update_guild = "/guilds/{command.guild_id}/commands/{command.id}"
    __delete_guild = "/guilds/{command.guild_id}/commands/{command.id}"

    def __init__(self, client: Client):
        self.client = client
        self._api_commands: List[AppCommand] = []
        _log.debug(
            "%i commands registered.", len(ChatCommandHandler.register.items())
        )

        self.__prefix = f"applications/{self.client.bot.id}"

    async def get_commands(self) -> List[AppCommand]:
        """|coro|

        Get a list of app commands from Discord

        Returns
        -------
        List[:class:`~pincer.objects.app.command.AppCommand`]
            List of commands.
        """
        # TODO: Update if discord adds bulk get guild commands

        guild_commands = await gather(*(
            self.client.http.get(
                self.__prefix + self.__get_guild.format(
                    guild_id=guild.id if isinstance(guild, Guild) else guild
                )
            ) for guild in self.client.guilds
        ))
        return list(
            map(
                AppCommand.from_dict,
                await self.client.http.get(self.__prefix + self.__get)
                + [cmd for guild in guild_commands for cmd in guild],
            )
        )

    async def remove_command(self, cmd: AppCommand):
        """|coro|

        Remove a specific command

        Parameters
        ----------
        cmd : :class:`~pincer.objects.app.command.AppCommand`
            What command to delete
        """
        # TODO: Update if discord adds bulk delete commands

        if cmd.guild_id:
            _log.info(
                "Removing command `%s` with guild id %d from Discord",
                cmd.name,
                cmd.guild_id,
            )
        else:
            _log.info("Removing global command `%s` from Discord", cmd.name)

        remove_endpoint = self.__delete_guild if cmd.guild_id else self.__delete

        await self.client.http.delete(
            self.__prefix + remove_endpoint.format(command=cmd)
        )

    async def add_command(self, cmd: AppCommand):
        """|coro|

        Add an app command

        Parameters
        ----------
        cmd : :class:`~pincer.objects.app.command.AppCommand`
            Command to add
        """
        _log.info("Updated or registered command `%s` to Discord", cmd.name)

        add_endpoint = self.__add

        if cmd.guild_id:
            add_endpoint = self.__add_guild.format(command=cmd)

        await self.client.http.post(
            self.__prefix + add_endpoint, data=cmd.to_dict()
        )

    async def add_commands(self, commands: List[AppCommand]):
        """|coro|

        Add a list of app commands

        Parameters
        ----------
        commands : List[:class:`~pincer.objects.app.command.AppCommand`]
            List of command objects to add
        """
        await gather(*map(self.add_command, commands))

    @staticmethod
    def __build_local_commands():
        """Builds the commands into the format that Discord expects. See class info
        for the reasoning.
        """

        # Reset the built register
        ChatCommandHandler.built_register = {}

        for cmd in ChatCommandHandler.register.values():

            if cmd.sub_group:
                # If a command has a sub_group, it must be nested 2 levels deep.
                #
                # command
                #     subcommand-group
                #         subcommand
                #
                # The children of the subcommand-group object are being set to include
                # `cmd` If that subcommand-group object does not exist, it will be
                # created here. The same goes for the top-level command.
                #
                # First make sure the command exists. This command will hold the
                # subcommand-group for `cmd`.

                # `key` represents the hash value for the top-level command that will
                # hold the subcommand.
                key = _hash_app_command_params(
                    cmd.group.name,
                    cmd.app.guild_id,
                    AppCommandType.CHAT_INPUT,
                    None,
                    None,
                )

                if key not in ChatCommandHandler.built_register:
                    ChatCommandHandler.built_register[key] = AppCommand(
                        name=cmd.group.name,
                        description=cmd.group.description,
                        type=AppCommandType.CHAT_INPUT,
                        guild_id=cmd.app.guild_id,
                        options=[]
                    )

                # The top-level command now exists. A subcommand group now if placed
                # inside the top-level command. This subcommand group will hold `cmd`.

                children = ChatCommandHandler.built_register[key].options

                sub_command_group = AppCommandOption(
                    name=cmd.sub_group.name,
                    description=cmd.sub_group.description,
                    type=AppCommandOptionType.SUB_COMMAND_GROUP,
                    options=[]
                )

                # This for-else makes sure that sub_command_group will hold a reference
                # to the subcommand group that we want to modify to hold `cmd`

                for cmd_in_children in children:
                    if (
                        cmd_in_children.name == sub_command_group.name

                        and cmd_in_children.description == sub_command_group.description

                        and cmd_in_children.type == sub_command_group.type

                    ):
                        sub_command_group = cmd_in_children
                        break
                else:
                    children.append(sub_command_group)

                sub_command_group.options.append(AppCommandOption(
                    name=cmd.app.name,
                    description=cmd.app.description,
                    type=AppCommandOptionType.SUB_COMMAND,
                    options=cmd.app.options,
                ))

                continue

            if cmd.group:
                # Any command at this point will only have one level of nesting.
                #
                # Command
                #    subcommand
                #
                # A subcommand object is what is being generated here. If there is no
                # top level command, it will be created here.

                # `key` represents the hash value for the top-level command that will
                # hold the subcommand.

                key = _hash_app_command_params(
                    cmd.group.name,
                    cmd.app.guild_id,
                    AppCommandOptionType.SUB_COMMAND,
                    None,
                    None
                )

                if key not in ChatCommandHandler.built_register:
                    ChatCommandHandler.built_register[key] = AppCommand(
                        name=cmd.group.name,
                        description=cmd.group.description,
                        type=AppCommandOptionType.SUB_COMMAND,
                        guild_id=cmd.app.guild_id,
                        options=[]
                    )

                # No checking has to be done before appending `cmd` since it is the
                # lowest level.
                ChatCommandHandler.built_register[key].options.append(
                    AppCommandOption(
                        name=cmd.app.name,
                        description=cmd.app.description,
                        type=AppCommandType.CHAT_INPUT,
                        options=cmd.app.options
                    )
                )

                continue

            # All single-level commands are registered here.
            ChatCommandHandler.built_register[
                _hash_app_command(cmd.app, cmd.group, cmd.sub_group)
            ] = cmd.app

    @staticmethod
    def get_local_registered_commands() -> ValuesView[AppCommand]:

        return ChatCommandHandler.built_register.values()

    async def __get_existing_commands(self):
        """|coro|

        Get AppCommand objects for all commands registered to discord.
        """
        try:
            self._api_commands = await self.get_commands()

        except ForbiddenError:
            logging.error("Cannot retrieve slash commands, skipping...")
            return

    async def __remove_unused_commands(self):
        """|coro|

        Remove commands that are registered by discord but not in use
        by the current client
        """
        local_registered_commands = self.get_local_registered_commands()

        def should_be_removed(target: AppCommand) -> bool:
            for reg_cmd in local_registered_commands:
                # Commands have endpoints based on their `name` amd `guild_id`. Other
                # parameters can be updated instead of deleting and re-registering the
                # command.
                if (
                    target.name == reg_cmd.name

                    and target.guild_id == reg_cmd.guild_id

                ):
                    return False
            return True

        # NOTE: Cannot be generator since it can't be consumed due to lines 743-745
        to_remove = [*filter(should_be_removed, self._api_commands)]

        await gather(
            *(self.remove_command(cmd) for cmd in to_remove)
        )

        self._api_commands = [
            cmd for cmd in self._api_commands
            if cmd not in to_remove
        ]

    async def __add_commands(self):
        """|coro|

        Add all new commands which have been registered by the decorator to Discord.

        .. code-block::

            Because commands have unique names within a type and scope, we treat POST
            requests for new commands as upserts. That means making a new command with
            an already-used name for your application will update the existing command.
            `<https://discord.dev/interactions/application-commands#updating-and-deleting-a-command>`_


        Therefore, we don't need to use a separate loop for updating and adding
        commands.
        """
        local_registered_commands = self.get_local_registered_commands()

        def should_be_updated_or_uploaded(target):
            for command in self._api_commands:
                if target == command:
                    return False
            return True

        changed_commands = filter(
            should_be_updated_or_uploaded, local_registered_commands
        )

        for command in changed_commands:
            await self.add_command(command)

    async def initialize(self):
        """|coro|

        Call methods of this class to refresh all app commands
        """
        if ChatCommandHandler.has_been_initialized:
            # Only first shard should be initialized.
            return

        ChatCommandHandler.has_been_initialized = True

        self.__build_local_commands()
        await self.__get_existing_commands()
        await self.__remove_unused_commands()
        await self.__add_commands()


def _hash_app_command(
    command: AppCommand,

    group: Optional[str],

    sub_group: Optional[str]

) -> int:
    """
    See :func:`~pincer.commands.commands._hash_app_command_params` for information.
    """
    return _hash_app_command_params(
        command.name,
        command.guild_id,
        command.type,
        group,
        sub_group
    )


def _hash_app_command_params(
    name: str,

    guild_id: Union[Snowflake, None, MISSING],

    app_command_type: AppCommandType,

    group: Optional[str],

    sub_group: Optional[str]

) -> int:
    """
    The group layout in Pincer is very different from what discord has on their docs.
    You can think of the Pincer group layout like this:

    name: The name of the function that is being called.

    group: The :class:`~pincer.commands.groups.Group` object that this function is
        using.
    sub_option: The :class:`~pincer.commands.groups.Subgroup` object that this
        functions is using.

    Abstracting away this part of the Discord API allows for a much cleaner
    transformation between what users want to input and what commands Discord
    expects.

    Parameters
    ----------
    name : str
        The name of the function for the command
    guild_id : Union[:class:`~pincer.utils.snowflake.Snowflake`, None, MISSING]
        The ID of a guild, None, or MISSING.
    app_command_type : :class:`~pincer.objects.app.command_types.AppCommandType`
        The app command type of the command. NOT THE OPTION TYPE.
    group : str
        The highest level of organization the command is it. This should always be the
        name of the base command. :data:`None` or :data:`MISSING` if not there.
    sub_option : str
        The name of the group that holds the lowest level of options. :data:`None` or
        :data:`MISSING` if not there.
    """
    return hash((name, guild_id, app_command_type, group, sub_group))


1			# Copyright Pincer 2021-Present
			0 ignored issues – show introduced 2022-01-16 12:07 UTC by Report Bug Copy Issue Report Missing module docstring Loading history...
2			# Full MIT License can be found in `LICENSE` at the project root.
3
4			from __future__ import annotations
5
6			from typing import TYPE_CHECKING
7
8			import logging
9			from asyncio import gather
10
11			from ..utils.types import MISSING, Singleton
12
13			from ..exceptions import ForbiddenError
14			from ..objects.guild.guild import Guild
15			from ..objects.app.command import AppCommand, AppCommandOption
16			from ..objects.app.command_types import AppCommandOptionType, AppCommandType
17
18			if TYPE_CHECKING:
19			from typing import List, Dict, Optional, ValuesView, Union
			0 ignored issues – show introduced 2022-01-16 12:07 UTC by Report Bug Copy Issue Report Imports from package typing are not grouped Loading history...
20			from .interactable import Interactable
21			from ..client import Client
22			from ..utils.snowflake import Snowflake
23			from ..objects.app.command import InteractableStructure
24
25			_log = logging.getLogger(__name__)
26
27
28			class ChatCommandHandler(metaclass=Singleton):
29			"""Singleton containing methods used to handle various commands
30
31			The register and built_register
32			-------------------------------
33			I found the way Discord expects commands to be registered to be very different than
34			how you want to think about command registration. i.e. Discord wants nesting but we
35			don't want any nesting. Nesting makes it hard to think about commands and also will
36			increase lookup time.
37			The way this problem is avoided is by storing a version of the commands that we can
38			deal with as library developers and a version of the command that Discord thinks we
39			should provide. That is where the register and the built_register help simplify the
40			design of the library.
41			The register is simply where the "Pincer version" of commands gets saved to memory.
42			The built_register is where the version of commands that Discord requires is saved.
43			The register allows for O(1) lookups by storing commands in a Python dictionary. It
44			does cost some memory to save two copies in the current iteration of the system but
45			we should be able to drop the built_register in runtime if we want to. I don't feel
46			that lost maintainability from this is optimal. We can index by in O(1) by checking
47			the register but can still use the built_register if we need to do a nested lookup.
48
49			Attributes
50			----------
51			client: :class:`Client`
52			The client object
53			managers: Dict[:class:`str`, :class:`~typing.Any`]
54			Dictionary of managers
55			register: Dict[:class:`str`, :class:`~pincer.objects.app.command.InteractableStructure`]
56			Dictionary of ``InteractableStructure``
57			built_register: Dict[:class:`str`, :class:`~pincer.objects.app.command.InteractableStructure`]
58			Dictionary of ``InteractableStructure`` where the commands are converted to
59			the format that Discord expects for sub commands and sub command groups.
60			""" # noqa: E501
61
62			has_been_initialized = False
63			managers: List[Interactable] = []
64			register: Dict[str, InteractableStructure] = {}
65			built_register: Dict[str, AppCommand] = {}
66
67			# Endpoints:
68			__get = "/commands"
69			__delete = "/commands/{command.id}"
70			__update = "/commands/{command.id}"
71			__add = "/commands"
72			__add_guild = "/guilds/{command.guild_id}/commands"
73			__get_guild = "/guilds/{guild_id}/commands"
74			__update_guild = "/guilds/{command.guild_id}/commands/{command.id}"
75			__delete_guild = "/guilds/{command.guild_id}/commands/{command.id}"
76
77			def __init__(self, client: Client):
78			self.client = client
79			self._api_commands: List[AppCommand] = []
80			_log.debug(
81			"%i commands registered.", len(ChatCommandHandler.register.items())
82			)
83
84			self.__prefix = f"applications/{self.client.bot.id}"
85
86			async def get_commands(self) -> List[AppCommand]:
87			"""\|coro\|
88
89			Get a list of app commands from Discord
90
91			Returns
92			-------
93			List[:class:`~pincer.objects.app.command.AppCommand`]
94			List of commands.
95			"""
96			# TODO: Update if discord adds bulk get guild commands
			0 ignored issues – show Coding Style introduced 2021-11-24 09:33 UTC by Report Bug Copy Issue Report `TODO` and `FIXME` comments should generally be avoided. Loading history...
97			guild_commands = await gather(*(
98			self.client.http.get(
99			self.__prefix + self.__get_guild.format(
100			guild_id=guild.id if isinstance(guild, Guild) else guild
101			)
102			) for guild in self.client.guilds
103			))
104			return list(
105			map(
106			AppCommand.from_dict,
107			await self.client.http.get(self.__prefix + self.__get)
108			+ [cmd for guild in guild_commands for cmd in guild],
109			)
110			)
111
112			async def remove_command(self, cmd: AppCommand):
113			"""\|coro\|
114
115			Remove a specific command
116
117			Parameters
118			----------
119			cmd : :class:`~pincer.objects.app.command.AppCommand`
120			What command to delete
121			"""
122			# TODO: Update if discord adds bulk delete commands
			0 ignored issues – show Coding Style introduced 2021-11-24 09:33 UTC by Report Bug Copy Issue Report `TODO` and `FIXME` comments should generally be avoided. Loading history...
123			if cmd.guild_id:
124			_log.info(
125			"Removing command `%s` with guild id %d from Discord",
126			cmd.name,
127			cmd.guild_id,
128			)
129			else:
130			_log.info("Removing global command `%s` from Discord", cmd.name)
131
132			remove_endpoint = self.__delete_guild if cmd.guild_id else self.__delete
133
134			await self.client.http.delete(
135			self.__prefix + remove_endpoint.format(command=cmd)
136			)
137
138			async def add_command(self, cmd: AppCommand):
139			"""\|coro\|
140
141			Add an app command
142
143			Parameters
144			----------
145			cmd : :class:`~pincer.objects.app.command.AppCommand`
146			Command to add
147			"""
148			_log.info("Updated or registered command `%s` to Discord", cmd.name)
149
150			add_endpoint = self.__add
151
152			if cmd.guild_id:
153			add_endpoint = self.__add_guild.format(command=cmd)
154
155			await self.client.http.post(
156			self.__prefix + add_endpoint, data=cmd.to_dict()
157			)
158
159			async def add_commands(self, commands: List[AppCommand]):
160			"""\|coro\|
161
162			Add a list of app commands
163
164			Parameters
165			----------
166			commands : List[:class:`~pincer.objects.app.command.AppCommand`]
167			List of command objects to add
168			"""
169			await gather(*map(self.add_command, commands))
170
171			@staticmethod
172			def __build_local_commands():
173			"""Builds the commands into the format that Discord expects. See class info
174			for the reasoning.
175			"""
176
177			# Reset the built register
178			ChatCommandHandler.built_register = {}
179
180			for cmd in ChatCommandHandler.register.values():
181
182			if cmd.sub_group:
183			# If a command has a sub_group, it must be nested 2 levels deep.
184			#
185			# command
186			# subcommand-group
187			# subcommand
188			#
189			# The children of the subcommand-group object are being set to include
190			# `cmd` If that subcommand-group object does not exist, it will be
191			# created here. The same goes for the top-level command.
192			#
193			# First make sure the command exists. This command will hold the
194			# subcommand-group for `cmd`.
195
196			# `key` represents the hash value for the top-level command that will
197			# hold the subcommand.
198			key = _hash_app_command_params(
199			cmd.group.name,
200			cmd.app.guild_id,
201			AppCommandType.CHAT_INPUT,
202			None,
203			None,
204			)
205
206			if key not in ChatCommandHandler.built_register:
207			ChatCommandHandler.built_register[key] = AppCommand(
208			name=cmd.group.name,
209			description=cmd.group.description,
210			type=AppCommandType.CHAT_INPUT,
211			guild_id=cmd.app.guild_id,
212			options=[]
213			)
214
215			# The top-level command now exists. A subcommand group now if placed
216			# inside the top-level command. This subcommand group will hold `cmd`.
217
218			children = ChatCommandHandler.built_register[key].options
219
220			sub_command_group = AppCommandOption(
221			name=cmd.sub_group.name,
222			description=cmd.sub_group.description,
223			type=AppCommandOptionType.SUB_COMMAND_GROUP,
224			options=[]
225			)
226
227			# This for-else makes sure that sub_command_group will hold a reference
228			# to the subcommand group that we want to modify to hold `cmd`
229
230			for cmd_in_children in children:
231			if (
232			cmd_in_children.name == sub_command_group.name
			0 ignored issues – show Coding Style introduced 2021-11-24 09:33 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
233			and cmd_in_children.description == sub_command_group.description
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
234			and cmd_in_children.type == sub_command_group.type
			0 ignored issues – show Coding Style introduced 2022-01-13 22:22 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
235			):
236			sub_command_group = cmd_in_children
237			break
238			else:
239			children.append(sub_command_group)
240
241			sub_command_group.options.append(AppCommandOption(
242			name=cmd.app.name,
243			description=cmd.app.description,
244			type=AppCommandOptionType.SUB_COMMAND,
245			options=cmd.app.options,
246			))
247
248			continue
249
250			if cmd.group:
251			# Any command at this point will only have one level of nesting.
252			#
253			# Command
254			# subcommand
255			#
256			# A subcommand object is what is being generated here. If there is no
257			# top level command, it will be created here.
258
259			# `key` represents the hash value for the top-level command that will
260			# hold the subcommand.
261
262			key = _hash_app_command_params(
263			cmd.group.name,
264			cmd.app.guild_id,
265			AppCommandOptionType.SUB_COMMAND,
266			None,
267			None
268			)
269
270			if key not in ChatCommandHandler.built_register:
271			ChatCommandHandler.built_register[key] = AppCommand(
272			name=cmd.group.name,
273			description=cmd.group.description,
274			type=AppCommandOptionType.SUB_COMMAND,
275			guild_id=cmd.app.guild_id,
276			options=[]
277			)
278
279			# No checking has to be done before appending `cmd` since it is the
280			# lowest level.
281			ChatCommandHandler.built_register[key].options.append(
282			AppCommandOption(
283			name=cmd.app.name,
284			description=cmd.app.description,
285			type=AppCommandType.CHAT_INPUT,
286			options=cmd.app.options
287			)
288			)
289
290			continue
291
292			# All single-level commands are registered here.
293			ChatCommandHandler.built_register[
294			_hash_app_command(cmd.app, cmd.group, cmd.sub_group)
295			] = cmd.app
296
297			@staticmethod
298			def get_local_registered_commands() -> ValuesView[AppCommand]:
			0 ignored issues – show introduced 2021-12-18 18:37 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
299			return ChatCommandHandler.built_register.values()
300
301			async def __get_existing_commands(self):
302			"""\|coro\|
303
304			Get AppCommand objects for all commands registered to discord.
305			"""
306			try:
307			self._api_commands = await self.get_commands()
308
309			except ForbiddenError:
310			logging.error("Cannot retrieve slash commands, skipping...")
311			return
312
313			async def __remove_unused_commands(self):
314			"""\|coro\|
315
316			Remove commands that are registered by discord but not in use
317			by the current client
318			"""
319			local_registered_commands = self.get_local_registered_commands()
320
321			def should_be_removed(target: AppCommand) -> bool:
322			for reg_cmd in local_registered_commands:
323			# Commands have endpoints based on their `name` amd `guild_id`. Other
324			# parameters can be updated instead of deleting and re-registering the
325			# command.
326			if (
327			target.name == reg_cmd.name
			0 ignored issues – show Coding Style introduced 2022-01-16 19:09 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
328			and target.guild_id == reg_cmd.guild_id
			0 ignored issues – show Coding Style introduced 2022-01-16 19:09 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
329			):
330			return False
331			return True
332
333			# NOTE: Cannot be generator since it can't be consumed due to lines 743-745
334			to_remove = [*filter(should_be_removed, self._api_commands)]
335
336			await gather(
337			*(self.remove_command(cmd) for cmd in to_remove)
338			)
339
340			self._api_commands = [
341			cmd for cmd in self._api_commands
342			if cmd not in to_remove
343			]
344
345			async def __add_commands(self):
346			"""\|coro\|
347
348			Add all new commands which have been registered by the decorator to Discord.
349
350			.. code-block::
351
352			Because commands have unique names within a type and scope, we treat POST
353			requests for new commands as upserts. That means making a new command with
354			an already-used name for your application will update the existing command.
355			`<https://discord.dev/interactions/application-commands#updating-and-deleting-a-command>`_
			0 ignored issues – show Coding Style introduced 2021-12-20 17:00 UTC by Report Bug Copy Issue Report This line is too long as per the coding-style (102/100). This check looks for lines that are too long. You can specify the maximum line length. Loading history...
356
357			Therefore, we don't need to use a separate loop for updating and adding
358			commands.
359			"""
360			local_registered_commands = self.get_local_registered_commands()
361
362			def should_be_updated_or_uploaded(target):
363			for command in self._api_commands:
364			if target == command:
365			return False
366			return True
367
368			changed_commands = filter(
369			should_be_updated_or_uploaded, local_registered_commands
370			)
371
372			for command in changed_commands:
373			await self.add_command(command)
374
375			async def initialize(self):
376			"""\|coro\|
377
378			Call methods of this class to refresh all app commands
379			"""
380			if ChatCommandHandler.has_been_initialized:
381			# Only first shard should be initialized.
382			return
383
384			ChatCommandHandler.has_been_initialized = True
385
386			self.__build_local_commands()
387			await self.__get_existing_commands()
388			await self.__remove_unused_commands()
389			await self.__add_commands()
390
391
392			def _hash_app_command(
393			command: AppCommand,
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
394			group: Optional[str],
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
395			sub_group: Optional[str]
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
396			) -> int:
397			"""
398			See :func:`~pincer.commands.commands._hash_app_command_params` for information.
399			"""
400			return _hash_app_command_params(
401			command.name,
402			command.guild_id,
403			command.type,
404			group,
405			sub_group
406			)
407
408
409			def _hash_app_command_params(
410			name: str,
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
411			guild_id: Union[Snowflake, None, MISSING],
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
412			app_command_type: AppCommandType,
			0 ignored issues – show Coding Style introduced 2022-01-02 01:15 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
413			group: Optional[str],
			0 ignored issues – show Coding Style introduced 2021-11-24 09:33 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
414			sub_group: Optional[str]
			0 ignored issues – show Coding Style introduced 2022-01-16 12:07 UTC by Report Bug Copy Issue Report Wrong hanging indentation before block (add 4 spaces). Loading history...
415			) -> int:
416			"""
417			The group layout in Pincer is very different from what discord has on their docs.
418			You can think of the Pincer group layout like this:
419
420			name: The name of the function that is being called.
421
422			group: The :class:`~pincer.commands.groups.Group` object that this function is
423			using.
424			sub_option: The :class:`~pincer.commands.groups.Subgroup` object that this
425			functions is using.
426
427			Abstracting away this part of the Discord API allows for a much cleaner
428			transformation between what users want to input and what commands Discord
429			expects.
430
431			Parameters
432			----------
433			name : str
434			The name of the function for the command
435			guild_id : Union[:class:`~pincer.utils.snowflake.Snowflake`, None, MISSING]
436			The ID of a guild, None, or MISSING.
437			app_command_type : :class:`~pincer.objects.app.command_types.AppCommandType`
438			The app command type of the command. NOT THE OPTION TYPE.
439			group : str
440			The highest level of organization the command is it. This should always be the
441			name of the base command. :data:`None` or :data:`MISSING` if not there.
442			sub_option : str
443			The name of the group that holds the lowest level of options. :data:`None` or
444			:data:`MISSING` if not there.
445			"""
446			return hash((name, guild_id, app_command_type, group, sub_group))
447

Pincer-org / Pincer

Pull Request — main (#389)

ChatCommandHandler.__build_local_commands() C

Complexity

Size

Duplication

Importance

How to fix Long Method Complexity

Long Method

Complexity

Duplication Side-by-Side

Filter issues like