pincer.utils.convert_message

Complexity

Total Complexity

6

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
B	convert_message()	0	15	6

from __future__ import annotations

Missing module docstring

from typing import TYPE_CHECKING
from typing import Union

from ..objects.message.embed import Embed
from ..objects.message.file import File
from ..objects.message.message import Message
from ..objects.app.interaction_flags import InteractionFlags

PILLOW_IMPORT = True

try:
    from PIL.Image import Image
except (ModuleNotFoundError, ImportError):
    PILLOW_IMPORT = False

if TYPE_CHECKING:
    from pincer import Client


def convert_message(
        client: Client,
        message: Union[Embed, Message, str]
) -> Message:
    """Converts a message to a Message object"""
    if isinstance(message, Embed):
        message = Message(embeds=[message])
    elif PILLOW_IMPORT and isinstance(message, (File, Image)):
        message = Message(attachments=[message])
    elif not isinstance(message, Message):
        message = Message(message) if message else Message(
            client.received_message,
            flags=InteractionFlags.EPHEMERAL
        )
    return message