mirror of
https://github.com/rommapp/romm.git
synced 2026-06-30 07:45:52 +00:00
b61bb11bce
This change avoids blocking requests when retrieving covers from SteamGridDB, which is the main bottleneck as the current implementation iterates over paginated results for multiple games. Using an asynchronous client like `httpx` provides a good performance improvement, and reduces the latency when calling this endpoint. Also, the inclusion of FastAPI `lifespan` allows instantiating a single client on startup. When testing with "Final Fantasy V Advance", the endpoint goes from ~9s to ~1.5s to retrieve all covers.
106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
import asyncio
|
|
import itertools
|
|
from typing import Any, Final
|
|
|
|
import httpx
|
|
from config import STEAMGRIDDB_API_KEY
|
|
from logger.logger import log
|
|
|
|
# Used to display the Mobygames API status in the frontend
|
|
STEAMGRIDDB_API_ENABLED: Final = bool(STEAMGRIDDB_API_KEY)
|
|
|
|
# SteamGridDB dimensions
|
|
STEAMVERTICAL: Final = "600x900"
|
|
GALAXY342: Final = "342x482"
|
|
GALAXY660: Final = "660x930"
|
|
SQUARE512: Final = "512x512"
|
|
SQUARE1024: Final = "1024x1024"
|
|
|
|
# SteamGridDB types
|
|
STATIC: Final = "static"
|
|
ANIMATED: Final = "animated"
|
|
|
|
SGDB_API_COVER_LIMIT: Final = 50
|
|
|
|
|
|
class SGDBBaseHandler:
|
|
def __init__(self) -> None:
|
|
self.BASE_URL = "https://www.steamgriddb.com/api/v2"
|
|
self.search_endpoint = f"{self.BASE_URL}/search/autocomplete"
|
|
self.grid_endpoint = f"{self.BASE_URL}/grids/game"
|
|
self.headers = {
|
|
"Authorization": f"Bearer {STEAMGRIDDB_API_KEY}",
|
|
"Accept": "*/*",
|
|
}
|
|
|
|
async def get_details(
|
|
self, requests_client: httpx.AsyncClient, search_term: str
|
|
) -> list[dict[str, Any]]:
|
|
search_response = (
|
|
await requests_client.get(
|
|
f"{self.search_endpoint}/{search_term}",
|
|
headers=self.headers,
|
|
timeout=120,
|
|
)
|
|
).json()
|
|
|
|
if len(search_response["data"]) == 0:
|
|
log.warning(f"Could not find '{search_term}' on SteamGridDB")
|
|
return []
|
|
|
|
tasks = [
|
|
self._get_game_covers(
|
|
requests_client=requests_client,
|
|
game_id=game["id"],
|
|
game_name=game["name"],
|
|
)
|
|
for game in search_response["data"]
|
|
]
|
|
results = await asyncio.gather(*tasks)
|
|
|
|
return list(filter(None, results))
|
|
|
|
async def _get_game_covers(
|
|
self, requests_client: httpx.AsyncClient, game_id: int, game_name: str
|
|
) -> dict[str, Any] | None:
|
|
game_covers = []
|
|
for page in itertools.count(start=0):
|
|
covers_response = (
|
|
await requests_client.get(
|
|
f"{self.grid_endpoint}/{game_id}",
|
|
headers=self.headers,
|
|
timeout=120,
|
|
params={
|
|
"dimensions": f"{STEAMVERTICAL},{GALAXY342},{GALAXY660},{SQUARE512},{SQUARE1024}",
|
|
"types": f"{STATIC},{ANIMATED}",
|
|
"limit": SGDB_API_COVER_LIMIT,
|
|
"page": page,
|
|
},
|
|
)
|
|
).json()
|
|
page_covers = covers_response["data"]
|
|
|
|
game_covers.extend(page_covers)
|
|
if len(page_covers) < SGDB_API_COVER_LIMIT:
|
|
break
|
|
|
|
if not game_covers:
|
|
return None
|
|
|
|
return {
|
|
"name": game_name,
|
|
"resources": [
|
|
{
|
|
"thumb": cover["thumb"],
|
|
"url": cover["url"],
|
|
"type": (
|
|
"animated" if cover["thumb"].endswith(".webm") else "static"
|
|
),
|
|
}
|
|
for cover in game_covers
|
|
],
|
|
}
|
|
|
|
|
|
sgdb_handler = SGDBBaseHandler()
|