youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-29 06:58:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: update raise error

Browse Source
This commit is contained in:
Joe
2024-09-26 15:49:37 +08:00
parent 9e4ee2beb1
commit b249f2b9f9
3 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/controllers/console/auth/error.py
View File

@ -49,3 +49,9 @@ class EmailPasswordLoginLimitError(BaseHTTPException):
"Too many incorrect password attempts. Please verify your identity with the email code to complete login."
)
code = 429
class EmailCodeLoginRateLimitExceededError(BaseHTTPException):
error_code = "email_code_login_rate_limit_exceeded"
description = "Too many login emails have been sent. Please try again in 5 minutes."
code = 429

12
api/controllers/console/auth/forgot_password.py
View File

@ -43,14 +43,10 @@ class ForgotPasswordSendEmailApi(Resource):
)
else:
raise NotAllowedRegister()
elif account:
try:
token = AccountService.send_reset_password_email(
account=account, email=args["email"], language=args["language"] or "en-US"
)
except RateLimitExceededError:
logging.warning(f"Rate limit exceeded for email: {args['email']}")
raise PasswordResetRateLimitExceededError()
else:
token = AccountService.send_reset_password_email(
account=account, email=args["email"], language=args["language"] or "en-US"
)
return {"result": "success", "data": token}