From 9075d46efdc9d1b430558d7d4d67816a26d04c28 Mon Sep 17 00:00:00 2001 From: Rapptz Date: Thu, 17 Dec 2015 18:22:16 -0500 Subject: Add on_member_ban and on_member_unban events. Fixes #58. --- discord/client.py | 3 ++- discord/state.py | 20 ++++++++++++++++++++ 2 files changed, 22 insertions(+), 1 deletion(-) (limited to 'discord') 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', {}) -- cgit v1.2.3