Handle TYPING_START event.

This commit is contained in:
Rapptz 2015-11-24 21:11:53 -05:00
parent 2e03927f2c
commit 7f94f75223
2 changed files with 34 additions and 2 deletions

View File

@ -47,7 +47,7 @@ from ws4py.client import WebSocketBaseClient
import sys
import logging
import itertools
import datetime
log = logging.getLogger(__name__)
request_logging_format = '{response.request.method} {response.url} has returned {response.status_code}'
@ -131,7 +131,7 @@ class WebSocket(WebSocketBaseClient):
'CHANNEL_DELETE', 'CHANNEL_UPDATE', 'CHANNEL_CREATE',
'GUILD_MEMBER_ADD', 'GUILD_MEMBER_REMOVE',
'GUILD_MEMBER_UPDATE', 'GUILD_CREATE', 'GUILD_DELETE',
'GUILD_ROLE_CREATE', 'GUILD_ROLE_DELETE',
'GUILD_ROLE_CREATE', 'GUILD_ROLE_DELETE', 'TYPING_START',
'GUILD_ROLE_UPDATE', 'VOICE_STATE_UPDATE'):
self.dispatch('socket_update', event, data)
@ -404,6 +404,26 @@ class ConnectionState(object):
updated_member = self._update_voice_state(server, data)
self.dispatch('voice_state_update', updated_member)
def handle_typing_start(self, data):
channel = self.get_channel(data.get('channel_id'))
if channel is not None:
member = None
user_id = data.get('user_id')
if not getattr(channel, 'is_private', True):
members = channel.server.members
member = utils.find(lambda m: m.id == user_id, members)
else:
# At the moment we can make the assumption that if we are
# in a private channel then the user belongs to one of our
# already existing server member lists.
# This might change when we get friend lists.
gen = (m for s in self.servers for m in s.members)
member = utils.find(lambda m: m.id == user_id, gen)
if member is not None:
timestamp = datetime.datetime.utcfromtimestamp(data.get('timestamp'))
self.dispatch('typing', channel, member, timestamp)
def get_channel(self, id):
if id is None:
return None

View File

@ -246,6 +246,18 @@ to handle it, which defaults to print a traceback and ignore the exception.
:param member: The :class:`Member` whose voice state changed.
.. function:: on_typing(channel, user, when)
Called when someone begins typing a message.
The ``channel`` parameter could either be a :class:`PrivateChannel` or a
:class:`Channel`.
:param channel: The location where the typing originated from.
:param user: The :class:`Member` that started typing.
:param when: A ``datetime.datetime`` object representing when typing started.
Utility Functions
-----------------