velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

pubsub.lazy_guild: add more checks for new group being None 

presence updates aren't checking that value yet.
This commit is contained in:
Luna 2018-12-02 18:52:16 -03:00
parent 300f948ab2
commit bb36d7e669
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
litecord/pubsub/lazy_guild.py
View File

@ -384,6 +384,11 @@ class GuildMemberList:
member_id, presence['roles'], presence['status'] member_id, presence['roles'], presence['status']
) )
# skip members that don't have any group assigned.
# (members without read messages)
if group_id is None:
continue
member = await self.storage.get_member_data_one( member = await self.storage.get_member_data_one(
self.guild_id, member_id self.guild_id, member_id
) )
@ -849,6 +854,11 @@ class GuildMemberList:
group_id = await self.get_group_for_member( group_id = await self.get_group_for_member(
user_id, member['roles'], pres['status']) user_id, member['roles'], pres['status'])
if group_id is None:
log.warning('lazy: not adding uid {}, no group',
user_id)
return
self.list.data[group_id].append(user_id) self.list.data[group_id].append(user_id)
await self._sort_groups() await self._sort_groups()
@ -1009,6 +1019,7 @@ class GuildMemberList:
status = partial_presence.get('status', old_presence['status']) status = partial_presence.get('status', old_presence['status'])
# calculate a possible new group # calculate a possible new group
# TODO: handle when new_group is None (member loses perms)
new_group = await self.get_group_for_member( new_group = await self.get_group_for_member(
user_id, roles, status) user_id, roles, status)