youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 00:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add billing subscription plan api

Browse Source
This commit is contained in:
hj24
2025-12-18 10:33:25 +08:00
parent 0bbf8f72b1
commit bffd67f3a4
2 changed files with 80 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
api/services/billing_service.py
View File

@ -291,15 +291,19 @@ class BillingService:
chunk_size = 200
for i in range(0, len(tenant_ids), chunk_size):
chunk = tenant_ids[i : i + chunk_size]
resp = cls._send_request("POST", "/subscription/plan/batch", json={"tenant_ids": chunk})
data = resp.get("data", {})
for tenant_id, plan in data.items():
if isinstance(plan, dict) and "plan" in plan and "expiration_date" in plan:
subscription_plan: SubscriptionPlan = {
"plan": str(plan["plan"]),
"expiration_date": int(plan["expiration_date"]),
}
results[tenant_id] = subscription_plan
try:
resp = cls._send_request("POST", "/subscription/plan/batch", json={"tenant_ids": chunk})
data = resp.get("data", {})
for tenant_id, plan in data.items():
if isinstance(plan, dict) and "plan" in plan and "expiration_date" in plan:
subscription_plan: SubscriptionPlan = {
"plan": str(plan["plan"]),
"expiration_date": int(plan["expiration_date"]),
}
results[tenant_id] = subscription_plan
except Exception:
logger.exception("Failed to fetch billing info batch for tenants: %s", chunk)
continue
return results
@ -307,4 +311,7 @@ class BillingService:
def get_expired_subscription_cleanup_whitelist(cls) -> Sequence[str]:
resp = cls._send_request("GET", "/subscription/cleanup/whitelist")
data = resp.get("data", [])
return data
tenant_whitelist = []
for item in data:
tenant_whitelist.append(item["tenant_id"])
return tenant_whitelist