fix: status 522 cannot reload

PaiGramTeam · Jan 16, 2024 · b37691c · b37691c
1 parent c78b941
commit b37691c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/func/client.py b/func/client.py
@@ -1,6 +1,6 @@
 import asyncio
 
-from httpx import AsyncClient, TimeoutException
+from httpx import AsyncClient, TimeoutException, HTTPStatusError
 
 headers = {
     "authority": "api-static.mihoyo.com",
@@ -22,9 +22,15 @@
 
 def retry(func):
     async def wrapper(*args, **kwargs):
-        for i in range(3):
+        for i in range(5):
             try:
                 return await func(*args, **kwargs)
+            except HTTPStatusError as exc:
+                if exc.response.status_code == 522:
+                    print(f"重试 {func.__name__} {i + 1} 次")
+                    await asyncio.sleep(1)
+                else:
+                    raise exc
             except TimeoutException:
                 print(f"重试 {func.__name__} {i + 1} 次")
                 await asyncio.sleep(1)