new

osinaweb/osinacore/consumers.py

@@ -48,11 +48,9 @@ class OnlineUserConsumer(WebsocketConsumer):
         online_users_ids = [connection.user.id for connection in online_connections]
         customer_connections = []
         staff_connections = []
-        for connection in sorted_connections:
-            if hasattr(connection.user, 'customerprofile'):
-               print('hi')
-            elif hasattr(connection.user, 'staffprofile'):
-               print('bye')
+        customer_connections = [connection for connection in sorted_connections if hasattr(connection.user, 'customerprofile')]
+        staff_connections = [connection for connection in sorted_connections if hasattr(connection.user, 'staffprofile')]
+
         event = {
             'type': 'online_user_connection_handler',
             'staff_connections': staff_connections,