aboutsummaryrefslogtreecommitdiffstats
path: root/bot/utils/commands.py
diff options
context:
space:
mode:
authorGravatar ChrisJL <[email protected]>2022-09-18 22:03:34 +0100
committerGravatar GitHub <[email protected]>2022-09-18 22:03:34 +0100
commit46d1e8ddb217f1bb5e07179b32db50b6a04b6de8 (patch)
treec3f4db63c2751c51acfee97d016551b8f677e29b /bot/utils/commands.py
parentRemove unnecessary hasattr check (diff)
parentFix Poetry 1.2 Support (#1099) (diff)
Merge branch 'main' into fix-whitelist-inheritance
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]