CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-12-07 09:52:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix poor fuzzy results due to changes in rapidfuzz 3.0 (#6224)

Browse Source 
Co-authored-by: aikaterna <20862007+aikaterna@users.noreply.github.com>
This commit is contained in:
Jakub Kuczys
2023-08-10 07:03:21 +02:00
committed by GitHub
parent 3ac2512c14
commit dbb91dfce8
4 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
redbot/cogs/audio/core/utilities/local_tracks.py
View File

@@ -7,7 +7,7 @@ import discord
import lavalink
from red_commons.logging import getLogger
from rapidfuzz import process
import rapidfuzz
from redbot.core import commands
from redbot.core.i18n import Translator
from redbot.core.utils import AsyncIter
@@ -116,7 +116,9 @@ class LocalTrackUtilities(MixinMeta, metaclass=CompositeMetaClass):
to_search_string = {
i.local_track_path.name for i in to_search if i.local_track_path is not None
}
search_results = process.extract(search_words, to_search_string, limit=50)
search_results = rapidfuzz.process.extract(
search_words, to_search_string, limit=50, processor=rapidfuzz.utils.default_process
)
search_list = []
async for track_match, percent_match, __ in AsyncIter(search_results):
if percent_match > 85:

6
redbot/cogs/audio/core/utilities/queue.py
View File

@@ -7,7 +7,7 @@ import discord
import lavalink
from red_commons.logging import getLogger
from rapidfuzz import process
import rapidfuzz
from redbot.core import commands
from redbot.core.i18n import Translator
from redbot.core.utils import AsyncIter
@@ -132,7 +132,9 @@ class QueueUtilities(MixinMeta, metaclass=CompositeMetaClass):
track_title = track.title
tracks[queue_idx] = track_title
search_results = process.extract(search_words, tracks, limit=50)
search_results = rapidfuzz.process.extract(
search_words, tracks, limit=50, processor=rapidfuzz.utils.default_process
)
search_list = []
async for title, percent_match, queue_position in AsyncIter(search_results):
if percent_match > 89: