estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: List not found in account service (#2434)

Browse Source
This commit is contained in:
takatost
2024-02-12 00:56:17 +08:00
committed by GitHub
parent 063191889d
commit 476eb90a90
6 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/services/account_service.py
View File

@@ -243,7 +243,7 @@ class TenantService:
return ta
@staticmethod
def get_join_tenants(account: Account) -> List[Tenant]:
def get_join_tenants(account: Account) -> list[Tenant]:
"""Get account join tenants"""
return db.session.query(Tenant).join(
TenantAccountJoin, Tenant.id == TenantAccountJoin.tenant_id
@@ -282,7 +282,7 @@ class TenantService:
account.current_tenant_id = tenant_account_join.tenant_id
@staticmethod
def get_tenant_members(tenant: Tenant) -> List[Account]:
def get_tenant_members(tenant: Tenant) -> list[Account]:
"""Get tenant members"""
query = (
db.session.query(Account, TenantAccountJoin.role)
@@ -303,7 +303,7 @@ class TenantService:
return updated_accounts
@staticmethod
def has_roles(tenant: Tenant, roles: List[TenantAccountJoinRole]) -> bool:
def has_roles(tenant: Tenant, roles: list[TenantAccountJoinRole]) -> bool:
"""Check if user has any of the given roles for a tenant"""
if not all(isinstance(role, TenantAccountJoinRole) for role in roles):
raise ValueError('all roles must be TenantAccountJoinRole')