From fe944f487587eb26d6344f9c8b4ae2702a007cda Mon Sep 17 00:00:00 2001
From: Artyom Kravchenko <krav-88@mail.ru>
Date: Sat, 6 Mar 2021 01:36:19 +0300
Subject: [PATCH] Fixed info bug on page

---
 main/templates/pages/work.html |  4 ++--
 main/views.py                  | 14 ++++++++++++--
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/main/templates/pages/work.html b/main/templates/pages/work.html
index 2fcb4cf..a588ef5 100644
--- a/main/templates/pages/work.html
+++ b/main/templates/pages/work.html
@@ -22,13 +22,13 @@
             <h6 class="table-title">Список сотрудников с правами инженера</h6>
             <table class="light-table">
                 <thead>
-                    <th>ID</th>
+                    <th>Email</th>
                     <th>Name</th>
                 </thead>
                 <tbody>
                     {% for engineer in engineers %}
                         <tr>
-                            <td>{{ engineer.id }}</td>
+                            <td>{{ engineer.email }}</td>
                             <td>{{ engineer.name }}</td>
                         </tr>
                     {% endfor %}
diff --git a/main/views.py b/main/views.py
index ca8c9c2..0008255 100644
--- a/main/views.py
+++ b/main/views.py
@@ -128,10 +128,20 @@ def auth_user(request):
 
 @login_required()
 def work_page(request, id):
+    users = get_users_list()
     if request.user.id == id:
+        engineers = []
+        light_agents = []
+        for user in users:
+
+            if user.custom_role_id == ZENDESK_ROLES['engineer']:
+                engineers.append((user))
+            elif user.custom_role_id == ZENDESK_ROLES['light_agent']:
+                light_agents.append(user)
+
         context = {
-            'engineers': UserProfile.objects.filter(role="admin"),
-            'agents': UserProfile.objects.filter(role="agent"),
+            'engineers': engineers,
+            'agents': light_agents,
             'pagename': 'Управление правами'
         }
         return render(request, 'pages/work.html', context)