PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-09-28 03:46:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Add handling for InvalidDevice exception to PublicCookiesService

Browse Source
This commit is contained in:
luoshuijs 2023-11-23 00:51:46 +08:00 committed by GitHub
parent c4c452fb60
commit 30ad0151e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core/services/cookies/services.py
View File

@ -11,7 +11,7 @@ from gram_core.services.cookies.services import (
) )
from simnet import GenshinClient, Region, Game from simnet import GenshinClient, Region, Game
from simnet.errors import InvalidCookies, TooManyRequests, BadRequest as SimnetBadRequest, NeedChallenge from simnet.errors import InvalidCookies, TooManyRequests, BadRequest as SimnetBadRequest, NeedChallenge, InvalidDevice
from utils.log import logger from utils.log import logger
@ -79,6 +79,11 @@ class PublicCookiesService(BaseService, BasePublicCookiesService):
await self.set_device_valid(client.account_id, False) await self.set_device_valid(client.account_id, False)
await self._cache.delete_public_cookies(cookies.user_id, region) await self._cache.delete_public_cookies(cookies.user_id, region)
raise NeedContinue raise NeedContinue
except InvalidDevice:
logger.warning("用户 [%s] 设备信息无效", public_id)
await self.set_device_valid(client.account_id, False)
await self._cache.delete_public_cookies(cookies.user_id, region)
raise NeedContinue
except SimnetBadRequest as exc: except SimnetBadRequest as exc:
if "invalid content type" in exc.message: if "invalid content type" in exc.message:
raise exc raise exc