From 5f90b17b0e954ef3fd6a507e947016b9d21e0b91 Mon Sep 17 00:00:00 2001
From: Joshua Arulsamy <joshua.gf.arul@gmail.com>
Date: Thu, 13 Aug 2020 01:58:12 -0600
Subject: [PATCH] :zap: Switch to PlexAPI for search

---
 PlexBot/bot.py        | 34 ++++++++++++----------------------
 PlexBot/exceptions.py |  2 ++
 2 files changed, 14 insertions(+), 22 deletions(-)
 create mode 100644 PlexBot/exceptions.py

diff --git a/PlexBot/bot.py b/PlexBot/bot.py
index 4436f62..19b29b5 100644
--- a/PlexBot/bot.py
+++ b/PlexBot/bot.py
@@ -9,10 +9,11 @@ from async_timeout import timeout
 from discord import FFmpegPCMAudio
 from discord.ext import commands
 from discord.ext.commands import command
-from fuzzywuzzy import fuzz
 from plexapi.exceptions import Unauthorized
 from plexapi.server import PlexServer
 
+from .exceptions import TrackNotFoundError
+
 root_log = logging.getLogger()
 plex_log = logging.getLogger("Plex")
 bot_log = logging.getLogger("Bot")
@@ -152,32 +153,22 @@ class Plex(commands.Cog):
         self.bot.loop.create_task(self._audio_player_task())
 
     def _search_tracks(self, title: str):
-        """Search the Plex music db
-
-        Uses a fuzzy search algorithm to find the closest matching song
-        title in the Plex music database.
+        """Search the Plex music db for track
 
         Args:
             title: str title of song to search for
 
         Returns:
             plexapi.audio.Track pointing to closest matching title
-            None if song can't be found.
 
         Raises:
-            None
+            TrackNotFoundError: Title of track can't be found in plex db.
         """
-        tracks = self.music.searchTracks()
-        score = [None, -1]
-        for i in tracks:
-            ratio = fuzz.ratio(title.lower(), i.title.lower())
-            if ratio > score[1]:
-                score[0] = i
-                score[1] = ratio
-            elif ratio == score[1]:
-                score[0] = i
-
-        return score[0]
+        results = self.music.searchTracks(title=title, maxresults=1)
+        try:
+            return results[0]
+        except IndexError:
+            raise TrackNotFoundError
 
     async def _play(self):
         """Heavy lifting of playing songs
@@ -332,10 +323,9 @@ class Plex(commands.Cog):
             return
 
         title = " ".join(args)
-        track = self._search_tracks(title)
-
-        # Fail if song title can't be found
-        if not track:
+        try:
+            track = self._search_tracks(title)
+        except TrackNotFoundError:
             await ctx.send(f"Can't find song: {title}")
             bot_log.debug("Failed to play, can't find song - %s", title)
             return
diff --git a/PlexBot/exceptions.py b/PlexBot/exceptions.py
new file mode 100644
index 0000000..782496a
--- /dev/null
+++ b/PlexBot/exceptions.py
@@ -0,0 +1,2 @@
+class TrackNotFoundError(Exception):
+    pass