{dev bugfix} issue - RMS45

app/applicationlayer/download/accounts/views.py

@@ -5,13 +5,11 @@ from drf_renderer_xlsx.renderers import XLSXRenderer
 from app.entities.models import (
     User
 )
-# from app.applicationlayer.form_listing_ import headers
-from app.applicationlayer.download.accounts.serializers import headers
+from app.applicationlayer.master.download.accounts.serializers import headers
 from django.db.models import CharField, Value
 from django.db.models import Q
 from rest_framework.response import Response
 from rest_framework.permissions import AllowAny
-# from cms.applicationlayer.utilities import logged_user
 from rest_framework.exceptions import ParseError
 from rest_framework import status
 
@@ -25,22 +23,27 @@ class UserDownloadRequest(XLSXFileMixin, ReadOnlyModelViewSet):
     
     def list(self, request, *args, **kwargs):
 
-        # id_number = self.request.user
-        
-        data = User.objects.all().exclude(id=1)
-        user_type = data.values_list('user_type', flat=True)[0]
+        user_type = self.request.user.user_type
+        company = self.request.user.department.company.code
+        department = self.request.user.department.code
         
+        print(user_type)
+        print(company)
+        print(department)
+
         if user_type == 'CUA':
-            code = data.values_list('department__company', flat=True)[0]
+            print('cua')
             data = User.objects.filter(
-                department__company=code).exclude(id=1)
+                department__company=company).exclude(id=1)
         elif user_type == 'DUA':
-            code = data.values_list('department', flat=True)[0]
+            print('dua')
             data = User.objects.filter(
-                department=code).exclude(id=1)
+                department=department).exclude(id=1)
         elif user_type == 'SU':
+            print('su')
             data = User.objects.all().exclude(id=1)
         elif user_type == 'OUA':
+            print('oua')
             data = User.objects.all().exclude(id=1)
         else:
             return Response(