Underground-Digital/Workflow-Engine

1import datetime2 3from flask import request4from flask_restful import Resource, reqparse5 6from constants.languages import supported_language7from controllers.console import api8from controllers.console.error import AlreadyActivateError9from extensions.ext_database import db10from libs.helper import StrLen, email, extract_remote_ip, timezone11from models.account import AccountStatus, Tenant12from services.account_service import AccountService, RegisterService13 14 15class ActivateCheckApi(Resource):16    def get(self):17        parser = reqparse.RequestParser()18        parser.add_argument("workspace_id", type=str, required=False, nullable=True, location="args")19        parser.add_argument("email", type=email, required=False, nullable=True, location="args")20        parser.add_argument("token", type=str, required=True, nullable=False, location="args")21        args = parser.parse_args()22 23        workspaceId = args["workspace_id"]24        reg_email = args["email"]25        token = args["token"]26 27        invitation = RegisterService.get_invitation_if_token_valid(workspaceId, reg_email, token)28        if invitation:29            data = invitation.get("data", {})30            tenant: Tenant = invitation.get("tenant", None)31            workspace_name = tenant.name if tenant else None32            workspace_id = tenant.id if tenant else None33            invitee_email = data.get("email") if data else None34            return {35                "is_valid": invitation is not None,36                "data": {"workspace_name": workspace_name, "workspace_id": workspace_id, "email": invitee_email},37            }38        else:39            return {"is_valid": False}40 41 42class ActivateApi(Resource):43    def post(self):44        parser = reqparse.RequestParser()45        parser.add_argument("workspace_id", type=str, required=False, nullable=True, location="json")46        parser.add_argument("email", type=email, required=False, nullable=True, location="json")47        parser.add_argument("token", type=str, required=True, nullable=False, location="json")48        parser.add_argument("name", type=StrLen(30), required=True, nullable=False, location="json")49        parser.add_argument(50            "interface_language", type=supported_language, required=True, nullable=False, location="json"51        )52        parser.add_argument("timezone", type=timezone, required=True, nullable=False, location="json")53        args = parser.parse_args()54 55        invitation = RegisterService.get_invitation_if_token_valid(args["workspace_id"], args["email"], args["token"])56        if invitation is None:57            raise AlreadyActivateError()58 59        RegisterService.revoke_token(args["workspace_id"], args["email"], args["token"])60 61        account = invitation["account"]62        account.name = args["name"]63 64        account.interface_language = args["interface_language"]65        account.timezone = args["timezone"]66        account.interface_theme = "light"67        account.status = AccountStatus.ACTIVE.value68        account.initialized_at = datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None)69        db.session.commit()70 71        token_pair = AccountService.login(account, ip_address=extract_remote_ip(request))72 73        return {"result": "success", "data": token_pair.model_dump()}74 75 76api.add_resource(ActivateCheckApi, "/activate/check")77api.add_resource(ActivateApi, "/activate")78