diff options
Diffstat (limited to 'bot/exts/valentines/myvalenstate.py')
| -rw-r--r-- | bot/exts/valentines/myvalenstate.py | 82 | 
1 files changed, 0 insertions, 82 deletions
| diff --git a/bot/exts/valentines/myvalenstate.py b/bot/exts/valentines/myvalenstate.py deleted file mode 100644 index 52a61011..00000000 --- a/bot/exts/valentines/myvalenstate.py +++ /dev/null @@ -1,82 +0,0 @@ -import collections -import json -import logging -from pathlib import Path -from random import choice - -import discord -from discord.ext import commands - -from bot.bot import Bot -from bot.constants import Colours - -log = logging.getLogger(__name__) - -STATES = json.loads(Path("bot/resources/valentines/valenstates.json").read_text("utf8")) - - -class MyValenstate(commands.Cog): -    """A Cog to find your most likely Valentine's vacation destination.""" - -    def levenshtein(self, source: str, goal: str) -> int: -        """Calculates the Levenshtein Distance between source and goal.""" -        if len(source) < len(goal): -            return self.levenshtein(goal, source) -        if len(source) == 0: -            return len(goal) -        if len(goal) == 0: -            return len(source) - -        pre_row = list(range(0, len(source) + 1)) -        for i, source_c in enumerate(source): -            cur_row = [i + 1] -            for j, goal_c in enumerate(goal): -                if source_c != goal_c: -                    cur_row.append(min(pre_row[j], pre_row[j + 1], cur_row[j]) + 1) -                else: -                    cur_row.append(min(pre_row[j], pre_row[j + 1], cur_row[j])) -            pre_row = cur_row -        return pre_row[-1] - -    @commands.command() -    async def myvalenstate(self, ctx: commands.Context, *, name: str = None) -> None: -        """Find the vacation spot(s) with the most matching characters to the invoking user.""" -        eq_chars = collections.defaultdict(int) -        if name is None: -            author = ctx.author.name.lower().replace(" ", "") -        else: -            author = name.lower().replace(" ", "") - -        for state in STATES.keys(): -            lower_state = state.lower().replace(" ", "") -            eq_chars[state] = self.levenshtein(author, lower_state) - -        matches = [x for x, y in eq_chars.items() if y == min(eq_chars.values())] -        valenstate = choice(matches) -        matches.remove(valenstate) - -        embed_title = "But there are more!" -        if len(matches) > 1: -            leftovers = f"{', '.join(matches[:-2])}, and {matches[-1]}" -            embed_text = f"You have {len(matches)} more matches, these being {leftovers}." -        elif len(matches) == 1: -            embed_title = "But there's another one!" -            embed_text = f"You have another match, this being {matches[0]}." -        else: -            embed_title = "You have a true match!" -            embed_text = "This state is your true Valenstate! There are no states that would suit" \ -                         " you better" - -        embed = discord.Embed( -            title=f"Your Valenstate is {valenstate} \u2764", -            description=STATES[valenstate]["text"], -            colour=Colours.pink -        ) -        embed.add_field(name=embed_title, value=embed_text) -        embed.set_image(url=STATES[valenstate]["flag"]) -        await ctx.send(embed=embed) - - -def setup(bot: Bot) -> None: -    """Load the Valenstate Cog.""" -    bot.add_cog(MyValenstate()) | 
