support check api error

This commit is contained in:
xtaodada 2023-05-24 23:39:55 +08:00
parent c9c9ef4f99
commit cd6c283426
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659

View File

@ -89,7 +89,13 @@ async def send_to_user(user: User, tweet: Tweet):
@flood_wait() @flood_wait()
async def send_username_changed(user: str): async def send_username_changed(user: str):
text = f"获取 {user} 的数据失败,可能用户名已改变" text = f"获取 {user} 的数据失败,可能用户名已改变,请考虑移除该用户"
await bot.send_message(owner, text)
@flood_wait()
async def send_api_error():
text = "获取数据过多,可能 API 失效"
await bot.send_message(owner, text) await bot.send_message(owner, text)
@ -111,6 +117,7 @@ async def send_check(user_data: User):
async def check_update(): async def check_update():
logs.info("开始检查更新") logs.info("开始检查更新")
users = UserDB.get_all() users = UserDB.get_all()
failed_users = []
nums = len(users) nums = len(users)
for idx, user in enumerate(users): for idx, user in enumerate(users):
try: try:
@ -120,11 +127,17 @@ async def check_update():
await send_check(user_data) await send_check(user_data)
else: else:
logs.warning(f"获取 {user} 的数据失败,未知原因") logs.warning(f"获取 {user} 的数据失败,未知原因")
failed_users.append(user)
except UsernameNotFound: except UsernameNotFound:
logs.warning(f"获取 {user} 的数据失败,可能用户名已改变") logs.warning(f"获取 {user} 的数据失败,可能用户名已改变")
await send_username_changed(user) failed_users.append(user)
UserDB.remove(user)
except Exception: except Exception:
traceback.print_exc() traceback.print_exc()
logs.info(f"处理完成,剩余 {nums - idx - 1} 个用户") logs.info(f"处理完成,剩余 {nums - idx - 1} 个用户")
if len(failed_users) > 5:
logs.warning("失效数据过多,可能 API 失效")
await send_api_error()
else:
for user in failed_users:
await send_username_changed(user)
logs.info("检查更新完成") logs.info("检查更新完成")