Add filter_by_search_query support to get_movies, get_episodes and get_tv_shows.

pykodi/kodi.py

@@ -6,6 +6,7 @@
 import jsonrpc_base
 import jsonrpc_async
 import jsonrpc_websocket
+from rapidfuzz import fuzz, process, utils
 
 
 def get_kodi_connection(
@@ -411,6 +412,37 @@ async def get_players(self):
         """Return the active player objects."""
         return await self._server.Player.GetActivePlayers()
 
+    async def search_media(self, media_type, search_query=None, score_cutoff=80, **kwargs):
+        """ Filter the original result by the search query. """
+        result = None
+        match media_type:
+            case 'movies':
+                result = await self.get_movies(**kwargs)
+            case 'tvshows':
+                result = await self.get_tv_shows(**kwargs)
+            case 'episodes':
+                result = await self.get_episodes(**kwargs)
+            case _:
+                return []
+
+        media = result.get(media_type, [])
+
+        if search_query is None:
+            return media
+
+        media_labels = [i["label"] for i in media]
+        matching_media_indices = process.extract(
+            search_query,
+            media_labels,
+            scorer=fuzz.WRatio,
+            processor=utils.default_process,
+            score_cutoff=score_cutoff
+        )
+
+        filtered_media = [{**media[idx], 'fuzz_score':fuzz_score} for (label, fuzz_score, idx) in matching_media_indices]
+
+        return filtered_media
+
     async def send_notification(self, title, message, icon="info", displaytime=10000):
         """Display on-screen message."""
         await self._server.GUI.ShowNotification(title, message, icon, displaytime)