Spaces:
Sleeping
Sleeping
File size: 4,576 Bytes
3078fb9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import datetime
import os
from dotenv import load_dotenv
from APIs.api_types import IGDBType
from APIs.http_client import HttpClient
class IGDB:
def __init__(self):
load_dotenv()
self.client_id = os.getenv("IGDB_CLIENT_ID")
self.client_secret = os.getenv("IGDB_CLIENT_SECRET")
self.access_token = None
self.http = HttpClient()
# Format pattern to get release date 2025-04-24 00:00:00+00:00
self.format_pattern = "%Y-%m-%d %H:%M:%S+00:00"
def _get_access_token(self):
"""
Authenticates with Twitch to get the Bearer Token required for IGDB.
"""
response = self.http.post(
"https://id.twitch.tv/oauth2/token",
params={
"client_id": self.client_id,
"client_secret": self.client_secret,
"grant_type": "client_credentials",
},
)
self.access_token = response.json()["access_token"]
return self.access_token
def _query(self, body: str):
"""POST an APICALYPSE query; on 401 (expired token) refresh once and retry."""
import requests as _requests
headers = {
"Client-ID": self.client_id,
"Authorization": f"Bearer {self.access_token}",
"Content-Type": "text/plain",
}
try:
return self.http.post("https://api.igdb.com/v4/games", headers=headers, data=body).json()
except _requests.HTTPError as e:
if e.response is not None and e.response.status_code == 401:
self._get_access_token()
headers["Authorization"] = f"Bearer {self.access_token}"
return self.http.post("https://api.igdb.com/v4/games", headers=headers, data=body).json()
raise
def search(self, game_name: str, max_n: int = 1) -> list[IGDBType]:
"""
Searches for games by name and retrieves details.
Returns a list of IGDBType objects.
"""
if not self.access_token:
self._get_access_token()
# IGDB's APICALYPSE query language wants the raw search string with only
# backslashes and double-quotes escaped — NOT URL-encoded. Using quote()
# here (the old behavior) turned 'The Witcher 3: Wild Hunt' into
# 'The%20Witcher%203%3A...' which matched nothing; only punctuation-free
# names happened to work.
safe_name = game_name.replace("\\", "\\\\").replace('"', '\\"')
body = f"""
fields name, game_modes.name, game_type.type, keywords.name, language_supports.language.name, platforms.name, player_perspectives.name, themes.name, rating, summary, first_release_date, genres.name, cover.url, involved_companies.company.name, involved_companies.developer, involved_companies.publisher;
search "{safe_name}";
limit {max_n};
"""
results = self._query(body)
igdb_results = []
for game in results:
companies = game.get("involved_companies", []) or []
developers = [c["company"]["name"] for c in companies
if c.get("developer") and c.get("company", {}).get("name")]
publishers = [c["company"]["name"] for c in companies
if c.get("publisher") and c.get("company", {}).get("name")]
igdb_obj = IGDBType(
name=game.get("name"),
game_modes=[g["name"] for g in game.get("game_modes", [])],
game_type=game.get("game_type", {}).get("type"),
keywords=[g["name"] for g in game.get("keywords", [])],
language_supports=[g["language"]["name"] for g in game.get("language_supports", [])],
platforms=[g["name"] for g in game.get("platforms", [])],
player_perspectives=[g["name"] for g in game.get("player_perspectives", [])],
themes=[g["name"] for g in game.get("themes", [])],
igdb_rating=float(game.get("rating")) / 100.0 if game.get("rating") else 0.0,
release=datetime.datetime.fromtimestamp(int(game.get("first_release_date", "0")), datetime.timezone.utc).strftime("%Y-%m-%d"),
genres=[g["name"] for g in game.get("genres", [])],
cover_url=[game.get("cover", {}).get("url")],
description=game.get("summary"),
developers=developers,
publishers=publishers,
)
igdb_results.append(igdb_obj)
return igdb_results
|