mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-14 12:06:23 +00:00
1248458451
* perf: 支持自定义类型资产 * perf: 改名前 * perf: 优化支持 choices * perf: 优化自定义资产 * perf: 优化资产的详情 * perf: 修改完成自定义平台和资产 --------- Co-authored-by: ibuler <ibuler@qq.com> Co-authored-by: Jiangjie.Bai <bugatti_it@163.com>
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from assets.const import AllTypes
|
|
from assets.models import Platform
|
|
from assets.serializers import PlatformSerializer
|
|
from common.api import JMSModelViewSet
|
|
from common.serializers import GroupedChoiceSerializer
|
|
|
|
__all__ = ['AssetPlatformViewSet']
|
|
|
|
|
|
class AssetPlatformViewSet(JMSModelViewSet):
|
|
queryset = Platform.objects.all()
|
|
serializer_classes = {
|
|
'default': PlatformSerializer,
|
|
'categories': GroupedChoiceSerializer
|
|
}
|
|
filterset_fields = ['name', 'category', 'type']
|
|
search_fields = ['name']
|
|
rbac_perms = {
|
|
'categories': 'assets.view_platform',
|
|
'type_constraints': 'assets.view_platform',
|
|
'ops_methods': 'assets.view_platform'
|
|
}
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
queryset = queryset.filter(type__in=AllTypes.get_types_values())
|
|
return queryset
|
|
|
|
def get_object(self):
|
|
pk = self.kwargs.get('pk', '')
|
|
if pk.isnumeric():
|
|
return super().get_object()
|
|
return self.get_queryset().get(name=pk)
|
|
|
|
def check_object_permissions(self, request, obj):
|
|
if request.method.lower() in ['delete', 'put', 'patch'] and obj.internal:
|
|
self.permission_denied(
|
|
request, message={"detail": "Internal platform"}
|
|
)
|
|
return super().check_object_permissions(request, obj)
|