blob: 79d6b6d9a6373a193b60c781e09fd2ebf3340247 (
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
|
import random
from pathlib import Path
import yaml
from discord.ext.commands import Cog, Context, command
from bot.bot import Bot
class WonderTwins(Cog):
"""Cog for a Wonder Twins inspired command."""
def __init__(self):
with open(Path.cwd() / "bot" / "resources" / "fun" / "wonder_twins.yaml", "r", encoding="utf-8") as f:
info = yaml.load(f, Loader=yaml.FullLoader)
self.water_types = info["water_types"]
self.objects = info["objects"]
self.adjectives = info["adjectives"]
@staticmethod
def append_onto(phrase: str, insert_word: str) -> str:
"""Appends one word onto the end of another phrase in order to format with the proper determiner."""
if insert_word.endswith("s"):
phrase = phrase.split()
del phrase[0]
phrase = " ".join(phrase)
insert_word = insert_word.split()[-1]
return " ".join([phrase, insert_word])
def format_phrase(self) -> str:
"""Creates a transformation phrase from available words."""
adjective = random.choice((None, random.choice(self.adjectives)))
object_name = random.choice(self.objects)
water_type = random.choice(self.water_types)
if adjective:
object_name = self.append_onto(adjective, object_name)
return f"{object_name} of {water_type}"
@command(name="formof", aliases=("wondertwins", "wondertwin", "fo"))
async def form_of(self, ctx: Context) -> None:
"""Command to send a Wonder Twins inspired phrase to the user invoking the command."""
await ctx.send(f"Form of {self.format_phrase()}!")
def setup(bot: Bot) -> None:
"""Load the WonderTwins cog."""
bot.add_cog(WonderTwins())
|
