blob: 2f9bf38b9321bd2e4e302fbded1cd72bb99194c3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import random
from typing import Optional, Tuple
from discord.ext import commands
from bot.bot import Bot
from bot.constants import Emojis
class CoinSide(commands.Converter):
"""Class used to convert the `side` parameter of coinflip command."""
HEADS: Tuple[str] = ("h", "head", "heads")
TAILS: Tuple[str] = ("t", "tail", "tails")
async def convert(self, ctx: commands.Context, side: str) -> str:
"""Converts the provided `side` into the corresponding string."""
if side in CoinSide.HEADS:
return "heads"
elif side in CoinSide.TAILS:
return "tails"
else:
raise commands.BadArgument(f"{side!r} is not a valid coin side.")
class CoinFlip(commands.Cog):
"""Cog for the CoinFlip command."""
@commands.command(name="coinflip", aliases=("flip", "coin", "cf"))
async def coinflip_command(self, ctx: commands.Context, side: Optional[CoinSide]) -> None:
"""
Flips a coin.
If `coin_side` is provided will state whether you guessed the side correctly.
"""
flipped_side = random.choice(["heads", "tails"])
if not side:
await ctx.send(f"{ctx.author.mention} flipped **{flipped_side}**!")
return
message = f"{ctx.author.mention} flipped **{flipped_side}**. "
if side == flipped_side:
message += f"You guessed correctly! {Emojis.lemon_hyperpleased}"
else:
message += f"You guessed incorrectly. {Emojis.lemon_pensive}"
await ctx.send(message)
def setup(bot: Bot) -> None:
"""Loads the coinflip cog."""
bot.add_cog(CoinFlip())
|
