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:
@ -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)
|
||||
|
Reference in New Issue
Block a user