(Games Cog): Created new helper function `get_best_results` to match not correct genres.

1 files changed, 13 insertions, 0 deletions

diff --git a/bot/seasons/evergreen/game.py b/bot/seasons/evergreen/game.py
index e6700937..79fe8ae9 100644
--- a/bot/seasons/evergreen/game.py
+++ b/bot/seasons/evergreen/game.py
@@ -1,6 +1,7 @@
 import difflib
 import logging
 import random
+import re
 from datetime import datetime as dt
 from enum import IntEnum
 from typing import Any, Dict, List, Optional, Tuple
@@ -25,6 +26,8 @@ HEADERS = {
 
 logger = logging.getLogger(__name__)
 
+REGEX_NON_ALPHABET = re.compile(r"[^a-z0-9]", re.IGNORECASE)
+
 # ---------
 # TEMPLATES
 # ---------
@@ -385,6 +388,16 @@ class Games(Cog):
 
         return page, url
 
+    async def get_best_results(self, query: str, nr: int = 4) -> List[Tuple[float, str]]:
+        """Get best match result of genre when original genre is invalid."""
+        results = []
+        for genre in self.genres:
+            ratios = [difflib.SequenceMatcher(None, query, genre).ratio()]
+            for word in REGEX_NON_ALPHABET.split(genre):
+                ratios.append(difflib.SequenceMatcher(None, query, word).ratio())
+            results.append((round(max(ratios), 2), genre))
+        return sorted(results, reverse=True)[:nr]
+
 
 def setup(bot: SeasonalBot) -> None:
     """Add/Load Games cog."""