async def leaderboard(self, ctx):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
all_members = {}
for user in self.bot.data[str(ctx.guild.id)]:
all_members[user] = len(self.bot.data[str(
ctx.guild.id)][user]["Vouches"])
# Sort the users by their vouches
top10 = {
k: v
for k, v in sorted(all_members.items(), key=lambda item: item[1])
}
s = ''
for i in range(1, 11):
try:
if [*top10.values()][-i] < 1:
continue
s += f"{i}) <@{[*top10.keys()][-i]}> - **{[*top10.values()][-i]}** Vouches\n"
except IndexError:
pass
if s == '':
await utils.error_embed(
ctx, "Looks like nobody in this server has any vouches.", None)
return
await utils.embed(ctx, 'Vouch Leaderboard', s)
async def help(self, ctx):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
embed = discord.Embed(
title="Commands / Help",
description=f"**Start your commands with `{utils.prefix}`**",
color=utils.embed_color)
embed.add_field(
name=f"{utils.prefix}vouch",
value=
f"Vouch for a user. Usage: `{utils.prefix}vouch @user Message here`",
inline=False)
embed.add_field(
name=f"{utils.prefix}vouches",
value=
f"Display all vouches a user has. Usage: `{utils.prefix}vouches @user`",
inline=False)
embed.add_field(
name=f"{utils.prefix}leaderboard",
value=f"Display the leaderboard for users with the most vouches.",
inline=False)
embed.add_field(
name=f"{utils.prefix}delvouch",
value=
f"**(ADMIN)** Delete a users vouch from another user. Usage: `{utils.prefix}delvouch @user1 @user2`",
inline=False)
embed.add_field(
name=f"{utils.prefix}delallvouches",
value=
f"**(ADMIN)** Delete ALL of a users vouches. Usage: `{utils.prefix}delallvouches @user`",
inline=False)
await ctx.send(embed=embed)
async def reload(self, ctx):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
for filename in os.listdir('./cogs'):
if filename.endswith('.py'):
self.bot.reload_extension(f'cogs.{filename[:-3]}')
importlib.reload(utils)
await utils.embed(ctx, 'Reloaded all cogs successfully.', None)
def UPNPScan() -> tuple:
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
socket.IPPROTO_UDP) as server:
multicast_req = struct.pack('4sl', socket.inet_aton(M_GROUP),
socket.INADDR_ANY)
server.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP,
multicast_req)
server_init(server, ('', LISTEN_PORT), TIMEOUT)
return scan(server)
def start(self):
try:
server_init(self._server, self._address, TIMEOUT)
except Exception as e:
self.log('TCP binding error {}:{}: {}'.format(*self._address, e),
logger.ERROR)
else:
self._server.listen(1)
self._work = True
super().start()
async def on_message(self, ctx):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
# Update username
if ctx.author.id != self.bot.user.id and str(
ctx.author) != self.bot.data[str(ctx.guild.id)][str(
ctx.author.id)]["Username"]:
self.bot.data[str(ctx.guild.id)][str(
ctx.author.id)]["Username"] = str(ctx.author)
utils.write(self)
def start(self):
try:
# FIXME: OSError: [Errno 19] No such device
self._server.setsockopt(socket.IPPROTO_IP,
socket.IP_ADD_MEMBERSHIP,
self._multicast_req)
except OSError as e:
self.log('Unexpected \'setsockopt\' error: {}'.format(e),
logger.ERROR)
return
try:
server_init(self._server, self._address, TIMEOUT)
except Exception as e:
self.log('UDP binding error: {}'.format(e), logger.ERROR)
else:
self.work = True
self._http.start()
super().start()
self.log('start', logger.INFO)
async def delvouch(self,
ctx,
member: discord.Member = None,
member1: discord.Member = None):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
if not member or not member1:
await utils.error_embed(
ctx, "Please mention two members (remove x's vouch from y)",
None)
return
if not str(member.id) in self.bot.data[str(ctx.guild.id)]:
self.bot.data[str(ctx.guild.id)][str(member.id)] = {
"Username": str(member),
"Vouches": {}
}
utils.write(self)
if not str(member1.id) in self.bot.data[str(ctx.guild.id)]:
self.bot.data[str(ctx.guild.id)][str(member1.id)] = {
"Username": str(member1),
"Vouches": {}
}
utils.write(self)
if not str(member.id) in self.bot.data[str(ctx.guild.id)][str(
member1.id)]["Vouches"]:
await utils.error_embed(ctx, f'{member} has not vouched {member1}',
None)
return
del self.bot.data[str(ctx.guild.id)][str(member1.id)]["Vouches"][str(
member.id)]
utils.write(self)
await utils.embed(ctx, f"✅ Removed {member}'s vouch from {member1}",
None)
async def vouches(self, ctx, member: discord.Member = None):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
if not member:
member = ctx.author
if member.id == self.bot.user.id:
await utils.error_embed(ctx, "I don't have any vouches!", None)
return
vouches = self.bot.data[str(ctx.guild.id)][str(member.id)]["Vouches"]
if len(vouches) < 1:
await utils.error_embed(
ctx, "It looks like you don't have any vouches :(" if member.id
== ctx.author.id else "This user does not have any vouches.",
None)
return
pages = []
pages_count = math.ceil(len(vouches) / 5)
for i in range(1, pages_count + 1):
page_contents = ''
for user in {
k: vouches[k]
for k in list(vouches)[(5 * i) - 5 if i > 1 else 0:5 * i]
}:
page_contents += f"<@{user}> - *{vouches[user]['Date']}*\n```{vouches[user]['Message']}```\n"
pages.append(page_contents)
cur_page = 1
embed = discord.Embed(
title=f"All vouches for {member}",
description=
f"__Total Vouches:__ **{len(vouches)}**\n\n{pages[cur_page-1]}",
color=utils.embed_color).set_footer(
text=f'Page {cur_page}/{pages_count}').set_thumbnail(
url=member.avatar_url).set_author(
name=ctx.author.name, icon_url=ctx.author.avatar_url)
message = await ctx.send(embed=embed)
back = "⬅️"
forward = "➡️"
await message.add_reaction(back)
await message.add_reaction(forward)
def check(reaction, user):
return user == ctx.author and str(reaction.emoji) in [
back, forward
] and reaction.message.id == message.id
# This makes sure nobody except the command sender can interact with the "menu"
while True:
try:
reaction, user = await self.bot.wait_for("reaction_add",
timeout=30,
check=check)
# waiting for a reaction to be added - times out after 30 seconds
if str(reaction.emoji) == forward and cur_page != pages_count:
cur_page += 1
await message.edit(embed=discord.Embed(
title=f"All vouches for {member}",
description=
f"__Total Vouches:__ **{len(vouches)}**\n\n{pages[cur_page-1]}",
color=utils.embed_color).set_footer(
text=f'Page {cur_page}/{pages_count}'
).set_thumbnail(url=member.avatar_url).set_author(
name=ctx.author.name,
icon_url=ctx.author.avatar_url))
await message.remove_reaction(reaction, user)
elif str(reaction.emoji) == back and cur_page > 1:
cur_page -= 1
await message.edit(embed=discord.Embed(
title=f"All vouches for {member}",
description=
f"__Total Vouches:__ **{len(vouches)}**\n\n{pages[cur_page-1]}",
color=utils.embed_color).set_footer(
text=f'Page {cur_page}/{pages_count}'
).set_thumbnail(url=member.avatar_url).set_author(
name=ctx.author.name,
icon_url=ctx.author.avatar_url))
await message.remove_reaction(reaction, user)
else:
await message.remove_reaction(reaction, user)
# removes reactions if the user tries to go forward on the last page or
# backwards on the first page
except asyncio.TimeoutError:
await message.delete()
break
async def vouch(self,
ctx,
member: discord.Member = None,
*,
message: str = None):
utils.server_init(self, ctx)
utils.author_init(self, ctx)
if not member:
await utils.error_embed(ctx, 'Please mention a member to vouch.',
None)
return
if member.id == ctx.author.id:
await utils.error_embed(ctx, 'You cannot vouch yourself.', None)
return
if member.id == self.bot.user.id:
await utils.error_embed(ctx, 'I am not vouchable!', None)
return
if not message:
await utils.error_embed(
ctx,
f'Please add a message to your vouch. Usage: `{utils.prefix}vouch @user Message here`',
None)
return
if not str(member.id) in self.bot.data[str(ctx.guild.id)]:
# print(f'User {member} in {ctx.guild.name} has no vouch data, creating fresh vouch data for them.')
self.bot.data[str(ctx.guild.id)][str(member.id)] = {
"Username": str(member),
"Vouches": {}
}
utils.write(self)
if str(ctx.author.id) in self.bot.data[str(ctx.guild.id)][str(
member.id)]["Vouches"]:
await utils.error_embed(ctx, 'You have already vouched this user.',
None)
return
if len(message) > 250:
await utils.error_embed(
ctx, 'Please limit your vouch message to 250 characters.',
None)
return
if len(message.splitlines()) > 3:
await utils.error_embed(
ctx, 'Please limit your vouch message to 3 newlines.', None)
return
self.bot.data[str(ctx.guild.id)][str(member.id)]["Vouches"][str(
ctx.author.id)] = {
"Message": message,
"Username": str(ctx.author),
"Date": datetime.now().strftime('%Y-%m-%d')
}
utils.write(self)
await utils.embed(
ctx, f'✅ Vouched __{member}__',
f'**With message:**\n```{message}```'
if message is not None else None)