mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-05 07:34:05 +00:00
395b868dcf
* perf: swagger upgrade * perf: upgrade to drf-spectacular * perf: 添加部分注解 * perf: swagger done --------- Co-authored-by: ibuler <ibuler@qq.com>
23 lines
765 B
Python
23 lines
765 B
Python
from django.shortcuts import get_object_or_404
|
|
|
|
from accounts.models import VirtualAccount
|
|
from accounts.serializers import VirtualAccountSerializer
|
|
from common.utils import is_uuid
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
|
|
|
|
class VirtualAccountViewSet(OrgBulkModelViewSet):
|
|
serializer_class = VirtualAccountSerializer
|
|
search_fields = ('alias',)
|
|
filterset_fields = ('alias',)
|
|
|
|
def get_queryset(self):
|
|
if getattr(self, "swagger_fake_view", False):
|
|
return VirtualAccount.objects.none()
|
|
return VirtualAccount.get_or_init_queryset()
|
|
|
|
def get_object(self, ):
|
|
pk = self.kwargs.get('pk')
|
|
kwargs = {'pk': pk} if is_uuid(pk) else {'alias': pk}
|
|
return get_object_or_404(VirtualAccount, **kwargs)
|