2019-09-04 12:48:49 -07:00
|
|
|
import random
|
2019-08-31 20:28:21 -07:00
|
|
|
import discord
|
|
|
|
from discord.ext import commands
|
|
|
|
import asyncio
|
|
|
|
|
2020-01-26 22:23:17 -08:00
|
|
|
async def take_vote(ctx, question:str, vote_time, min_voters):
|
2019-08-31 20:28:21 -07:00
|
|
|
|
|
|
|
"""
|
|
|
|
take_vote(ctx, question:str) - Collects votes
|
|
|
|
ctx: pass from command function
|
|
|
|
question: what to ask
|
|
|
|
|
|
|
|
returns [<all who want>, <all who don't want>].
|
|
|
|
It's up to the context/use case to decide how these should be used.
|
|
|
|
"""
|
2020-01-26 22:23:17 -08:00
|
|
|
|
2019-08-31 20:28:21 -07:00
|
|
|
embed_title="NEW VOTE"
|
|
|
|
votey_message = await ctx.send(
|
|
|
|
embed=discord.Embed(
|
|
|
|
type="rich",
|
|
|
|
title=embed_title,
|
|
|
|
# description="{}\n\n✅ - Yes\n\n❌ - No".format(question)
|
|
|
|
description="{0}\n\n✅ - Yes\n❌ - No\nMinimum {1} votes required.".format(question, min_voters)
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
await votey_message.add_reaction('✅')
|
|
|
|
await votey_message.add_reaction('❌')
|
|
|
|
|
|
|
|
passed = False
|
2020-01-26 22:23:17 -08:00
|
|
|
curr_time = 0
|
|
|
|
while curr_time < vote_time:
|
|
|
|
await asyncio.sleep(5)
|
|
|
|
all_in_favor = 0
|
|
|
|
not_in_favor = 0
|
|
|
|
finished_votey = await votey_message.channel.fetch_message(votey_message.id)
|
|
|
|
for reaction in finished_votey.reactions:
|
|
|
|
if str(reaction.emoji) == '✅':
|
|
|
|
all_in_favor += reaction.count-1 # don't include bot's reaction
|
|
|
|
elif str(reaction.emoji) == '❌':
|
|
|
|
not_in_favor += reaction.count-1
|
|
|
|
|
|
|
|
if all_in_favor > not_in_favor and all_in_favor >= min_voters:
|
|
|
|
passed = True
|
|
|
|
break
|
|
|
|
|
|
|
|
await asyncio.sleep(5)
|
|
|
|
curr_time += 5
|
|
|
|
|
|
|
|
question += "\nVERDICT: **Vote passed!**" if passed else "\nVERDICT: **Vote failed!**"
|
2019-08-31 20:28:21 -07:00
|
|
|
|
|
|
|
await votey_message.edit(embed=discord.Embed(
|
|
|
|
type="rich",
|
|
|
|
title=embed_title,
|
|
|
|
description=question
|
|
|
|
))
|
|
|
|
return passed
|
|
|
|
|
2020-07-21 20:21:00 -07:00
|
|
|
class CommandBreakerException(Exception):
|
|
|
|
pass
|
|
|
|
|
|
|
|
async def error_invalid_usage(ctx):
|
|
|
|
await ctx.send("⚠ `Invalid command usage! See `>>help` for more.`")
|
|
|
|
|
|
|
|
async def error_admin_targeted(ctx):
|
|
|
|
await ctx.send("⚠ `Cannot perform this action on adminstrators or users with a higher or equal role.`")
|
|
|
|
|
|
|
|
async def require_lower_permissions(ctx, user:discord.User, bot:discord.ext.commands.Bot):
|
|
|
|
memb = await ctx.guild.fetch_member(user.id)
|
|
|
|
bot_memb = await ctx.guild.fetch_member(bot.user.id)
|
|
|
|
is_admin = memb.permissions_in(ctx.channel).administrator
|
|
|
|
is_higher = memb.top_role.position >= bot_memb.top_role.position
|
|
|
|
|
|
|
|
if is_admin or is_higher:
|
|
|
|
await error_admin_targeted(ctx)
|
|
|
|
raise CommandBreakerException
|
2019-09-04 12:48:49 -07:00
|
|
|
|
|
|
|
def imgfun(msg:str, img_url:str):
|
|
|
|
return discord.Embed(
|
|
|
|
title=msg, color=random.randint(0, 16777215)
|
|
|
|
).set_image(url=img_url)
|