pincer.objects.message.message.Message.__post_init__() - Code Metrics - Inspection of "✨ Guild registration, guild commands fix, command..." - Pincer-org/Pincer - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — main (#158)

unknown

created 2021-10-22 14:23 UTC

Message.__post_init__() C

↳ Parent: pincer.objects.message.message

Complexity

Conditions

Size

Total Lines	26
Code Lines	18

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	18
dl	0
loc	26
rs	6.6666
c	0
b	0
f	0
cc	9
nop	1

# Copyright Pincer 2021-Present

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

from __future__ import annotations

from dataclasses import dataclass
from json import dumps
from typing import Dict, Tuple, Union, List, Optional, TYPE_CHECKING

from aiohttp import FormData, Payload

from ...exceptions import CommandReturnIsEmpty
from ...utils.api_object import APIObject


PILLOW_IMPORT = True

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

if TYPE_CHECKING:
    from ..guild.role import Role
    from ..message.embed import Embed
    from ..message.file import File
    from ..message.user_message import AllowedMentionTypes
    from ..user import User
    from ...utils.snowflake import Snowflake
    from ..app import InteractionFlags
    from .component import MessageComponent


@dataclass

class AllowedMentions(APIObject):
    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
        }


@dataclass

class Message:
    # TODO: Docs for tts, allowed_mentions, components, flags, and type.


    """
    A discord message that will be send to discord

    :param content:
        The text in the message.

    :param attachments:
        Attachments on the message. This is a File object. You can also attach
        a Pillow Image or string. Pillow images will be converted to PNGs. They
        will use the naming sceme ``image%`` where % is the images index in the
        attachments array. Strings will be read as a filepath. The name of the
        file that the string points to will be used as the name.

    :param embeds:
        Embed attached to the message. This is an Embed object.
    """
    content: str = ''
    attachments: Optional[List[File]] = None
    tts: Optional[bool] = False
    embeds: Optional[List[Embed]] = None
    allowed_mentions: Optional[AllowedMentions] = None
    components: Optional[List[MessageComponent]] = None
    flags: Optional[InteractionFlags] = None
    delete_after: Optional[float] = None

    def __post_init__(self):
        if self.delete_after and self.delete_after < 0:
            raise ValueError(
                "Message can not be deleted after a negative amount of "
                "seconds!"
            )

        if not self.attachments:
            return

        attch = []

        for count, value in enumerate(self.attachments):
            if isinstance(value, File):
                attch.append(value)
            elif PILLOW_IMPORT and isinstance(value, Image):
                attch.append(File.from_pillow_image(
                    value,
                    f"image{count}.png",
                ))
            elif isinstance(value, str):
                attch.append(File.from_file(value))
            else:
                raise ValueError(f"Attachment {count} is invalid type.")

        self.attachments = attch

    @property
    def isempty(self) -> bool:
        """
        :return:
            Returns true if a message is empty.
        """

        return (
                len(self.content) < 1

                and not self.embeds

                and not self.attachments

        )

    def to_dict(self):


        allowed_mentions = (
            self.allowed_mentions.to_dict()
            if self.allowed_mentions else {}
        )

        # Attachments aren't serialized
        # because they are not sent as part of the json
        resp = {
            "content": self.content,
            "tts": self.tts,
            "flags": self.flags,
            "embeds": [embed.to_dict() for embed in (self.embeds or [])],
            "allowed_mentions": allowed_mentions,
            "components": [
                components.to_dict() for components in (self.components or [])
            ]
        }

        # IDE does not recognise return type of filter properly.
        # noinspection PyTypeChecker
        return dict(filter(
            lambda kv: kv[1],
            resp.items()
        ))

    def serialize(self) -> Tuple[str, Union[Payload, Dict]]:
        """
        Creates the data that the discord API wants for the message object

        :return: (content_type, data)

        :raises CommandReturnIsEmpty:
            Command does not have content, an embed, or attachment.
        """

        if self.isempty:
            raise CommandReturnIsEmpty("Cannot return empty message.")

        if not self.attachments:
            return "application/json", self.to_dict()

        form = FormData()
        form.add_field("payload_json", dumps(self.to_dict()))

        for file in self.attachments:
            form.add_field("file", file.content, filename=file.filename)

        payload = form()
        return payload.headers["Content-Type"], payload


1			# Copyright Pincer 2021-Present
			0 ignored issues – show introduced 2021-10-03 07:58 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 dataclasses import dataclass
7			from json import dumps
8			from typing import Dict, Tuple, Union, List, Optional, TYPE_CHECKING
9
10			from aiohttp import FormData, Payload
11
12			from ...exceptions import CommandReturnIsEmpty
13			from ...utils.api_object import APIObject
			0 ignored issues – show introduced 2021-10-22 14:25 UTC by Report Bug Copy Issue Report Cannot import 'utils.api_object' due to syntax error 'invalid syntax (<unknown>, line 77)' Loading history... Bug introduced 2021-10-21 08:55 UTC by Report Bug Copy Issue Report The name `api_object` does not seem to exist in module `pincer.utils`. Loading history...
14
15			PILLOW_IMPORT = True
16
17			try:
18			from PIL.Image import Image
19			except (ModuleNotFoundError, ImportError):
20			PILLOW_IMPORT = False
21
22			if TYPE_CHECKING:
23			from ..guild.role import Role
24			from ..message.embed import Embed
25			from ..message.file import File
26			from ..message.user_message import AllowedMentionTypes
27			from ..user import User
28			from ...utils.snowflake import Snowflake
29			from ..app import InteractionFlags
30			from .component import MessageComponent
31
32
33			@dataclass
			0 ignored issues – show introduced 2021-10-03 07:58 UTC by Report Bug Copy Issue Report Missing class docstring Loading history...
34			class AllowedMentions(APIObject):
35			parse: List[AllowedMentionTypes]
36			roles: List[Union[Role, Snowflake]]
37			users: List[Union[User, Snowflake]]
38			reply: bool = True
39
40			def to_dict(self):
			0 ignored issues – show introduced 2021-10-03 07:58 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
41			def get_str_id(obj: Union[Snowflake, User, Role]) -> str:
42			if hasattr(obj, "id"):
43			obj = obj.id
44
45			return str(obj)
46
47			return {
48			"parse": self.parse,
49			"roles": list(map(get_str_id, self.roles)),
50			"users": list(map(get_str_id, self.users)),
51			"replied_user": self.reply
52			}
53
54
55			@dataclass
			0 ignored issues – show best-practice introduced 2021-10-04 01:39 UTC by Report Bug Copy Issue Report Too many instance attributes (8/7) Loading history...
56			class Message:
57			# TODO: Docs for tts, allowed_mentions, components, flags, and type.
			0 ignored issues – show Coding Style introduced 2021-10-05 21:22 UTC by Report Bug Copy Issue Report `TODO` and `FIXME` comments should generally be avoided. Loading history...
58
59			"""
60			A discord message that will be send to discord
61
62			:param content:
63			The text in the message.
64
65			:param attachments:
66			Attachments on the message. This is a File object. You can also attach
67			a Pillow Image or string. Pillow images will be converted to PNGs. They
68			will use the naming sceme ``image%`` where % is the images index in the
69			attachments array. Strings will be read as a filepath. The name of the
70			file that the string points to will be used as the name.
71
72			:param embeds:
73			Embed attached to the message. This is an Embed object.
74			"""
75			content: str = ''
76			attachments: Optional[List[File]] = None
77			tts: Optional[bool] = False
78			embeds: Optional[List[Embed]] = None
79			allowed_mentions: Optional[AllowedMentions] = None
80			components: Optional[List[MessageComponent]] = None
81			flags: Optional[InteractionFlags] = None
82			delete_after: Optional[float] = None
83
84			def __post_init__(self):
85			if self.delete_after and self.delete_after < 0:
86			raise ValueError(
87			"Message can not be deleted after a negative amount of "
88			"seconds!"
89			)
90
91			if not self.attachments:
92			return
93
94			attch = []
95
96			for count, value in enumerate(self.attachments):
97			if isinstance(value, File):
98			attch.append(value)
99			elif PILLOW_IMPORT and isinstance(value, Image):
100			attch.append(File.from_pillow_image(
101			value,
102			f"image{count}.png",
103			))
104			elif isinstance(value, str):
105			attch.append(File.from_file(value))
106			else:
107			raise ValueError(f"Attachment {count} is invalid type.")
108
109			self.attachments = attch
110
111			@property
112			def isempty(self) -> bool:
113			"""
114			:return:
115			Returns true if a message is empty.
116			"""
117
118			return (
119			len(self.content) < 1
			0 ignored issues – show Coding Style introduced 2021-10-07 14:58 UTC by Report Bug Copy Issue Report Wrong hanging indentation (remove 4 spaces). Loading history...
120			and not self.embeds
			0 ignored issues – show Coding Style introduced 2021-10-07 14:58 UTC by Report Bug Copy Issue Report Wrong hanging indentation (remove 4 spaces). Loading history...
121			and not self.attachments
			0 ignored issues – show Coding Style introduced 2021-10-07 14:58 UTC by Report Bug Copy Issue Report Wrong hanging indentation (remove 4 spaces). Loading history...
122			)
123
124			def to_dict(self):
			0 ignored issues – show introduced 2021-10-21 08:55 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
125
126			allowed_mentions = (
127			self.allowed_mentions.to_dict()
128			if self.allowed_mentions else {}
129			)
130
131			# Attachments aren't serialized
132			# because they are not sent as part of the json
133			resp = {
134			"content": self.content,
135			"tts": self.tts,
136			"flags": self.flags,
137			"embeds": [embed.to_dict() for embed in (self.embeds or [])],
138			"allowed_mentions": allowed_mentions,
139			"components": [
140			components.to_dict() for components in (self.components or [])
141			]
142			}
143
144			# IDE does not recognise return type of filter properly.
145			# noinspection PyTypeChecker
146			return dict(filter(
147			lambda kv: kv[1],
148			resp.items()
149			))
150
151			def serialize(self) -> Tuple[str, Union[Payload, Dict]]:
152			"""
153			Creates the data that the discord API wants for the message object
154
155			:return: (content_type, data)
156
157			:raises CommandReturnIsEmpty:
158			Command does not have content, an embed, or attachment.
159			"""
160
161			if self.isempty:
162			raise CommandReturnIsEmpty("Cannot return empty message.")
163
164			if not self.attachments:
165			return "application/json", self.to_dict()
166
167			form = FormData()
168			form.add_field("payload_json", dumps(self.to_dict()))
169
170			for file in self.attachments:
171			form.add_field("file", file.content, filename=file.filename)
172
173			payload = form()
174			return payload.headers["Content-Type"], payload
175

Pincer-org / Pincer

Pull Request — main (#158)

Message.__post_init__() C

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like