diff options
Diffstat (limited to 'bot')
| -rw-r--r-- | bot/exts/fun/fun.py | 31 | ||||
| -rw-r--r-- | bot/exts/fun/uwu.py | 143 |
2 files changed, 143 insertions, 31 deletions
diff --git a/bot/exts/fun/fun.py b/bot/exts/fun/fun.py index b148f1f3..773c2ffb 100644 --- a/bot/exts/fun/fun.py +++ b/bot/exts/fun/fun.py @@ -17,20 +17,6 @@ from bot.utils import helpers log = logging.getLogger(__name__) -UWU_WORDS = { - "fi": "fwi", - "l": "w", - "r": "w", - "some": "sum", - "th": "d", - "thing": "fing", - "tho": "fo", - "you're": "yuw'we", - "your": "yur", - "you": "yuw", -} - - def caesar_cipher(text: str, offset: int) -> Iterable[str]: """ Implements a lazy Caesar Cipher algorithm. @@ -74,23 +60,6 @@ class Fun(Cog): else: raise BadArgument(f"`{Client.prefix}roll` only supports between 1 and 6 rolls.") - @commands.command(name="uwu", aliases=("uwuwize", "uwuify",)) - async def uwu_command(self, ctx: Context, *, text: clean_content(fix_channel_mentions=True)) -> None: - """Converts a given `text` into it's uwu equivalent.""" - conversion_func = functools.partial( - utils.replace_many, replacements=UWU_WORDS, ignore_case=True, match_case=True - ) - text, embed = await Fun._get_text_and_embed(ctx, text) - # Convert embed if it exists - if embed is not None: - embed = Fun._convert_embed(conversion_func, embed) - converted_text = conversion_func(text) - converted_text = helpers.suppress_links(converted_text) - # Don't put >>> if only embed present - if converted_text: - converted_text = f">>> {converted_text.lstrip('> ')}" - await ctx.send(content=converted_text, embed=embed) - @commands.command(name="randomcase", aliases=("rcase", "randomcaps", "rcaps",)) async def randomcase_command(self, ctx: Context, *, text: clean_content(fix_channel_mentions=True)) -> None: """Randomly converts the casing of a given `text`.""" diff --git a/bot/exts/fun/uwu.py b/bot/exts/fun/uwu.py new file mode 100644 index 00000000..2b063e1c --- /dev/null +++ b/bot/exts/fun/uwu.py @@ -0,0 +1,143 @@ +#!/usr/bin/env python3 + +import re +import random +import re +from functools import partial + +from bot.utils import helpers + +from bot.bot import Bot + +from bot.exts.fun.fun import Fun + +from discord import Message +from discord.ext import commands +from discord.ext.commands import Cog, Context, clean_content + +# Wepwacement +WEPWACE_HASH = { + "small": "smol", + "cute": "kawaii~", + "fluff": "floof", + "love": "luv", + "stupid": "baka", + "idiot": "baka", + "what": "nani", + "meow": "nya~", + "roar": "rawrr~", +} + +EMOJI_LUT = [ + "rawr x3", + "OwO", + "UwU", + "o.O", + "-.-", + ">w<", + "(⑅˘꒳˘)", + "(ꈍᴗꈍ)", + "(˘ω˘)", + "(U ᵕ U❁)", + "σωσ", + "òωó", + "(///ˬ///✿)", + "(U ﹏ U)", + "( ͡o ω ͡o )", + "ʘwʘ", + ":3", + ":3", # important enough to have twice + "XD", + "nyaa~~", + "mya", + ">_<", + "😳", + "🥺", + "😳😳😳", + "rawr", + "uwu", + "^^", + "^^;;", + "(ˆ ﻌ ˆ)♡", + "^•ﻌ•^", + "/(^•ω•^)", + "(✿oωo)", +] + +wepwace_regex = re.compile(r"(?<![w])[lr](?![w])") + +def word_replace(input_string: str) -> str: + for word in WEPWACE_HASH: + input_string = input_string.replace(word, WEPWACE_HASH[word]) + + return input_string + +def char_replace(input_string: str) -> str: + return wepwace_regex.sub("w", input_string) + +# Stuttering +stutter_regex = re.compile(r"(\s)([a-zA-Z])") +stutter_subst = "\\g<1>\\g<2>-\\g<2>" + +def stutter(strength: float, input_string: str): + return stutter_regex.sub(partial(stutter_replace, strength=strength), input_string, 0) + +def stutter_replace(match, strength = 0.0): + match_string = match.string[slice(*match.span())] + if random.random() < strength: + char = match_string[-1] + return f"{match_string}-{char}" + return match_string + +# Nyaification +nya_regex = re.compile(r"n([aeou])([^aeiou])") +nya_subst = "ny\\g<1>\\g<2>" + +def nyaify(input_string): + return nya_regex.sub(nya_subst, input_string, 0) + +# Emoji +punctuation_regex = re.compile(r"\s+") + +def emoji(strength: float, input_string: str): + return punctuation_regex.sub(partial(emoji_replace, strength=strength), input_string, 0) + + +def emoji_replace(match, strength = 0.0): + match_string = match.string[slice(*match.span())] + if random.random() < strength: + return f" {EMOJI_LUT[random.randint(0, len(EMOJI_LUT) - 1)]} " + + return match_string + +# Main + +def uwuify(input_string: str, *, stutter_strength: float = 0.2, emoji_strength: float = 0.2) -> str: + input_string = input_string.lower() + input_string = word_replace(input_string) + input_string = nyaify(input_string) + input_string = char_replace(input_string) + input_string = stutter(stutter_strength, input_string) + input_string = emoji(emoji_strength, input_string) + return input_string + +class Uwu(Cog): + """ + Cog for uwuification. + """ + def __init__(self, bot: Bot): + self.bot = bot + + @commands.command(name="uwu", aliases=("uwuwize", "uwuify",)) + async def uwu_command(self, ctx: Context, *, text: clean_content(fix_channel_mentions=True)) -> None: + """Converts a given `text` into it's uwu equivalent.""" + text, embed = await Fun._get_text_and_embed(ctx, text) + # Convert embed if it exists + if embed is not None: + embed = Fun._convert_embed(uwuify, embed) + converted_text = uwuify(text) + converted_text = helpers.suppress_links(converted_text) + # Don't put >>> if only embed present + if converted_text: + converted_text = f">>> {converted_text.lstrip('> ')}" + await ctx.send(content=converted_text, embed=embed) |