ConnectionState is now constructed in Client.__init__.

This should reduce the amount of checks for None if someone doesn't
want a websocket connection. The connection state is instead cleared
rather than reconstructed.
This commit is contained in:
Rapptz
2015-12-29 11:59:18 -05:00
parent 7696a566e1
commit c11bd9b8f4
2 changed files with 15 additions and 15 deletions

View File

@ -40,13 +40,15 @@ import datetime
class ConnectionState:
def __init__(self, dispatch, max_messages):
self.max_messages = max_messages
self.dispatch = dispatch
self.clear()
def clear(self):
self.user = None
self.email = None
self.servers = []
self.private_channels = []
self.max_messages = max_messages
self.messages = deque(maxlen=self.max_messages)
self.dispatch = dispatch
def _get_message(self, msg_id):
return utils.find(lambda m: m.id == msg_id, self.messages)