mirror of https://github.com/jumpserver/jumpserver
30 lines
913 B
Python
30 lines
913 B
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from rest_framework import status
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
from ..models import UserGroup, User
|
|
from ..serializers import UserGroupSerializer
|
|
|
|
__all__ = ['UserGroupViewSet']
|
|
|
|
|
|
class UserGroupViewSet(OrgBulkModelViewSet):
|
|
model = UserGroup
|
|
filterset_fields = ("name",)
|
|
search_fields = filterset_fields
|
|
serializer_class = UserGroupSerializer
|
|
ordering = ('name',)
|
|
rbac_perms = (
|
|
("add_all_users", "users.add_usergroup"),
|
|
)
|
|
|
|
@action(methods=['post'], detail=True, url_path='add-all-users')
|
|
def add_all_users(self, request, *args, **kwargs):
|
|
instance = self.get_object()
|
|
users = User.get_org_users().exclude(groups__id=instance.id)
|
|
instance.users.add(*users)
|
|
return Response(status=status.HTTP_200_OK)
|