mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-02 06:00:36 +08:00
feat(license): refactor available team members handling in feature service.
- add tenant_id support to info request methods
This commit is contained in:
parent
2e419ca3ba
commit
18362c38e7
@ -17,6 +17,7 @@ from libs.login import login_required
|
|||||||
from models.account import Account, TenantAccountRole
|
from models.account import Account, TenantAccountRole
|
||||||
from services.account_service import RegisterService, TenantService
|
from services.account_service import RegisterService, TenantService
|
||||||
from services.errors.account import AccountAlreadyInTenantError
|
from services.errors.account import AccountAlreadyInTenantError
|
||||||
|
from services.feature_service import FeatureService
|
||||||
|
|
||||||
|
|
||||||
class MemberListApi(Resource):
|
class MemberListApi(Resource):
|
||||||
@ -55,7 +56,7 @@ class MemberInviteEmailApi(Resource):
|
|||||||
invitation_results = []
|
invitation_results = []
|
||||||
console_web_url = dify_config.CONSOLE_WEB_URL
|
console_web_url = dify_config.CONSOLE_WEB_URL
|
||||||
if (FeatureService.get_system_features().license.product_id == "DIFY_ENTERPRISE_STANDARD" and
|
if (FeatureService.get_system_features().license.product_id == "DIFY_ENTERPRISE_STANDARD" and
|
||||||
len(invitee_emails) > FeatureService.get_system_features().available_team_members):
|
len(invitee_emails) > FeatureService.get_features(tenant_id=inviter.current_tenant.id).available_team_members):
|
||||||
return {
|
return {
|
||||||
"code": "limit-exceeded",
|
"code": "limit-exceeded",
|
||||||
"message": "Limit exceeded",
|
"message": "Limit exceeded",
|
||||||
@ -80,6 +81,7 @@ class MemberInviteEmailApi(Resource):
|
|||||||
)
|
)
|
||||||
break
|
break
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
print(str(e))
|
||||||
invitation_results.append({"status": "failed", "email": invitee_email, "message": str(e)})
|
invitation_results.append({"status": "failed", "email": invitee_email, "message": str(e)})
|
||||||
|
|
||||||
return {
|
return {
|
||||||
|
@ -585,7 +585,7 @@ class TenantService:
|
|||||||
if not FeatureService.get_system_features().is_allow_create_workspace and not is_setup:
|
if not FeatureService.get_system_features().is_allow_create_workspace and not is_setup:
|
||||||
raise WorkSpaceNotAllowedCreateError()
|
raise WorkSpaceNotAllowedCreateError()
|
||||||
|
|
||||||
if FeatureService.get_system_features().license.product_id == "DIFY_ENTERPRISE_STANDARD" and FeatureService.get_system_features().available_workspaces == 0:
|
if FeatureService.get_system_features().license.product_id == "DIFY_ENTERPRISE_STANDARD" and FeatureService.get_system_features().available_workspaces <= 0:
|
||||||
raise WorkSpaceNotAllowedCreateError()
|
raise WorkSpaceNotAllowedCreateError()
|
||||||
|
|
||||||
if name:
|
if name:
|
||||||
|
@ -35,7 +35,6 @@ class LicenseModel(BaseModel):
|
|||||||
status: LicenseStatus = LicenseStatus.NONE
|
status: LicenseStatus = LicenseStatus.NONE
|
||||||
expired_at: str = ""
|
expired_at: str = ""
|
||||||
product_id: str = ""
|
product_id: str = ""
|
||||||
available_team_members: int = 0
|
|
||||||
available_workspaces: int = 0
|
available_workspaces: int = 0
|
||||||
|
|
||||||
|
|
||||||
@ -59,6 +58,7 @@ class FeatureModel(BaseModel):
|
|||||||
model_load_balancing_enabled: bool = False
|
model_load_balancing_enabled: bool = False
|
||||||
dataset_operator_enabled: bool = False
|
dataset_operator_enabled: bool = False
|
||||||
webapp_copyright_enabled: bool = False
|
webapp_copyright_enabled: bool = False
|
||||||
|
available_team_members: int = 0
|
||||||
|
|
||||||
# pydantic configs
|
# pydantic configs
|
||||||
model_config = ConfigDict(protected_namespaces=())
|
model_config = ConfigDict(protected_namespaces=())
|
||||||
@ -92,6 +92,7 @@ class FeatureService:
|
|||||||
|
|
||||||
if dify_config.ENTERPRISE_ENABLED:
|
if dify_config.ENTERPRISE_ENABLED:
|
||||||
features.webapp_copyright_enabled = True
|
features.webapp_copyright_enabled = True
|
||||||
|
cls._fulfill_parms_from_license(features, tenant_id)
|
||||||
|
|
||||||
return features
|
return features
|
||||||
|
|
||||||
@ -123,6 +124,11 @@ class FeatureService:
|
|||||||
features.model_load_balancing_enabled = dify_config.MODEL_LB_ENABLED
|
features.model_load_balancing_enabled = dify_config.MODEL_LB_ENABLED
|
||||||
features.dataset_operator_enabled = dify_config.DATASET_OPERATOR_ENABLED
|
features.dataset_operator_enabled = dify_config.DATASET_OPERATOR_ENABLED
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _fulfill_parms_from_license(cls, features: FeatureModel, tenant_id: str):
|
||||||
|
license_info = EnterpriseService.get_info(tenant_id)["License"]
|
||||||
|
features.available_team_members = license_info["availableTeamMembers"]
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _fulfill_params_from_billing_api(cls, features: FeatureModel, tenant_id: str):
|
def _fulfill_params_from_billing_api(cls, features: FeatureModel, tenant_id: str):
|
||||||
billing_info = BillingService.get_info(tenant_id)
|
billing_info = BillingService.get_info(tenant_id)
|
||||||
@ -209,8 +215,8 @@ class FeatureService:
|
|||||||
if "productId" in license_info:
|
if "productId" in license_info:
|
||||||
features.license.product_id = license_info["productId"]
|
features.license.product_id = license_info["productId"]
|
||||||
|
|
||||||
if "availableTeamMembers" in license_info:
|
# if "availableTeamMembers" in license_info:
|
||||||
features.license.available_team_members = license_info["availableTeamMembers"]
|
# features.license.available_team_members = license_info["availableTeamMembers"]
|
||||||
|
|
||||||
if "availableWorkspaces" in license_info:
|
if "availableWorkspaces" in license_info:
|
||||||
features.license.available_workspaces = license_info["availableWorkspaces"]
|
features.license.available_workspaces = license_info["availableWorkspaces"]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user