lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix out of range bug

Browse source
This commit is contained in:
cuom1999 2021-12-20 18:55:26 -06:00
parent 4a614bec65
commit 20c2bd71f8
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
chat_box/views.py
View file

@ -408,11 +408,13 @@ def get_unread_count(rooms, user):
.order_by().values('room')\ .order_by().values('room')\
.annotate(unread_count=Count('pk')).values('unread_count') .annotate(unread_count=Count('pk')).values('unread_count')
return UserRoom.objects\ res = UserRoom.objects\
.filter(user=user, room__isnull=True)\ .filter(user=user, room__isnull=True)\
.annotate( .annotate(
unread_count=Coalesce(Subquery(mess, output_field=IntegerField()), 0), unread_count=Coalesce(Subquery(mess, output_field=IntegerField()), 0),
).values_list('unread_count', flat=True)[0] ).values_list('unread_count', flat=True)
return res[0] if len(res) else 0
@login_required @login_required