-
Notifications
You must be signed in to change notification settings - Fork 0
/
events.py
55 lines (40 loc) · 1.79 KB
/
events.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from typing import Set, List, Union
from discord import Member, AuditLogAction, Role, Object
from discord.ext.commands import Cog
class RoleHandler(Cog):
"""
Class used to handle persisting roles when members join
"""
@Cog.listener()
async def on_member_join(self, member: Member):
"""
Called when a Member joins or leaves a Guild
Will add roles that were never added or removed
:param member: the member who joined
"""
roles_not_removed = set()
top_role = member.guild.me.top_role
async for entry in member.guild.audit_logs(action=AuditLogAction.member_role_update, oldest_first=True):
if entry.target == member:
before_roles = self.filter_roles(top_role, entry.changes.before.roles)
after_roles = self.filter_roles(top_role, entry.changes.after.roles)
roles_added = after_roles - before_roles
roles_removed = before_roles - after_roles
roles_not_removed.update(roles_added)
roles_not_removed.difference_update(roles_removed)
await member.add_roles(*roles_not_removed, reason='Persisted roles')
@staticmethod
def filter_roles(top_role: Role, roles: List[Union[Role, Object]]) -> Set[Role]:
"""
Filters a list of possible roles on whether they still exist
and whether they are not above a top role
Will return a set
:param top_role: the top role
:param roles: list of possible roles, each element is either a Role or Object
:return: the filtered set of roles
"""
filtered_roles = set()
for role in roles:
if isinstance(role, Role) and role < top_role:
filtered_roles.add(role)
return filtered_roles