From 7b61972401dcb8d659d0564260a5b975a084e243 Mon Sep 17 00:00:00 2001 From: Georges-Antoine Assi Date: Tue, 21 May 2024 10:00:35 -0400 Subject: [PATCH] Fix search by ID for mobygames --- backend/handler/metadata/igdb_handler.py | 3 ++- backend/handler/metadata/moby_handler.py | 7 ++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/backend/handler/metadata/igdb_handler.py b/backend/handler/metadata/igdb_handler.py index 7dd6cf232..4f7698e54 100644 --- a/backend/handler/metadata/igdb_handler.py +++ b/backend/handler/metadata/igdb_handler.py @@ -416,7 +416,8 @@ def get_matched_roms_by_id(self, igdb_id: int) -> list[IGDBRom]: if not IGDB_API_ENABLED: return [] - return [self.get_rom_by_id(igdb_id)] + rom = self.get_rom_by_id(igdb_id) + return [rom] if rom["igdb_id"] else [] @check_twitch_token def get_matched_roms_by_name( diff --git a/backend/handler/metadata/moby_handler.py b/backend/handler/metadata/moby_handler.py index 92d1c6d70..3ee6f1172 100644 --- a/backend/handler/metadata/moby_handler.py +++ b/backend/handler/metadata/moby_handler.py @@ -253,14 +253,14 @@ async def get_rom(self, file_name: str, platform_moby_id: int) -> MobyGamesRom: def get_rom_by_id(self, moby_id: int) -> MobyGamesRom: if not MOBY_API_ENABLED: - return MobyGamesRom(moby_id=moby_id) + return MobyGamesRom(moby_id=None) url = yarl.URL(self.games_url).with_query(id=moby_id) roms = self._request(str(url)).get("games", []) res = pydash.get(roms, "[0]", None) if not res: - return MobyGamesRom(moby_id=moby_id) + return MobyGamesRom(moby_id=None) rom = { "moby_id": res["game_id"], @@ -278,7 +278,8 @@ def get_matched_roms_by_id(self, moby_id: int) -> list[MobyGamesRom]: if not MOBY_API_ENABLED: return [] - return [self.get_rom_by_id(moby_id)] + rom = self.get_rom_by_id(moby_id) + return [rom] if rom["moby_id"] else [] def get_matched_roms_by_name( self, search_term: str, platform_moby_id: int