404 when logo_path does not exist

supakeen · Mar 9, 2024 · 7003162 · 7003162
1 parent c9abe81
commit 7003162
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/pinnwand/handler/website.py b/src/pinnwand/handler/website.py
@@ -595,7 +595,10 @@ def initialize(self, path: str) -> None:
         self.path = path
 
     async def get(self) -> None:
-        self.set_header("Content-Type", "image/png")
 
-        with open(self.path, "rb") as f:
-            self.write(f.read())
+        try:
+            with open(self.path, "rb") as f:
+                self.write(f.read())
+                self.set_header("Content-Type", "image/png")
+        except FileNotFoundError:
+            raise tornado.web.HTTPError(404)