Browse Source

member_count tracking (#165)

* member_count tracking

* Remove unset checks.
staging/v1.0.0
Luke 5 years ago
committed by Andrei Zbikowski
parent
commit
bbddbc4956
  1. 6
      disco/state.py

6
disco/state.py

@ -285,6 +285,10 @@ class State(object):
if event.member.guild_id not in self.guilds: if event.member.guild_id not in self.guilds:
return return
# Avoid adding duplicate events to member_count.
if event.member.id not in self.guilds[event.member.guild_id].members:
self.guilds[event.member.guild_id].member_count += 1
self.guilds[event.member.guild_id].members[event.member.id] = event.member self.guilds[event.member.guild_id].members[event.member.id] = event.member
def on_guild_member_update(self, event): def on_guild_member_update(self, event):
@ -303,6 +307,8 @@ class State(object):
if event.user.id not in self.guilds[event.guild_id].members: if event.user.id not in self.guilds[event.guild_id].members:
return return
self.guilds[event.guild_id].member_count -= 1
del self.guilds[event.guild_id].members[event.user.id] del self.guilds[event.guild_id].members[event.user.id]
def on_guild_members_chunk(self, event): def on_guild_members_chunk(self, event):

Loading…
Cancel
Save