From e1c1a72a5007e08216bafdc7b61e4c2bcd50d9da Mon Sep 17 00:00:00 2001 From: z03h <7235242+z03h@users.noreply.github.com> Date: Sat, 21 Oct 2023 01:14:29 -0700 Subject: [PATCH] Fix AutoModRuleAction attribute error --- discord/automod.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/discord/automod.py b/discord/automod.py index 2c952afaf..b20f9016a 100644 --- a/discord/automod.py +++ b/discord/automod.py @@ -135,6 +135,10 @@ class AutoModRuleAction: raise ValueError('Only one of channel_id, duration, or custom_message can be passed.') self.type: AutoModRuleActionType + self.channel_id: Optional[int] = None + self.duration: Optional[datetime.timedelta] = None + self.custom_message: Optional[str] = None + if type is not None: self.type = type elif channel_id is not None: @@ -147,14 +151,15 @@ class AutoModRuleAction: if self.type is AutoModRuleActionType.send_alert_message: if channel_id is None: raise ValueError('channel_id cannot be None if type is send_alert_message') - self.channel_id: Optional[int] = channel_id + self.channel_id = channel_id if self.type is AutoModRuleActionType.timeout: if duration is None: raise ValueError('duration cannot be None set if type is timeout') - self.duration: Optional[datetime.timedelta] = duration + self.duration = duration - self.custom_message: Optional[str] = custom_message + if self.type is AutoModRuleActionType.block_message: + self.custom_message = custom_message def __repr__(self) -> str: return f''