added whimsy

Author: GlowDustWL

src/calculator.py

@@ -290,6 +290,9 @@ def get_disable_calculation(self):
 # If boolean is true, result was successful and str contains result.
 # If boolean is false, result was unsuccessful and str contains error message
 def general_kill_handler(weapon_fuzzy_name, target_fuzzy_name):
+    if (result := easter_egg_check(weapon_fuzzy_name,target_fuzzy_name)) is not None:
+        return result
+
     args = None
     if weapon_fuzzy_name in parse.weapons_dictionary:
         weapon_name = parse.weapons_dictionary[weapon_fuzzy_name]
@@ -446,3 +449,40 @@ def statsheet_handler(entity_name):
         return "I could not process a request because the entity is not a valid weapon, structure or vehicle."
     except:
         return bot.EntityNotFoundError(entity_name)
+
+# whimsy
+def easter_egg_check(weapon_fuzzy_name, target_fuzzy_name):
+    easter_eggs = [
+        ("Watch Tower", "Scout Plane", 1, 95)
+    ]
+    for weapon, target, hits, threshold in easter_eggs:
+        if fuzz.normalized_ratio_equal(weapon_fuzzy_name,weapon,threshold) and fuzz.normalized_partial_ratio_equal(target_fuzzy_name,target,threshold):
+            return f"It takes {hits} {weapon} to kill a {target}"
+    return None
+
+#                                                .--.
+#                                                `.  \
+#                                                  \  \
+#                                                   .  \
+#                                                   :   .
+#                                                   |    .
+#                                                   |    :
+#                                                   |    |
+#   ..._  ___                                       |    |
+#  `."".`''''""--..___                              |    |
+#  ,-\  \             ""-...__         _____________/    |
+#  / ` " '                    `""""""""                  .
+#  \                                                      L
+#  (>                                                      \
+# /                                                         \
+# \_    ___..---.                                            L
+#   `--'         '.                                           \
+#                  .                                           \_
+#                 _/`.                                           `.._
+#              .'     -.                                             `.
+#             /     __.-Y     /''''''-...___,...--------.._            |
+#            /   _."    |    /                ' .      \   '---..._    |
+#           /   /      /    /                _,. '    ,/           |   |
+#           \_,'     _.'   /              /''     _,-'            _|   |
+#                   '     /               `-----''               /     |
+#                   `...-'                                       `...-'

src/fuzzy.py

@@ -5,6 +5,7 @@
 import utils
 import bot
 from fuzzywuzzy import process
+import re
 
 
 
@@ -156,4 +157,27 @@ def fuzzy_match_weapon_name_command(name):
             if score > 60:
                  good_score_list.append(key)
         return good_score_list
-        
+
+# no space, all lowercase, singular only
+def normalize(text: str) -> str:
+    text = re.sub(r"\s+", " ", text.lower().strip()) 
+    words = text.split()
+    words = [singularize(w) for w in words]
+    return "".join(words)
+
+def singularize(word: str) -> str:
+    if word.endswith("ies"):
+        return word[:-3] + "y"
+    if word.endswith("ses"):
+        return word[:-2]  # e.g. classes -> class
+    if word.endswith("s") and not word.endswith("ss"):
+        return word[:-1]
+    return word
+
+def normalized_partial_ratio_equal(a: str, b: str, c: int = 98) -> bool:
+    print(fuzz.partial_ratio(normalize(a), normalize(b)))
+    return fuzz.partial_ratio(normalize(a), normalize(b)) >= c
+
+def normalized_ratio_equal(a: str, b: str, c: int = 98) -> bool:
+    print(fuzz.ratio(normalize(a), normalize(b)))
+    return fuzz.ratio(normalize(a), normalize(b)) >= c