From 1a5279a3ef3a9dfb850e7c2ff85a126ac2eb0942 Mon Sep 17 00:00:00 2001 From: zxhlyh Date: Mon, 18 Dec 2023 17:54:16 +0800 Subject: [PATCH] fix: get billing info in self-hosted edition from current workspace (#1774) --- api/services/workspace_service.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/api/services/workspace_service.py b/api/services/workspace_service.py index 360eb25559..6aa7198a04 100644 --- a/api/services/workspace_service.py +++ b/api/services/workspace_service.py @@ -1,3 +1,4 @@ +from flask import current_app from flask_login import current_user from extensions.ext_database import db from models.account import Tenant, TenantAccountJoin, TenantAccountJoinRole @@ -31,10 +32,12 @@ class WorkspaceService: ).first() tenant_info['role'] = tenant_account_join.role - billing_info = BillingService.get_info(tenant_info['id']) + edition = current_app.config['EDITION'] + if edition == 'CLOUD': + billing_info = BillingService.get_info(tenant_info['id']) - if billing_info['can_replace_logo'] and TenantService.has_roles(tenant, [TenantAccountJoinRole.OWNER, TenantAccountJoinRole.ADMIN]): - tenant_info['custom_config'] = tenant.custom_config_dict + if billing_info['can_replace_logo'] and TenantService.has_roles(tenant, [TenantAccountJoinRole.OWNER, TenantAccountJoinRole.ADMIN]): + tenant_info['custom_config'] = tenant.custom_config_dict # Get providers providers = db.session.query(Provider).filter(