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

Fix crashing for now when a category is created.

Browse Source 
A full implementation will come later.
This commit is contained in:
Rapptz 2017-09-08 21:28:45 -04:00
parent 8912625428
commit 1367877d36
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
discord/guild.py
View File

@ -249,10 +249,11 @@ class Guild(Hashable):
for c in channels: for c in channels:
if c['type'] == ChannelType.text.value: if c['type'] == ChannelType.text.value:
channel = TextChannel(guild=self, data=c, state=self._state) channel = TextChannel(guild=self, data=c, state=self._state)
else: self._add_channel(channel)
elif c['type'] == ChannelType.voice.value:
channel = VoiceChannel(guild=self, data=c, state=self._state) channel = VoiceChannel(guild=self, data=c, state=self._state)
self._add_channel(channel)
self._add_channel(channel)
@property @property
def channels(self): def channels(self):

4
discord/state.py
View File

@ -496,6 +496,10 @@ class ConnectionState:
guild_id = utils._get_as_snowflake(data, 'guild_id') guild_id = utils._get_as_snowflake(data, 'guild_id')
guild = self._get_guild(guild_id) guild = self._get_guild(guild_id)
if guild is not None: if guild is not None:
if factory is None:
log.warning('CHANNEL_CREATE referencing an unknown channel type %s. Discarding.', data['type'])
return
channel = factory(guild=guild, state=self, data=data) channel = factory(guild=guild, state=self, data=data)
guild._add_channel(channel) guild._add_channel(channel)
self.dispatch('guild_channel_create', channel) self.dispatch('guild_channel_create', channel)