jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Feat/billing enhancement (#2239)

Browse source 
Co-authored-by: takatost <takatost@gmail.com>
This commit is contained in:
Garfield Dai 2024-01-26 18:26:15 +08:00 • committed by GitHub
commit bb5d5fc683
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 43 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/controllers/console/billing/billing.py
View file

@ -20,7 +20,7 @@ class Subscription(Resource):
parser.add_argument('interval', type=str, required=True, location='args', choices=['month', 'year'])
args = parser.parse_args()
BillingService.is_tenant_owner(current_user)
BillingService.is_tenant_owner_or_admin(current_user)
return BillingService.get_subscription(args['plan'],
args['interval'],
@ -35,8 +35,8 @@ class Invoices(Resource):
@account_initialization_required
@only_edition_cloud
def get(self):
BillingService.is_tenant_owner(current_user)
return BillingService.get_invoices(current_user.email)
BillingService.is_tenant_owner_or_admin(current_user)
return BillingService.get_invoices(current_user.email, current_user.current_tenant_id)
api.add_resource(Subscription, '/billing/subscription')