matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Fix bug that made member roles go missing.

Browse Source
This commit is contained in:
Rapptz
2017-01-03 20:15:56 -05:00
parent 3f06cf9e81
commit c863d93f69
3 changed files with 22 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
discord/member.py
View File

@ -145,12 +145,12 @@ class Member(discord.abc.Messageable):
def __init__(self, *, data, guild, state):
self._state = state
self._user = state.store_user(data['user'])
self.guild = guild
self.joined_at = utils.parse_time(data.get('joined_at'))
self.roles = data.get('roles', [])
self._update_roles(data)
self.status = Status.offline
game = data.get('game', {})
self.game = Game(**game) if game else None
self.guild = guild
self.nick = data.get('nick', None)
def __str__(self):
@ -177,6 +177,22 @@ class Member(discord.abc.Messageable):
def _get_guild_id(self):
return None
def _update_roles(self, data):
# update the roles
self.roles = [self.guild.default_role]
for role in self.guild.roles:
if role.id in data['roles']:
self.roles.append(role)
self.roles = [self.guild.default_role]
for roleid in map(int, data['roles']):
role = utils.find(lambda r: r.id == roleid, self.guild.roles)
if role is not None:
self.roles.append(role)
# sort the roles by ID since they can be "randomised"
self.roles.sort(key=lambda r: r.id)
def _update(self, data, user=None):
if user:
self._user.name = user['username']
@ -191,14 +207,7 @@ class Member(discord.abc.Messageable):
except KeyError:
pass
# update the roles
self.roles = [self.guild.default_role]
for role in self.guild.roles:
if role.id in data['roles']:
self.roles.append(role)
# sort the roles by ID since they can be "randomised"
self.roles.sort(key=lambda r: int(r.id))
self._update_roles(data)
def _presence_update(self, data, user):
self.status = try_enum(Status, data['status'])