24 lines
855 B
Python
24 lines
855 B
Python
|
from rest_framework import mixins, viewsets
|
||
|
|
from rest_framework.filters import SearchFilter
|
||
|
|
from rest_framework.permissions import IsAuthenticated
|
||
|
|
|
||
|
|
from apps.api.serializers.user_group import UserGroupSerializer
|
||
|
|
from apps.auth_token.auth import PluginAuthentication
|
||
|
|
from apps.slack.models import SlackUserGroup
|
||
|
|
|
||
|
|
|
||
|
|
class UserGroupViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
|
||
|
|
authentication_classes = (PluginAuthentication,)
|
||
|
|
permission_classes = (IsAuthenticated,)
|
||
|
|
serializer_class = UserGroupSerializer
|
||
|
|
|
||
|
|
filter_backends = (SearchFilter,)
|
||
|
|
search_fields = ("name", "handle")
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
slack_team_identity = self.request.auth.organization.slack_team_identity
|
||
|
|
if slack_team_identity is None:
|
||
|
|
return SlackUserGroup.objects.none()
|
||
|
|
|
||
|
|
return slack_team_identity.usergroups.all()
|