mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-11-02 06:53:37 +00:00
7e7e24f51f
* reactor: 重构applications模块 & 删除applications、perms中已不再使用的模块 * reactor: 1. 针对application.attrs字段的view-serializer映射逻辑，采用DynamicMapping的方案重写; * reactor: 2. 删除applications和perms模块中已不再使用的database-app/k8s-app/remote-app模块; * reactor: 添加迁移文件(删除perms/databaseperrmission/remoteapppermission/k8sapppermission) * reactor: 修改细节 Co-authored-by: Bai <bugatti_it@163.com>
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
from orgs.models import Organization
|
|
from ..serializers.utils import get_dynamic_mapping_fields_mapping_rule_by_view
|
|
|
|
|
|
__all__ = ['ApplicationViewMixin', 'SerializeApplicationToTreeNodeMixin']
|
|
|
|
|
|
class ApplicationViewMixin:
|
|
""" 实现 `get_dynamic_mapping_fields_mapping_rule` 方法, 供其他和 Application 相关的 View 继承使用"""
|
|
|
|
def get_dynamic_mapping_fields_mapping_rule(self):
|
|
fields_mapping_rule = get_dynamic_mapping_fields_mapping_rule_by_view(view=self)
|
|
return fields_mapping_rule
|
|
|
|
|
|
class SerializeApplicationToTreeNodeMixin:
|
|
|
|
@staticmethod
|
|
def _serialize_db(db):
|
|
return {
|
|
'id': db.id,
|
|
'name': db.name,
|
|
'title': db.name,
|
|
'pId': '',
|
|
'open': False,
|
|
'iconSkin': 'database',
|
|
'meta': {'type': 'database_app'}
|
|
}
|
|
|
|
@staticmethod
|
|
def _serialize_remote_app(remote_app):
|
|
return {
|
|
'id': remote_app.id,
|
|
'name': remote_app.name,
|
|
'title': remote_app.name,
|
|
'pId': '',
|
|
'open': False,
|
|
'isParent': False,
|
|
'iconSkin': 'chrome',
|
|
'meta': {'type': 'remote_app'}
|
|
}
|
|
|
|
@staticmethod
|
|
def _serialize_cloud(cloud):
|
|
return {
|
|
'id': cloud.id,
|
|
'name': cloud.name,
|
|
'title': cloud.name,
|
|
'pId': '',
|
|
'open': False,
|
|
'isParent': False,
|
|
'iconSkin': 'k8s',
|
|
'meta': {'type': 'k8s_app'}
|
|
}
|
|
|
|
def _serialize_application(self, application):
|
|
method_name = f'_serialize_{application.category}'
|
|
data = getattr(self, method_name)(application)
|
|
data.update({
|
|
'pId': application.org.id,
|
|
'org_name': application.org_name
|
|
})
|
|
return data
|
|
|
|
def serialize_applications(self, applications):
|
|
data = [self._serialize_application(application) for application in applications]
|
|
return data
|
|
|
|
@staticmethod
|
|
def _serialize_organization(org):
|
|
return {
|
|
'id': org.id,
|
|
'name': org.name,
|
|
'title': org.name,
|
|
'pId': '',
|
|
'open': True,
|
|
'isParent': True,
|
|
'meta': {
|
|
'type': 'node'
|
|
}
|
|
}
|
|
|
|
def serialize_organizations(self, organizations):
|
|
data = [self._serialize_organization(org) for org in organizations]
|
|
return data
|
|
|
|
@staticmethod
|
|
def filter_organizations(applications):
|
|
organizations_id = set(applications.values_list('org_id', flat=True))
|
|
organizations = [Organization.get_instance(org_id) for org_id in organizations_id]
|
|
return organizations
|
|
|
|
def serialize_applications_with_org(self, applications):
|
|
organizations = self.filter_organizations(applications)
|
|
data_organizations = self.serialize_organizations(organizations)
|
|
data_applications = self.serialize_applications(applications)
|
|
data = data_organizations + data_applications
|
|
return data
|