mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-05-12 17:53:53 +00:00
a56ac7b34e
* perf(orgs): 默认组织改为实体组织 * perf: 添加获取当前组织信息的api * perf: 资产列表在 root 组织下的表现 * fix: 修复 root 组织引起的问题 * perf: 优化OrgModelMixin save; org_root获取; org_roles获取; UserCanUseCurrentOrg权限类 Co-authored-by: ibuler <ibuler@qq.com> Co-authored-by: Bai <bugatti_it@163.com>
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from .cache import OrgRelatedCache, IntegerField
|
|
from users.models import UserGroup, User
|
|
from assets.models import Node, AdminUser, SystemUser, Domain, Gateway
|
|
from applications.models import Application
|
|
from perms.models import AssetPermission, ApplicationPermission
|
|
from .models import OrganizationMember
|
|
|
|
|
|
class OrgResourceStatisticsCache(OrgRelatedCache):
|
|
users_amount = IntegerField()
|
|
groups_amount = IntegerField(queryset=UserGroup.objects)
|
|
|
|
assets_amount = IntegerField()
|
|
nodes_amount = IntegerField(queryset=Node.objects)
|
|
admin_users_amount = IntegerField(queryset=AdminUser.objects)
|
|
system_users_amount = IntegerField(queryset=SystemUser.objects)
|
|
domains_amount = IntegerField(queryset=Domain.objects)
|
|
gateways_amount = IntegerField(queryset=Gateway.objects)
|
|
|
|
applications_amount = IntegerField(queryset=Application.objects)
|
|
|
|
asset_perms_amount = IntegerField(queryset=AssetPermission.objects)
|
|
app_perms_amount = IntegerField(queryset=ApplicationPermission.objects)
|
|
|
|
def __init__(self, org):
|
|
super().__init__()
|
|
self.org = org
|
|
|
|
def get_key_suffix(self):
|
|
return f'<org:{self.org.id}>'
|
|
|
|
def get_current_org(self):
|
|
return self.org
|
|
|
|
def compute_users_amount(self):
|
|
if self.org.is_root():
|
|
users_amount = User.objects.all().count()
|
|
else:
|
|
users_amount = OrganizationMember.objects.values(
|
|
'user_id'
|
|
).filter(org_id=self.org.id).distinct().count()
|
|
return users_amount
|
|
|
|
def compute_assets_amount(self):
|
|
node = Node.org_root()
|
|
return node.assets_amount
|