Audit log role add and removes are lists, not single item.

Breaking change. role -> roles and it's now a list.
This commit is contained in:
Rapptz
2017-05-06 16:24:14 -04:00
parent d7153b1b5c
commit 7916878eb5
2 changed files with 19 additions and 14 deletions

View File

@ -121,10 +121,10 @@ class AuditLogChanges:
# special cases for role add/remove
if attr == '$add':
self._handle_role(self.before, self.after, entry, elem)
self._handle_role(self.before, self.after, entry, elem['new_value'])
continue
elif attr == '$remove':
self._handle_role(self.after, self.before, entry, elem)
self._handle_role(self.after, self.before, entry, elem['new_value'])
continue
transformer = self.TRANSFORMERS.get(attr)
@ -157,17 +157,22 @@ class AuditLogChanges:
self.before.color = self.before.colour
def _handle_role(self, first, second, entry, elem):
setattr(first, 'role', None)
setattr(first, 'roles', None)
# TODO: partial data?
role_id = int(elem['id'])
role = utils.find(lambda r: r.id == role_id, entry.guild.roles)
data = []
roles = entry.guild.roles
if role is None:
role = discord.Object(id=role_id)
role.name = elem['name']
for e in elem:
role_id = int(e['id'])
role = utils.find(lambda r: r.id == role_id, roles)
setattr(second, 'role', role)
if role is None:
role = discord.Object(id=role_id)
role.name = e['name']
data.append(role)
setattr(second, 'roles', data)
class AuditLogEntry:
"""Represents an Audit Log entry.