mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-07-01 07:40:07 +00:00
Code Issues Projects Releases Wiki Activity

Add overloads for AutoModRuleAction.__init__

Browse Source
This commit is contained in:
Puncher 2023-03-14 12:34:43 +01:00 committed by GitHub
parent 8ba830eeb8
commit 879c8b9340
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
discord/automod.py
View File

@ -25,7 +25,7 @@ DEALINGS IN THE SOFTWARE.
from __future__ import annotations from __future__ import annotations
import datetime import datetime
from typing import TYPE_CHECKING, Any, Dict, Optional, List, Sequence, Set, Union, Sequence from typing import TYPE_CHECKING, Any, Dict, Optional, List, Set, Union, Sequence, overload
from .enums import AutoModRuleTriggerType, AutoModRuleActionType, AutoModRuleEventType, try_enum from .enums import AutoModRuleTriggerType, AutoModRuleActionType, AutoModRuleEventType, try_enum
from .flags import AutoModPresets from .flags import AutoModPresets
@ -58,6 +58,9 @@ __all__ = (
class AutoModRuleAction: class AutoModRuleAction:
"""Represents an auto moderation's rule action. """Represents an auto moderation's rule action.
.. note::
Only one of ``channel_id``, ``duration``, or ``custom_message`` can be used.
.. versionadded:: 2.0 .. versionadded:: 2.0
Attributes Attributes
@ -81,6 +84,18 @@ class AutoModRuleAction:
__slots__ = ('type', 'channel_id', 'duration', 'custom_message') __slots__ = ('type', 'channel_id', 'duration', 'custom_message')
@overload
def __init__(self, *, channel_id: Optional[int] = ...) -> None:
...
@overload
def __init__(self, *, duration: Optional[datetime.timedelta] = ...) -> None:
...
@overload
def __init__(self, *, custom_message: Optional[str] = ...) -> None:
...
def __init__( def __init__(
self, self,
*, *,
@ -95,19 +110,17 @@ class AutoModRuleAction:
if sum(v is None for v in (channel_id, duration, custom_message)) < 2: if sum(v is None for v in (channel_id, duration, custom_message)) < 2:
raise ValueError('Only one of channel_id, duration, or custom_message can be passed.') raise ValueError('Only one of channel_id, duration, or custom_message can be passed.')
self.type: AutoModRuleActionType = AutoModRuleActionType.block_message
if channel_id: if channel_id:
self.type = AutoModRuleActionType.send_alert_message self.type = AutoModRuleActionType.send_alert_message
elif duration: elif duration:
self.type = AutoModRuleActionType.timeout self.type = AutoModRuleActionType.timeout
else:
self.type = AutoModRuleActionType.block_message
def __repr__(self) -> str: def __repr__(self) -> str:
return f'<AutoModRuleAction type={self.type.value} channel={self.channel_id} duration={self.duration}>' return f'<AutoModRuleAction type={self.type.value} channel={self.channel_id} duration={self.duration}>'
@classmethod @classmethod
def from_data(cls, data: AutoModerationActionPayload) -> Self: def from_data(cls, data: AutoModerationActionPayload) -> Self:
type_ = try_enum(AutoModRuleActionType, data['type'])
if data['type'] == AutoModRuleActionType.timeout.value: if data['type'] == AutoModRuleActionType.timeout.value:
duration_seconds = data['metadata']['duration_seconds'] duration_seconds = data['metadata']['duration_seconds']
return cls(duration=datetime.timedelta(seconds=duration_seconds)) return cls(duration=datetime.timedelta(seconds=duration_seconds))