mirror of
https://github.com/langgenius/dify.git
synced 2025-12-12 10:51:00 +00:00
cced33d068
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
from flask_restx import Resource, reqparse
|
|
|
|
from controllers.console import console_ns
|
|
from controllers.console.wraps import account_initialization_required, only_edition_cloud, setup_required
|
|
from libs.login import current_account_with_tenant, login_required
|
|
from services.billing_service import BillingService
|
|
|
|
|
|
@console_ns.route("/billing/subscription")
|
|
class Subscription(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@only_edition_cloud
|
|
def get(self):
|
|
current_user, current_tenant_id = current_account_with_tenant()
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("plan", type=str, required=True, location="args", choices=["professional", "team"])
|
|
parser.add_argument("interval", type=str, required=True, location="args", choices=["month", "year"])
|
|
args = parser.parse_args()
|
|
BillingService.is_tenant_owner_or_admin(current_user)
|
|
return BillingService.get_subscription(args["plan"], args["interval"], current_user.email, current_tenant_id)
|
|
|
|
|
|
@console_ns.route("/billing/invoices")
|
|
class Invoices(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@only_edition_cloud
|
|
def get(self):
|
|
current_user, current_tenant_id = current_account_with_tenant()
|
|
BillingService.is_tenant_owner_or_admin(current_user)
|
|
return BillingService.get_invoices(current_user.email, current_tenant_id)
|