mirror of
https://github.com/langgenius/dify.git
synced 2026-04-15 00:00:21 -04:00
e37aaa482d
Co-authored-by: ai-hpc <ai-hpc@users.noreply.github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
from flask import request
|
|
from flask_restx import Resource
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
from constants.languages import supported_language
|
|
from controllers.common.schema import register_schema_models
|
|
from controllers.console import console_ns
|
|
from controllers.console.error import AlreadyActivateError
|
|
from extensions.ext_database import db
|
|
from libs.datetime_utils import naive_utc_now
|
|
from libs.helper import EmailStr, timezone
|
|
from models import AccountStatus
|
|
from services.account_service import RegisterService
|
|
|
|
|
|
class ActivateCheckQuery(BaseModel):
|
|
workspace_id: str | None = Field(default=None)
|
|
email: EmailStr | None = Field(default=None)
|
|
token: str
|
|
|
|
|
|
class ActivatePayload(BaseModel):
|
|
workspace_id: str | None = Field(default=None)
|
|
email: EmailStr | None = Field(default=None)
|
|
token: str
|
|
name: str = Field(..., max_length=30)
|
|
interface_language: str = Field(...)
|
|
timezone: str = Field(...)
|
|
|
|
@field_validator("interface_language")
|
|
@classmethod
|
|
def validate_lang(cls, value: str) -> str:
|
|
return supported_language(value)
|
|
|
|
@field_validator("timezone")
|
|
@classmethod
|
|
def validate_tz(cls, value: str) -> str:
|
|
return timezone(value)
|
|
|
|
|
|
class ActivationCheckResponse(BaseModel):
|
|
is_valid: bool = Field(description="Whether token is valid")
|
|
data: dict | None = Field(default=None, description="Activation data if valid")
|
|
|
|
|
|
class ActivationResponse(BaseModel):
|
|
result: str = Field(description="Operation result")
|
|
|
|
|
|
register_schema_models(console_ns, ActivateCheckQuery, ActivatePayload, ActivationCheckResponse, ActivationResponse)
|
|
|
|
|
|
@console_ns.route("/activate/check")
|
|
class ActivateCheckApi(Resource):
|
|
@console_ns.doc("check_activation_token")
|
|
@console_ns.doc(description="Check if activation token is valid")
|
|
@console_ns.expect(console_ns.models[ActivateCheckQuery.__name__])
|
|
@console_ns.response(
|
|
200,
|
|
"Success",
|
|
console_ns.models[ActivationCheckResponse.__name__],
|
|
)
|
|
def get(self):
|
|
args = ActivateCheckQuery.model_validate(request.args.to_dict(flat=True)) # type: ignore
|
|
|
|
workspaceId = args.workspace_id
|
|
token = args.token
|
|
|
|
invitation = RegisterService.get_invitation_with_case_fallback(workspaceId, args.email, token)
|
|
if invitation:
|
|
data = invitation.get("data", {})
|
|
tenant = invitation.get("tenant", None)
|
|
|
|
# Check workspace permission
|
|
if tenant:
|
|
from libs.workspace_permission import check_workspace_member_invite_permission
|
|
|
|
check_workspace_member_invite_permission(tenant.id)
|
|
|
|
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}
|
|
|
|
|
|
@console_ns.route("/activate")
|
|
class ActivateApi(Resource):
|
|
@console_ns.doc("activate_account")
|
|
@console_ns.doc(description="Activate account with invitation token")
|
|
@console_ns.expect(console_ns.models[ActivatePayload.__name__])
|
|
@console_ns.response(
|
|
200,
|
|
"Account activated successfully",
|
|
console_ns.models[ActivationResponse.__name__],
|
|
)
|
|
@console_ns.response(400, "Already activated or invalid token")
|
|
def post(self):
|
|
args = ActivatePayload.model_validate(console_ns.payload)
|
|
|
|
normalized_request_email = args.email.lower() if args.email else None
|
|
invitation = RegisterService.get_invitation_with_case_fallback(args.workspace_id, args.email, args.token)
|
|
if invitation is None:
|
|
raise AlreadyActivateError()
|
|
|
|
RegisterService.revoke_token(args.workspace_id, normalized_request_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
|
|
account.initialized_at = naive_utc_now()
|
|
db.session.commit()
|
|
|
|
return {"result": "success"}
|