100 lines
3.7 KiB
Python
Executable file
100 lines
3.7 KiB
Python
Executable file
import asyncio
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
|
|
class Sondage(commands.Cog):
|
|
"""Commandes sondage."""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command(pass_context=True)
|
|
async def sondage(self, ctx, *, msg="help"):
|
|
if msg != "help":
|
|
await ctx.message.delete()
|
|
options = msg.split(" | ")
|
|
|
|
times = [x for x in options if x.startswith("time=")]
|
|
|
|
if times:
|
|
time = int(times[0].strip("time="))
|
|
options.remove(times[0])
|
|
else:
|
|
time = 0
|
|
|
|
if len(options) <= 1:
|
|
raise commands.errors.MissingRequiredArgument
|
|
if len(options) >= 22:
|
|
return await ctx.send(f"{ctx.message.author.mention}> "
|
|
f":octagonal_sign: Vous ne pouvez pas "
|
|
f"mettre plus de 20 réponses !")
|
|
|
|
emoji = ['1⃣', '2⃣', '3⃣', '4⃣', '5⃣', '6⃣', '7⃣', '8⃣', '9⃣', '🔟', '0⃣',
|
|
'🇦', '🇧', '🇨', '🇩', '🇪', '🇫', '🇬', '🇭', '🇮']
|
|
to_react = []
|
|
confirmation_msg = f"**{options[0].rstrip('?')}?**:\n\n"
|
|
|
|
for idx, option in enumerate(options[1:]):
|
|
confirmation_msg += f"{emoji[idx]} - {option}\n"
|
|
to_react.append(emoji[idx])
|
|
|
|
confirmation_msg += "*Sondage proposé par* " + \
|
|
str(ctx.message.author.mention)
|
|
if time == 0:
|
|
confirmation_msg += ""
|
|
else:
|
|
confirmation_msg += f"\n\nVous avez {time} secondes pour voter!"
|
|
|
|
poll_msg = await ctx.send(confirmation_msg)
|
|
for emote in to_react:
|
|
await poll_msg.add_reaction(emote)
|
|
|
|
if time != 0:
|
|
await asyncio.sleep(time)
|
|
async for message in ctx.message.channel.history():
|
|
if message.id == poll_msg.id:
|
|
poll_msg = message
|
|
|
|
results = {}
|
|
|
|
for reaction in poll_msg.reactions:
|
|
if reaction.emoji in to_react:
|
|
results[reaction.emoji] = reaction.count - 1
|
|
end_msg = "Le sondage est términé. Les résultats sont:\n\n"
|
|
|
|
for result in results:
|
|
end_msg += "{} {} - {} votes\n". \
|
|
format(result,
|
|
options[emoji.index(result)+1],
|
|
results[result])
|
|
|
|
top_result = max(results, key=lambda key: results[key])
|
|
|
|
if len([x for x in results
|
|
if results[x] == results[top_result]]) > 1:
|
|
top_results = []
|
|
for key, value in results.items():
|
|
if value == results[top_result]:
|
|
top_results.append(options[emoji.index(key)+1])
|
|
end_msg += "\nLes gagnants sont : {}". \
|
|
format(", ".join(top_results))
|
|
else:
|
|
top_result = options[emoji.index(top_result)+1]
|
|
end_msg += "\n\"{}\" est le gagnant!".format(top_result)
|
|
await ctx.send(end_msg)
|
|
else:
|
|
await ctx.send("please use `@tuxbot poll` (this is rewrite version in beta")
|
|
await ctx.message.delete()
|
|
|
|
text = open('texts/rpoll.md').read()
|
|
em = discord.Embed(title='Aide sur le sondage',
|
|
description=text,
|
|
colour=0xEEEEEE)
|
|
await ctx.send(embed=em)
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Sondage(bot))
|