refactor: use get() instead of first()

danjac · Jan 10, 2025 · 26c6efb · 26c6efb
1 parent e170edc
commit 26c6efb
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/radiofeed/episodes/middleware.py b/radiofeed/episodes/middleware.py
@@ -1,3 +1,4 @@
+import contextlib
 import dataclasses
 
 from django.http import HttpRequest, HttpResponse
@@ -44,12 +45,9 @@ def pop(self) -> int | None:
     def audio_log(self) -> AudioLog | None:
         """Returns audio log for the current episode in player."""
         if self.request.user.is_authenticated and (episode_id := self.get()):
-            return (
-                self.request.user.audio_logs.filter(episode_id=episode_id)
-                .select_related(
+            with contextlib.suppress(AudioLog.DoesNotExist):
+                return self.request.user.audio_logs.select_related(
                     "episode",
                     "episode__podcast",
-                )
-                .first()
-            )
+                ).get(episode_id=episode_id)
         return None