b6abd5b314
* main: (141 commits) fix(workflow/hooks/use-shortcuts): resolve issue of copy shortcut not working in workflow debug and preview panel (#8249) chore: cleanup pycodestyle E rules (#8269) let claude models in bedrock support the response_format parameter (#8220) enhance: improve empty data display for detail panel (#8266) chore: remove useless code (#8198) chore: apply pep8-naming rules for naming convention (#8261) fix:ollama text embedding 500 error (#8252) Update Gitlab query field, add query by path (#8244) editor can also create api key (#8214) fix: upload img icon mis-align in the chat input area (#8263) fix: truthy value (#8208) fix(workflow): IF-ELSE nodes connected to the same subsequent node cause execution to stop (#8247) fix: workflow parallel limit in ifelse node (#8242) fix: CHECK_UPDATE_URL comment (#8235) fix:error when adding the ollama embedding model (#8236) fix: improving the regionalization of translation (#8231) feat: add from_variable_selector for stream chunk / message event (#8228) fix(workflow): answers are output simultaneously across different braches in the question classifier node. (#8225) fix(workflow): in multi-parallel execution with multiple conditional branches (#8221) fix(docker/docker-compose.yaml): Set default value for `REDIS_SENTINEL_SOCKET_TIMEOUT` and `CELERY_SENTINEL_SOCKET_TIMEOUT` (#8218) ...
80 lines
3.4 KiB
Python
80 lines
3.4 KiB
Python
import datetime
|
|
|
|
from flask import request
|
|
from flask_restful import Resource, reqparse
|
|
|
|
from constants.languages import supported_language
|
|
from controllers.console import api
|
|
from controllers.console.error import AlreadyActivateError
|
|
from extensions.ext_database import db
|
|
from libs.helper import StrLen, email, get_remote_ip, timezone
|
|
from libs.password import valid_password
|
|
from models.account import AccountStatus, Tenant
|
|
from services.account_service import AccountService, RegisterService
|
|
|
|
|
|
class ActivateCheckApi(Resource):
|
|
def get(self):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("workspace_id", type=str, required=False, nullable=True, location="args")
|
|
parser.add_argument("email", type=email, required=False, nullable=True, location="args")
|
|
parser.add_argument("token", type=str, required=True, nullable=False, location="args")
|
|
args = parser.parse_args()
|
|
|
|
workspaceId = args["workspace_id"]
|
|
reg_email = args["email"]
|
|
token = args["token"]
|
|
|
|
invitation = RegisterService.get_invitation_if_token_valid(workspaceId, reg_email, token)
|
|
if invitation:
|
|
data = invitation.get("data", {})
|
|
tenant: Tenant = invitation.get("tenant", None)
|
|
workspace_name = tenant.name if tenant else None
|
|
workspace_id = tenant.id if tenant else None
|
|
invitee_email = data.get("email") if data else None
|
|
return {
|
|
"is_valid": invitation is not None,
|
|
"data": {"workspace_name": workspace_name, "workspace_id": workspace_id, "email": invitee_email},
|
|
}
|
|
else:
|
|
return {"is_valid": False}
|
|
|
|
|
|
class ActivateApi(Resource):
|
|
def post(self):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("workspace_id", type=str, required=False, nullable=True, location="json")
|
|
parser.add_argument("email", type=email, required=False, nullable=True, location="json")
|
|
parser.add_argument("token", type=str, required=True, nullable=False, location="json")
|
|
parser.add_argument("name", type=StrLen(30), required=True, nullable=False, location="json")
|
|
parser.add_argument("password", type=valid_password, required=True, nullable=False, location="json")
|
|
parser.add_argument(
|
|
"interface_language", type=supported_language, required=True, nullable=False, location="json"
|
|
)
|
|
parser.add_argument("timezone", type=timezone, required=True, nullable=False, location="json")
|
|
args = parser.parse_args()
|
|
|
|
invitation = RegisterService.get_invitation_if_token_valid(args["workspace_id"], args["email"], args["token"])
|
|
if invitation is None:
|
|
raise AlreadyActivateError()
|
|
|
|
RegisterService.revoke_token(args["workspace_id"], args["email"], args["token"])
|
|
|
|
account = invitation["account"]
|
|
account.name = args["name"]
|
|
|
|
account.interface_language = args["interface_language"]
|
|
account.timezone = args["timezone"]
|
|
account.interface_theme = "light"
|
|
account.status = AccountStatus.ACTIVE.value
|
|
account.initialized_at = datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None)
|
|
db.session.commit()
|
|
|
|
token = AccountService.login(account, ip_address=get_remote_ip(request))
|
|
|
|
return {"result": "success", "data": token}
|
|
|
|
|
|
api.add_resource(ActivateCheckApi, "/activate/check")
|
|
api.add_resource(ActivateApi, "/activate")
|