aboutsummaryrefslogtreecommitdiffstats
path: root/bot/utils/commands.py
diff options
context:
space:
mode:
Diffstat (limited to 'bot/utils/commands.py')
-rw-r--r--bot/utils/commands.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/bot/utils/commands.py b/bot/utils/commands.py
new file mode 100644
index 00000000..7c04a25a
--- /dev/null
+++ b/bot/utils/commands.py
@@ -0,0 +1,11 @@
+from typing import Optional
+
+from rapidfuzz import process
+
+
+def get_command_suggestions(
+ all_commands: list[str], query: str, *, cutoff: int = 60, limit: int = 3
+) -> Optional[list]:
+ """Get similar command names."""
+ results = process.extract(query, all_commands, score_cutoff=cutoff, limit=limit)
+ return [result[0] for result in results]