39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
from rest_framework.filters import (
|
|
SearchFilter,
|
|
OrderingFilter
|
|
)
|
|
from rest_framework.generics import ListAPIView
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from counter.models import DeviceInfo
|
|
from .serializers import DeviceInfoMobileSerializer
|
|
from utils.pagination import DeviceInfoMobilePagination
|
|
from ...mixins.role import RoleMixin
|
|
|
|
|
|
class DeviceInfoMobileListAPIView(ListAPIView, RoleMixin):
|
|
serializer_class = DeviceInfoMobileSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
filter_backends = [SearchFilter, OrderingFilter]
|
|
pagination_class = DeviceInfoMobilePagination
|
|
search_fields = ['device_id', 'online']
|
|
|
|
def get_queryset(self, *args, **kwargs):
|
|
user_roles = self.get_role()
|
|
queryset = DeviceInfo.objects.get_queryset().order_by('-last_offline_time')
|
|
if 'staff' in user_roles or 'manager' in user_roles:
|
|
child = self.request.user.get_child()
|
|
device_ids = [query.device_id
|
|
for query in queryset
|
|
if query.org == self.request.user.org or query.org.id in child]
|
|
queryset = queryset.filter(device_id__in=device_ids)
|
|
device_id = self.request.GET.get('device_id')
|
|
status = self.request.GET.get('status')
|
|
|
|
if device_id:
|
|
queryset = queryset.filter(device_id__icontains=device_id)
|
|
|
|
if status:
|
|
queryset = queryset.filter(online=int(status))
|
|
return queryset
|