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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233 | class PersistentRoles(Extension):
def __init__(self, bot: Client):
self.bot = bot
persistent_roles = SlashCommand(name='persistentroles', description='Manage persistent roles', default_member_permissions=Permissions.ADMINISTRATOR)
@persistent_roles.subcommand(sub_cmd_name='add', sub_cmd_description="Make a role persistent")
@role()
async def persistent_roles_add(self, ctx, role:OptionType.ROLE=None):
"""/persistentroles add
Description:
Add a role to the persistent roles list. This means that when a user leaves and rejoins, they will be given this role again.
Args:
role: Role
"""
# if role is None:
# return await ctx.send(embed=Embed(color=0xDD2222, description=":x: Please provide a role"), ephemeral=True)
if role == ctx.guild.my_role:
return await ctx.send(embed=Embed(color=0xDD2222, description=":x: This is my role, you cannot manage this role"), ephemeral=True)
elif role == ctx.guild.default_role:
return await ctx.send(embed=Embed(color=0xDD2222, description=":x: This is a default role, you cannot manage this role"), ephemeral=True)
ranks = await db.leveling_roles.find_one({'guildid':ctx.guild_id, 'roleid':role.id})
if ranks is not None:
return await ctx.send(embed=Embed(color=0xDD2222, description=f":x: {role.mention} is a rank, which makes it already a persistant role."), ephemeral=True)
pers_roles = await db.persistent_roles_settings.find_one({'guildid':ctx.guild_id, 'roleid':role.id})
if pers_roles is not None:
return await ctx.send(embed=Embed(color=0xDD2222, description=f":x: {role.mention} is already a persistent role."), ephemeral=True)
await db.persistent_roles_settings(guildid=ctx.guild_id, roleid=role.id).insert()
if ctx.author.guild_avatar is not None:
avatarurl = f'{ctx.author.guild_avatar.url}.png'
else:
avatarurl = f'{ctx.author.avatar.url}.png'
embed = Embed(description=f"I have made {role.mention} a persistent role.",
color=0xffcc50)
embed.set_footer(text=f'{ctx.author}|{ctx.author.id}',icon_url=avatarurl)
await ctx.send(embed=embed)
@persistent_roles.subcommand(sub_cmd_name='remove', sub_cmd_description="Remove role from persistent roles")
@role()
async def persistent_roles_remove(self, ctx, role:OptionType.ROLE=None):
"""/persistentroles remove
Description:
Removes a role from the persistent roles list.
Args:
role: Get the role object from the user's input
"""
if role is None:
return await ctx.send(embed=Embed(color=0xDD2222, description=":x: Please provide a role"), ephemeral=True)
elif role == ctx.guild.my_role:
return await ctx.send(embed=Embed(color=0xDD2222, description=":x: This is my role, you cannot manage this role"), ephemeral=True)
elif role == ctx.guild.default_role:
return await ctx.send(embed=Embed(color=0xDD2222, description=":x: This is a default role, you cannot manage this role"), ephemeral=True)
ranks = await db.leveling_roles.find_one({'guildid':ctx.guild_id, 'roleid':role.id})
if ranks.roleid == role.id:
return await ctx.send(embed=Embed(color=0xDD2222, description=f":x: {role.mention} is a rank, which makes it a persistant role that you can't remove from persistent roles."), ephemeral=True)
pers_roles = await db.persistent_roles_settings.find_one({'guildid':ctx.guild_id, 'roleid':role.id})
if pers_roles.roleid != role.id:
return await ctx.send(embed=Embed(color=0xDD2222, description=f":x: {role.mention} is not a persistent role."), ephemeral=True)
await pers_roles.delete()
if ctx.author.guild_avatar is not None:
avatarurl = f'{ctx.author.guild_avatar.url}.png'
else:
avatarurl = f'{ctx.author.avatar.url}.png'
embed = Embed(description=f"I have removed {role.mention} from persistent roles.",
color=0xffcc50)
embed.set_footer(text=f'{ctx.author}|{ctx.author.id}',icon_url=avatarurl)
await ctx.send(embed=embed)
@persistent_roles.subcommand(sub_cmd_name='list', sub_cmd_description="List all the persistent roles")
async def persistent_roles_list(self, ctx):
"""/persistentroles list
Description:
Display a list of all persistent roles for the guild.
"""
from interactions.ext.paginators import Paginator
def chunks(l, n):
n = max(1, n)
return (l[i:i+n] for i in range(0, len(l), n))
def mlis(lst, s, e):
nc = list(chunks(lst, 20))
mc = ''
for l in nc[s:e]:
for m in l:
mc = mc + m
return mc
def newpage(title, roles):
embed = Embed(title=title,
color=0xffcc50)
embed.add_field(name='Roles', value=roles, inline=True)
return embed
leveling_roles = db.leveling_roles.find({"guildid":ctx.guild.id})
level_roles = []
async for lvl in leveling_roles:
level_roles.append(lvl.roleid)
pr = db.persistent_roles_settings.find({'guildid':ctx.guild.id})
pers_roles = []
async for r in pr:
pers_roles.append(r.roleid)
all_persistent_roles = level_roles+pers_roles
if all_persistent_roles == []:
embed = Embed(description=f"There are no persistent roles for {ctx.guild.name}.",
color=0xffcc50)
await ctx.send(embed=embed)
return
roles = []
for r in all_persistent_roles:
role = ctx.guild.get_role(r)
if role is None:
roles.append(f'{r}[ROLE NOT FOUND]\n')
else:
roles.append(f"{role.mention}\n")
s = -1
e = 0
embedcount = 1
nc = list(chunks(roles, 20))
embeds = []
while embedcount <= len(nc):
s = s+1
e = e+1
embeds.append(newpage(f'List of persistent roles for {ctx.guild.name}', mlis(roles, s, e)))
embedcount = embedcount+1
paginator = Paginator(
client=self.bot,
pages=embeds,
timeout_interval=80,
show_select_menu=False)
await paginator.send(ctx)
@listen()
async def on_member_remove(self, event):
guild= event.member.guild
member = event.member
member_roles = [role.id for role in member.roles if role.name != '@everyone']
if member_roles == []:
return
p_r = await db.persistent_roles.find_one({'guildid':guild.id, 'user':member.id})
if p_r is not None:
await p_r.delete()
leveling_roles = db.leveling_roles.find({"guildid":guild.id})
level_roles = []
async for lvl in leveling_roles:
level_roles.append(lvl.roleid)
persistent_roles = db.persistent_roles_settings.find({'guildid':guild.id})
pers_roles = []
async for r in persistent_roles:
pers_roles.append(r.roleid)
all_persistent_roles = level_roles+pers_roles
member_roles = [str(role.id) for role in member.roles if role.id in all_persistent_roles]
if member_roles == []:
return
roles = ','.join(member_roles)
await db.persistent_roles(guildid=guild.id, user=member.id, roles=roles).insert()
@listen()
async def on_member_add(self, event):
guild = event.member.guild
member = event.member
warnings = db.strikes.find({'guildid':guild.id, 'user':member.id, 'action':{'$regex':'^warn$', '$options':'i'}})
warncount = []
async for warn in warnings:
if warn.type != 'Minor':
warncount.append(warn.strikeid)
if warncount != []:
wrc = 0
while wrc != len(warncount):
warnrolename = f'Warning-{wrc+1}'
wrc = wrc+1
warn_role = [role for role in guild.roles if role.name == warnrolename]
if warn_role == []:
role = await guild.create_role(name=warnrolename, reason='[automod]|[warn]created new warnrole as warnrole with this number did not exist yet')
await member.add_role(role, '[automod] Given back warnrole on server rejoin')
else:
for role in warn_role:
await member.add_role(role, '[automod] Given back warnrole on server rejoin')
p_r = await db.persistent_roles.find_one({'guildid':guild.id, 'user':member.id})
if p_r is not None:
roles = [guild.get_role(int(id_)) for id_ in p_r.roles.split(",") if len(id_)]
for role in roles:
if role not in member.roles:
await member.add_role(role, '[pt]persistent_role stored in db was added back to member on rejoin')
else:
p_r = await db.persistentroles.find_one({'guildid':guild.id, 'userid':member.id})
if p_r is not None:
roles = [guild.get_role(int(id_)) for id_ in p_r.roles.split(",") if len(id_)]
leveling_roles = db.leveling_roles.find({"guildid":guild.id})
level_roles = []
async for lvl in leveling_roles:
level_roles.append(lvl.roleid)
persistent_roles = db.persistent_roles_settings.find({'guildid':guild.id})
pers_roles = []
async for r in persistent_roles:
pers_roles.append(r.roleid)
all_persistent_roles = level_roles+pers_roles
member_roles = [role.id for role in roles if role.id in all_persistent_roles]
for role in member_roles:
if role not in member.roles:
await member.add_role(role, '[pt]persistent_role stored in db was added back to member on rejoin')
else:
mem_lvl = await db.leveling.find_one({'guildid':guild.id, 'memberid':member.id})
if mem_lvl is not None:
level_roles = db.leveling_roles.find({"guildid":guild.id, 'level':{'$lte':mem_lvl.level}})
roles = []
async for role in level_roles:
roles.append(role.roleid)
if level_roles != []:
for role_id in roles:
role = guild.get_role(role_id)
if role not in member.roles:
await member.add_role(role)
|