Simon/romm
1
0
Fork 0
mirror of https://github.com/rommapp/romm.git synced 2026-06-28 06:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert IGDB handler and test changes

Browse Source 
Co-authored-by: gantoine <3247106+gantoine@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-05-27 00:06:50 +00:00
committed by GitHub
parent 536f6ac815
commit 511f5e4272
2 changed files with 8 additions and 143 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
backend/handler/metadata/igdb_handler.py
View File

@@ -133,48 +133,8 @@ def build_related_game(
)
def select_preferred_release_date(
rom: Game,
preferred_regions: list[int],
platform_igdb_id: int | None,
) -> int | None:
release_dates = rom.get("release_dates", [])
assert mark_list_expanded(release_dates)
candidates: list[tuple[int | None, int]] = []
for release_date in release_dates:
date = release_date.get("date")
if not isinstance(date, int):
continue
platform = release_date.get("platform")
if platform_igdb_id is not None:
if isinstance(platform, dict):
if platform.get("id") != platform_igdb_id:
continue
elif isinstance(platform, int) and platform != platform_igdb_id:
continue
region = release_date.get("region")
region_id = region if isinstance(region, int) else None
candidates.append((region_id, date))
if not candidates:
return rom.get("first_release_date", None)
for preferred_region in preferred_regions:
preferred_dates = [date for region_id, date in candidates if region_id == preferred_region]
if preferred_dates:
return min(preferred_dates)
return min(date for _, date in candidates)
def extract_metadata_from_igdb_rom(
self: MetadataHandler,
rom: Game,
platform_igdb_id: int | None,
preferred_release_regions: list[int],
self: MetadataHandler, rom: Game, platform_igdb_id: int | None
) -> IGDBMetadata:
age_ratings = rom.get("age_ratings", [])
alternative_names = rom.get("alternative_names", [])
@@ -250,9 +210,7 @@ def extract_metadata_from_igdb_rom(
"youtube_video_id": videos[0].get("video_id") if videos else None,
"total_rating": str(round(rom.get("total_rating", 0.0), 2)),
"aggregated_rating": str(round(rom.get("aggregated_rating", 0.0), 2)),
"first_release_date": select_preferred_release_date(
rom, preferred_release_regions, platform_igdb_id
),
"first_release_date": rom.get("first_release_date", None),
"genres": [g.get("name", "") for g in genres if g.get("name")],
"franchises": pydash.compact(
[franchise.get("name") if franchise else None]
@@ -368,17 +326,6 @@ REGION_TO_IGDB_LOCALE: dict[str, str | None] = {
"tw": "zh-TW", # Taiwan (Traditional Chinese)
}
REGION_TO_IGDB_RELEASE_DATE: dict[str, int] = {
"eu": 1, # Europe
"us": 2, # North America
"jp": 5, # Japan
"cn": 6, # China
"tw": 7, # Asia
"wor": 8, # Worldwide
"kr": 9, # Korea
"br": 10, # Brazil
}
def get_igdb_preferred_locale(rom: Rom | None = None) -> str | None:
"""Get IGDB locale, preferring the rom's own region tag when available.
@@ -403,30 +350,6 @@ def get_igdb_preferred_locale(rom: Rom | None = None) -> str | None:
return None
def get_igdb_preferred_release_regions(rom: Rom | None = None) -> list[int]:
config = cm.get_config()
priority = config.SCAN_REGION_PRIORITY
rom_codes: list[str] = []
if rom is not None and isinstance(rom.regions, list):
for region_name in rom.regions:
code = region_name_to_provider_shortcode(region_name)
if code:
rom_codes.append(code)
rom_codes.sort(
key=lambda code: priority.index(code) if code in priority else len(priority)
)
ordered_codes = list(
dict.fromkeys(rom_codes + priority + ["us", "wor", "eu", "jp", "kr", "cn", "tw"])
)
return [
REGION_TO_IGDB_RELEASE_DATE[code]
for code in ordered_codes
if code in REGION_TO_IGDB_RELEASE_DATE
]
def extract_localized_data(rom: Game, preferred_locale: str | None) -> tuple[str, str]:
"""Extract localized name and cover URL based on preferred locale.
@@ -475,7 +398,6 @@ def build_igdb_rom(
handler: "IGDBHandler",
rom: Game,
preferred_locale: str | None,
preferred_release_regions: list[int],
platform_igdb_id: int | None,
) -> "IGDBRom":
"""Build an IGDBRom from IGDB game data with localization support.
@@ -506,9 +428,7 @@ def build_igdb_rom(
handler.normalize_cover_url(s.get("url", "")).replace("t_thumb", "t_720p")
for s in rom_screenshots
],
igdb_metadata=extract_metadata_from_igdb_rom(
handler, rom, platform_igdb_id, preferred_release_regions
),
igdb_metadata=extract_metadata_from_igdb_rom(handler, rom, platform_igdb_id),
)
@@ -787,11 +707,7 @@ class IGDBHandler(MetadataHandler):
return fallback_rom
return build_igdb_rom(
self,
res,
get_igdb_preferred_locale(rom=rom),
get_igdb_preferred_release_regions(rom=rom),
platform_igdb_id,
self, res, get_igdb_preferred_locale(rom=rom), platform_igdb_id
)
async def get_rom_by_id(self, rom: Rom, igdb_id: int) -> IGDBRom:
@@ -806,13 +722,7 @@ class IGDBHandler(MetadataHandler):
if not roms:
return IGDBRom(igdb_id=None)
return build_igdb_rom(
self,
roms[0],
get_igdb_preferred_locale(rom=rom),
get_igdb_preferred_release_regions(rom=rom),
None,
)
return build_igdb_rom(self, roms[0], get_igdb_preferred_locale(rom=rom), None)
async def get_matched_rom_by_id(self, rom: Rom, igdb_id: int) -> IGDBRom | None:
if not self.is_enabled():
@@ -874,11 +784,8 @@ class IGDBHandler(MetadataHandler):
]
preferred_locale = get_igdb_preferred_locale(rom=rom)
preferred_release_regions = get_igdb_preferred_release_regions(rom=rom)
return [
build_igdb_rom(
self, rom, preferred_locale, preferred_release_regions, platform_igdb_id
)
build_igdb_rom(self, rom, preferred_locale, platform_igdb_id)
for rom in matched_roms
]
@@ -967,9 +874,6 @@ GAMES_FIELDS = (
"total_rating",
"aggregated_rating",
"first_release_date",
"release_dates.date",
"release_dates.region",
"release_dates.platform.id",
"artworks.url",
"cover.url",
"screenshots.url",

43
backend/tests/handler/metadata/test_igdb_handler.py
View File

@@ -1,15 +1,11 @@
"""Tests for the IGDB metadata handler."""
from unittest.mock import AsyncMock, MagicMock, patch
from unittest.mock import AsyncMock, patch
import pytest
from adapters.services.igdb_types import GameType
from handler.metadata.igdb_handler import (
IGDBHandler,
build_igdb_rom,
get_igdb_preferred_release_regions,
)
from handler.metadata.igdb_handler import IGDBHandler
GENESIS_IGDB_ID = 29
@@ -46,7 +42,6 @@ def _make_game(game_id: int, name: str) -> dict:
"age_ratings": [],
"multiplayer_modes": [],
"game_localizations": [],
"release_dates": [],
}
@@ -160,37 +155,3 @@ class TestSearchRomGameTypeFilter:
f"Expected Ecco: The Tides of Time (id=5379), got {result.get('name')} (id={result.get('id')}). "
"The expanded search must consider ALL results, not just the first."
)
class TestIgdbReleaseDates:
def test_build_igdb_rom_prefers_region_specific_release_date(self):
handler = IGDBHandler()
game = _make_game(5379, "Ecco: The Tides of Time")
game["first_release_date"] = 672537600 # 1991-04-16
game["release_dates"] = [
{"date": 632448000, "region": 2, "platform": {"id": GENESIS_IGDB_ID}}, # US
{"date": 593568000, "region": 5, "platform": {"id": GENESIS_IGDB_ID}}, # JP
]
rom = build_igdb_rom(
handler=handler,
rom=game,
preferred_locale="ja-JP",
preferred_release_regions=[5, 2, 1],
platform_igdb_id=GENESIS_IGDB_ID,
)
assert rom["igdb_metadata"]["first_release_date"] == 593568000
def test_get_igdb_preferred_release_regions_prefers_rom_tags(self):
rom = MagicMock()
rom.regions = ["Japan", "USA"]
with patch(
"handler.metadata.igdb_handler.cm.get_config",
return_value=MagicMock(SCAN_REGION_PRIORITY=["us", "eu"]),
):
regions = get_igdb_preferred_release_regions(rom=rom)
assert regions[0] == 2 # North America (US)
assert regions[1] == 5 # Japan