aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Gintautas Kazlauskas <[email protected]>2021-09-22 01:09:39 +0300
committerGravatar Gintautas Kazlauskas <[email protected]>2021-09-22 01:09:39 +0300
commit54b9b03c6fbfe0affba9d95b2abb796c588d028a (patch)
treeb3d2622470594d5717f6804a5ce53824ab62f8aa
parentLinting (diff)
Further linting
-rw-r--r--bot/exts/fun/uwu.py49
1 files changed, 24 insertions, 25 deletions
diff --git a/bot/exts/fun/uwu.py b/bot/exts/fun/uwu.py
index ef838d56..ce2ed45d 100644
--- a/bot/exts/fun/uwu.py
+++ b/bot/exts/fun/uwu.py
@@ -1,15 +1,15 @@
-import re
import random
+import re
from functools import partial
+from typing import Callable
-from bot.utils import helpers
+from discord.ext import commands
+from discord.ext.commands import Cog, Context, clean_content
from bot.bot import Bot
-
from bot.exts.fun.fun import Fun
+from bot.utils import helpers
-from discord.ext import commands
-from discord.ext.commands import Cog, Context, clean_content
# Wepwacement
WEPWACE_HASH = {
@@ -63,14 +63,14 @@ EMOJI_LUT = [
wepwace_regex = re.compile(r"(?<![w])[lr](?![w])")
-def word_replace(input_string: str) -> str:
+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:
+def _char_replace(input_string: str) -> str:
return wepwace_regex.sub("w", input_string)
@@ -79,11 +79,11 @@ 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(strength: float, input_string: str) -> str:
+ return stutter_regex.sub(partial(_stutter_replace, strength=strength), input_string, 0)
-def stutter_replace(match, strength=0.0):
+def _stutter_replace(match: Callable, strength: float = 0.0) -> str:
match_string = match.string[slice(*match.span())]
if random.random() < strength:
char = match_string[-1]
@@ -96,7 +96,7 @@ nya_regex = re.compile(r"n([aeou])([^aeiou])")
nya_subst = "ny\\g<1>\\g<2>"
-def nyaify(input_string):
+def _nyaify(input_string: str) -> str:
return nya_regex.sub(nya_subst, input_string, 0)
@@ -104,11 +104,11 @@ def nyaify(input_string):
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(strength: float, input_string: str) -> str:
+ return punctuation_regex.sub(partial(_emoji_replace, strength=strength), input_string, 0)
-def emoji_replace(match, strength=0.0):
+def _emoji_replace(match: Callable, strength: float = 0.0) -> str:
match_string = match.string[slice(*match.span())]
if random.random() < strength:
return f" {EMOJI_LUT[random.randint(0, len(EMOJI_LUT) - 1)]} "
@@ -118,20 +118,19 @@ def emoji_replace(match, strength=0.0):
# Main
-def uwuify(input_string: str, *, stutter_strength: float = 0.2, emoji_strength: float = 0.2) -> str:
+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)
+ 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.
- """
+ """Cog for uwuification."""
+
def __init__(self, bot: Bot):
self.bot = bot
@@ -141,8 +140,8 @@ class Uwu(Cog):
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)
+ 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: