|
@@ -26,6 +26,7 @@ from libs.helper import TimestampField
|
|
|
from libs.login import login_required
|
|
|
from models.account import Tenant, TenantStatus
|
|
|
from services.account_service import TenantService
|
|
|
+from services.feature_service import FeatureService
|
|
|
from services.file_service import FileService
|
|
|
from services.workspace_service import WorkspaceService
|
|
|
|
|
@@ -68,6 +69,11 @@ class TenantListApi(Resource):
|
|
|
tenants = TenantService.get_join_tenants(current_user)
|
|
|
|
|
|
for tenant in tenants:
|
|
|
+ features = FeatureService.get_features(tenant.id)
|
|
|
+ if features.billing.enabled:
|
|
|
+ tenant.plan = features.billing.subscription.plan
|
|
|
+ else:
|
|
|
+ tenant.plan = "sandbox"
|
|
|
if tenant.id == current_user.current_tenant_id:
|
|
|
tenant.current = True # Set current=True for current tenant
|
|
|
return {"workspaces": marshal(tenants, tenants_fields)}, 200
|