PDU
/
Mosqkiller-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

# add org in device log history

This commit is contained in:
xianfuxing 2018-10-10 17:44:38 +08:00
parent 4e55433573
commit 8b35461dda
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/counter/api/views.py
View File

@ -123,9 +123,23 @@ class DeviceLogHistoryListAPIView(ListAPIView):
pagination_class = PostPageNumberPagination
search_fields = ['device_id', 'date']
# def get_queryset(self, *args, **kwargs):
# queryset_list = DevicePostStatistic.objects.get_queryset().order_by('-date')
# device_id = self.request.GET.get('device_id')
# if device_id:
# queryset_list = queryset_list.filter(device_id__icontains=device_id)
# return queryset_list
def get_queryset(self, *args, **kwargs):
user = self.request.user
user_roles = user.role.split(',')
queryset_list = DevicePostStatistic.objects.get_queryset().order_by('-date')
if 'staff' in user_roles or 'admin' in user_roles:
queryset_ids = [query.device_id for query in queryset_list if query.org == user.org]
queryset_list = queryset_list.filter(device_id__in=queryset_ids)
device_id = self.request.GET.get('device_id')
if device_id:
queryset_list = queryset_list.filter(device_id__icontains=device_id)
return queryset_list