turtlebasket
/
comrade-bot
mirror of https://github.com/turtlebasket/comrade-bot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

short circuit vote eval

pull/1/head
michael 2020-01-26 22:23:17 -08:00
parent a7949bb5a7
commit 1197e512e2
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
bot_utils.py
View File

@ -3,7 +3,7 @@ import discord
from discord.ext import commands from discord.ext import commands
import asyncio import asyncio
async def take_vote(ctx, question:str, wait_time, min_voters): async def take_vote(ctx, question:str, vote_time, min_voters):
""" """
take_vote(ctx, question:str) - Collects votes take_vote(ctx, question:str) - Collects votes
@ -13,6 +13,7 @@ async def take_vote(ctx, question:str, wait_time, min_voters):
returns [<all who want>, <all who don't want>]. returns [<all who want>, <all who don't want>].
It's up to the context/use case to decide how these should be used. It's up to the context/use case to decide how these should be used.
""" """
embed_title="NEW VOTE" embed_title="NEW VOTE"
votey_message = await ctx.send( votey_message = await ctx.send(
embed=discord.Embed( embed=discord.Embed(
@ -26,24 +27,32 @@ async def take_vote(ctx, question:str, wait_time, min_voters):
await votey_message.add_reaction('✅') await votey_message.add_reaction('✅')
await votey_message.add_reaction('❌') await votey_message.add_reaction('❌')
await asyncio.sleep(wait_time) # TODO: Short-circuit eval loop, old stuff temp commented
# await asyncio.sleep(vote_time)
finished_votey = await votey_message.channel.fetch_message(votey_message.id)
all_in_favor = 0
not_in_favor = 0
for reaction in finished_votey.reactions:
if str(reaction.emoji) == '✅':
all_in_favor += reaction.count-1 # don't include bot's reaction
if str(reaction.emoji) == '❌':
not_in_favor += reaction.count-1
# await ctx.send(embed=discord.Embed(type='rich', title="VOTE RESULTS", description="✅ - {0}\n\n❌ - {1}\n".format(all_in_favor, not_in_favor)))
passed = False passed = False
if all_in_favor > not_in_favor and all_in_favor >= min_voters: curr_time = 0
question += "\nVERDICT: **Vote passed!**" while curr_time < vote_time:
passed = True await asyncio.sleep(5)
else: all_in_favor = 0
question += "\nVERDICT: **Vote failed!**" 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
print("check ({0} ✅, {1} ❌)".format(all_in_favor, not_in_favor))
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!**"
await votey_message.edit(embed=discord.Embed( await votey_message.edit(embed=discord.Embed(
type="rich", type="rich",