mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
662 B
21 lines
662 B
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):
|
|
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)
|