diff options
Diffstat (limited to 'discord')
| -rw-r--r-- | discord/client.py | 3 | ||||
| -rw-r--r-- | discord/state.py | 20 |
2 files changed, 22 insertions, 1 deletions
diff --git a/discord/client.py b/discord/client.py index 9483c5f9..9e5f3c64 100644 --- a/discord/client.py +++ b/discord/client.py @@ -344,7 +344,8 @@ class Client: 'GUILD_MEMBER_ADD', 'GUILD_MEMBER_REMOVE', 'GUILD_UPDATE' 'GUILD_MEMBER_UPDATE', 'GUILD_CREATE', 'GUILD_DELETE', 'GUILD_ROLE_CREATE', 'GUILD_ROLE_DELETE', 'TYPING_START', - 'GUILD_ROLE_UPDATE', 'VOICE_STATE_UPDATE'): + 'GUILD_ROLE_UPDATE', 'VOICE_STATE_UPDATE', + 'GUILD_BAN_ADD', 'GUILD_BAN_REMOVE'): parser = 'parse_' + event.lower() getattr(self.connection, parser)(data) else: diff --git a/discord/state.py b/discord/state.py index a200fc83..a4ba1526 100644 --- a/discord/state.py +++ b/discord/state.py @@ -251,6 +251,26 @@ class ConnectionState: else: self.dispatch('server_remove', server) + def parse_guild_ban_add(self, data): + # we make the assumption that GUILD_BAN_ADD is done + # before GUILD_MEMBER_REMOVE is called + # hence we don't remove it from cache or do anything + # strange with it, the main purpose of this event + # is mainly to dispatch to another event worth listening to for logging + server = self._get_server(data.get('guild_id')) + if server is not None: + user_id = data.get('user', {}).get('id') + member = utils.get(server.members, id=user_id) + if member is not None: + self.dispatch('member_ban', member) + + def parse_guild_ban_remove(self, data): + server = self._get_server(data.get('guild_id')) + if server is not None: + if 'user' in data: + user = User(**data['user']) + self.dispatch('member_unban', server, user) + def parse_guild_role_create(self, data): server = self._get_server(data.get('guild_id')) role_data = data.get('role', {}) |