Skip to content

Commit

Permalink
feat(api): filter software to users organizations
Browse files Browse the repository at this point in the history
!17 closes #45
  • Loading branch information
jon-nfc committed Jun 2, 2024
1 parent 89a5e0f commit 4151e0a
Showing 1 changed file with 12 additions and 4 deletions.
16 changes: 12 additions & 4 deletions app/api/views/itam/software.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
# from django.contrib.auth.mixins import PermissionRequiredMixin, LoginRequiredMixin
from django.db.models import Q
from django.shortcuts import get_object_or_404
from rest_framework import generics, viewsets

from itam.models.software import Software
from rest_framework import generics, viewsets

from api.serializers.itam.software import SoftwareSerializer
from api.views.mixin import OrganizationPermissionAPI

from itam.models.software import Software



class SoftwareViewSet(viewsets.ModelViewSet):
Expand All @@ -26,7 +27,14 @@ def get_object(self, queryset=None, **kwargs):


def get_queryset(self):
return Software.objects.all()

if self.request.user.is_superuser:

return self.queryset.filter().order_by('name')

else:

return self.queryset.filter(Q(organization__in=self.user_organizations()) | Q(is_global = True)).order_by('name')


def get_view_name(self):
Expand Down

0 comments on commit 4151e0a

Please sign in to comment.