chore(deps): auto fix by pre-commit hooks
This commit is contained in:
committed by
MingxuanGame
parent
b4fd4e0256
commit
7625cd99f5
@@ -68,9 +68,7 @@ class BaseFetcher:
|
||||
if response.status_code == 401:
|
||||
logger.warning(f"Received 401 error for {url}")
|
||||
await self._clear_tokens()
|
||||
raise TokenAuthError(
|
||||
f"Authentication failed. Please re-authorize using: {self.authorize_url}"
|
||||
)
|
||||
raise TokenAuthError(f"Authentication failed. Please re-authorize using: {self.authorize_url}")
|
||||
|
||||
response.raise_for_status()
|
||||
return response.json()
|
||||
@@ -146,7 +144,7 @@ class BaseFetcher:
|
||||
清除所有 token
|
||||
"""
|
||||
logger.warning(f"Clearing tokens for client {self.client_id}")
|
||||
|
||||
|
||||
# 清除内存中的 token
|
||||
self.access_token = ""
|
||||
self.refresh_token = ""
|
||||
@@ -167,4 +165,4 @@ class BaseFetcher:
|
||||
"has_refresh_token": bool(self.refresh_token),
|
||||
"token_expired": self.is_token_expired(),
|
||||
"authorize_url": self.authorize_url,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -14,12 +14,13 @@ from app.utils import bg_tasks
|
||||
|
||||
from ._base import BaseFetcher
|
||||
|
||||
import redis.asyncio as redis
|
||||
from httpx import AsyncClient
|
||||
import redis.asyncio as redis
|
||||
|
||||
|
||||
class RateLimitError(Exception):
|
||||
"""速率限制异常"""
|
||||
|
||||
pass
|
||||
|
||||
|
||||
@@ -73,9 +74,7 @@ class BeatmapsetFetcher(BaseFetcher):
|
||||
if response.status_code == 401:
|
||||
logger.warning(f"Received 401 error for {url}")
|
||||
await self._clear_tokens()
|
||||
raise TokenAuthError(
|
||||
f"Authentication failed. Please re-authorize using: {self.authorize_url}"
|
||||
)
|
||||
raise TokenAuthError(f"Authentication failed. Please re-authorize using: {self.authorize_url}")
|
||||
|
||||
response.raise_for_status()
|
||||
return response.json()
|
||||
@@ -205,7 +204,9 @@ class BeatmapsetFetcher(BaseFetcher):
|
||||
try:
|
||||
await self.prefetch_next_pages(query, api_response["cursor"], redis_client, pages=1)
|
||||
except RateLimitError:
|
||||
logger.opt(colors=True).info("<yellow>[BeatmapsetFetcher]</yellow> Prefetch skipped due to rate limit")
|
||||
logger.opt(colors=True).info(
|
||||
"<yellow>[BeatmapsetFetcher]</yellow> Prefetch skipped due to rate limit"
|
||||
)
|
||||
|
||||
bg_tasks.add_task(delayed_prefetch)
|
||||
|
||||
@@ -352,4 +353,4 @@ class BeatmapsetFetcher(BaseFetcher):
|
||||
except Exception as e:
|
||||
logger.opt(colors=True).error(
|
||||
f"<red>[BeatmapsetFetcher]</red> Failed to warmup cache for {query.sort}: {e}"
|
||||
)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user