Add on_member_ban and on_member_unban events.

Fixes #58.

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', {})