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

extract enum type for tenant account role (#3788)

Browse source
This commit is contained in:
Bowen Liang 2024-04-25 18:20:08 +08:00 • committed by GitHub
commit c54fcfb45d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 36 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/services/billing_service.py
View file

@ -3,7 +3,7 @@ import os
import requests
from extensions.ext_database import db
from models.account import TenantAccountJoin
from models.account import TenantAccountJoin, TenantAccountRole
class BillingService:
@ -74,5 +74,5 @@ class BillingService:
TenantAccountJoin.account_id == current_user.id
).first()
if join.role not in ['owner', 'admin']:
if TenantAccountRole.is_privileged_role(join.role):
raise ValueError('Only team owner or team admin can perform this action')