Add helper bool magic method for state validity

Reported in #102
This commit is contained in:
Luna 2020-05-01 11:39:21 -03:00
parent 82d793ae07
commit 373b727084
2 changed files with 5 additions and 4 deletions

View File

@ -94,5 +94,9 @@ class GatewayState:
self.large: int = kwargs.get("large") or 50 self.large: int = kwargs.get("large") or 50
def __bool__(self):
"""Return if the given state is a valid state to be used."""
return self.ws is not None
def __repr__(self): def __repr__(self):
return f"GatewayState<seq={self.seq} shard={self.current_shard},{self.shard_count} uid={self.user_id}>" return f"GatewayState<seq={self.seq} shard={self.current_shard},{self.shard_count} uid={self.user_id}>"

View File

@ -593,10 +593,7 @@ class GuildMemberList:
dispatched = [] dispatched = []
for state in states: for state in states:
if state is None: if not state:
continue
if state.ws is None:
continue continue
await state.ws.dispatch("GUILD_MEMBER_LIST_UPDATE", payload) await state.ws.dispatch("GUILD_MEMBER_LIST_UPDATE", payload)