Add support for configuring allowed mentions per message or bot wide.

This commit is contained in:
Rapptz
2020-04-04 07:40:51 -04:00
parent 6c764465c0
commit 041785937e
8 changed files with 152 additions and 7 deletions

View File

@ -39,6 +39,7 @@ from .guild import Guild
from .activity import BaseActivity
from .user import User, ClientUser
from .emoji import Emoji
from .mentions import AllowedMentions
from .partial_emoji import PartialEmoji
from .message import Message
from .relationship import Relationship
@ -78,6 +79,12 @@ class ConnectionState:
self._fetch_offline = options.get('fetch_offline_members', True)
self.heartbeat_timeout = options.get('heartbeat_timeout', 60.0)
self.guild_subscriptions = options.get('guild_subscriptions', True)
mentions = options.get('mentions')
if mentions is not None and not isinstance(mentions, AllowedMentions):
raise TypeError('mentions parameter must be AllowedMentions')
self.mentions = mentions
# Only disable cache if both fetch_offline and guild_subscriptions are off.
self._cache_members = (self._fetch_offline or self.guild_subscriptions)
self._listeners = []