diff --git a/bookwyrm/tests/views/test_updates.py b/bookwyrm/tests/views/test_updates.py index 3fe5b944c..dff730e6d 100644 --- a/bookwyrm/tests/views/test_updates.py +++ b/bookwyrm/tests/views/test_updates.py @@ -47,9 +47,9 @@ class UpdateViews(TestCase): request = self.factory.get("") request.user = self.local_user - with patch('bookwyrm.activitystreams.ActivityStream.get_unread_count') as mock: + with patch("bookwyrm.activitystreams.ActivityStream.get_unread_count") as mock: mock.return_value = 3 - result = views.get_unread_status_count(request, 'home') + result = views.get_unread_status_count(request, "home") self.assertIsInstance(result, JsonResponse) data = json.loads(result.getvalue()) diff --git a/bookwyrm/views/updates.py b/bookwyrm/views/updates.py index 5bcb26b92..b095c965a 100644 --- a/bookwyrm/views/updates.py +++ b/bookwyrm/views/updates.py @@ -4,14 +4,16 @@ from django.http import JsonResponse from bookwyrm import activitystreams + @login_required def get_notification_count(request): """ any notifications waiting? """ - return JsonResponse({ - "count": request.user.notification_set.filter( - read=False - ).count(), - }) + return JsonResponse( + { + "count": request.user.notification_set.filter(read=False).count(), + } + ) + @login_required def get_unread_status_count(request, stream): @@ -19,6 +21,4 @@ def get_unread_status_count(request, stream): stream = activitystreams.streams.get(stream) if not stream: return JsonResponse({}) - return JsonResponse({ - "count": stream.get_unread_count(request.user) - }) + return JsonResponse({"count": stream.get_unread_count(request.user)})