Commit 6c401938 authored by John Red Medrano's avatar John Red Medrano

Merge pull request #154 in RMS/api-main-service from red-develop to RMSv2

* commit '2ec865b6':
  fixing
parents 0e5c2d8d 2ec865b6
...@@ -12,6 +12,7 @@ from django_filters.rest_framework import DjangoFilterBackend ...@@ -12,6 +12,7 @@ from django_filters.rest_framework import DjangoFilterBackend
from app.applicationlayer.master.Account import serializer from app.applicationlayer.master.Account import serializer
from app.applicationlayer.management.account.table_filters import AccountFilterset from app.applicationlayer.management.account.table_filters import AccountFilterset
from app.helper.decorators import rms from app.helper.decorators import rms
from django.db.models import Q
class AdminAccountViewSet(viewsets.ModelViewSet): class AdminAccountViewSet(viewsets.ModelViewSet):
...@@ -29,13 +30,13 @@ class AdminAccountViewSet(viewsets.ModelViewSet): ...@@ -29,13 +30,13 @@ class AdminAccountViewSet(viewsets.ModelViewSet):
# @rms.department_list # @rms.department_list
def list(self, request, *args, **kwargs): def list(self, request, *args, **kwargs):
try: # try:
queryset = self.filter_queryset(self.get_queryset()) queryset = self.filter_queryset(self.get_queryset())
if len(self.request.query_params) > 0: if len(self.request.query_params) > 0:
dept = self.request.query_params['department_code'] dept = self.request.query_params['department_code']
company = self.request.query_params['company_code'] company = self.request.query_params['company_code']
queryset = Users.objects.filter( queryset = queryset.filter(
Q(department__company=company) & Q(department__company=company) &
Q(department=dept) Q(department=dept)
) )
...@@ -54,8 +55,8 @@ class AdminAccountViewSet(viewsets.ModelViewSet): ...@@ -54,8 +55,8 @@ class AdminAccountViewSet(viewsets.ModelViewSet):
serializer = self.get_serializer(queryset, many=True) serializer = self.get_serializer(queryset, many=True)
return Response(serializer.data) return Response(serializer.data)
except Exception as e: # except Exception as e:
return Response( # return Response(
{'message': "query params department_code and company_code are both expected"}, # {'message': "query params department_code and company_code are both expected"},
status=status.HTTP_400_BAD_REQUEST # status=status.HTTP_400_BAD_REQUEST
) # )
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment