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

Handle CHANNEL_UPDATE for group direct messages.

Browse Source
This commit is contained in:
Rapptz
2016-07-13 17:23:35 -04:00
parent ddd3fd0a3d
commit 69c506d7ae
3 changed files with 27 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
discord/state.py
View File

@ -32,7 +32,7 @@ from .channel import Channel, PrivateChannel
from .member import Member
from .role import Role
from . import utils, compat
from .enums import Status
from .enums import Status, ChannelType, try_enum
from collections import deque, namedtuple
@ -138,11 +138,13 @@ class ConnectionState:
def _add_private_channel(self, channel):
self._private_channels[channel.id] = channel
self._private_channels_by_user[channel.user.id] = channel
if channel.type is ChannelType.private:
self._private_channels_by_user[channel.user.id] = channel
def _remove_private_channel(self, channel):
self._private_channels.pop(channel.id, None)
self._private_channels_by_user.pop(channel.user.id, None)
if channel.type is ChannelType.private:
self._private_channels_by_user.pop(channel.user.id, None)
def _get_message(self, msg_id):
return utils.find(lambda m: m.id == msg_id, self.messages)
@ -289,9 +291,17 @@ class ConnectionState:
self.dispatch('channel_delete', channel)
def parse_channel_update(self, data):
channel_type = try_enum(ChannelType, data.get('type'))
channel_id = data.get('id')
if channel_type is ChannelType.group:
channel = self._get_private_channel(channel_id)
old_channel = copy.copy(channel)
channel._update_group(**data)
self.dispatch('channel_update', old_channel, channel)
return
server = self._get_server(data.get('guild_id'))
if server is not None:
channel_id = data.get('id')
channel = server.get_channel(channel_id)
if channel is not None:
old_channel = copy.copy(channel)