matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Fix typing for private channel by user lookup

Browse Source
This commit is contained in:
Rapptz 2021-08-22 07:36:39 -04:00
parent 4a6670c062
commit d17551f51f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
discord/state.py
View File

@ -261,7 +261,7 @@ class ConnectionState:
# LRU of max size 128 # LRU of max size 128
self._private_channels: OrderedDict[int, PrivateChannel] = OrderedDict() self._private_channels: OrderedDict[int, PrivateChannel] = OrderedDict()
# extra dict to look up private channels by user id # extra dict to look up private channels by user id
self._private_channels_by_user: Dict[int, PrivateChannel] = {} self._private_channels_by_user: Dict[int, DMChannel] = {}
if self.max_messages is not None: if self.max_messages is not None:
self._messages: Optional[Deque[Message]] = deque(maxlen=self.max_messages) self._messages: Optional[Deque[Message]] = deque(maxlen=self.max_messages)
else: else:
@ -421,7 +421,7 @@ class ConnectionState:
self._private_channels.move_to_end(channel_id) # type: ignore self._private_channels.move_to_end(channel_id) # type: ignore
return value return value
def _get_private_channel_by_user(self, user_id: Optional[int]) -> Optional[PrivateChannel]: def _get_private_channel_by_user(self, user_id: Optional[int]) -> Optional[DMChannel]:
# the keys of self._private_channels are ints # the keys of self._private_channels are ints
return self._private_channels_by_user.get(user_id) # type: ignore return self._private_channels_by_user.get(user_id) # type: ignore