mirror of
https://github.com/Rapptz/discord.py.git
synced 2025-04-20 16:00:29 +00:00
Fallback to an unavailable guild instead of an Object in interactions
This commit is contained in:
parent
781b297ac9
commit
573bb32a1e
@ -168,7 +168,7 @@ class Namespace:
|
||||
state = interaction._state
|
||||
members = resolved.get('members', {})
|
||||
guild_id = interaction.guild_id
|
||||
guild = (state._get_guild(guild_id) or Object(id=guild_id)) if guild_id is not None else None
|
||||
guild = state._get_or_create_unavailable_guild(guild_id) if guild_id is not None else None
|
||||
type = AppCommandOptionType.user.value
|
||||
for (user_id, user_data) in resolved.get('users', {}).items():
|
||||
try:
|
||||
|
@ -452,6 +452,10 @@ class Guild(Hashable):
|
||||
|
||||
return role
|
||||
|
||||
@classmethod
|
||||
def _create_unavailable(cls, *, state: ConnectionState, guild_id: int) -> Guild:
|
||||
return cls(state=state, data={'id': guild_id, 'unavailable': True}) # type: ignore
|
||||
|
||||
def _from_data(self, guild: GuildPayload) -> None:
|
||||
try:
|
||||
self._member_count = guild['member_count']
|
||||
|
@ -38,7 +38,6 @@ from .channel import PartialMessageable, ChannelType
|
||||
from .user import User
|
||||
from .member import Member
|
||||
from .message import Message, Attachment
|
||||
from .object import Object
|
||||
from .permissions import Permissions
|
||||
from .http import handle_message_parameters
|
||||
from .webhook.async_ import async_context, Webhook, interaction_response_params, interaction_message_response_params
|
||||
@ -184,16 +183,14 @@ class Interaction:
|
||||
self.user: Union[User, Member] = MISSING
|
||||
self._permissions: int = 0
|
||||
|
||||
# TODO: there's a potential data loss here
|
||||
if self.guild_id:
|
||||
guild = self.guild or Object(id=self.guild_id)
|
||||
guild = self._state._get_or_create_unavailable_guild(self.guild_id)
|
||||
try:
|
||||
member = data['member'] # type: ignore # The key is optional and handled
|
||||
except KeyError:
|
||||
pass
|
||||
else:
|
||||
# The fallback to Object for guild causes a type check error but is explicitly allowed here
|
||||
self.user = Member(state=self._state, guild=guild, data=member) # type: ignore
|
||||
self.user = Member(state=self._state, guild=guild, data=member)
|
||||
self._permissions = self.user._permissions or 0
|
||||
else:
|
||||
try:
|
||||
|
@ -381,6 +381,9 @@ class ConnectionState:
|
||||
# the keys of self._guilds are ints
|
||||
return self._guilds.get(guild_id) # type: ignore
|
||||
|
||||
def _get_or_create_unavailable_guild(self, guild_id: int) -> Guild:
|
||||
return self._guilds.get(guild_id) or Guild._create_unavailable(state=self, guild_id=guild_id)
|
||||
|
||||
def _add_guild(self, guild: Guild) -> None:
|
||||
self._guilds[guild.id] = guild
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user