Parcourir la source

Feat/account not found (#10804)

tags/0.11.2
Garfield Dai il y a 11 mois
Parent
révision
c49efc0c22
Aucun compte lié à l'adresse e-mail de l'auteur

+ 2
- 2
api/controllers/console/auth/forgot_password.py Voir le fichier

InvalidTokenError, InvalidTokenError,
PasswordMismatchError, PasswordMismatchError,
) )
from controllers.console.error import EmailSendIpLimitError, NotAllowedRegister
from controllers.console.error import AccountNotFound, EmailSendIpLimitError
from controllers.console.wraps import setup_required from controllers.console.wraps import setup_required
from events.tenant_event import tenant_was_created from events.tenant_event import tenant_was_created
from extensions.ext_database import db from extensions.ext_database import db
token = AccountService.send_reset_password_email(email=args["email"], language=language) token = AccountService.send_reset_password_email(email=args["email"], language=language)
return {"result": "fail", "data": token, "code": "account_not_found"} return {"result": "fail", "data": token, "code": "account_not_found"}
else: else:
raise NotAllowedRegister()
raise AccountNotFound()
else: else:
token = AccountService.send_reset_password_email(account=account, email=args["email"], language=language) token = AccountService.send_reset_password_email(account=account, email=args["email"], language=language)



+ 4
- 4
api/controllers/console/auth/login.py Voir le fichier

) )
from controllers.console.error import ( from controllers.console.error import (
AccountBannedError, AccountBannedError,
AccountNotFound,
EmailSendIpLimitError, EmailSendIpLimitError,
NotAllowedCreateWorkspace, NotAllowedCreateWorkspace,
NotAllowedRegister,
) )
from controllers.console.wraps import setup_required from controllers.console.wraps import setup_required
from events.tenant_event import tenant_was_created from events.tenant_event import tenant_was_created
token = AccountService.send_reset_password_email(email=args["email"], language=language) token = AccountService.send_reset_password_email(email=args["email"], language=language)
return {"result": "fail", "data": token, "code": "account_not_found"} return {"result": "fail", "data": token, "code": "account_not_found"}
else: else:
raise NotAllowedRegister()
raise AccountNotFound()
# SELF_HOSTED only have one workspace # SELF_HOSTED only have one workspace
tenants = TenantService.get_join_tenants(account) tenants = TenantService.get_join_tenants(account)
if len(tenants) == 0: if len(tenants) == 0:
if FeatureService.get_system_features().is_allow_register: if FeatureService.get_system_features().is_allow_register:
token = AccountService.send_reset_password_email(email=args["email"], language=language) token = AccountService.send_reset_password_email(email=args["email"], language=language)
else: else:
raise NotAllowedRegister()
raise AccountNotFound()
else: else:
token = AccountService.send_reset_password_email(account=account, language=language) token = AccountService.send_reset_password_email(account=account, language=language)


if FeatureService.get_system_features().is_allow_register: if FeatureService.get_system_features().is_allow_register:
token = AccountService.send_email_code_login_email(email=args["email"], language=language) token = AccountService.send_email_code_login_email(email=args["email"], language=language)
else: else:
raise NotAllowedRegister()
raise AccountNotFound()
else: else:
token = AccountService.send_email_code_login_email(account=account, language=language) token = AccountService.send_email_code_login_email(account=account, language=language)



+ 2
- 2
api/controllers/console/error.py Voir le fichier

code = 400 code = 400




class NotAllowedRegister(BaseHTTPException):
error_code = "unauthorized"
class AccountNotFound(BaseHTTPException):
error_code = "account_not_found"
description = "Account not found." description = "Account not found."
code = 400 code = 400



+ 2
- 2
api/services/account_service.py Voir le fichier

) -> Account: ) -> Account:
"""create account""" """create account"""
if not FeatureService.get_system_features().is_allow_register and not is_setup: if not FeatureService.get_system_features().is_allow_register and not is_setup:
from controllers.console.error import NotAllowedRegister
from controllers.console.error import AccountNotFound


raise NotAllowedRegister()
raise AccountNotFound()
account = Account() account = Account()
account.email = email account.email = email
account.name = name account.name = name

Chargement…
Annuler
Enregistrer