57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from enum import StrEnum
|
||
|
|
|
||
|
|
|
||
|
|
class LineErrorCode(StrEnum):
|
||
|
|
INVALID_REPLY_TOKEN = "invalid_reply_token"
|
||
|
|
REPLY_TOKEN_EXPIRED = "reply_token_expired"
|
||
|
|
INVALID_CHANNEL_ACCESS_TOKEN = "invalid_channel_access_token"
|
||
|
|
CHANNEL_ACCESS_TOKEN_EXPIRED = "channel_access_token_expired"
|
||
|
|
RATE_LIMITED = "rate_limited"
|
||
|
|
INVALID_MESSAGE = "invalid_message"
|
||
|
|
USER_NOT_FOUND = "user_not_found"
|
||
|
|
GROUP_NOT_FOUND = "group_not_found"
|
||
|
|
NETWORK_ERROR = "network_error"
|
||
|
|
UNKNOWN = "unknown"
|
||
|
|
|
||
|
|
|
||
|
|
ERROR_PATTERN_MAP: dict[str, LineErrorCode] = {
|
||
|
|
"Invalid reply token": LineErrorCode.INVALID_REPLY_TOKEN,
|
||
|
|
"Reply token expired": LineErrorCode.REPLY_TOKEN_EXPIRED,
|
||
|
|
"Reply token has already been used": LineErrorCode.REPLY_TOKEN_EXPIRED,
|
||
|
|
"Invalid channel access token": LineErrorCode.INVALID_CHANNEL_ACCESS_TOKEN,
|
||
|
|
"The channel access token expired": LineErrorCode.CHANNEL_ACCESS_TOKEN_EXPIRED,
|
||
|
|
"Rate limit exceeded": LineErrorCode.RATE_LIMITED,
|
||
|
|
"Invalid message": LineErrorCode.INVALID_MESSAGE,
|
||
|
|
"user not found": LineErrorCode.USER_NOT_FOUND,
|
||
|
|
"group not found": LineErrorCode.GROUP_NOT_FOUND,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def classify_line_error(status_code: int, error_message: str) -> LineErrorCode:
|
||
|
|
msg_lower = error_message.lower()
|
||
|
|
|
||
|
|
for pattern, code in ERROR_PATTERN_MAP.items():
|
||
|
|
if pattern.lower() in msg_lower:
|
||
|
|
return code
|
||
|
|
|
||
|
|
if status_code == 401:
|
||
|
|
return LineErrorCode.INVALID_CHANNEL_ACCESS_TOKEN
|
||
|
|
if status_code == 429:
|
||
|
|
return LineErrorCode.RATE_LIMITED
|
||
|
|
if status_code >= 500:
|
||
|
|
return LineErrorCode.NETWORK_ERROR
|
||
|
|
|
||
|
|
return LineErrorCode.UNKNOWN
|
||
|
|
|
||
|
|
|
||
|
|
def is_reply_token_expired(error: LineErrorCode) -> bool:
|
||
|
|
return error in (LineErrorCode.REPLY_TOKEN_EXPIRED, LineErrorCode.INVALID_REPLY_TOKEN)
|
||
|
|
|
||
|
|
|
||
|
|
def is_retryable(error: LineErrorCode) -> bool:
|
||
|
|
return error in (
|
||
|
|
LineErrorCode.RATE_LIMITED,
|
||
|
|
LineErrorCode.NETWORK_ERROR,
|
||
|
|
)
|