From 3436792614d6478e25e43a717c4f393a16dd43bb Mon Sep 17 00:00:00 2001
From: Rapptz <rapptz@gmail.com>
Date: Sat, 20 May 2017 15:19:47 -0400
Subject: Allow setting a presence upon logging in.

---
 discord/gateway.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'discord/gateway.py')

diff --git a/discord/gateway.py b/discord/gateway.py
index 56d26c02..8cc13f04 100644
--- a/discord/gateway.py
+++ b/discord/gateway.py
@@ -275,6 +275,15 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
         if self.shard_id is not None and self.shard_count is not None:
             payload['d']['shard'] = [self.shard_id, self.shard_count]
 
+        state = self._connection
+        if state._game is not None or state._status is not None:
+            payload['d']['presence'] = {
+                'status': state._status,
+                'game': state._game,
+                'since': 0,
+                'afk': False
+            }
+
         yield from self.send_as_json(payload)
         log.info('Shard ID %s has sent the IDENTIFY payload.', self.shard_id)
 
-- 
cgit v1.2.3