🐛 Add TooManyRequestPublicCookies to error handler

This commit is contained in:
xtaodada 2024-06-09 21:30:44 +08:00
parent 46aa8461db
commit 8b17c74d9e
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659

View File

@ -25,7 +25,7 @@ from telegram.helpers import create_deep_linked_url
from core.config import config from core.config import config
from core.plugin import Plugin, error_handler from core.plugin import Plugin, error_handler
from gram_core.services.cookies.error import CookieServiceError as PublicCookieServiceError from gram_core.services.cookies.error import TooManyRequestPublicCookies
from gram_core.services.players.error import PlayerNotFoundError from gram_core.services.players.error import PlayerNotFoundError
from modules.apihelper.error import APIHelperException, APIHelperTimedOut, ResponseException, ReturnCodeError from modules.apihelper.error import APIHelperException, APIHelperTimedOut, ResponseException, ReturnCodeError
from modules.errorpush import ( from modules.errorpush import (
@ -274,9 +274,9 @@ class ErrorHandler(Plugin):
@error_handler() @error_handler()
async def process_public_cookies(self, update: object, context: CallbackContext): async def process_public_cookies(self, update: object, context: CallbackContext):
if not isinstance(context.error, PublicCookieServiceError) or not isinstance(update, Update): if not isinstance(context.error, TooManyRequestPublicCookies) or not isinstance(update, Update):
return return
self.create_notice_task(update, context, "公共Cookies池已经耗尽请稍后重试或者绑定账号") self.create_notice_task(update, context, config.notice.user_not_found)
raise ApplicationHandlerStop raise ApplicationHandlerStop
@error_handler(block=False) @error_handler(block=False)