Sfoglia il codice sorgente

fix: get billing info in self-hosted edition from current workspace (#1774)

zxhlyh 1 anno fa
parent
commit
1a5279a3ef
1 ha cambiato i file con 6 aggiunte e 3 eliminazioni
  1. 6 3
      api/services/workspace_service.py

+ 6 - 3
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(