135 lines
4 KiB
Python
135 lines
4 KiB
Python
import discord
|
|
import random
|
|
import pendulum
|
|
import asyncio
|
|
|
|
from discord.ext import commands
|
|
|
|
import utils
|
|
|
|
|
|
class Roulette(commands.Cog):
|
|
"""A fun game that ends in someone getting kicked!"""
|
|
|
|
roulettes = []
|
|
|
|
def get_game(self, server):
|
|
for x in self.roulettes:
|
|
if x.server == server:
|
|
return x
|
|
|
|
def start_game(self, server, time):
|
|
game = self.get_game(server)
|
|
if game:
|
|
return False
|
|
else:
|
|
game = Game(server, time)
|
|
self.roulettes.append(game)
|
|
return game
|
|
|
|
def end_game(self, server):
|
|
game = self.get_game(server)
|
|
member = game.choose()
|
|
self.roulettes.remove(game)
|
|
return member
|
|
|
|
@commands.group(invoke_without_command=True)
|
|
@commands.guild_only()
|
|
@utils.can_run(send_messages=True)
|
|
async def roulette(self, ctx):
|
|
"""Joins the current running roulette
|
|
|
|
EXAMPLE: !roulette
|
|
RESULT: You're probably going to get kicked..."""
|
|
r = self.get_game(ctx.message.guild)
|
|
if not r:
|
|
await ctx.send("There is no roulette game running on this server!")
|
|
else:
|
|
result = r.join(ctx.message.author)
|
|
time_left = r.time_left
|
|
if result:
|
|
await ctx.send(
|
|
"You have joined this roulette game! Good luck~ This roulette will end in "
|
|
+ time_left
|
|
)
|
|
else:
|
|
await ctx.send("This roulette will end in " + time_left)
|
|
|
|
@roulette.command(name="start", aliases=["create"])
|
|
@commands.guild_only()
|
|
@utils.can_run(kick_members=True)
|
|
async def roulette_start(self, ctx, time: int = 5):
|
|
"""Starts a roulette, that will end in one of the entrants being kicked from the server
|
|
By default, the roulette will end in 5 minutes; provide a number (up to 30)
|
|
to change how many minutes until it ends
|
|
|
|
EXAMPLE: !roulette start
|
|
RESULT: A new roulette game!"""
|
|
if time < 1 or time > 30:
|
|
await ctx.send(
|
|
"Invalid time! The roulette must be set to run between 1 and 30 minutes"
|
|
)
|
|
return
|
|
else:
|
|
game = self.start_game(ctx.message.guild, time)
|
|
if game:
|
|
await ctx.send(
|
|
"A new roulette game has just started! A random entrant will be kicked in {} minutes."
|
|
" Type {}roulette to join this roulette...good luck~".format(
|
|
game.time_left, ctx.prefix
|
|
)
|
|
)
|
|
else:
|
|
await ctx.send(
|
|
"There is already a roulette game running on this server!"
|
|
)
|
|
return
|
|
|
|
await asyncio.sleep(time * 60)
|
|
member = self.end_game(ctx.message.guild)
|
|
|
|
if member is None:
|
|
await ctx.send("Well no one joined the roulette. That was boring.")
|
|
return
|
|
|
|
try:
|
|
fmt = "The unlucky member to be kicked is {}; hopefully someone invites them back :)".format(
|
|
member.display_name
|
|
)
|
|
await member.kick()
|
|
except discord.Forbidden:
|
|
fmt = "Well, the unlucky member chosen was {} but I can't kick you...so kick yourself please?".format(
|
|
member.display_name
|
|
)
|
|
|
|
await ctx.send(fmt)
|
|
|
|
|
|
class Game:
|
|
def __init__(self, guild, time):
|
|
self.entrants = []
|
|
self.server = guild
|
|
self.end_time = pendulum.now(tz="UTC").add(minutes=time)
|
|
|
|
@property
|
|
def time_left(self):
|
|
return (self.end_time - pendulum.now(tz="UTC")).in_words()
|
|
|
|
def join(self, member):
|
|
"""Adds a member to the list of entrants"""
|
|
if member in self.entrants:
|
|
return False
|
|
else:
|
|
self.entrants.append(member)
|
|
return True
|
|
|
|
def choose(self):
|
|
try:
|
|
return random.SystemRandom().choice(self.entrants)
|
|
except IndexError:
|
|
return None
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Roulette(bot))
|