fix: wrong params

This commit is contained in:
GareArc 2025-03-10 02:30:17 -04:00
parent 31fed12bbc
commit d8575a4537
2 changed files with 19 additions and 17 deletions

View File

@ -3,20 +3,25 @@ import datetime
import pytz import pytz
from flask import request from flask import request
from flask_login import current_user # type: ignore from flask_login import current_user # type: ignore
from flask_restful import (Resource, fields, marshal_with, # type: ignore from flask_restful import Resource, fields, marshal_with, reqparse # type: ignore
reqparse)
from configs import dify_config from configs import dify_config
from constants.languages import supported_language from constants.languages import supported_language
from controllers.console import api from controllers.console import api
from controllers.console.workspace.error import ( from controllers.console.workspace.error import (
AccountAlreadyInitedError, CurrentPasswordIncorrectError, AccountAlreadyInitedError,
InvalidAccountDeletionCodeError, InvalidInvitationCodeError, CurrentPasswordIncorrectError,
RepeatPasswordNotMatchError) InvalidAccountDeletionCodeError,
from controllers.console.wraps import (account_initialization_required, InvalidInvitationCodeError,
cloud_edition_billing_enabled, RepeatPasswordNotMatchError,
enterprise_license_required, )
only_edition_cloud, setup_required) from controllers.console.wraps import (
account_initialization_required,
cloud_edition_billing_enabled,
enterprise_license_required,
only_edition_cloud,
setup_required,
)
from extensions.ext_database import db from extensions.ext_database import db
from fields.member_fields import account_fields from fields.member_fields import account_fields
from libs.helper import TimestampField, timezone from libs.helper import TimestampField, timezone
@ -24,8 +29,7 @@ from libs.login import login_required
from models import AccountIntegrate, InvitationCode from models import AccountIntegrate, InvitationCode
from services.account_service import AccountService from services.account_service import AccountService
from services.billing_service import BillingService from services.billing_service import BillingService
from services.errors.account import \ from services.errors.account import CurrentPasswordIncorrectError as ServiceCurrentPasswordIncorrectError
CurrentPasswordIncorrectError as ServiceCurrentPasswordIncorrectError
class AccountInitApi(Resource): class AccountInitApi(Resource):

View File

@ -2,8 +2,7 @@ import os
from typing import Literal, Optional from typing import Literal, Optional
import httpx import httpx
from tenacity import (retry, retry_if_exception_type, stop_before_delay, from tenacity import retry, retry_if_exception_type, stop_before_delay, wait_fixed
wait_fixed)
from extensions.ext_database import db from extensions.ext_database import db
from libs.helper import RateLimiter from libs.helper import RateLimiter
@ -95,8 +94,8 @@ class BillingService:
return cls._send_request("POST", "/account/delete-feedback", json=json) return cls._send_request("POST", "/account/delete-feedback", json=json)
class EducationIdentity: class EducationIdentity:
verification_rate_limit = RateLimiter(prefix="edu_verification_rate_limit", limit=10, period=60) verification_rate_limit = RateLimiter(prefix="edu_verification_rate_limit", max_attempts=10, time_window=60)
activation_rate_limit = RateLimiter(prefix="edu_activation_rate_limit", limit=10, period=60) activation_rate_limit = RateLimiter(prefix="edu_activation_rate_limit", max_attempts=10, time_window=60)
@classmethod @classmethod
def verify(cls, account_id: str, account_email: str): def verify(cls, account_id: str, account_email: str):
@ -118,8 +117,7 @@ class BillingService:
@classmethod @classmethod
def activate(cls, account: Account, token: str, institution: str): def activate(cls, account: Account, token: str, institution: str):
if cls.activation_rate_limit.is_rate_limited(account.email): if cls.activation_rate_limit.is_rate_limited(account.email):
from controllers.console.error import \ from controllers.console.error import EducationActivateLimitError
EducationActivateLimitError
raise EducationActivateLimitError() raise EducationActivateLimitError()