Skip to content

Commit e94ed9f

Browse files
feat: add user & member props to auto mod action (#1358)
* feat: add `user` & `member` props to auto mod action * ci: correct from checks. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 92dfbf4 commit e94ed9f

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

interactions/models/discord/auto_mod.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,7 @@
1616
from interactions.models.discord.snowflake import to_snowflake_list, to_snowflake
1717

1818
if TYPE_CHECKING:
19-
from interactions import (
20-
Snowflake_Type,
21-
Guild,
22-
GuildText,
23-
Message,
24-
Client,
25-
Member,
26-
)
19+
from interactions import Snowflake_Type, Guild, GuildText, Message, Client, Member, User
2720

2821
__all__ = ("AutoModerationAction", "AutoModRule")
2922

@@ -319,6 +312,7 @@ class AutoModerationAction(ClientObject):
319312
_guild_id: "Snowflake_Type" = attrs.field(
320313
repr=False,
321314
)
315+
_user_id: "Snowflake_Type" = attrs.field(repr=False)
322316

323317
@classmethod
324318
def _process_dict(cls, data: dict, client: "Client") -> dict:
@@ -338,6 +332,14 @@ def channel(self) -> "Optional[GuildText]":
338332
def message(self) -> "Optional[Message]":
339333
return self._client.cache.get_message(self._channel_id, self._message_id)
340334

335+
@property
336+
def user(self) -> "User":
337+
return self._client.cache.get_user(self._user_id)
338+
339+
@property
340+
def member(self) -> "Optional[Member]":
341+
return self._client.cache.get_member(self._guild_id, self._user_id)
342+
341343

342344
ACTION_MAPPING = {
343345
AutoModAction.BLOCK_MESSAGE: BlockMessage,

0 commit comments

Comments
 (0)