From d42bd612d0e857c2979c2df1c42294fcc13dacac Mon Sep 17 00:00:00 2001
From: Simon <simobilleter@gmail.com>
Date: Sat, 26 Aug 2023 21:05:17 +0700
Subject: [PATCH] handle 404 video sponsorblock, #526

---
 tubearchivist/api/views.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tubearchivist/api/views.py b/tubearchivist/api/views.py
index 4b455b1..34762d2 100644
--- a/tubearchivist/api/views.py
+++ b/tubearchivist/api/views.py
@@ -228,6 +228,10 @@ class VideoSponsorView(ApiBaseView):
         # pylint: disable=unused-argument
 
         self.get_document(video_id)
+        if not self.response.get("data"):
+            message = {"message": "video not found"}
+            return Response(message, status=404)
+
         sponsorblock = self.response["data"].get("sponsorblock")
 
         return Response(sponsorblock)