Backend: Fix stats view

1 files changed, 7 insertions, 2 deletions

diff --git a/backend/views.py b/backend/views.py
index 8d28674..d96fabf 100644
--- a/backend/views.py
+++ b/backend/views.py
@@ -1067,10 +1067,14 @@ def stats(request):
     likes_total = SJLike.objects.filter(content__content_type='video').count()
     users_total = auth.models.User.objects.filter(is_active=True).count()
     users_today = auth.models.User.objects.filter(is_active=True, last_login__gt=yesterday_datetime())
-    users_in_main_room = auth.models.User.objects.filter(sjroom__name='main', sjuserprofile__lastseen_chat__gt=lastseen_datetime())
+    try:
+        main_room = SJRoom.objects.get(name='main')
+        users_in_main_room = main_room.users.all()
+    except:
+        users_in_main_room = []
     new_users_today = auth.models.User.objects.filter(is_active=True, date_joined__gt=yesterday_datetime()).order_by('-date_joined')
     rooms_total = SJRoom.objects.all().count()
-    main_room_peak = auth.models.User.objects.filter(sjroom__name='main').count()
+    main_room_peak = len(users_in_main_room)
     return render_to_response('stats.html', {'users_total': users_total,
                                              'users_today': users_today,
                                              'new_users_today': new_users_today,
@@ -1080,6 +1084,7 @@ def stats(request):
                                              'likes_today': likes_today,
                                              'rooms_total': rooms_total,
                                              'main_room_peak': main_room_peak,
+                                             'users_in_main_room': users_in_main_room,
                                              'uptime': uptime})
 
 def redirect(request):