Revert "Update DailyRewardClient error handler"

This commit is contained in:
洛水居室 2023-07-19 13:27:55 +08:00
parent 409e0058f9
commit abd73f4050
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC

View File

@ -81,7 +81,7 @@ class DailyRewardClient(BaseClient):
url = base_url / endpoint
daily_reward = await self.request_lab(
return await self.request_lab(
url,
method,
params=params,
@ -89,20 +89,6 @@ 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,
*,
@ -232,7 +218,7 @@ class DailyRewardClient(BaseClient):
Returns:
If `reward` is True, a DailyReward object representing the claimed reward. Otherwise, None.
"""
await self.request_daily_reward(
daily_reward = await self.request_daily_reward(
"sign",
method="POST",
game=game or self.game,
@ -241,6 +227,17 @@ 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