feat(fetcher): refresh access_token automatically

app/fetcher/_base.py

@@ -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())
 

app/fetcher/beatmap.py

@@ -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())

app/fetcher/beatmapset.py

@@ -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(
+        return BeatmapsetResp.model_validate(
-                f"https://osu.ppy.sh/api/v2/beatmapsets/{beatmap_set_id}",
+            await self.request_api(
-                headers=self.header,
+                f"https://osu.ppy.sh/api/v2/beatmapsets/{beatmap_set_id}"
             )
-            response.raise_for_status()
+        )
-            return BeatmapsetResp.model_validate(response.json())