PaiGramTeam/SIMNet
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/SIMNet.git synced 2024-11-22 06:17:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Update DailyRewardClient error handler

Browse Source
This commit is contained in:
洛水居室 2023-07-19 10:22:10 +08:00
parent cc06be92fb
commit 9970561cf7
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
simnet/client/components/daily.py
View File

@ -80,7 +80,7 @@ class DailyRewardClient(BaseClient):
url = base_url / endpoint
return await self.request_lab(
daily_reward = await self.request_lab(
url,
method,
params=params,
@ -88,6 +88,20 @@ class DailyRewardClient(BaseClient):
lang=lang,
)
if endpoint == "sign":
if self.region == Region.CHINESE and daily_reward.get("success", 0) == 1:
gt = daily_reward.get("gt", "")
challenge = daily_reward.get("challenge", "")
raise GeetestTriggered(gt, challenge)
if self.region == Region.OVERSEAS:
gt_result = daily_reward.get("gt_result")
if gt_result is not None and gt_result["success"] != 0:
gt = gt_result.get("gt", "")
challenge = gt_result.get("challenge", "")
raise GeetestTriggered(gt, challenge)
return daily_reward
async def get_reward_info(
self,
*,
@ -217,7 +231,7 @@ class DailyRewardClient(BaseClient):
Returns:
If `reward` is True, a DailyReward object representing the claimed reward. Otherwise, None.
"""
daily_reward = await self.request_daily_reward(
await self.request_daily_reward(
"sign",
method="POST",
game=game or self.game,
@ -226,17 +240,6 @@ class DailyRewardClient(BaseClient):
validate=validate,
)
if self.region == Region.CHINESE and daily_reward.get("success", 0) == 1:
gt = daily_reward.get("gt", "")
challenge = daily_reward.get("challenge", "")
raise GeetestTriggered(gt, challenge)
if self.region == Region.OVERSEAS:
gt_result = daily_reward.get("gt_result")
if gt_result is not None and gt_result["success"] != 0:
gt = gt_result.get("gt", "")
challenge = gt_result.get("challenge", "")
raise GeetestTriggered(gt, challenge)
if not reward:
return None