Pincer-org / Pincer

    PILLOW_IMPORT = False

@dataclass

class AllowedMentions(APIObject):

    """Represents the entities the client can mention

    Attributes

    ----------

    parse: List[:class:`~pincer.objects.message.user_message.AllowedMentionTypes`]

        An array of allowed mention types to parse from the content.

    roles: List[Union[:class:`~pincer.objects.guild.role.Role`, :class:`~pincer.utils.snowflake.Snowflake`]]

        List of ``Role`` objects or snowflakes of allowed mentions.

    users: List[Union[:class:`~pincer.objects.user.user.User` :class:`~pincer.utils.snowflake.Snowflake`]]

        List of ``user`` objects or snowflakes of allowed mentions.

    reply: :class:`bool`

        If replies should mention the author.

        |default| :data:`True`

    """  # noqa: E501

    parse: List[AllowedMentionTypes]

    roles: List[Union[Role, Snowflake]]

    users: List[Union[User, Snowflake]]

    reply: bool = True

    def to_dict(self):

        def get_str_id(obj: Union[Snowflake, User, Role]) -> str:

            if hasattr(obj, "id"):

                obj = obj.id

            return str(obj)

        return {

            "parse": self.parse,

            "roles": list(map(get_str_id, self.roles)),

            "users": list(map(get_str_id, self.users)),

            "replied_user": self.reply

        }

    EVERYONE = "everyone"

@dataclass

class AllowedMentions(APIObject):

    parse: List[AllowedMentionTypes]

    roles: List[Union[Role, Snowflake]]

    users: List[Union[User, Snowflake]]

    reply: bool = True

    @staticmethod

    def get_str_id(obj: Union[Snowflake, User, Role]) -> str:

        if hasattr(obj, "id"):

            obj = obj.id

        return str(obj)

    def to_dict(self):

        return {

            "parse": self.parse,

            "roles": list(map(self.get_str_id, self.roles)),

            "users": list(map(self.get_str_id, self.users)),

            "replied_user": self.reply

        }