feat(fetcher): refresh access_token automatically

This commit is contained in:
MingxuanGame
2025-08-10 11:44:09 +00:00
parent 02051289e9
commit f6c375caf3
3 changed files with 25 additions and 15 deletions

View File

@@ -38,6 +38,22 @@ class BaseFetcher:
"Content-Type": "application/json",
}
async def request_api(self, url: str, method: str = "GET", **kwargs) -> dict:
if self.is_token_expired():
await self.refresh_access_token()
header = kwargs.pop("headers", {})
header = self.header
async with AsyncClient() as client:
response = await client.request(
method,
url,
headers=header,
**kwargs,
)
response.raise_for_status()
return response.json()
def is_token_expired(self) -> bool:
return self.token_expiry <= int(time.time())

View File

@@ -5,8 +5,6 @@ from app.log import logger
from ._base import BaseFetcher
from httpx import AsyncClient
class BeatmapFetcher(BaseFetcher):
async def get_beatmap(
@@ -21,11 +19,10 @@ class BeatmapFetcher(BaseFetcher):
logger.opt(colors=True).debug(
f"<blue>[BeatmapFetcher]</blue> get_beatmap: <y>{params}</y>"
)
async with AsyncClient() as client:
response = await client.get(
return BeatmapResp.model_validate(
await self.request_api(
"https://osu.ppy.sh/api/v2/beatmaps/lookup",
headers=self.header,
params=params,
)
response.raise_for_status()
return BeatmapResp.model_validate(response.json())
)

View File

@@ -5,18 +5,15 @@ from app.log import logger
from ._base import BaseFetcher
from httpx import AsyncClient
class BeatmapsetFetcher(BaseFetcher):
async def get_beatmapset(self, beatmap_set_id: int) -> BeatmapsetResp:
logger.opt(colors=True).debug(
f"<blue>[BeatmapsetFetcher]</blue> get_beatmapset: <y>{beatmap_set_id}</y>"
)
async with AsyncClient() as client:
response = await client.get(
f"https://osu.ppy.sh/api/v2/beatmapsets/{beatmap_set_id}",
headers=self.header,
return BeatmapsetResp.model_validate(
await self.request_api(
f"https://osu.ppy.sh/api/v2/beatmapsets/{beatmap_set_id}"
)
response.raise_for_status()
return BeatmapsetResp.model_validate(response.json())
)